BadBuilder/BadBuilder/Program.cs

global using DownloadItem = (string name, string url);
global using ArchiveItem = (string name, string path);
global using HomebrewApp = (string name, string folder, string entryPoint);

using Spectre.Console;
using BadBuilder.Models;
using BadBuilder.Helpers;

using static BadBuilder.Utilities.Constants;
using BadBuilder.Utilities;

namespace BadBuilder
{
    internal partial class Program
    {
        static readonly Style OrangeStyle = new Style(new Color(255, 114, 0));
        static readonly Style LightOrangeStyle = new Style(new Color(255, 172, 77));
        static readonly Style PeachStyle = new Style(new Color(255, 216, 153));

        static readonly Style GreenStyle = new Style(new Color(118, 185, 0));
        static readonly Style GrayStyle = new Style(new Color(132, 133, 137));

        static string XexToolPath = string.Empty;
        static string TargetDriveLetter = string.Empty;

        static ActionQueue actionQueue = new();

        static void Main(string[] args)
        {
            ShowWelcomeMessage();

            while (true)
            {
                Console.WriteLine();
                string action = PromptForAction();

                if (action == "Exit") Environment.Exit(0);

                List<DiskInfo> disks = DiskHelper.GetDisks();
                string selectedDisk = PromptDiskSelection(disks);
                TargetDriveLetter = selectedDisk.Substring(0, 3);

                bool confirmation = PromptFormatConfirmation(selectedDisk);
                if (confirmation)
                {
                    if (!FormatDisk(disks, selectedDisk)) continue; 
                    break;
                }
            }

            ClearConsole();

            List<ArchiveItem> downloadedFiles = DownloadRequiredFiles().Result;
            ExtractFiles(downloadedFiles).Wait();


            AnsiConsole.MarkupLine("\n\n[#76B900]{0}[/] Copying requried files and folders.", Markup.Escape("[*]"));
            foreach (var folder in Directory.GetDirectories($@"{EXTRACTED_DIR}"))
            {
                var folderName = folder.Split("\\").Last();

                switch (folderName)
                {
                    case "XEXMenu":
                        EnqueueMirrorDirectory(
                            Path.Combine(folder, $"{ContentFolder}C0DE9999"),
                            Path.Combine(TargetDriveLetter, $"{ContentFolder}C0DE9999"),
                            7
                        );
                        break;

                    case "FreeMyXe":
                        EnqueueFileCopy(
                            Path.Combine(folder, "FreeMyXe.xex"),
                            Path.Combine(TargetDriveLetter, "BadUpdatePayload", "default.xex"),
                            9
                        );
                        break;

                    case "BadUpdate":
                        actionQueue.EnqueueAction(async () =>
                        {
                            using (StreamWriter writer = new(Path.Combine(TargetDriveLetter, "name.txt")))
                            {
                                writer.WriteLine("USB Storage Device");
                            }

                            using (StreamWriter writer = new(Path.Combine(TargetDriveLetter, "info.txt")))
                            {
                                writer.WriteLine("This drive was created with BadBuilder by Pdawg.\nFind more info here: https://github.com/Pdawg-bytes/BadBuilder");
                            }

                            Directory.CreateDirectory(Path.Combine(TargetDriveLetter, "Apps"));
                            await FileSystemHelper.MirrorDirectoryAsync(Path.Combine(folder, "Rock Band Blitz"), TargetDriveLetter);
                        }, 10);
                        break;

                    case "BadUpdate Tools":
                        XexToolPath = Path.Combine(folder, "XePatcher", "XexTool.exe");
                        break;

                    case "Rock Band Blitz":
                        EnqueueMirrorDirectory(
                            Path.Combine(folder, $"{ContentFolder}5841122D\\000D0000"),
                            Path.Combine(TargetDriveLetter, $"{ContentFolder}5841122D\\000D0000"),
                            8
                        );
                        break;

                    case "Simple 360 NAND Flasher":
                        actionQueue.EnqueueAction(async () =>
                        {
                            await PatchHelper.PatchXexAsync(Path.Combine(folder, "Simple 360 NAND Flasher", "Default.xex"), XexToolPath);
                            await FileSystemHelper.MirrorDirectoryAsync(Path.Combine(folder, "Simple 360 NAND Flasher"), Path.Combine(TargetDriveLetter, "Apps", "Simple 360 NAND Flasher"));
                        }, 6);
                        break;

                    default: throw new Exception($"Unexpected directory in working folder: {folder}");
                }
            }
            actionQueue.ExecuteActionsAsync().Wait();

            ClearConsole();
            if (!PromptAddHomebrew())
            {
                AnsiConsole.MarkupLine("\n[#76B900]{0}[/] Your USB drive is ready to go.", Markup.Escape("[+]"));
                Environment.Exit(0);
            }

            Console.WriteLine();

            List<HomebrewApp> homebrewApps = ManageHomebrewApps();

            AnsiConsole.Status()
                .SpinnerStyle(OrangeStyle)
                .StartAsync("Copying and patching homebrew apps.", async ctx =>
                {
                    await Task.WhenAll(homebrewApps.Select(async item =>
                    {
                        await FileSystemHelper.MirrorDirectoryAsync(item.folder, Path.Combine(TargetDriveLetter, "Apps", item.name));
                        await PatchHelper.PatchXexAsync(Path.Combine(TargetDriveLetter, "Apps", item.name, Path.GetFileName(item.entryPoint)), XexToolPath);
                    }));
                }).Wait();

            string status = "[+]";
            AnsiConsole.MarkupInterpolated($"[#76B900]{status}[/] [bold]{homebrewApps.Count()}[/] apps copied.\n");

            AnsiConsole.MarkupLine("\n[#76B900]{0}[/] Your USB drive is ready to go.", Markup.Escape("[+]"));
        }

