214 lines
7.2 KiB
C#
214 lines
7.2 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Xml.Serialization;
|
|
|
|
public class Program
|
|
{
|
|
static void Main(string[] args)
|
|
{
|
|
Installer installer = new Installer();
|
|
installer.GetUserInput();
|
|
}
|
|
}
|
|
|
|
public class Installer
|
|
{
|
|
public string programfolder;
|
|
public bool copyfiles;
|
|
public bool overwrite;
|
|
|
|
public Installer()
|
|
{
|
|
programfolder = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), "GeckoLoader");
|
|
copyfiles = true;
|
|
overwrite = true;
|
|
}
|
|
|
|
void CopyAll (DirectoryInfo source, DirectoryInfo destination, string wildcard, string[] exclude, int maxdepth = 16)
|
|
{
|
|
if (maxdepth <= 0) return;
|
|
|
|
DirectoryInfo[] subdirs = source.GetDirectories();
|
|
|
|
foreach (DirectoryInfo dirPath in source.EnumerateDirectories())
|
|
Directory.CreateDirectory(dirPath.FullName.Replace(source.FullName, destination.FullName));
|
|
|
|
foreach (FileInfo filePath in source.EnumerateFiles(wildcard))
|
|
{
|
|
bool include = true;
|
|
foreach (string exc in exclude)
|
|
{
|
|
if (exc.ToLower().Contains(filePath.Name.ToLower()))
|
|
{
|
|
include = false;
|
|
break;
|
|
}
|
|
}
|
|
if (include) File.Copy(filePath.FullName, filePath.FullName.Replace(source.FullName, destination.FullName), true);
|
|
}
|
|
|
|
foreach (DirectoryInfo dir in subdirs)
|
|
{
|
|
DirectoryInfo dest = new DirectoryInfo(Path.Combine(destination.FullName, dir.Name));
|
|
DirectoryInfo src = new DirectoryInfo(dir.FullName);
|
|
|
|
CopyAll(src, dest, wildcard, exclude, maxdepth - 1);
|
|
}
|
|
}
|
|
|
|
private static void ClearConsoleLine(int index, bool moveto)
|
|
{
|
|
int currentLineCursor = Console.CursorTop;
|
|
Console.SetCursorPosition(0, index);
|
|
Console.Write(new string(' ', Console.WindowWidth));
|
|
if (moveto) Console.SetCursorPosition(0, index);
|
|
else Console.SetCursorPosition(0, currentLineCursor);
|
|
}
|
|
|
|
private static string HandleConsoleQuestion(string msg, string[] options)
|
|
{
|
|
bool handled = false;
|
|
string input = String.Empty;
|
|
|
|
while (handled == false)
|
|
{
|
|
Console.Write("{0} ({1}): ", msg, String.Join("|", options));
|
|
|
|
input = Console.ReadLine();
|
|
if (options.Any(s => s.Contains(input.ToLower())))
|
|
{
|
|
handled = true;
|
|
}
|
|
else
|
|
{
|
|
ClearConsoleLine(Console.CursorTop - 1, true);
|
|
}
|
|
}
|
|
return input;
|
|
}
|
|
|
|
private void SetProgramFolder(string folderName)
|
|
{
|
|
this.programfolder = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), folderName);
|
|
}
|
|
|
|
private void SetProgramFolderToPath()
|
|
{
|
|
if (!Directory.Exists(this.programfolder)) Directory.CreateDirectory(this.programfolder);
|
|
|
|
var scope = EnvironmentVariableTarget.User;
|
|
var curPATH = Environment.GetEnvironmentVariable("PATH", scope);
|
|
|
|
if (!curPATH.Contains(this.programfolder))
|
|
{
|
|
var newValue = curPATH + ";" + this.programfolder;
|
|
Environment.SetEnvironmentVariable("PATH", newValue, scope);
|
|
}
|
|
}
|
|
|
|
private void RemoveProgramFolderFromPath()
|
|
{
|
|
var scope = EnvironmentVariableTarget.User;
|
|
var curPATH = Environment.GetEnvironmentVariable("PATH", scope);
|
|
|
|
if (curPATH.Contains(this.programfolder))
|
|
{
|
|
var newValue = curPATH.Replace(";" + this.programfolder, "");
|
|
Environment.SetEnvironmentVariable("PATH", newValue, scope);
|
|
}
|
|
}
|
|
|
|
private bool MoveFilesToprogramfolder(string wildcard, bool copy = true, bool overwrite = false)
|
|
{
|
|
DirectoryInfo cwd = new DirectoryInfo(Path.Combine(Directory.GetCurrentDirectory()));
|
|
DirectoryInfo programspace = new DirectoryInfo(this.programfolder);
|
|
|
|
bool validinstall = false;
|
|
|
|
try
|
|
{
|
|
foreach (FileInfo file in programspace.EnumerateFiles())
|
|
{
|
|
file.Delete();
|
|
}
|
|
foreach (DirectoryInfo dir in programspace.EnumerateDirectories())
|
|
{
|
|
dir.Delete(true);
|
|
}
|
|
|
|
//Copy top level files
|
|
foreach (FileInfo file in cwd.EnumerateFiles(wildcard, SearchOption.TopDirectoryOnly))
|
|
{
|
|
string ext = ".exe.py.bin.dll";
|
|
|
|
if (ext.Contains(file.Extension.ToLower()))
|
|
{
|
|
if (file.Extension.ToLower() == ".exe" && file.Name != "GeckoLoader.exe") continue;
|
|
if (file.Name.ToLower() == "geckoloader.py" || file.Name.ToLower() == "geckoloader.exe") validinstall = true;
|
|
file.CopyTo(Path.Combine(programspace.FullName, file.Name), true);
|
|
}
|
|
}
|
|
//Copy dependancies
|
|
string[] exclude = { "installer.exe" };
|
|
this.CopyAll(cwd, programspace, wildcard, exclude);
|
|
}
|
|
catch (UnauthorizedAccessException e)
|
|
{
|
|
Console.WriteLine(String.Format("Insufficient privledges provided! {0}\nTry running with administrator privledges", e));
|
|
return false;
|
|
}
|
|
|
|
return validinstall;
|
|
}
|
|
|
|
private void DeleteProgramFolder()
|
|
{
|
|
if (Directory.Exists(this.programfolder)) Directory.Delete(this.programfolder, true);
|
|
}
|
|
|
|
public void GetUserInput()
|
|
{
|
|
string status;
|
|
string[] continueoptions = { "y", "n" };
|
|
string[] actionoptions = { "install", "uninstall" };
|
|
|
|
Console.SetWindowSize(84, 20);
|
|
Console.Title = "GeckoLoader Installer";
|
|
Console.WriteLine("This installer modifies the Windows User PATH variable\n");
|
|
|
|
status = HandleConsoleQuestion("Are you sure you want to continue?", continueoptions);
|
|
|
|
if (status.ToLower() == (string)continueoptions.GetValue(0))
|
|
{
|
|
this.SetProgramFolder("GeckoLoader");
|
|
|
|
status = HandleConsoleQuestion("What do you want to do?", actionoptions);
|
|
if (status.ToLower() == (string)actionoptions.GetValue(0))
|
|
{
|
|
this.SetProgramFolderToPath();
|
|
if (this.MoveFilesToprogramfolder("*", this.copyfiles, this.overwrite) == false)
|
|
{
|
|
Console.WriteLine("Failed to install :( Is Geckoloader and its dependancies in the same directory?");
|
|
}
|
|
else
|
|
{
|
|
Console.WriteLine("Finished installation successfully! You can run GeckoLoader from anywhere\nby simply calling \"GeckoLoader <dol> <gct|txt|folder> [options]\"");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
this.RemoveProgramFolderFromPath();
|
|
this.DeleteProgramFolder();
|
|
Console.WriteLine("Uninstalled successfully!");
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
Console.WriteLine("That's okay! You can always run this program again when you feel ready.");
|
|
}
|
|
Console.Write("Press any key to exit . . . ");
|
|
Console.ReadKey();
|
|
}
|
|
}
|