Implemented RegisterBSA

erri120 · Nov 28, 2019 · 4c068c9 · 4c068c9
1 parent c26ba28
commit 4c068c9
Showing 1 changed file with 28 additions and 2 deletions.
diff --git a/OMODFramework/Scripting/OBMMScriptHandler.cs b/OMODFramework/Scripting/OBMMScriptHandler.cs
@@ -445,10 +445,10 @@ internal static ScriptReturnData Execute(string inputScript, string dataPath, st
                         FunctionModifyInstallFolder(line, true);
                         break;
                     case "RegisterBSA":
-                        //TODO: FunctionRegisterBSA(line, true);
+                        FunctionRegisterBSA(line, true);
                         break;
                     case "UnregisterBSA":
-                        //TODO: FunctionRegisterBSA(line, false);
+                        FunctionRegisterBSA(line, false);
                         break;
                     case "FatalError":
                         srd.CancelInstall = true;
@@ -2309,5 +2309,31 @@ private static void FunctionDisplayFile(IReadOnlyCollection<string> line, bool i
                 _scriptFunctions.DisplayText(text, title);
             }
         }
+
+        private static void FunctionRegisterBSA(IReadOnlyCollection<string> line, bool register)
+        {
+            var funcName = register ? "Register" : "Unregister";
+            funcName += "BSA";
+
+            if (line.Count == 1)
+            {
+                Warn($"Missing arguments for '{funcName}'");
+                return;
+            }
+
+            var esp = line.ElementAt(1).ToLower();
+            if (esp.Contains(",") || esp.Contains(";") || esp.Contains("="))
+            {
+                Warn($"Invalid argument for '{funcName}'\nBSA file names are not allowed to include the characters ',' '=' or ';'");
+                return;
+            }
+
+            if(line.Count > 2) Warn($"Unexpected arguments after '{funcName}'");
+
+            if (register && !srd.RegisterBSASet.Contains(esp))
+                srd.RegisterBSASet.Add(esp);
+            else
+                srd.RegisterBSASet.RemoveWhere(s => s == esp);
+        }
     }
 }