Program.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Windows.Forms;
using System.Diagnostics;
using System.IO;
using System.Text.RegularExpressions;
using System.Text;

namespace LVLTool
{
    static class Program
    {
        // default values
        private static string input_lvl  = null;
        private static string output_lvl = null;
        private static string files = null;
        private static Platform platform = Platform.PC;
        private static OperationMode operation = OperationMode.ShowHelp;
        private static string merge_strings_search_dir = null;
        private static string old_name = null;
        private static string new_name = null;
        public static bool Verbose = false;

        private static string sLuaSourceDir = null;

        public static string LuaSourceDir
        {
            get { return sLuaSourceDir; }
            set
            {
                if (!value.EndsWith("\\"))
                    sLuaSourceDir = value + "\\";
                else
                    sLuaSourceDir = value;
            }
        }

        private static string sModToolsDir = "C:\\BF2_ModTools\\";

        public static string ModToolsDir
        {
            get { return sModToolsDir; }
            set
            {
                if (!value.EndsWith("\\"))
                    sModToolsDir = value + "\\";
                else
                    sModToolsDir = value;
            }
        }

        public static string TmpDir
        {
            get
            {
                string retVal = ".\\Temp\\";
                if (!Directory.Exists(retVal))
                    Directory.CreateDirectory(retVal);
                return retVal;
            }
        }

        public static string Version { get { return System.Reflection.Assembly.GetEntryAssembly().GetName().Version.ToString(); } }


