Ejemplo n.º 1
0
        public async Task <bool> UpdateMods(List <BaseMod> mods, bool forceNoBackup = false)
        {
            this.Updating = true;
            UpdaterForm.SetStatus($"Starting SteamCMD...");

            int    statusCount = 0;
            double totalSize   = 0;

            mods.ForEach(x => totalSize += x.Details?.file_size ?? 0);

            double totalSizeTemp = totalSize;

            UpdaterForm.SetProgress(0, ProgressBarStyle.Blocks);
            UpdaterForm.SetProgressBounds(mods.Count);

            var downloader = new Downloader();

            downloader.ModDownloaded += delegate(object sender, BaseMod mod, string folder)
            {
                statusCount++;
                totalSizeTemp -= mod.Details.file_size;

                UpdaterForm.SetStatus($"Downloaded {statusCount}/{mods.Count} mods ({Math.Round(totalSizeTemp / 1024d / 1024d, 2)} MB)");
                UpdaterForm.UpdateProgress();
            };

            downloader.SteamStarted += sender =>
            {
                UpdaterForm.SetStatus($"Downloaded {statusCount}/{mods.Count} mods ({Math.Round(totalSizeTemp / 1024d / 1024d, 2)} MB)");
            };

            bool ok = await downloader.DownloadMods(mods);

            if (!ok)
            {
                this.Updating = false;
                MessageBox.Show("An error occurred while downloading workshop mods; Check the log file.", "Uh oh", MessageBoxButtons.OK, MessageBoxIcon.Error);

                return(false);
            }

            Log.Information("Updating mod files...");
            UpdaterForm.SetStatus("Updating mod files...");
            string modsPath            = Path.Combine(ModsPath);
            string toolPath            = Path.GetFullPath("./");
            string downloadContentPath = Path.Combine(toolPath, "steamcmd", "steamapps", "workshop", "content", "294100");

            UpdaterForm.SetProgress(0, ProgressBarStyle.Blocks);
            UpdaterForm.SetProgressBounds(mods.Count);

            foreach (var obj in mods)
            {
                if (!UpdateMod(obj, downloadContentPath, forceNoBackup))
                {
                    UpdaterForm.UpdateProgress();
                    continue;
                }

                UpdaterForm.UpdateProgress();
                obj.Updated = true;

                if (!forceNoBackup)
                {
                    UpdaterForm.UpdateObject(obj);
                }
            }

            UpdaterForm.SetStatus($"Finished updating {mods.Count} mods! A total of {Math.Round(totalSize / 1024d / 1024d, 2)} MB was downloaded.");
            Log.Information($"Finished updating {mods.Count} mods.");

            this.Updating = false;
            return(true);
        }
Ejemplo n.º 2
0
        private void UpdaterForm_Load(object s, EventArgs e)
        {
            _instance = this;

            this.Text = "Rimworld Mod Updater " + Utils.GetProgramVersion();

            // It's times like this when I really question myself
            new ToolTip().SetToolTip(btnManualUpdate, "Select a mod in your mods folder to update manually.\nWill replace files regardless of whether it's outdated or not.");
            new ToolTip().SetToolTip(btnReset, "Reset the updater to it's inital state, allowing you to query for outdated mods again.");
            new ToolTip().SetToolTip(btnOpenFileDialog, "Opens a file dialog for you to choose your rimworld installation folder.\nIf you put the updater in your rimworld folder, it should be set for you.");
            new ToolTip().SetToolTip(textGamePath, "The rimworld game path. Shit wont work if this isn't set.");
            new ToolTip().SetToolTip(btnViewDownloadMod, "Start downloading this mod/collection.\nIf it's a collection, it may take a while to query the collection & all of the mods in it depending on the size.");

            // Fixes an issue where context menu isn't opened on first try. Not completely sure why this fixes it.
            listMenu.Items.Add("");
            downloadedModMenu.Items.Add("");

            if (Directory.Exists("./Mods"))
            {
                // Talk about a cluttered rimworld folder if you had this piece of shit in the main folder.
                textGamePath.Text = Path.GetFullPath("./");
            }
            else if (Directory.Exists("../Mods"))
            {
                textGamePath.Text = Path.GetFullPath("../");
            }

            listObjectsParsed.ItemsChanged += delegate(object sender, ItemsChangedEventArgs args)
            {
                lblObjectsParsed.Text = $"Mods with updates available: {listObjectsParsed.GetItemCount()}";
            };

            textViewUrlBar.Text    = "Loading browser...";
            textViewUrlBar.Enabled = false;

            var clicker = new WorkshopJSClicker();

            clicker.Clicked += (id) =>
            {
                if (id == null)
                {
                    StartModDownload(null);
                }
                else
                {
                    StartModDownload("https://steamcommunity.com/sharedfiles/filedetails/?id=" + id);
                }
            };

            chromiumWebBrowser.JavascriptObjectRepository.Register("_____updater", clicker, false);
            if (File.Exists("steamHelper.js"))
            {
                chromiumWebBrowser.ExecuteScriptAsyncWhenPageLoaded(File.ReadAllText("steamHelper.js"), false);
            }
            chromiumWebBrowser.Load("https://steamcommunity.com/app/294100/workshop/");

            chromiumWebBrowser.LifeSpanHandler = new LifeSpanHandler();

            currentUpdater     = new Updater(textGamePath.Text);
            downloadTabManager = new DownloadTabManager(downloaderTabProgressBar, downloaderTabStatus, currentUpdater);

            Utils.CheckForUpdates();
        }
