1
0
Fork 0
This repository has been archived on 2024-02-06. You can view files and clone it, but cannot push or open issues or pull requests.
GeckoLoader/installer.cs
2020-08-25 07:06:56 -05:00

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();
}
}