        /// <summary>
        /// The main entry point for the application.
        /// </summary>
        [STAThread]
        public static void Main(string[] args)
        {
            LoadSettings();
            if (args.Length == 0)
            {
                Application.EnableVisualStyles();
                Application.SetCompatibleTextRenderingDefault(false);
                Application.Run(new MainForm());
            }
            else
            {
                try
                {
                    ProcessArgs(args);
                    if (input_lvl != null && !File.Exists(input_lvl))
                    {
                        Console.WriteLine("Could not find input file '{0}'", input_lvl);
                        return;
                    }

                    string[] files_ = GetFiles(files);
                    switch (operation)
                    {
                        case OperationMode.ShowHelp:
                            PrintHelp();
                            break;
                        case OperationMode.Extract:
                            if (input_lvl != null)
                            {
                                UcfbHelper helper = new UcfbHelper();
                                helper.FileName = input_lvl;
                                helper.ExtractContents();
                            }
                            else
                                Console.WriteLine("Error! Must specify input file");
                            break;
                        case OperationMode.Add:
                            if (output_lvl == null)
                                output_lvl = input_lvl;
                            if (input_lvl != null)
                            {
                                UcfbHelper helper = new UcfbHelper();
                                helper.FileName = input_lvl;
                                foreach (string f in files_)
                                {
                                    string fileName = Munger.EnsureMungedFile(f, platform);
                                    helper.AddItemToEnd(fileName);
                                }
                                helper.SaveData(output_lvl);
                            }
                            else
                                Console.WriteLine("Error! Must specify input file");
                            break;
                        case OperationMode.Replace:
                            if (output_lvl == null)
                                output_lvl = input_lvl;
                            if (input_lvl != null)
                            {
                                ReplaceItem_WithForm(input_lvl, output_lvl, files_);
                            }
                            else
                                Console.WriteLine("Error! Must specify input file");
                            break;
                        case OperationMode.Rename:
                            if (input_lvl == null)
                            {
                                Console.WriteLine("Rename Error: no input file specified");
                                return;
                            }
                            if (old_name != null && new_name != null && old_name.Length == new_name.Length)
                            {
                                UcfbHelper helper = new UcfbHelper();
                                helper.FileName = input_lvl;
                                helper.InitializeRead();
                                Chunk c = null;
                                while ((c = helper.RipChunk(false)) != null)
                                {
                                    if (c.Name == old_name)
                                        break;
                                }
                                if (c != null)
                                {
                                    int start = (int)BinUtils.GetLocationOfGivenBytes(c.Start, ASCIIEncoding.ASCII.GetBytes(c.Name), helper.Data, 80);
                                    for (int i = 0; i < c.Name.Length; i++)
                                    {
                                        helper.Data[start + i] = (byte)new_name[i];
                                    }
                                    helper.SaveData(output_lvl);
                                    Console.WriteLine("Changed {0} to {1}, saved to {2}", old_name, new_name, output_lvl);
                                }
                                else
                                {
                                    Console.WriteLine("Coould not find '{0}'. No action taken", old_name);
                                }
                            }
                            else
                            {
                                Console.WriteLine("Rename Error: old name and new name must be specified and be equal length");
                            }
                            break;
                        case OperationMode.ListContents:
                            UcfbHelper listHelper = new UcfbHelper();
                            listHelper.FileName = input_lvl;
                            listHelper.InitializeRead();
                            Chunk cur = null;
                            while ((cur = listHelper.RipChunk(false)) != null)
                            {
                                Console.WriteLine("{0}.{1}", cur.Name, cur.Type);
                            }
                            break;
                        case OperationMode.MergeCore:
                            List<string> theFiles = new List<string>(Directory.GetFiles(
                                merge_strings_search_dir, "core.lvl", SearchOption.AllDirectories));

                            //english,spanigh,italian,french,german,japanese,uk_english 
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "english.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "spanish.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "italian.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "french.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "german.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "japanese.txt", SearchOption.AllDirectories));
                            theFiles.AddRange(Directory.GetFiles(merge_strings_search_dir, "uk_english.txt", SearchOption.AllDirectories));

                            CoreMerge.MergeLoc(input_lvl, output_lvl, theFiles);
                            break;
                        case OperationMode.PS2_BF1AddonScript:
                            // This code operates as expected; but currently the PS2 crashes when reading the generated .lvl .
                            string[] neededFiles = new String[] { 
                            ".\\bin\\luac_4.exe",
                            ".\\bin\\BF1_LevelPack.exe",
                            ".\\bin\\BF1_ScriptMunge.exe"
                        };
                            foreach (string nf in neededFiles)
                            {
                                if (!File.Exists(nf))
                                {
                                    MessageBox.Show("Required files for operation:\r\n" + String.Join("\r\n", neededFiles), "Error, missing required files");
                                    return;
                                }
                            }
                            StringBuilder builder = new StringBuilder();
                            builder.Append("print('PROCESS_ADDONS_PS2_BF1 start', 'debug')\r\n");

                            //string debugFile = "DEBUG\\DEBUG.lvl";
                            //if (File.Exists(debugFile))
                            //{
                            //    builder.Append("  ReadDataFile('addon\\debug\\debug.lvl'); \r\n");
                            //    builder.Append("  ScriptCB_DoFile('debug'); \r\n");
                            //}
                            string current = "";

                            for (int i = 0; i < 1000; i++)
                            {
                                current = string.Format("{0:D3}\\ADDON{0:D3}.lvl", i);
                                if (File.Exists(current))
                                {
                                    builder.Append(String.Format("  ReadDataFile('addon\\{0:D3}\\addon{0:D3}.lvl'); \r\n", i));
                                    builder.Append(String.Format("  ScriptCB_DoFile('addon{0:D3}'); \r\n", i));
                                }
                            }
                            builder.Append("print('PROCESS_ADDONS_PS2_BF1 end', 'debug')");

                            // copy to 'PROCESS_ADDONS_PS2_BF1.LUA'
                            String ps2LuaFileName = "PROCESS_ADDONS_PS2_BF1.LUA";

                            File.WriteAllText(Path.GetFullPath(ps2LuaFileName), builder.ToString());
                            // create the lvl with the 'CreateLvlForm'
                            CreateLvlForm lvlMaker = new CreateLvlForm();
                            lvlMaker.OverrideScriptMunge = Path.GetFullPath("bin\\BF1_ScriptMunge.exe");
                            lvlMaker.OverrideLevelPack = Path.GetFullPath("bin\\BF1_LevelPack.exe");
                            File.Copy("bin\\luac_4.exe", ".\\luac.exe", true); // ensure lua4 for BF1
                            lvlMaker.AddItem(ps2LuaFileName);
                            try
                            {
                                lvlMaker.CreateLvl(".\\", "PROCESS_ADDONS_PS2_BF1.LVL", "PS2", true);
                                RemoveLogFiles();
                            }
                            catch (Exception lvlExc)
                            {
                                Console.WriteLine("Error Creating PROCESS_ADDONS_PS2_BF1.LVL!" + lvlExc.Message);
                            }
                            finally
                            {
                                File.Delete(".\\luac.exe");
                                lvlMaker.Dispose();
                            }
                            break;
                    }

                }
                catch (Exception ex)
                {
                    Console.Error.WriteLine("Error! Check arguments!\n{0}", ex.Message);
                    return;
                }
            }
        }

