QuickLook/QuickLook.Plugin/QuickLook.Plugin.ArchiveViewer/ChromiumResourcePackage/PakInfoPanel.xaml.cs

// Copyright © 2017-2026 QL-Win Contributors
//
// This file is part of QuickLook program.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

using QuickLook.Common.ExtensionMethods;
using QuickLook.Common.Helpers;
using QuickLook.Plugin.ArchiveViewer.ArchiveFile;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Threading.Tasks;
using System.Windows.Controls;

namespace QuickLook.Plugin.ArchiveViewer.ChromiumResourcePackage;

public partial class PakInfoPanel : UserControl, IDisposable, INotifyPropertyChanged
{
    private readonly Dictionary<string, ArchiveFileEntry> _fileEntries = [];
    private bool _disposed;
    private double _loadPercent;
    private ulong _totalSize;

    public PakInfoPanel(string path)
    {
        InitializeComponent();
        Resources.MergedDictionaries.Clear();
        BeginLoadPak(path);
    }

    public double LoadPercent
    {
        get => _loadPercent;
        private set
        {
            if (value == _loadPercent) return;
            _loadPercent = value;
            OnPropertyChanged();
        }
    }

    public void Dispose()
    {
        GC.SuppressFinalize(this);
        _disposed = true;
    }

    public event PropertyChangedEventHandler PropertyChanged;

    private void BeginLoadPak(string path)
    {
        new Task(() =>
        {
            _totalSize = (ulong)new FileInfo(path).Length;
            var root = new ArchiveFileEntry(Path.GetFileName(path), true);
            _fileEntries.Add(string.Empty, root);
            try
            {
                LoadItemsFromPak(path);
            }
            catch (Exception e)
            {
                ProcessHelper.WriteLog(e.ToString());
                Dispatcher.Invoke(() => { lblLoading.Content = "Preview failed. See log for more details."; });
                return;
            }

            var folders = -1; // do not count root node
            var files = 0;
            ulong sizeU = 0L;
            foreach (var item in _fileEntries)
            {
                if (item.Value.IsFolder)
                    folders++;
                else
                    files++;
                sizeU += item.Value.Size;
            }

            string t;
            var d = folders != 0 ? $"{folders} folders" : string.Empty;
            var f = files != 0 ? $"{files} files" : string.Empty;
            if (!string.IsNullOrEmpty(d) && !string.IsNullOrEmpty(f))
                t = $", {d} folders and {f} files";
            else if (string.IsNullOrEmpty(d) && string.IsNullOrEmpty(f))
                t = string.Empty;
            else
                t = $", {d}{f}";

            Dispatcher.Invoke(() =>
            {
                if (_disposed)
                    return;
                fileListView?.DataContext = _fileEntries[string.Empty].Children.Keys;
                archiveCount.Content = $"PAK File{t}";
                archiveSizeC.Content = string.Empty;
                archiveSizeU.Content = $"Total resource size {((long)sizeU).ToPrettySize(2)}";
            });
            LoadPercent = 100d;
        }).Start();
    }

    private void LoadItemsFromPak(string path)
    {
        var dict = PakExtractor.ExtractToDictionary(path, true);
        var modifiedDate = File.GetLastWriteTime(path);

        foreach (var kv in dict)
        {
            var fragments = kv.Key.Split(['/', '\\'], StringSplitOptions.RemoveEmptyEntries);
            string currentPath = string.Empty;
            ArchiveFileEntry parent = _fileEntries[string.Empty];

            for (int i = 0; i < fragments.Length - 1; i++)
            {
                var dirName = fragments[i];
                currentPath = string.IsNullOrEmpty(currentPath) ? dirName : currentPath + "\\" + dirName;
                if (!_fileEntries.TryGetValue(currentPath, out var dirEntry))
                {
                    dirEntry = new ArchiveFileEntry(dirName, true, parent)
                    {
                        ModifiedDate = modifiedDate,
                    };
                    _fileEntries.Add(currentPath, dirEntry);
                }
                parent = dirEntry;
            }

            var fileName = fragments.Last();
            var filePath = fragments.Length > 1 ? currentPath + "\\" + fileName : fileName;
            if (!_fileEntries.ContainsKey(filePath))
            {
                var entry = new ArchiveFileEntry(fileName, false, parent)
                {
                    Size = (ulong)kv.Value.Length,
                    ModifiedDate = modifiedDate,
                };
                _fileEntries.Add(filePath, entry);
            }
        }
    }

    protected virtual void OnPropertyChanged([CallerMemberName] string propertyName = null)
    {
        PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName));
    }
}