extension.swift

extension Dictionary where Key: String, Value: NSFileWrapper
{
	mutating func renameBundleResource(oldName: String, newName: String)
	{
		// remove extension
		let prefix = (oldName as NSString).stringByDeletingPathExtension
		
		// find all files belonging to this resource
		let fileNames = keys.sort().filter { (key) -> Bool in
			return key.hasPrefix(prefix)
		}
		
		// rename page image and user files
		for fileName in fileNames
		{
			// remove prefix
			let suffix = fileName.stringByReplacingOccurrencesOfString(prefix, withString: "")
			let newPrefix = (newName as NSString).stringByDeletingPathExtension
			let newName = newPrefix + suffix
			
			var fileWrapper = dict[fileName]!
			
			// if the writing file name has changed we need a fresh file wrapper
			if newName.hasPrefix("Proof") && fileWrapper.filename != newName
			{
				fileWrapper = NSFileWrapper(regularFileWithContents: fileWrapper.regularFileContents!)
				fileWrapper.filename = newName
			}
			
			self[fileName] = nil
			self[newName] = fileWrapper
		}
	}
}