Example #1
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            loader.RegisterModifiedFile(TargetFile);
            var source_path = mod.GetFile(SourceFile);

            var parsed_script = LuaParser.ParseFile(loader.ExpandTargetPathDefaulted(TargetFile));

            if (parsed_script == null)
            {
                throw new Exception($"Failed parsing '{TargetFile}'");
            }
            var ast_search = new ASTSearchWalker();

            ast_search.Match(LuaInsertTarget, parsed_script);

            if (!ast_search.HasSelection)
            {
                throw new Exception($"AST search failed");
            }

            var offset = ast_search.SelectedSpan.Value.Location.Position;

            string s;

            using (var target_file = new StreamReader(File.OpenRead(loader.ExpandTargetPathDefaulted(TargetFile)))) {
                using (var source_file = new StreamReader(File.OpenRead(source_path))) {
                    s = InsertAtPos(target_file, offset, source_file, Trim);
                }
            }

            loader.DeleteFile(TargetFile);

            using (var output = new StreamWriter(loader.OpenWriteTarget(TargetFile))) {
                output.Write(loader.ProcessNewlines(s.ToString()));
            }

            if (loader.ExtraChecks)
            {
                Logger.Debug($"Checking result to see if it's still valid");
                var parsed_new_script = LuaParser.ParseFile(loader.ExpandTargetPath(TargetFile));
                if (parsed_new_script == null)
                {
                    Logger.Warn($"Insertion broke Lua script");
                }
                else
                {
                    var test_ast_search = new ASTSearchWalker();
                    test_ast_search.Match(LuaInsertTarget, parsed_new_script);

                    if (!test_ast_search.HasSelection)
                    {
                        Logger.Warn($"AST query no longer resolves after insertion");
                    }
                }
            }

            DoReplaceIfRequested(mod, loader, TargetFile);
        }
Example #2
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            loader.RegisterModifiedFile(TargetFile);

            loader.DeleteFile(TargetFile);
            loader.CopyFile(mod, SourceFile, TargetFile);

            DoReplaceIfRequested(mod, loader, TargetFile);
        }
Example #3
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            if (loader.FileExists(TargetFile))
            {
                throw new Exception($"Can't copy as file '{TargetFile}', because it already exists at that path");
            }

            loader.RegisterNewFile(TargetFile);

            loader.DeleteFile(TargetFile);
            loader.CopyFile(mod, SourceFile, TargetFile);

            DoReplaceIfRequested(mod, loader, TargetFile);
        }
Example #4
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            var source_path = mod.GetFile(SourceFile);

            loader.RegisterModifiedFile(TargetFile);

            XMLMerger merger = null;

            using (var source_file = new StreamReader(File.OpenRead(source_path))) {
                using (var target_file = new StreamReader(File.OpenRead(loader.ExpandTargetPathDefaulted(TargetFile)))) {
                    merger = new XMLMerger(source_file, target_file); // patch, target
                }
            }

            var doc = merger.CreateMergedDocument();

            loader.DeleteFile(TargetFile);
            using (var target_file = new StreamWriter(loader.OpenWriteTarget(TargetFile))) {
                XMLMerger.WriteDocumentToFile(doc, target_file);
            }

            DoReplaceIfRequested(mod, loader, TargetFile);
        }