        private static void ReplaceItem_WithForm(string inputLvlName,string outputLvlName, string[] files_)
        {
            MainForm form = new MainForm();// lame to use the form, but it works.
            form.SetCodeDisplayStyle("summary");
            int count = form.SetLvlFile(inputLvlName);
            if (count < 1)
            {
                Console.WriteLine("Error! nothing found in input file!");
                return;
            }
            if (form.ReplaceItem(files_[0], platform))
                Console.WriteLine("Replaced Item with " + files_[0]);
            else
            {
                Console.WriteLine("Error during file replacement, output not saved");
                return;
            }
            form.SaveLvl(outputLvlName);
            //Console.WriteLine("Saved to: " + outputLvlName);
        }

        private static void RemoveLogFiles()
        {
            string[] logFiles = Directory.GetFiles(".", "*.log", SearchOption.TopDirectoryOnly);
            foreach (string log in logFiles)
            {
                File.Delete(log);
            }
        }

        public static Chunk FindChunk(string lvlFile, string name)
        {
            Chunk cur = null;
            UcfbHelper helper = new UcfbHelper();
            helper.FileName = lvlFile;
            helper.InitializeRead();
            while ((cur = helper.RipChunk(false)) != null)
            {
                if (cur.Name == name)
                    return cur;
            }
            return null;
        }

        private static string GetChunkName(string fileName)
        {
            int start = fileName.LastIndexOf(Path.DirectorySeparatorChar) + 1;
            int end = fileName.LastIndexOf('.');
            if (end < start)
                end = fileName.Length - 1;
            string retVal = fileName.Substring(start, end - start);
            return retVal;
        }

        private static string[] GetFiles(string files)
        {
            string[] retVal = null;
            if (files != null)
            {
                retVal = files.Split(";".ToCharArray());
            }
            return retVal;
        }

        private static void ProcessArgs(string[] args)
        {
            int i = 0;
            for (; i < args.Length; i++)
            {
                switch (args[i].ToLower())
                {
                    case "-file":
                        input_lvl = args[i + 1];
                        i++;
                        break;
                    case "-o":
                        output_lvl = args[i + 1]; 
                        i++;
                        break;
                    case "-r":
                        operation = OperationMode.Replace;
                        files = args[i + 1];
                        i++;
                        break;
                    case "-rename":
                        operation = OperationMode.Rename;
                        old_name = args[i + 1];
                        new_name = args[i + 2];
                        i += 2;
                        break;
                    case "-a":
                        operation = OperationMode.Add;
                        files = args[i + 1];
                        i++;
                        break;
                    case "-e":
                        operation = OperationMode.Extract;
                        break;
                    case "-h":
                    case "--help":
                    case "/h":
                    case "/?":
                        operation = OperationMode.ShowHelp;
                        break;
                    case "-p":
                        string p = args[i + 1].ToUpper();
                        platform = (Platform) Enum.Parse(typeof(Platform), p);
                        i++;
                        break;
                    case "-mod_tools_folder":
                        ModToolsDir = args[i + 1];
                        i++;
                        break;
                    case "-l":
                        operation = OperationMode.ListContents;
                        break;
                    case "-verbose":
                        Verbose = true;
                        break;
                    case "-merge_strings":
                        operation = OperationMode.MergeCore;
                        merge_strings_search_dir = args[i + 1];
                        i++;
                        break;
                    case "-ps2_bf1_addon_lvl":
                        operation = OperationMode.PS2_BF1AddonScript;
                        i++;
                        break;
                }
            }
        }

