CodenameEngine/source/funkin/backend/system/MainState.hx

package funkin.backend.system;

#if MOD_SUPPORT
import sys.FileSystem;
#end
import flixel.FlxState;
import funkin.backend.assets.AssetsLibraryList;
import funkin.backend.assets.ModsFolder;
import funkin.backend.assets.ModsFolderLibrary;
import funkin.backend.assets.ZipFolderLibrary;
import funkin.backend.chart.EventsData;
import funkin.backend.system.framerate.Framerate;
import funkin.editors.ModConfigWarning;
import funkin.menus.TitleState;
import haxe.io.Path;


@dox(hide)
typedef AddonInfo = {
	var name:String;
	var path:String;
}

/**
 * Simple state used for loading the game
 */
class MainState extends FlxState {
	public static var initiated:Bool = false;
	public override function create() {
		super.create();
		if (!initiated) {
			Main.loadGameSettings();
		}

		initiated = true;

		#if sys
		CoolUtil.deleteFolder('./.temp/'); // delete temp folder
		#end
		Options.save();

		ControlsUtil.resetCustomControls();
		FlxG.bitmap.reset();
		FlxG.sound.destroy(true);

		Paths.assetsTree.reset();

		#if MOD_SUPPORT
		inline function isDirectory(path:String):Bool
			return FileSystem.exists(path) && FileSystem.isDirectory(path);

		inline function ltrim(str:String, prefix:String):String
			return str.substr(prefix.length).ltrim();

		inline function loadLib(path:String, name:String)
			Paths.assetsTree.addLibrary(ModsFolder.loadModLib(path, name));

		var _lowPriorityAddons:Array<AddonInfo> = [];
		var _highPriorityAddons:Array<AddonInfo> = [];
		var _noPriorityAddons:Array<AddonInfo> = [];

		var quick_modsPath = ModsFolder.modsPath + ModsFolder.currentModFolder;

		// handing if the loading mod (before it's properly loaded) is a compressed mod
		// we just need to use `Paths.assetsTree.hasCompressedLibrary` to complete valid checks for actual loaded compressed mods
		var isZipMod = false;
		
		// If we know it's a compressed mod, then we can check if it's using the `cnemod` folder path.
		// All it is really is a folder with the mod's name, then a compressed file called "cnemod.[zip|7z|rar|etc]"
		var isCneMod = false;

		// We are doing it like this because think about it: it's 1 for loop lol
		// We just need to know if any of these values is true, so if only one is true and we are not close to being done in the loop, that's fine.
		// 
		for (ext in Flags.ALLOWED_ZIP_EXTENSIONS) {
			if (FileSystem.exists(quick_modsPath+"."+ext)) isZipMod = true;
			if (FileSystem.exists(quick_modsPath+"/cnemod."+ext)) isCneMod = true;
			if (isZipMod && isCneMod) break;
		}
		
		// We get the addons folder from relative space (`./`) and then our mod's addons.
		var addonPaths = [
			ModsFolder.addonsPath,
			// So to check the mod's addons folder, we need to decompress it. Which is impossible* in this stage of the loading library process.
			// TODO: Write a function when the library is loaded to decompress the contents and then load the libraries :)
			( (ModsFolder.currentModFolder != null && !isZipMod) ?
				quick_modsPath + "/addons/" : null
			)
		];

		for (path in addonPaths) {
			if (path == null) continue;
			if (!isDirectory(path)) continue;

			for (addon in FileSystem.readDirectory(path)) {
				if (!FileSystem.isDirectory(path + addon)) {
					if (Flags.ALLOWED_ZIP_EXTENSIONS.contains(Path.extension(addon))) addon = Path.withoutExtension(addon);
					else continue;
				}

				var data:AddonInfo = {
					name: addon,
					path: path + addon
				};

				if (addon.startsWith("[LOW]")) _lowPriorityAddons.insert(0, data);
				else if (addon.startsWith("[HIGH]")) _highPriorityAddons.insert(0, data);
				else _noPriorityAddons.insert(0, data);
			}
		}
		#end

		#if GLOBAL_SCRIPT
		funkin.backend.scripting.GlobalScript.destroy();
		#end
		funkin.backend.scripting.Script.staticVariables.clear();

		#if MOD_SUPPORT
		for (addon in _lowPriorityAddons)
			loadLib(addon.path, ltrim(addon.name, "[LOW]"));
		
		if (ModsFolder.currentModFolder != null) {
			// isCneMod is a guarentee to be a zip mod because we just checked for it, so this will always load as a CompressedLibrary
			if (isCneMod)
				loadLib(quick_modsPath + "/cnemod", ModsFolder.currentModFolder);
			else
				loadLib(quick_modsPath, ModsFolder.currentModFolder);
		}

		for (addon in _noPriorityAddons)
			loadLib(addon.path, addon.name);

		for (addon in _highPriorityAddons)
			loadLib(addon.path, ltrim(addon.name, "[HIGH]"));
		#end

		Flags.reset();
		Flags.load();
		funkin.savedata.FunkinSave.init();

		TranslationUtil.findAllLanguages();
		TranslationUtil.setLanguage(Flags.DISABLE_LANGUAGES ? Flags.DEFAULT_LANGUAGE : null);
		ModsFolder.onModSwitch.dispatch(ModsFolder.currentModFolder); // Loads global.hx
		MusicBeatTransition.script = Flags.DEFAULT_TRANSITION_SCRIPT;
		WindowUtils.resetAffixes(false);
		WindowUtils.setWindow();
		Main.refreshAssets();
		DiscordUtil.init();
		EventsData.reloadEvents();
		ControlsUtil.loadCustomControls();
		TitleState.initialized = false;

		if (Framerate.isLoaded)
			Framerate.instance.reload();

		#if sys
		CoolUtil.safeAddAttributes('./.temp/', NativeAPI.FileAttribute.HIDDEN);
		#end

		#if MOD_SUPPORT
		for (lib in ModsFolder.getLoadedModsLibs()) {
			if (!(lib is ZipFolderLibrary)) continue;
			if (cast(lib, ZipFolderLibrary).PRELOAD_VIDEOS) cast(lib, ZipFolderLibrary).precacheVideos();
		}
		#end

		var startState:Class<FlxState> = Flags.DISABLE_WARNING_SCREEN ? TitleState : funkin.menus.WarningState;

		#if MOD_SUPPORT
		// In this case if the mod we just loaded a compressed modpack, we can't edit or modify files without decompressing it.
		if (Options.devMode && Options.allowConfigWarning && !isZipMod) {
			var lib:ModsFolderLibrary;
			for (e in Paths.assetsTree.libraries) if ((lib = cast AssetsLibraryList.getCleanLibrary(e)) is ModsFolderLibrary
				&& lib.modName == ModsFolder.currentModFolder)
			{
				if (lib.exists(Paths.ini("config/modpack"), lime.utils.AssetType.TEXT)) break;

				FlxG.switchState(new ModConfigWarning(lib, startState));
				return;
			}
		}
		#end

		FlxG.switchState(cast Type.createInstance(startState, []));
	}
}