Ejemplo n.º 3
0
        public async Task QueryFiles()
        {
            var folders = Directory.GetDirectories(ModsPath);

            int numFolders = folders.Length;
            int count      = 0;

            if (folders.Length == 0)
            {
                Log.Error("Tried to query mod folders but found none.");
                MessageBox.Show("You have no mods installed!", "bruh", MessageBoxButtons.OK, MessageBoxIcon.Error);
                return;
            }

            UpdaterForm.SetProgressBounds(numFolders);
            UpdaterForm.SetProgress(0);
            UpdaterForm.SetStatus($"Querying {numFolders} mod folders for outdated mods...");

            Log.Information($"Querying {numFolders} mod folders for outdated mods...");

            List <BaseMod> mods = new List <BaseMod>();

            for (int i = 0; i < numFolders; i++)
            {
                string id = await QueryModId(folders[i]);

                if (!String.IsNullOrWhiteSpace(id))
                {
                    Log.Information("Got valid mod folder {0} ({1})", folders[i], id);

                    mods.Add(new BaseMod
                    {
                        ModPath = folders[i],
                        ModId   = id
                    });

                    count++;
                }

                UpdaterForm.UpdateProgress();
            }

            UpdaterForm.SetStatus($"Found {count}/{numFolders} valid (steam) mod folders.");
            Log.Information("Found {0}/{1} valid mod folders.", count, numFolders);

            int batchCount = 50;

            int num  = 0;
            int num2 = 0;
            int len  = mods.Count;

            UpdaterForm.SetStatus($"Querying {len} mods in {(int)Math.Ceiling((double)len / batchCount)} batches of {batchCount}");
            Log.Information("Querying {0} mods in {1} batches of {2}", len, (int)Math.Ceiling((double)len / batchCount), batchCount);

            UpdaterForm.ResetProgress();
            UpdaterForm.SetProgressBounds((int)Math.Ceiling((double)len / batchCount));

            bool retry = false;
            var  tasks = new List <Task>();

            for (int i = 0; i < len; i += batchCount)
            {
                var task = Task.Run(async() =>
                {
                    var batch = mods.GetRange(i, Math.Min(batchCount, len - i));

                    await QueryModBatch(batch);

                    num2 += batchCount;

                    Log.Information($"Queried {Math.Min(num2, len)}/{len} mods.");
                    UpdaterForm.SetStatus($"Queried {Math.Min(num2, len)}/{len} mods.");
                    UpdaterForm.UpdateProgress();
                });

                tasks.Add(task);

                await Task.Delay(250);
            }

            await Task.WhenAll(tasks);

            // Let's just send it all at once lol
            //var files = await SteamWorkshop.GetWorkshopFileDetails(fileIds.Keys.ToArray());

            Log.Information($"Got workshop file details for {mods.Count(x => x.Details != null)} mods.");


            UpdaterForm.SetProgressBounds(mods.Count);
            UpdaterForm.ResetProgress();

            int updateCount = 0;

            foreach (var mod in mods)
            {
                UpdaterForm.UpdateProgress();
                var    details    = mod.Details;
                string folderName = mod.Folder;

                if (details == null)
                {
                    Log.Error($"Couldn't get any file details for mod {mod.ModId} ({folderName}).");
                    continue;
                }

                string id = details.publishedfileid;
                if (details.result == 9)
                {
                    Log.Warning($"Tried to query workshop file {id} ({folderName}) but no file could be found. (Code 9). This could mean the mod has been removed/unlisted");
                    continue;
                }

                if (details.result != 1)
                {
                    Log.Warning($"Tried to query workshop file {id} ({folderName}) but steam returned code {details.result}");
                }

                if (details.visibility != 0)
                {
                    Log.Warning($"Got workshop file {id} ({folderName}) but it's a private file.");
                    continue;
                }

                if (details.banned == 1)
                {
                    Log.Warning($"Got workshop file {id} ({folderName}) but it's a banned file.");
                    continue;
                }

                if (details.creator_app_id != 294100)
                {
                    Log.Warning($"Got workshop file {id} ({folderName}) but it's not a rimworld mod! (Huh?)");
                    continue;
                }

                var modPath = mod.ModPath;

                var remoteDate      = details.TimeUpdated;
                var lastUpdatedDate = GetModLastUpdatedTime(modPath);

                //Log.Information("{0} < {1}: {2}", remoteDate.ToUnixTimeSeconds(), lastUpdatedDate.ToUnixTimeSeconds(), (remoteDate < lastUpdatedDate));
                if (remoteDate > lastUpdatedDate)
                {
                    updateCount++;
                    Log.Information($"Mod folder {folderName} ({details.publishedfileid}) has an update available.");

                    UpdaterForm.AddRow(mod);

                    await Task.Delay(1);
                }
            }

            Log.Information("There are {0} mods with updates available.", updateCount);
        }