        public static void PrintHelp()
        {
            string help = "LVLTool.exe  Version: "+Version +"\n"+
@"Use without arguments for the GUI.
Use at the command line with the following arguments:
 -file <lvl file>  The file to operate on
 -r <munged or mungable file(s)>  to replace munged files in the target .lvl file.
 -a <munged or mungable file(s)>  to add munged files at the end of the target .lvl file.
 -o <lvl file>  The output file name. (default is input lvl file)
 -e             Extract contents
 -rename  <old_name> <new_name>  Rename a UCFB chunk (names must be same size).
 -p <platform> (pc|xbox|ps2) only important if specifying .tga files (pc=default)
 -l List the contents of the munged/lvl file.
 -mod_tools <mod_tools_folder_no_spaces> The modtools folder (needed for .lua & .tga files, default is 'C:\BF2_ModTools')
 -merge_strings <search_dir> Read in the '-file' arg, add the strings found in the 'core.lvl's 
               under 'search_dir' save to file specified with the '-o' arg.
 " + //"-ps2_bf1_addon_lvl  Create a PS2 Battlefront alt addon (workaround) lvl for PS2 BF1 (PS2 BF1 only)
@"
 -verbose       Display more info as operations are occuring.
 -h --help  /? Print help message

Examples:
    (Replace shell_interface script inside shell.lvl)
LVLTool.exe -file shell.lvl -r shell_interface.lua 
     or 
LVLTool.exe -file shell.lvl -r shell_interface.script
    or 
LVLTool.exe -file shell.lvl -r shell_interface.lua -mod_tools_folder C:\BFBuilder
 
    (extract all (munged) contents from shell.lvl)
LVLTool.exe -file shell.lvl -e
 
    (Add XXXg_con and XXXc_con to the end of mission.lvl)
LVLTool.exe -file mission.lvl -a XXXg_con.script;XXXc_con.script

    (Replace mode_icon_con in xbox shell.lvl )
LVLTool.exe -file shell.lvl -r mode_icon_con.tga -p xbox  

    (add the strings from core.lvl files under 'top_folder_with_cores', to 'base.core.lvl', save as 'core.lvl' )
LVLTool.exe -file base.core.lvl -o core.lvl -merge_strings top_folder_with_cores

";
            Console.WriteLine(help);
        }

        public static string Luac
        {
            get
            {
                string retVal = FindLuac(ModToolsDir);
                return retVal;
            }
        }

        private static String FindLuac(string startDir)
        {
            string retVal = startDir + "ToolsFL\\bin\\Luac.exe";
            if (!File.Exists(retVal))
            {
                string test = startDir + "..\\" + "ToolsFL\\bin\\Luac.exe";
                if (File.Exists(test))
                    retVal = test;
            }
            return retVal;
        }

        public static bool COMMAND_LINE = false;

        public static string settignsFile = ".\\.lvlToolSettings";

        public static void MessageUser(string message)
        {
            if (COMMAND_LINE)
                Console.WriteLine(message);
            else
                MessageBox.Show(message);
        }

        public static string RunCommand(string programName, string args, bool includeStdErr)
        {
            Console.WriteLine("Running command: " + programName + " " + args);
            ProcessStartInfo processStartInfo = new ProcessStartInfo
            {
                FileName = programName,
                Arguments = args,
                RedirectStandardOutput = true,
                RedirectStandardError = true,
                UseShellExecute = false,
                WindowStyle = ProcessWindowStyle.Hidden
            };
            
            FileInfo progToRun = new FileInfo(programName);
            if (!progToRun.Exists)
            {
                Console.WriteLine("Could not find '{0}'",  programName);
            }

            if (progToRun.Exists) // Make PATH varialbe awesome
            {
                string newPath = progToRun.DirectoryName + ";" + processStartInfo.EnvironmentVariables["path"];
                processStartInfo.EnvironmentVariables["path"] = newPath;
            }
            var process = Process.Start(processStartInfo);
            string output = process.StandardOutput.ReadToEnd();
            string err = process.StandardError.ReadToEnd();
            process.WaitForExit();
            if (includeStdErr)
                output = output + "\r\n" + err;
            return output;
        }

        private static Regex ModToolsRegex = new Regex("ModToolsDir:(.*)");

        public static void SaveSettings()
        {
            string contents = string.Format("ModToolsDir:{0}\n", ModToolsDir);
            File.WriteAllText(settignsFile, contents);
        }

        public static void LoadSettings()
        {
            if (File.Exists(settignsFile))
            {
                string contents = File.ReadAllText(settignsFile);
                Match m = ModToolsRegex.Match(contents);
                if (m != Match.Empty)
                {
                    ModToolsDir = m.Groups[1].ToString();
                }
            }
        }
    }

    public enum Platform
    {
        None,
        PC,
        XBOX,
        PS2
    }

    public enum OperationMode
    {
        Extract,
        Add,
        Replace,
        Rename,
        ListContents,
        ShowHelp,
        MergeCore,
        PS2_BF1AddonScript
    }

}