Example #5
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            if (!ValidEvents.Contains(Event))
            {
                throw new Exception($"Invalid event '{Event}'");
            }

            if (Event == "init")
            {
                TargetFile = "data/scripts/perks/perk_list.lua";                              // lua entry point
            }
            if (!loader.FileExists(TargetFile))
            {
                throw new Exception("Target file doesn't exist");
            }

            var event_files     = loader.TryGetLuaModEvent(Event, TargetFile);
            var mod_source_file = mod.GetFile(SourceFile);

            if (event_files == null)
            {
                loader.RegisterModifiedFile(TargetFile);
                event_files = loader.RegisterLuaModEvent(Event, TargetFile);
                int offset = 0;

                if (Event == "enter")
                {
                    var lua_parser = new NetLua.Parser();
                    var block      = lua_parser.ParseFile(loader.ExpandTargetPathDefaulted(TargetFile));
                    NetLua.Ast.IStatement first_non_dofile = null;

                    for (var i = 0; i < block.Statements.Count; i++)
                    {
                        var statement = block.Statements[i];

                        if (statement is NetLua.Ast.FunctionCall)
                        {
                            var call = (NetLua.Ast.FunctionCall)statement;
                            if (call.Function is NetLua.Ast.Variable)
                            {
                                var func = (NetLua.Ast.Variable)call.Function;

                                if (func.Name == "dofile")
                                {
                                    continue;
                                }
                            }
                        }

                        first_non_dofile = statement;
                        Logger.Debug($"Matched AST");
                        break;
                    }

                    if (first_non_dofile == null)
                    {
                        Logger.Warn("Failed to match AST - empty file? Inserting call at the very beginning");
                    }
                    else
                    {
                        offset = first_non_dofile.Span.Location.Position;
                    }
                }

                var event_caller = new StringBuilder();
                event_caller.AppendLine("local ommelrt = loadfile(\"data/ommel/ommelrt.lua\")()");
                event_caller.Append("ommelrt.run_event(\"");
                event_caller.Append(Event);
                event_caller.AppendLine($"\", \"{TargetFile}\")");

                string new_target_content = null;

                using (var target_file = new StreamReader(loader.ExpandTargetPathDefaulted(TargetFile))) {
                    using (var source_file = new StringReader(event_caller.ToString())) {
                        if (Event == "enter" || Event == "top" || Event == "init")
                        {
                            new_target_content = InsertAtPos(target_file, offset, source_file, false);
                        }
                        else if (Event == "leave")
                        {
                            new_target_content  = target_file.ReadToEnd();
                            new_target_content += "\r\n";
                            new_target_content += source_file.ReadToEnd();
                        }
                    }
                }

                loader.DeleteFile(TargetFile);
                using (var f = new StreamWriter(loader.OpenWriteTarget(TargetFile))) {
                    f.Write(new_target_content);
                }
            }

            var target_source_file = loader.ConvertOmmeldataPath(SourceFile);

            event_files.Add(target_source_file);

            loader.RegisterNewFile(target_source_file);
            loader.DeleteFile(target_source_file);
            loader.CopyFile(mod, SourceFile, target_source_file);

            DoReplaceIfRequested(mod, loader, target_source_file);
        }
Example #6
0
        public override void OnExecute(Ommel loader, Mod mod)
        {
            var param_counter = 0;

            if (Line != null)
            {
                param_counter += 1;
            }
            if (Offset != null)
            {
                param_counter += 1;
            }
            if (param_counter == 0)
            {
                throw new Exception($"Missing location parameter - choose either 'Line' or 'Offset'");
            }
            if (param_counter > 1)
            {
                throw new Exception("Too many parameters given to TextInsert - choose either 'Line' or 'Offset'");
            }

            loader.RegisterModifiedFile(TargetFile);
            var source_path = mod.GetFile(SourceFile);

            StringBuilder s = new StringBuilder();

            string data_to_be_inserted = null;

            using (var source_file = new StreamReader(File.OpenRead(source_path))) {
                data_to_be_inserted = source_file.ReadToEnd();
                if (Trim)
                {
                    data_to_be_inserted = data_to_be_inserted.Trim();
                }
            }


            var string_map = mod.LoadReplacementMap(PlaceholderFile);

            if (string_map == null && Placeholders != null)
            {
                Logger.Warn($"File operation requested to use placeholders, but there is no string map file");
                return;
            }
            else if (string_map != null && Placeholders != null)
            {
                data_to_be_inserted = mod.ReplaceStrings(string_map, Placeholders.ToArray(), mod.Ommeldata.PlaceholderPrefix, mod.Ommeldata.PlaceholderSuffix, data_to_be_inserted);
            }

            if (Offset != null)
            {
                using (var target_file = new StreamReader(File.OpenRead(loader.ExpandTargetPathDefaulted(TargetFile)))) {
                    var target_pre_data = new char[Offset.Value];
                    target_file.Read(target_pre_data, 0, Offset.Value);
                    s.Append(target_pre_data);
                    s.Append(data_to_be_inserted);
                    s.Append(target_file.ReadToEnd());
                }
            }
            else if (Line != null)
            {
                using (var target_file = new StreamReader(File.OpenRead(loader.ExpandTargetPathDefaulted(TargetFile)))) {
                    var line_idx = 1;
                    while (!target_file.EndOfStream)
                    {
                        if (line_idx == Line.Value)
                        {
                            s.Append(data_to_be_inserted);
                        }
                        s.AppendLine(target_file.ReadLine());
                        line_idx += 1;
                    }
                }
            }

            loader.DeleteFile(TargetFile);

            using (var output = new StreamWriter(loader.OpenWriteTarget(TargetFile))) {
                output.Write(loader.ProcessNewlines(s.ToString()));
            }
        }