        static void EnqueueMirrorDirectory(string sourcePath, string destinationPath, int priority)
        {
            actionQueue.EnqueueAction(async () =>
            {
                await FileSystemHelper.MirrorDirectoryAsync(sourcePath, destinationPath);
            }, priority);
        }

        static void EnqueueFileCopy(string sourceFile, string destinationFile, int priority)
        {
            actionQueue.EnqueueAction(async () =>
            {
                await FileSystemHelper.CopyFileAsync(sourceFile, destinationFile);
            }, priority);
        }


        static void ShowWelcomeMessage() => AnsiConsole.Markup(
            """
            [#4D8C00]██████╗  █████╗ ██████╗ ██████╗ ██╗   ██╗██╗██╗     ██████╗ ███████╗██████╗[/]
            [#65A800]██╔══██╗██╔══██╗██╔══██╗██╔══██╗██║   ██║██║██║     ██╔══██╗██╔════╝██╔══██╗[/]
            [#76B900]██████╔╝███████║██║  ██║██████╔╝██║   ██║██║██║     ██║  ██║█████╗  ██████╔╝[/]
            [#A1CF3E]██╔══██╗██╔══██║██║  ██║██╔══██╗██║   ██║██║██║     ██║  ██║██╔══╝  ██╔══██╗[/]
            [#CCE388]██████╔╝██║  ██║██████╔╝██████╔╝╚██████╔╝██║███████╗██████╔╝███████╗██║  ██║[/]
            [#CCE388]╚═════╝ ╚═╝  ╚═╝╚═════╝ ╚═════╝  ╚═════╝ ╚═╝╚══════╝╚═════╝ ╚══════╝╚═╝  ╚═╝[/]

            [#76B900]────────────────────────────────────────────────────────────────────────────[/]
            ───────────────────────Xbox 360 [#FF7200]BadUpdate[/] USB Builder───────────────────────
                                        [#848589]Created by Pdawg[/]
            [#76B900]────────────────────────────────────────────────────────────────────────────[/]

            """);

        static string PromptForAction() => AnsiConsole.Prompt(
            new SelectionPrompt<string>()
            .HighlightStyle(GreenStyle)
            .AddChoices(
                "Build exploit USB",
                "Exit"
            )
        );

        static void ClearConsole()
        {
            AnsiConsole.Clear();
            ShowWelcomeMessage();
            Console.WriteLine();
        }
    }
}