initial commit

BlueberryUpdater/Program.cs

@@ -0,0 +1,195 @@
+using System.Diagnostics;
+using System.IO.Compression;
+using System.Reflection;
+
+namespace BlueberryUpdater
+{
+    internal class Program
+    {
+        static string appName = "Blueberry";
+        static string updaterName = appName + "Updater";
+        static string installPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ProgramFiles), appName);
+
+        static void Main(string[] args)
+        {
+            if (Directory.Exists(installPath) && File.Exists(Path.Combine(installPath, appName + ".exe")))
+                Uninstall();
+            else
+                Install();
+        }
+
+        private static void Uninstall()
+        {
+            Console.WriteLine("Would you like to uninstall Blueberry? [y/N]");
+            var key = Console.ReadLine();
+            if (key == null ||key.ToLower() != "y")
+                return;
+
+            var appdata = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), appName);
+
+            Console.WriteLine("Removing Blueberry...");
+            var dirs = Directory.GetDirectories(installPath);
+            var files = Directory.GetFiles(installPath);
+            var total = dirs.Length + files.Length;
+            var i = 0;
+
+            foreach (var dir in dirs)
+            {
+                i++;
+                Directory.Delete(dir, true);
+                DrawProgressBar((int)((double)i / total * 100), dir.Split('\\').Last());
+            }
+
+            foreach (var file in files)
+            {
+                i++;
+                if (file.Split('\\').Last() == updaterName + ".exe")
+                    continue;
+                File.Delete(file);
+                i++;
+                DrawProgressBar((int)((double)i / total * 100), file.Split('\\').Last());
+            }
+            Directory.Delete(appdata, true);
+
+            DrawProgressBar(100, "Done");
+            Console.WriteLine();
+            Console.WriteLine("Uninstall Complete!");
+
+            Console.WriteLine("Press any key to exit...");
+            Console.ReadKey();
+
+            SelfDelete();
+        }
+
+        private static void Install()
+        {
+            try
+            {
+                string resourceName = "BlueberryUpdater.AppPayload.zip"; // Format: Namespace.Filename
+
+                Console.WriteLine($"Installing to {installPath}...");
+
+                // 1. Clean existing install if necessary
+                if (Directory.Exists(installPath))
+                {
+                    Directory.Delete(installPath, true);
+                }
+                Directory.CreateDirectory(installPath);
+
+                // 2. Extract Embedded Resource
+                var assembly = Assembly.GetExecutingAssembly();
+
+                using (Stream stream = assembly.GetManifestResourceStream(resourceName))
+                {
+                    if (stream == null)
+                    {
+                        throw new Exception($"Resource '{resourceName}' not found. Check Embedded Resource settings.");
+                    }
+
+                    using (ZipArchive archive = new(stream))
+                    {
+                        int totalEntries = archive.Entries.Count;
+                        int currentEntry = 0;
+
+                        foreach (ZipArchiveEntry entry in archive.Entries)
+                        {
+                            currentEntry++;
+
+                            // Calculate percentage
+                            int percent = (int)((double)currentEntry / totalEntries * 100);
+
+                            // Draw Progress Bar
+                            DrawProgressBar(percent, entry.Name);
+
+                            // Create the full path
+                            string destinationPath = Path.GetFullPath(Path.Combine(installPath, entry.FullName));
+
+                            // Security check: prevent ZipSlip (writing outside target folder)
+                            if (!destinationPath.StartsWith(installPath, StringComparison.OrdinalIgnoreCase))
+                                continue;
+
+                            // Handle folders vs files
+                            if (string.IsNullOrEmpty(entry.Name)) // It's a directory
+                            {
+                                Directory.CreateDirectory(destinationPath);
+                            }
+                            else // It's a file
+                            {
+                                // Ensure the directory exists (zipped files might not list their dir first)
+                                Directory.CreateDirectory(Path.GetDirectoryName(destinationPath));
+                                entry.ExtractToFile(destinationPath, overwrite: true);
+                            }
+                        }
+                    }
+                }
+
+                MoveUpdater();
+
+                DrawProgressBar(100, "Done");
+                Console.WriteLine();
+                Console.WriteLine("Installation Complete!");
+
+                // Optional: Create Shortcut logic here
+            }
+            catch (Exception ex)
+            {
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine($"Error: {ex.Message}");
+            }
+
+            Console.WriteLine("Press any key to exit...");
+            Console.ReadKey();
+        }
+
+        public static void SelfDelete()
+        {
+            string exePath = Process.GetCurrentProcess().MainModule.FileName;
+            string directoryToDelete = Path.GetDirectoryName(exePath);
+
+            string args = $"/C timeout /t 2 /nobreak > Nul & del \"{exePath}\" & rmdir /q \"{directoryToDelete}\"";
+
+            ProcessStartInfo psi = new ProcessStartInfo
+            {
+                FileName = "cmd.exe",
+                Arguments = args,
+                WindowStyle = ProcessWindowStyle.Hidden,
+                CreateNoWindow = true,
+                UseShellExecute = false
+            };
+
+            Process.Start(psi);
+
+            Environment.Exit(0);
+        }
+
+        public static void MoveUpdater()
+        {
+            string currentExe = Process.GetCurrentProcess().MainModule.FileName;
+
+            var updaterPath = Path.Combine(installPath, updaterName + ".exe");
+
+            Directory.CreateDirectory(Path.GetDirectoryName(updaterPath));
+            File.Copy(currentExe, updaterPath, overwrite: true);
+        }
+
+        static void DrawProgressBar(int percent, string filename)
+        {
+            // Move cursor to start of line
+            Console.CursorLeft = 0;
+
+            // Limit filename length for clean display
+            string shortName = filename.Length > 20 ? filename.Substring(0, 17) + "..." : filename.PadRight(20);
+
+            Console.Write("[");
+            int width = Console.WindowWidth - 1; // Width of the bar
+            int progress = (int)((percent / 100.0) * width);
+
+            // Draw filled part
+            Console.Write(new string('#', progress));
+            // Draw empty part
+            Console.Write(new string('-', width - progress));
+
+            Console.Write($"] {percent}% {shortName}");
+        }
+    }
+}