FModel/FModel/Methods/PakHelper/PakHelper.cs

using csharp_wick;
using FModel.Properties;
using System;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Windows.Forms;

namespace FModel
{
    static class PakHelper
    {
        public static string[] PakAsTxt { get; set; }
        private static StringBuilder _sb { get; set; }

        /// <summary>
        /// for each main paks we generate a file list
        /// if the main key used to generate the file list for each pak is working, we also load the localization file to get the translated version of all strings
        /// for the dynamic paks we check if the key exist in the AES Manager, if so, we do the same steps as a main pak
        /// </summary>
        /// <param name="theSinglePak"></param>
        /// <param name="loadAllPaKs"></param>
        public static void RegisterPaKsinDict(ToolStripItemClickedEventArgs theSinglePak = null, bool loadAllPaKs = false)
        {
            _sb = new StringBuilder();
            bool bMainKeyWorking = false;

            for (int i = 0; i < ThePak.mainPaksList.Count; i++)
            {
                PakExtractor theExtractor = null;
                try
                {
                    if (!string.IsNullOrWhiteSpace(Settings.Default.AESKey))
                    {
                        theExtractor = new PakExtractor(Settings.Default.PAKsPath + "\\" + ThePak.mainPaksList[i].thePak, Settings.Default.AESKey);
                    }
                    else { if (theExtractor != null) { theExtractor.Dispose(); } break; }
                }
                catch (Exception)
                {
                    if (theExtractor != null) { theExtractor.Dispose(); }
                    break;
                }

                string[] CurrentUsedPakLines = theExtractor.GetFileList().ToArray();
                if (CurrentUsedPakLines != null)
                {
                    bMainKeyWorking = true;
                    string mountPoint = theExtractor.GetMountPoint();

                    ThePak.PaksExtractorDictionary.Add(ThePak.mainPaksList[i].thePak, theExtractor);
                    ThePak.PaksFileArrayDictionary.Add(theExtractor, CurrentUsedPakLines); //in RegisterInDict we add the mount point to CurrentUsedPakLines

                    RegisterInDict(ThePak.mainPaksList[i].thePak, CurrentUsedPakLines, mountPoint, theSinglePak, loadAllPaKs);
                }
            }
            if (bMainKeyWorking) { LoadLocRes.LoadMySelectedLocRes(Settings.Default.IconLanguage); }

            for (int i = 0; i < ThePak.dynamicPaksList.Count; i++)
            {
                PakExtractor theExtractor = null;
                string pakName = DynamicKeysManager.AESEntries.Where(x => x.thePak == ThePak.dynamicPaksList[i].thePak).Select(x => x.thePak).FirstOrDefault();
                string pakKey = DynamicKeysManager.AESEntries.Where(x => x.thePak == ThePak.dynamicPaksList[i].thePak).Select(x => x.theKey).FirstOrDefault();

                if (!string.IsNullOrEmpty(pakName) && !string.IsNullOrEmpty(pakKey))
                {
                    try
                    {
                        theExtractor = new PakExtractor(Settings.Default.PAKsPath + "\\" + pakName, pakKey);
                    }
                    catch (Exception)
                    {
                        new UpdateMyConsole("0x" + pakKey + " doesn't work with " + ThePak.dynamicPaksList[i].thePak, Color.Red, true).AppendToConsole();
                        if (theExtractor != null) { theExtractor.Dispose(); }
                        continue;
                    }

                    string[] CurrentUsedPakLines = theExtractor.GetFileList().ToArray();
                    if (CurrentUsedPakLines != null)
                    {
                        string mountPoint = theExtractor.GetMountPoint();

                        ThePak.PaksExtractorDictionary.Add(ThePak.dynamicPaksList[i].thePak, theExtractor);
                        ThePak.PaksFileArrayDictionary.Add(theExtractor, CurrentUsedPakLines); //in RegisterInDict we add the mount point to CurrentUsedPakLines

                        RegisterInDict(ThePak.dynamicPaksList[i].thePak, CurrentUsedPakLines, mountPoint, theSinglePak, loadAllPaKs);
                    }
                }
            }

            if (loadAllPaKs)
            {
                File.WriteAllText(App.DefaultOutputPath + "\\FortnitePAKs.txt", _sb.ToString()); //File will always exist
            }

            new UpdateMyState("Building tree, please wait...", "Loading").ChangeProcessState();
        }

        /// <summary>
        /// 1. we get the pak mount point and add it to the PaksMountPoint dictionary
        /// 2. for each file in a pak, we add the file name as the key and its pak name as the value (used later by SearchFiles form)
        /// </summary>
        /// <param name="thePakName"></param>
        /// <param name="thePakLines"></param>
        /// <param name="theSinglePak"></param>
        /// <param name="weLoadAll"></param>
        private static void RegisterInDict(string thePakName, string[] thePakLines, string mountPoint, ToolStripItemClickedEventArgs theSinglePak = null, bool weLoadAll = false)
        {
            for (int i = 0; i < thePakLines.Length; i++)
            {
                thePakLines[i] = mountPoint.Substring(9) + thePakLines[i];

                string CurrentUsedPakFileName = thePakLines[i].Substring(thePakLines[i].LastIndexOf("/", StringComparison.Ordinal) + 1);
                if (CurrentUsedPakFileName.Contains(".uasset") || CurrentUsedPakFileName.Contains(".uexp") || CurrentUsedPakFileName.Contains(".ubulk"))
                {
                    if (!ThePak.AllpaksDictionary.ContainsKey(CurrentUsedPakFileName.Substring(0, CurrentUsedPakFileName.LastIndexOf(".", StringComparison.Ordinal))))
                    {
                        ThePak.AllpaksDictionary.Add(CurrentUsedPakFileName.Substring(0, CurrentUsedPakFileName.LastIndexOf(".", StringComparison.Ordinal)), thePakName);
                    }
                }
                else
                {
                    if (!ThePak.AllpaksDictionary.ContainsKey(CurrentUsedPakFileName))
                    {
                        ThePak.AllpaksDictionary.Add(CurrentUsedPakFileName, thePakName);
                    }
                }

                if (weLoadAll)
                {
                    _sb.Append(thePakLines[i] + "\n");
                }
            }

            if (weLoadAll) { new UpdateMyState(".PAK mount point: " + mountPoint.Substring(9), "Waiting").ChangeProcessState(); }
            if (theSinglePak != null && thePakName == theSinglePak.ClickedItem.Text) { PakAsTxt = thePakLines; }
        }
    }
}