bloxstrap/Bloxstrap/Integrations/RbxFpsUnlocker.cs

using System;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Threading.Tasks;
using Bloxstrap.Helpers;

using Bloxstrap.Models;

namespace Bloxstrap.Integrations
{
    internal class RbxFpsUnlocker
    {
        public const string ApplicationName = "rbxfpsunlocker";
        public const string ProjectRepository = "axstin/rbxfpsunlocker";

        // default settings but with QuickStart set to true and CheckForUpdates set to false
        private static readonly string Settings =
            "UnlockClient=true\n" +
            "UnlockStudio=false\n" +
            "FPSCapValues=[30.000000, 60.000000, 75.000000, 120.000000, 144.000000, 165.000000, 240.000000, 360.000000]\n" +
            "FPSCapSelection=0\n" +
            "FPSCap=0.000000\n" +
            "CheckForUpdates=false\n" +
            "NonBlockingErrors=true\n" +
            "SilentErrors=false\n" +
            "QuickStart=true\n";

        public static void CheckIfRunning()
        {
            Process[] processes = Process.GetProcessesByName(ApplicationName);

            if (processes.Length == 0)
                return;

            App.Logger.WriteLine("[RbxFpsUnlocker::CheckIfRunning] Closing currently running rbxfpsunlocker processes...");

            try
            {
                foreach (Process process in processes)
                {
                    if (process.MainModule?.FileName is null)
                        continue;

                    if (!process.MainModule.FileName.Contains(Directories.Base))
                        continue;

                    process.Kill();
                    process.Close();
                }
            }
            catch (Exception ex)
            {
                App.Logger.WriteLine($"[RbxFpsUnlocker::CheckIfRunning] Could not close rbxfpsunlocker process! {ex}");
            }
        }

        public static async Task CheckInstall()
        {
            string folderLocation = Path.Combine(Directories.Base, "Integrations\\rbxfpsunlocker");
            string fileLocation = Path.Combine(folderLocation, "rbxfpsunlocker.exe");
            string settingsLocation = Path.Combine(folderLocation, "settings");

            if (!App.Settings.Prop.RFUEnabled)
            {
                // don't delete rbxfpsunlocker if rbxfpsunlocker and roblox is currently running
                if (Utilities.CheckIfProcessRunning(ApplicationName) && Utilities.CheckIfRobloxRunning())
                    return;

                App.State.Prop.RbxFpsUnlockerVersion = "";
                App.State.Save();

                if (Directory.Exists(folderLocation))
                {
                    CheckIfRunning();
                    Directory.Delete(folderLocation, true);
                }

                return;
            }

            var releaseInfo = await Utilities.GetJson<GithubRelease>($"https://api.github.com/repos/{ProjectRepository}/releases/latest");

            if (releaseInfo is null || releaseInfo.Assets is null)
                return;

            string downloadUrl = releaseInfo.Assets[0].BrowserDownloadUrl;

            DirectoryInfo directory = new(folderLocation);
            directory.Create();
            // i have no idea how the read only flag enables itself but apparently it just does
            directory.Attributes &= ~FileAttributes.ReadOnly;

            if (File.Exists(fileLocation))
            {
                // no new release published, return
                if (App.State.Prop.RbxFpsUnlockerVersion == releaseInfo.TagName)
                    return;

                CheckIfRunning();
                File.Delete(fileLocation);
            }

            App.Logger.WriteLine("[RbxFpsUnlocker::CheckInstall] Installing/Updating rbxfpsunlocker...");

            {
                byte[] bytes = await App.HttpClient.GetByteArrayAsync(downloadUrl);

                using MemoryStream zipStream = new(bytes);
                using ZipArchive archive = new(zipStream);

                archive.ExtractToDirectory(folderLocation, true);
            }

            if (!File.Exists(settingsLocation))
                await File.WriteAllTextAsync(settingsLocation, Settings);

            App.State.Prop.RbxFpsUnlockerVersion = releaseInfo.TagName;
            App.State.Save();
        }
    }
}