pkNX/pkNX.WinForms/Ripper/FileRipper.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using pkNX.Containers;

namespace pkNX.WinForms
{
    public enum RipResultCode
    {
        UnknownFormat,
        Success,
        FileExist,
        BadSize,
        ReadError,
    }

    public static class FileRipper
    {
        public static ContainerHandler DefaultHandler { private get; set; } = new ContainerHandler();

        public static readonly List<Func<BinaryReader, uint, string, ContainerHandler, RipResult>> Loaders =
            new List<Func<BinaryReader, uint, string, ContainerHandler, RipResult>>
            {
                GFPackDump
            };

        private static RipResult GFPackDump(BinaryReader br, uint header, string path, ContainerHandler handler)
        {
            if (header != 0x584C_4647)
                return new RipResult(RipResultCode.UnknownFormat);

            br.BaseStream.Position = 0;
            var gfp = new GFPack(br);

            var dir = Path.GetDirectoryName(path);
            var folder = Path.GetFileNameWithoutExtension(path);
            var resultPath = Path.Combine(dir, folder);
            Directory.CreateDirectory(resultPath);

            gfp.Dump(resultPath, handler);

            return new RipResult(RipResultCode.Success) {ResultPath = resultPath};
        }

        public class RipResult
        {
            public readonly RipResultCode Code;
            public string ResultPath;

            public RipResult(RipResultCode code) => Code = code;
        }

        public static RipResult TryOpenFile(string path, ContainerHandler handler = null)
        {
            if (!File.Exists(path))
                return new RipResult(RipResultCode.FileExist);

            if (handler == null)
                handler = DefaultHandler;

            try
            {
                using (var fs = new FileStream(path, FileMode.Open, FileAccess.Read))
                using (var br = new BinaryReader(fs))
                {
                    if (br.BaseStream.Length < 4)
                        return new RipResult(RipResultCode.BadSize);
                    var header = br.ReadUInt32();
                    var result = TryLoadFile(br, header, path, handler);
                    if (result.Code == RipResultCode.Success)
                        return result;
                }
            }
            catch (Exception e)
            {
                Debug.WriteLine(e.Message);
                return new RipResult(RipResultCode.ReadError);
            }
            return new RipResult(RipResultCode.UnknownFormat);
        }

        private static RipResult TryLoadFile(BinaryReader br, uint header, string path, ContainerHandler handler)
        {
            foreach (var method in Loaders)
            {
                try
                {
                    var result = method(br, header, path, handler);
                    if (result.Code == RipResultCode.Success)
                        return result;
                }
                catch (Exception e)
                {
                    Debug.WriteLine(e.Message);
                }
            }
            return new RipResult(RipResultCode.UnknownFormat);
        }
    }
}