TBMM/GCMM/MainUtils.cs

using GCMM.Properties;
using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Net;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace GCMM
{
    partial class MainForm
    {
        public void UpdateButton(Button button, bool enabled)
        {
            if (enabled)
            {
                button.ForeColor = Color.Lime;
                button.FlatAppearance.MouseOverBackColor = Color.FromArgb(0, 40, 0);
                button.FlatAppearance.MouseDownBackColor = Color.Green;
            }
            else
            {
                button.ForeColor = Color.Green;
                button.FlatAppearance.MouseOverBackColor = Color.Black;
                button.FlatAppearance.MouseDownBackColor = Color.Black;
            }
        }

        public string GetGameFolder()
        { //TODO
            string libs;
            if (Environment.OSVersion.Platform == PlatformID.Win32NT)
                libs = Settings.Default.SteamUserID + @"\steamapps\libraryfolders.vdf"; //TODO: Not the Steam folder anymore!
            else
                return null;
            foreach (var line in File.ReadAllLines(libs).Concat(new[] {@"C:\Program Files (x86)\Steam\"}))
            {
                var regex = new Regex("\\t\"\\d+\"\\t\\t\"(.+)\"");
                var match = regex.Match(line);
                if (!match.Success)
                    continue;
                string library = match.Groups[1].Value.Replace("\\\\", "\\");
                library += @"\steamapps\common\";
                if (GetExe(library + "Gamecraft") != null)
                    return library + "Gamecraft";
                if (GetExe(library + "RobocraftX") != null)
                    return library + "RobocraftX";
            }
            return null;
        }

        public string SelectGameFolder()
        {
            var ofd = new OpenFileDialog();
            ofd.Filter = "Gamecraft executable|Gamecraft.exe|Gamecraft Preview executable|GamecraftPreview.exe";
            ofd.Title = "Game location";
            ofd.InitialDirectory = @"C:\Program Files (x86)\Steam\steamapps\common\"; //TODO
            ofd.CheckFileExists = true;
            ofd.ShowDialog();
            if (string.IsNullOrWhiteSpace(ofd.FileName))
                return null;
            return Directory.GetParent(ofd.FileName).FullName;
        }

        public (string, int) AskForSteamLogin()
        {
            while (MessageBox.Show("Couid not find your Steam configuration to set launch options.\n\n" +
                "Please make sure you are logged into Steam and click Retry or click Cancel to skip setting this up.",
                "Steam config not found", MessageBoxButtons.RetryCancel) != DialogResult.Cancel)
            {
                var ret = GetSteamLocationAndUser();
                if (ret != (null, 0))
                    return ret;
            }
            return (null, 0);
        }

        private bool UpdateOrGetSteamConfigToAutoStart(bool? autoLaunch)
        {
            var regex = new Regex(@"(\t{6}""LaunchOptions""\t+"")(.*)("")");
            string path = steamPath + @"\userdata\" + Settings.Default.SteamUserID + @"\config\localconfig.vdf";
            var lines = File.ReadAllLines(path);
            bool shouldMatch = false;
            bool ret = false;
            for (int i = 0; i < lines.Length; i++)
            {
                if (lines[i] == "\t\t\t\t\t\"1078000\"")
                    shouldMatch = true; //Found the game
                else if(shouldMatch)
                {
                    var match = regex.Match(lines[i]);
                    if (!match.Success)
                        continue;
                    if (autoLaunch.HasValue)
                    {
                        if (autoLaunch.Value)
                            lines[i] = match.Groups[1].Value + Application.ExecutablePath + " -start %command%" + match.Groups[3].Value;
                        else
                            lines[i] = match.Groups[1].Value + match.Groups[2].Value;
                        File.WriteAllLines(path, lines);
                    }
                    ret = match.Groups[2].Value.Contains("GCMM.exe");
                    break;
                }
            }
            return ret;
        }

        private (string, int) GetSteamLocationAndUser()
        {
            if (Environment.OSVersion.Platform != PlatformID.Win32NT) return (null, 0);
            using (var key = Registry.CurrentUser.OpenSubKey(@"Software\Valve\Steam\ActiveProcess"))
            {
                string path = (string)key?.GetValue("SteamClientDll");
                path = path != null ? Directory.GetParent(path).FullName : null;
                return (path, (int)(key.GetValue("ActiveUser") ?? 0));
            }
        }

        private void DetectConfigLocationAndAutoStart(string steamPath, ref int user)
        {
            string path = steamPath + @"\userdata";
            if (user == 0)
            {
                var dirs = Directory.GetDirectories(path);
                var goodPaths = (from dir in dirs
                                 where File.Exists(dir + @"\config\localconfig.vdf")
                                 select dir).ToArray();
                if (goodPaths.Length != 1)
                {
                    (_, user) = AskForSteamLogin();
                    path += user;
                }
                else
                {
                    path = goodPaths[0];
                    user = int.Parse(Path.GetFileName(path));
                }
            }
            else
                path += user;
            path += @"\config\localconfig.vdf";
            if (path != null && user != 0 && File.Exists(path))
            {
                Settings.Default.SteamUserID = user;
                UpdateOrGetSteamConfigToAutoStart(true);
            }
            else
                Settings.Default.AutoLaunch = false;
        }

        private void CheckStartGame(object sender, EventArgs e)
        {
            Action act = () =>
            {
                if (((sender as Process)?.ExitCode ?? 0) != 0)
                {
                    status.Text = "Status: Patching failed";
                    return;
                }
                if (CheckIfPatched() == GameState.Patched || unpatched.Checked)
                    if (sender is string command)
                        Process.Start(command);
                    else if (Environment.OSVersion.Platform == PlatformID.Win32NT)
                        Process.Start("steam://run/1078000/");
                    else
                        Process.Start("xdg-open", "steam://run/1078000/");
            };
            if (InvokeRequired)
                Invoke(act);
            else
                act();
            EndWork(false);
        }

        public WebClient GetClient()
        {
            var client = new WebClient();
            if (!Settings.Default.UseProxy)
                client.Proxy = null;
            client.Headers.Clear();
            client.Headers[HttpRequestHeader.Accept] = "application/json";
            client.BaseAddress = "https://git.exmods.org";
            return client;
        }

        private bool working = false;
        /// <summary>
        /// Some simple "locking", only allow one operation at a time
        /// </summary>
        /// <returns>Whether the work can begin</returns>
        public bool BeginWork()
        {
            if (working) return false;
            working = true;
            UpdateButton(playbtn, false);
            UpdateButton(installbtn, false);
            UpdateButton(uninstallbtn, false);
            UpdateButton(settingsbtn, false);
            unpatched.Enabled = false;
            return true;
        }

        public void EndWork(bool desc = true)
        {
            working = false;
            UpdateButton(playbtn, true);
            UpdateButton(settingsbtn, true);
            if (desc)
                modlist_SelectedIndexChanged(modlist, null);
            unpatched.Enabled = true;
        }

        /// <summary>
        /// Path must start with \
        /// </summary>
        /// <param name="path"></param>
        /// <param name="gamepath"></param>
        /// <returns></returns>
        public string GamePath(string path, string gamepath = null)
        {
            return ((gamepath ?? Settings.Default.GamePath) + path).Replace('\\', Path.DirectorySeparatorChar);
        }

        public string GetExe(string path = null)
        {
            if (File.Exists(GamePath("\\Gamecraft.exe", path)))
                return "Gamecraft.exe";
            if (File.Exists(GamePath("\\GamecraftPreview.exe", path)))
                return "GamecraftPreview.exe";
            return null;
        }

        private bool CheckNoExe()
        {
            return CheckNoExe(out _);
        }

        private bool CheckNoExe(out string path)
        {
            path = GetExe();
            if (path == null)
            {
                MessageBox.Show("Gamecraft not found! Set the correct path in Settings.");
                return true;
            }
            return false;
        }

        public async Task<DateTime> GetLastGameUpdateTime()
        {
            /*using (var client = GetClient())
            {
                string html = await client.DownloadStringTaskAsync("https://api.steamcmd.net/v1/info/1078000");
                var regex = new Regex("<i>timeupdated:</i>[^<]*<b>([^<]*)</b>");
                var match = regex.Match(html);
                if (!match.Success)
                    return default;
                return new DateTime(1970, 1, 1).AddSeconds(long.Parse(match.Groups[1].Value));
            }*/
            //return new DateTime(2020, 12, 28);
            return default;
        }
    }
}