MCGalaxy/GUI/Popups/CustomCommands.cs

/*
Copyright 2010 MCSharp team (Modified for use with MCZall/MCLawl/MCGalaxy)
Dual-licensed under the Educational Community License, Version 2.0 and
the GNU General Public License, Version 3 (the "Licenses"); you may
not use this file except in compliance with the Licenses. You may
obtain a copy of the Licenses at
http://www.opensource.org/licenses/ecl2.php
http://www.gnu.org/licenses/gpl-3.0.html
Unless required by applicable law or agreed to in writing,
software distributed under the Licenses are distributed on an "AS IS"
BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
or implied. See the Licenses for the specific language governing
permissions and limitations under the Licenses.
 */
using System;
using System.Collections.Generic;
using System.CodeDom.Compiler;
using System.IO;
using System.Reflection;
using System.Windows.Forms;
using MCGalaxy.Commands;
using MCGalaxy.Scripting;

namespace MCGalaxy.Gui.Popups {
    public partial class CustomCommands : Form {
        
        public CustomCommands() {
            InitializeComponent();

            //Sigh. I wish there were SOME event to help me.
            foreach (Command cmd in Command.allCmds) {
                if (!Command.IsCore(cmd)) lstCommands.Items.Add(cmd.name);
            }
        }
        
        void btnCreate_Click(object sender, EventArgs e) {
            string cmdName = txtCmdName.Text.Trim();
            if (cmdName.Length == 0) {
                Popup.Warning("Command must have a name"); return;
            }
            
            IScripting engine = radVB.Checked ? IScripting.VB : IScripting.CS;
            string path = engine.SourcePath(cmdName);
            if (File.Exists(path)) {
                Popup.Warning("Command already exists"); return;
            }
            
            try {
                engine.CreateNew(path, cmdName);
            } catch (Exception ex) {
                Logger.LogError(ex);
                Popup.Error("Failed to generate command. Check error logs for more details.");
                return;
            }
            Popup.Message("Command Cmd" + cmdName + engine.Ext + " created.");
        }

        void btnLoad_Click(object sender, EventArgs e) {
            List<Command> commands = null;
            string fileName;
            using (FileDialog dialog = new OpenFileDialog()) {
                dialog.RestoreDirectory = true;
                dialog.Filter = "Accepted File Types (*.cs, *.vb, *.dll)|*.cs;*.vb;*.dll|C# Source (*.cs)|*.cs|Visual Basic Source (*.vb)|*.vb|.NET Assemblies (*.dll)|*.dll";
                if (dialog.ShowDialog() != DialogResult.OK) return;
                fileName = dialog.FileName;
            }
            
            if (fileName.CaselessEnds(".dll")) {
                byte[] data = File.ReadAllBytes(fileName);
                Assembly lib = Assembly.Load(data);
                commands = IScripting.LoadTypes<Command>(lib);
            } else {
                IScripting engine = fileName.CaselessEnds(".cs") ? IScripting.CS : IScripting.VB;
                if (!File.Exists(fileName)) return;
                
                CompilerParameters args = new CompilerParameters();
                args.GenerateInMemory = true;
                CompilerResults result = engine.CompileSource(File.ReadAllText(fileName), args);

                if (result.Errors.HasErrors) {
                    foreach (CompilerError err in result.Errors) {
                        Logger.Log(LogType.Warning, "Error #" + err.ErrorNumber);
                        Logger.Log(LogType.Warning, "Message: " + err.ErrorText);
                        Logger.Log(LogType.Warning, "Line: " + err.Line);
                        Logger.Log(LogType.Warning, "=================================");
                    }
                    Popup.Error("Error compiling from source. Check logs for more details.");
                    return;
                }
                commands = IScripting.LoadTypes<Command>(result.CompiledAssembly);
            }

            if (commands == null) { 
                Popup.Error("Error compiling files. Check logs for more details"); return; 
            }
            for (int i = 0; i < commands.Count; i++) {
                Command cmd = commands[i];

                if (lstCommands.Items.Contains(cmd.name)) {
                    Popup.Warning("Command " + cmd.name + " already exists, so was not loaded");
                    continue;
                }

                lstCommands.Items.Add(cmd.name);
                Command.Register(cmd);
                Logger.Log(LogType.SystemActivity, "Added " + cmd.name + " to commands");
            }
        }

        void btnUnload_Click(object sender, EventArgs e) {
            string cmdName = lstCommands.SelectedItem.ToString();
            Command cmd = Command.Find(cmdName);
            if (cmd == null) {
                Popup.Warning("Command " + cmdName + " is not loaded."); return;
            }

            lstCommands.Items.Remove(cmd.name);
            Command.Unregister(cmd);
            Popup.Message("Command successfully unloaded.");
        }
        
        void lstCommands_SelectedIndexChanged(object sender, EventArgs e) {
            btnUnload.Enabled = lstCommands.SelectedIndex != -1;
        }
    }
}