package funkin.util; import haxe.zip.Entry; import lime.utils.Bytes; import lime.ui.FileDialog; import openfl.net.FileFilter; import haxe.io.Path; #if html5 import openfl.net.FileReference; import openfl.events.Event; #end /** * Utilities for reading and writing files on various platforms. */ class FileUtil { /** * Browses for a single file, then calls `onSelect(path)` when a path chosen. * Note that on HTML5 this will immediately fail, you should call `openFile(onOpen:Resource->Void)` instead. * * @param typeFilter Filters what kinds of files can be selected. * @return Whether the file dialog was opened successfully. */ public static function browseForFile(?typeFilter:Array, ?onSelect:String->Void, ?onCancel:Void->Void, ?defaultPath:String, ?dialogTitle:String):Bool { #if desktop var filter:String = convertTypeFilter(typeFilter); var fileDialog:FileDialog = new FileDialog(); if (onSelect != null) fileDialog.onSelect.add(onSelect); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.browse(OPEN, filter, defaultPath, dialogTitle); return true; #elseif html5 onCancel(); return false; #else onCancel(); return false; #end } /** * Browses for a directory, then calls `onSelect(path)` when a path chosen. * Note that on HTML5 this will immediately fail. * * @param typeFilter TODO What does this do? * @return Whether the file dialog was opened successfully. */ public static function browseForDirectory(?typeFilter:Array, ?onSelect:String->Void, ?onCancel:Void->Void, ?defaultPath:String, ?dialogTitle:String):Bool { #if desktop var filter:String = convertTypeFilter(typeFilter); var fileDialog:FileDialog = new FileDialog(); if (onSelect != null) fileDialog.onSelect.add(onSelect); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.browse(OPEN_DIRECTORY, filter, defaultPath, dialogTitle); return true; #elseif html5 onCancel(); return false; #else onCancel(); return false; #end } /** * Browses for multiple file, then calls `onSelect(paths)` when a path chosen. * Note that on HTML5 this will immediately fail. * * @return Whether the file dialog was opened successfully. */ public static function browseForMultipleFiles(?typeFilter:Array, ?onSelect:Array->Void, ?onCancel:Void->Void, ?defaultPath:String, ?dialogTitle:String):Bool { #if desktop var filter:String = convertTypeFilter(typeFilter); var fileDialog:FileDialog = new FileDialog(); if (onSelect != null) fileDialog.onSelectMultiple.add(onSelect); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.browse(OPEN_MULTIPLE, filter, defaultPath, dialogTitle); return true; #elseif html5 onCancel(); return false; #else onCancel(); return false; #end } /** * Browses for a file location to save to, then calls `onSelect(path)` when a path chosen. * Note that on HTML5 you can't do much with this, you should call `saveFile(resource:haxe.io.Bytes)` instead. * * @param typeFilter TODO What does this do? * @return Whether the file dialog was opened successfully. */ public static function browseForSaveFile(?typeFilter:Array, ?onSelect:String->Void, ?onCancel:Void->Void, ?defaultPath:String, ?dialogTitle:String):Bool { #if desktop var filter:String = convertTypeFilter(typeFilter); var fileDialog:FileDialog = new FileDialog(); if (onSelect != null) fileDialog.onSelect.add(onSelect); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.browse(SAVE, filter, defaultPath, dialogTitle); return true; #elseif html5 onCancel(); return false; #else onCancel(); return false; #end } /** * Browses for a single file location, then reads it and passes it to `onOpen(resource:haxe.io.Bytes)`. * Works great on desktop and HTML5. * * @param typeFilter TODO What does this do? * @return Whether the file dialog was opened successfully. */ public static function openFile(?typeFilter:Array, ?onOpen:Bytes->Void, ?onCancel:Void->Void, ?defaultPath:String, ?dialogTitle:String):Bool { #if desktop var filter:String = convertTypeFilter(typeFilter); var fileDialog:FileDialog = new FileDialog(); if (onOpen != null) fileDialog.onOpen.add(onOpen); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.open(filter, defaultPath, dialogTitle); return true; #elseif html5 var onFileLoaded:Event->Void = function(event) { var loadedFileRef:FileReference = event.target; trace('Loaded file: ' + loadedFileRef.name); onOpen(loadedFileRef.data); } var onFileSelected:Event->Void = function(event) { var selectedFileRef:FileReference = event.target; trace('Selected file: ' + selectedFileRef.name); selectedFileRef.addEventListener(Event.COMPLETE, onFileLoaded); selectedFileRef.load(); } var fileRef:FileReference = new FileReference(); fileRef.addEventListener(Event.SELECT, onFileSelected); fileRef.browse(typeFilter); return true; #else onCancel(); return false; #end } /** * Browses for a single file location, then writes the provided `haxe.io.Bytes` data and calls `onSave(path)` when done. * Works great on desktop and HTML5. * * @return Whether the file dialog was opened successfully. */ public static function saveFile(data:Bytes, ?onSave:String->Void, ?onCancel:Void->Void, ?defaultFileName:String, ?dialogTitle:String):Bool { #if desktop var filter:String = defaultFileName != null ? Path.extension(defaultFileName) : null; var fileDialog:FileDialog = new FileDialog(); if (onSave != null) fileDialog.onSelect.add(onSave); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.save(data, filter, defaultFileName, dialogTitle); return true; #elseif html5 var filter:String = defaultFileName != null ? Path.extension(defaultFileName) : null; var fileDialog:FileDialog = new FileDialog(); if (onSave != null) fileDialog.onSave.add(onSave); if (onCancel != null) fileDialog.onCancel.add(onCancel); fileDialog.save(data, filter, defaultFileName, dialogTitle); return true; #else onCancel(); return false; #end } /** * Prompts the user to save multiple files. * On desktop, this will prompt the user for a directory, then write all of the files to there. * On HTML5, this will zip the files up and prompt the user to save that. * * @param typeFilter TODO What does this do? * @return Whether the file dialog was opened successfully. */ public static function saveMultipleFiles(resources:Array, ?onSaveAll:Array->Void, ?onCancel:Void->Void, ?defaultPath:String, force:Bool = false):Bool { #if desktop // Prompt the user for a directory, then write all of the files to there. var onSelectDir:String->Void = function(targetPath:String):Void { var paths:Array = []; for (resource in resources) { var filePath = haxe.io.Path.join([targetPath, resource.fileName]); try { if (resource.data == null) { trace('WARNING: File $filePath has no data or content. Skipping.'); continue; } else { writeBytesToPath(filePath, resource.data, force ? Force : Skip); } } catch (_) { trace('Failed to write file (probably already exists): $filePath' + filePath); continue; } paths.push(filePath); } onSaveAll(paths); } trace('Browsing for directory to save individual files to...'); #if mac defaultPath = null; #end browseForDirectory(null, onSelectDir, onCancel, defaultPath, 'Choose directory to save all files to...'); return true; #elseif html5 saveFilesAsZIP(resources, onSaveAll, onCancel, defaultPath, force); return true; #else onCancel(); return false; #end } /** * Takes an array of file entries and prompts the user to save them as a ZIP file. */ public static function saveFilesAsZIP(resources:Array, ?onSave:Array->Void, ?onCancel:Void->Void, ?defaultPath:String, force:Bool = false):Bool { // Create a ZIP file. var zipBytes:Bytes = createZIPFromEntries(resources); var onSave:String->Void = function(path:String) { onSave([path]); }; // Prompt the user to save the ZIP file. saveFile(zipBytes, onSave, onCancel, defaultPath, 'Save files as ZIP...'); return true; } /** * Takes an array of file entries and forcibly writes a ZIP to the given path. * Only works on desktop, because HTML5 doesn't allow you to write files to arbitrary paths. * Use `saveFilesAsZIP` instead. * @param force Whether to force overwrite an existing file. */ public static function saveFilesAsZIPToPath(resources:Array, path:String, force:Bool = false):Bool { #if desktop // Create a ZIP file. var zipBytes:Bytes = createZIPFromEntries(resources); // Write the ZIP. writeBytesToPath(path, zipBytes, force ? Force : Skip); return true; #else return false; #end } /** * Read string file contents directly from a given path. * Only works on desktop. * * @param path The path to the file. * @return The file contents. */ public static function readStringFromPath(path:String):String { #if sys return sys.io.File.getContent(path); #else return null; #end } /** * Read bytes file contents directly from a given path. * Only works on desktop. * * @param path The path to the file. * @return The file contents. */ public static function readBytesFromPath(path:String):Bytes { #if sys return Bytes.ofString(sys.io.File.getContent(path)); #else return null; #end } /** * Read JSON file contents directly from a given path. * Only works on desktop. * * @param path The path to the file. * @return The JSON data. */ public static function readJSONFromPath(path:String):Dynamic { #if sys try { return SerializerUtil.fromJSON(sys.io.File.getContent(path)); } catch (ex) { return null; } #else return null; #end } /** * Write string file contents directly to a given path. * Only works on desktop. * * @param path The path to the file. * @param data The string to write. * @param mode Whether to Force, Skip, or Ask to overwrite an existing file. */ public static function writeStringToPath(path:String, data:String, mode:FileWriteMode = Skip):Void { #if sys createDirIfNotExists(Path.directory(path)); switch (mode) { case Force: sys.io.File.saveContent(path, data); case Skip: if (!sys.FileSystem.exists(path)) { sys.io.File.saveContent(path, data); } else { throw 'File already exists: $path'; } case Ask: if (sys.FileSystem.exists(path)) { // TODO: We don't have the technology to use native popups yet. } else { sys.io.File.saveContent(path, data); } } #else throw 'Direct file writing by path not supported on this platform.'; #end } /** * Write byte file contents directly to a given path. * Only works on desktop. * * @param path The path to the file. * @param data The bytes to write. * @param mode Whether to Force, Skip, or Ask to overwrite an existing file. */ public static function writeBytesToPath(path:String, data:Bytes, mode:FileWriteMode = Skip):Void { #if sys createDirIfNotExists(Path.directory(path)); switch (mode) { case Force: sys.io.File.saveBytes(path, data); case Skip: if (!sys.FileSystem.exists(path)) { sys.io.File.saveBytes(path, data); } else { throw 'File already exists: $path'; } case Ask: if (sys.FileSystem.exists(path)) { // TODO: We don't have the technology to use native popups yet. } else { sys.io.File.saveBytes(path, data); } } #else throw 'Direct file writing by path not supported on this platform.'; #end } /** * Write string file contents directly to the end of a file at the given path. * Only works on desktop. * * @param path The path to the file. * @param data The string to append. */ public static function appendStringToPath(path:String, data:String):Void { #if sys sys.io.File.append(path, false).writeString(data); #else throw 'Direct file writing by path not supported on this platform.'; #end } /** * Create a directory if it doesn't already exist. * Only works on desktop. * * @param dir The path to the directory. */ public static function createDirIfNotExists(dir:String):Void { #if sys if (!sys.FileSystem.exists(dir)) { sys.FileSystem.createDirectory(dir); } #end } static var tempDir:String = null; static final TEMP_ENV_VARS:Array = ['TEMP', 'TMPDIR', 'TEMPDIR', 'TMP']; /** * Get the path to a temporary directory we can use for writing files. * Only works on desktop. * * @return The path to the temporary directory. */ public static function getTempDir():String { if (tempDir != null) return tempDir; #if sys #if windows var path:String = null; for (envName in TEMP_ENV_VARS) { path = Sys.getEnv(envName); if (path == '') path = null; if (path != null) break; } tempDir = Path.join([path, 'funkin/']); return tempDir; #else tempDir = '/tmp/funkin/'; return tempDir; #end #else return null; #end } /** * Create a Bytes object containing a ZIP file, containing the provided entries. * * @param entries The entries to add to the ZIP file. * @return The ZIP file as a Bytes object. */ public static function createZIPFromEntries(entries:Array):Bytes { var o:haxe.io.BytesOutput = new haxe.io.BytesOutput(); var zipWriter:haxe.zip.Writer = new haxe.zip.Writer(o); zipWriter.write(entries.list()); return o.getBytes(); } /** * Create a ZIP file entry from a file name and its string contents. * * @param name The name of the file. You can use slashes to create subdirectories. * @param content The string contents of the file. * @return The resulting entry. */ public static function makeZIPEntry(name:String, content:String):Entry { var data:Bytes = haxe.io.Bytes.ofString(content, UTF8); return makeZIPEntryFromBytes(name, data); } /** * Create a ZIP file entry from a file name and its string contents. * * @param name The name of the file. You can use slashes to create subdirectories. * @param data The byte data of the file. * @return The resulting entry. */ public static function makeZIPEntryFromBytes(name:String, data:haxe.io.Bytes):Entry { return { fileName: name, fileSize: data.length, data: data, dataSize: data.length, compressed: false, fileTime: Date.now(), crc32: null, extraFields: null, }; } static function convertTypeFilter(typeFilter:Array):String { var filter:String = null; if (typeFilter != null) { var filters:Array = []; for (type in typeFilter) { filters.push(StringTools.replace(StringTools.replace(type.extension, '*.', ''), ';', ',')); } filter = filters.join(';'); } return filter; } } enum FileWriteMode { /** * Forcibly overwrite the file if it already exists. */ Force; /** * Ask the user if they want to overwrite the file if it already exists. */ Ask; /** * Skip the file if it already exists. */ Skip; }