Example #1
0
        static void Main(string[] args)
        {
            // Set Title.
            Console.Title = "Script Junkie";

            // Add Services.
            ServiceManager.Services.Add(new LogService());
            ServiceManager.Services.Add(new ArgumentService());

            // Debug Arguments
            // Arguments passed through args variable over power below arguments.
#if (DEBUG)
            // Example of programatically added arguments.
            //ServiceManager.Services.ArgumentService.AddArgument(ArgumentService.XmlTemplatePath, Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments) + "\\ScriptJunkie\\Template.xml");
            //ServiceManager.Services.ArgumentService.AddArgument(ArgumentService.XmlPath, Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments) + "\\ScriptJunkie\\Template.xml");
#endif

            // Start Script Junkie.
            Version version = Assembly.GetEntryAssembly().GetName().Version;
            ServiceManager.Services.LogService.WriteHeader("Starting Script Junkie v({0})", version.ToString());
            ServiceManager.Services.LogService.WriteLine("If executable of script needs to be ran as administrator, run ScriptJunkie as admin.", ConsoleColor.Yellow);
            // Check if debug mode.
            if (ServiceManager.Services.ArgumentService.IsDebug)
            {
                ServiceManager.Services.LogService.WriteSubHeader("Paused for debug mode. Attach to debugger then press any key to continue.");
                Console.ReadKey(true);
            }

            // Initialization.
            Setup setup    = new Setup();
            int   exitCode = -1;

            // Display Arguments
            ServiceManager.Services.LogService.WriteSubHeader("Arguments");
            foreach (Argument arg in ServiceManager.Services.ArgumentService.Arguments)
            {
                if (string.IsNullOrEmpty(arg.Value))
                {
                    ServiceManager.Services.LogService.WriteLine("\"{0}\" = \"True\"", arg.Key, arg.Value);
                }
                else
                {
                    ServiceManager.Services.LogService.WriteLine("\"{0}\" = \"{1}\"", arg.Key, arg.Value);
                }
            }

            // Checks if it should generate an example xml template.
            Argument xmlTemplatePath;
            if (ServiceManager.Services.ArgumentService.TryGetArgument(ArgumentService.XmlTemplatePath, out xmlTemplatePath))
            {
                setup.GenerateXmlTemplate(xmlTemplatePath.Value);
            }

            // Checks if it should run setup execution.
            Argument xmlPath;
            if (ServiceManager.Services.ArgumentService.TryGetArgument(ArgumentService.XmlPath, out xmlPath))
            {
                if (setup.Initalize(xmlPath.Value))
                {
                    exitCode = setup.Execute();
                }
            }
            else
            {
                ServiceManager.Services.LogService.WriteSubHeader("Script Junkie needs an xml file to execute.", ConsoleColor.Red);
            }

            // End Script Junkie.
            ServiceManager.Services.LogService.WriteHeader("Script Junkie has completed.");
#if (RELEASE)
            Environment.Exit(exitCode);
#elif (DEBUG)
            Console.ReadLine();
#endif
        }
Example #2
0
        /// <summary>
        /// Generates a template xml to base custom scripts off of.
        /// </summary>
        /// <param name="savePath">The location the template will be saved.</param>
        public void GenerateXmlTemplate(string savePath)
        {
            ServiceManager.Services.LogService.WriteHeader("Generating Xml Template...");

            #region Script 1
            Script script = new Script();
            script.Name        = "Script 1";
            script.Description = "Does nothing";

            Executable executable = new Executable();
            executable.Path = "C:/Temp/nothing.ps1";

            ArgumentCollection argCollection = new ArgumentCollection();
            argCollection.Add(new Argument()
            {
                Key = "-i", Value = "C:/Temp/something.bin"
            });
            argCollection.Add(new Argument()
            {
                Key = "-x", Value = ""
            });

            ExitCodeCollection exitCollection = new ExitCodeCollection();
            exitCollection.Add(new ExitCode()
            {
                Value = 0, Message = "Files deleted", IsSuccess = true
            });
            exitCollection.Add(new ExitCode()
            {
                Value = 1, Message = "Files failed to delete"
            });
            exitCollection.Add(new ExitCode()
            {
                Value = 2, Message = "Couldn't find any files"
            });

            // Add all elements into the single script file.
            script.Arguments  = argCollection;
            script.ExitCodes  = exitCollection;
            script.Executable = executable;
            #endregion

            #region Script 2
            Script script2 = new Script();
            script2.Name        = "Script 2";
            script2.Description = "Does nothing";

            Executable executable2 = new Executable();
            executable2.Path = "C:/Temp/Downloads/Extracted/SomethingThatWasInAZip.exe";

            ArgumentCollection argCollection2 = new ArgumentCollection();
            argCollection2.Add(new Argument()
            {
                Key = "-install"
            });
            argCollection2.Add(new Argument()
            {
                Key = "-silent"
            });

            ExitCodeCollection exitCollection2 = new ExitCodeCollection();
            exitCollection2.Add(new ExitCode()
            {
                Value = 0, Message = "Script 2 has installed", IsSuccess = true
            });
            exitCollection2.Add(new ExitCode()
            {
                Value = 1, Message = "Failed to install."
            });
            exitCollection2.Add(new ExitCode()
            {
                Value = 2, Message = "Installed but limited."
            });

            // Add all elements into the single script file.
            script2.Arguments  = argCollection;
            script2.ExitCodes  = exitCollection;
            script2.Executable = executable;

            // Add the single script above into a collection of scripts.
            ScriptCollection scriptCollection = new ScriptCollection();
            scriptCollection.Add(script);
            scriptCollection.Add(script2);
            #endregion

            ServiceManager.Services.LogService.WriteLine("Generating Download Collection...");
            DownloadCollection downloadCollection = new DownloadCollection();
            downloadCollection.TimeOut     = 60;
            downloadCollection.RefreshRate = 10;
            downloadCollection.Add(new Download()
            {
                Name = "Nothing Powershell Script", Description = "This script does nothing", DownloadUrl = "www.blank.com/nothing.ps1", DestinationPath = "C:/Temp/Downloads/nothing.ps1"
            });
            downloadCollection.Add(new Download()
            {
                Name = "Test Zip File", Description = "This zip has nothing", DownloadUrl = "www.blank.com/nothing.zip", DestinationPath = "C:/Temp/Downloads", ExtractionPath = "C:/Temp/Downloads/Extracted"
            });

            // Add the 2 main elements, the scripts to run and the downloads to download.
            Setup setup = new Setup();
            setup.Scripts   = scriptCollection;
            setup.Downloads = downloadCollection;

            ServiceManager.Services.LogService.WriteLine("Saving file to: \"{0}\"", savePath);
            setup.Serialize(savePath);
        }