using KKdBaseLib;
using KKdBaseLib.F2;
using KKdMainLib.IO;

namespace KKdMainLib
{
    public static class HeaderExtensions
    {
        public static Header ReadHeader(this Stream stream, bool seek, bool readSectionSignature = true)
        {
            if (seek)
                if (stream.PI64 > 4) stream.PI64 -= 4;
                else                 stream.PI64  = 0;
            return stream.ReadHeader(readSectionSignature);
        }

        public static Header ReadHeader(this Stream stream, bool readSectionSignature = true)
        {
            Header header = new Header { Format = Format.F2LE, Signature = stream.RI32(),
                DataSize = stream.RI32(), Length = stream.RI32(), Flags = stream.RI32(),
                Depth = stream.RI32(), SectionSize = stream.RI32(),
                Mode = stream.RI32() };
            stream.RI32();
            if ((header.Flags & 0x08000000) == 0x08000000) header.Format = Format.F2BE;
            header.NotUseDataSizeAsSectionSize = (header.Flags & 0x10000000) != 0x10000000;
            if (header.Length == 0x40)
            {
                stream.RI64();
                stream.RI64();
                header.InnerSignature = stream.RI32();
                stream.RI32();
                stream.RI64();
            }
            stream.Format = header.Format;
            if (readSectionSignature) header.SectionSignature = stream.RI32E();
            return header;
        }

        public static void W(this Stream stream, Header header, bool extended = false)
        {
            header.Length = (header.Format < Format.X && extended) ? 0x40 : 0x20;
            header.Flags = (!header.NotUseDataSizeAsSectionSize ? 0x10000000 : 0) |
                            (header.Format == Format.F2BE ? 0x08000000 : 0);

            stream.W(header.Signature);
            stream.W(header.DataSize);
            stream.W(header.Length);
            stream.W(header.Flags);
            stream.W(header.Depth);
            stream.W(header.SectionSize);
            stream.W(header.Mode);
            stream.W(0x00);
            if (header.Length == 0x40)
            {
                stream.W(header.Format < Format.MGF ? (int)((header.SectionSignature ^
                    (header.DataSize * (long)header.Signature)) - header.Depth + header.SectionSize) : 0);
                stream.W(0x00);
                stream.W(0x00L);
                stream.W(header.InnerSignature);
                stream.W(0x00);
                stream.W(0x00L);
            }
        }

        public static void WEOFC(this Stream stream, int depth = 0) =>
            stream.W(new Header { Depth = depth, Length = 0x20, Signature = 0x43464F45 });
    }

    public static class POFExtensions
    {
        public static void W(this Stream stream, POF pof, bool shiftX = false, int depth = 0)
        {
            byte[] data = pof.Write(shiftX);
            Header header = new Header { Depth = depth, Format = Format.F2LE,
                Length = 0x20, Signature = shiftX ? 0x31464F50 : 0x30464F50 };
            header.DataSize = header.SectionSize = data.Length;
            stream.W(header);
            stream.W(data);
        }
    }

    public static class ENRSExtensions
    {
        public static void W(this Stream stream, ENRS enrs, int depth = 0)
        {
            byte[] data = enrs.Write();
            Header header = new Header { Depth = depth, Format = Format.F2LE,
                Length = 0x20, Signature = 0x53524E45 };
            header.DataSize = header.SectionSize = data.Length;
            stream.W(header);
            stream.W(data);
        }
    }

    public static class StructExtensions
    {
        public static Struct RSt(this byte[] data)
        {
            if (data == null || data.Length < 1) return default;
            Struct @struct;
            using (Stream stream = File.OpenReader(data))
                @struct = stream.RSt(stream.ReadHeader(false));
            return @struct;
        }

        public static Struct RSt(this Stream stream, Header header)
        {
            Struct @struct = new Struct { Header = header, DataOffset =
                stream.P, Data = stream.RBy(header.SectionSize) };
            int depth = header.Depth;

            int lastSig = 0, sig;
            long length = stream.L - stream.P;
            long position = 0;
            KKdList<Struct> subStructs = KKdList<Struct>.New;
            while (length > position)
            {
                header = stream.ReadHeader(false);
                sig = header.Signature;
                position += header.Length + header.SectionSize;
                if (sig == 0x43464F45 && header.Depth == depth + 1) break;
                else if (sig == 0x53524E45 || (sig & 0xF0FFFFFF) == 0x30464F50)
                {
                    byte[] Data = stream.RBy(header.SectionSize);
                    if (sig == 0x53524E45) @struct.ENRS.Read(Data);
                    else                   @struct.POF .Read(Data, sig == 0x31464F50);
                }
                else if (header.Depth == 0 && sig == 0x43505854)
                { subStructs.Add(new Struct { Header = header, DataOffset =
                    stream.P, Data = stream.RBy(header.SectionSize) }); }
                else if (header.Depth <= depth) { stream.PI64 -= header.Length; break; }
                else subStructs.Add(stream.RSt(header));
                lastSig = sig;
            }

            if (subStructs.Capacity > 0) @struct.SubStructs = subStructs.ToArray();
            return @struct;
        }

        public static byte[] W(this Struct Struct, bool shiftX = false, bool useDepth = true)
        {
            byte[] Data;
            using (Stream stream = File.OpenWriter()) { Struct.Update(shiftX);
                stream.W(Struct, shiftX, useDepth); stream.WEOFC(); Data = stream.ToArray(); }
            return Data;
        }

        public static void W(this Stream stream, Struct @struct, bool shiftX = false, bool useDepth = true)
        {
            stream.W(@struct.Header);
            stream.W(@struct.Data  );
            if (@struct.HasPOF ) stream.W(@struct.POF , shiftX, useDepth ? @struct.Depth + 1 : 0);
            if (@struct.HasENRS) stream.W(@struct.ENRS,         useDepth ? @struct.Depth + 1 : 0);
            if (@struct.HasSubStructs)
            {
                for (int i = 0; i < @struct.SubStructs.Length; i++)
                    stream.W(@struct.SubStructs[i], shiftX);
                stream.WEOFC(@struct.Depth + 1);
            }
        }
    }

    public static class MPExt
    {
        public static MsgPack ReadMP(this byte[] array, bool json = false)
        {
            MsgPack MsgPack;
            if (json) using (JSON _IO = new JSON(File.OpenReader(array))) MsgPack = _IO.Read(    );
            else      using (  MP _IO = new   MP(File.OpenReader(array))) MsgPack = _IO.Read(true);
            return MsgPack;
        }

        public static MsgPack ReadMPAllAtOnce(this string file, bool json = false)
        {
            MsgPack MsgPack;
            if (json) using (JSON _IO = new JSON(File.OpenReader(file + ".json", true))) MsgPack = _IO.Read(    );
            else      using (  MP _IO = new   MP(File.OpenReader(file + ".mp"  , true))) MsgPack = _IO.Read(true);
            return MsgPack;
        }

        public static MsgPack ReadMP(this string file, bool json = false)
        {
            MsgPack MsgPack;
            if (json) using (JSON _IO = new JSON(File.OpenReader(file + ".json"))) MsgPack = _IO.Read(    );
            else      using (  MP _IO = new   MP(File.OpenReader(file + ".mp"  ))) MsgPack = _IO.Read(true);
            return MsgPack;
        }

        public static void Write(this MsgPack mp, bool temp, string file, bool json = false)
        { if (temp) MsgPack.New.Add(mp).Write(file, json).Dispose();
          else                      mp .Write(file, json); }

        public static MsgPack Write(this MsgPack mp, string file, bool json = false)
        {
            if (json) using (JSON _IO = new JSON(File.OpenWriter(file + ".json", true))) _IO.W(mp, "\n", "  ");
            else      using (  MP _IO = new   MP(File.OpenWriter(file + ".mp"  , true))) _IO.W(mp);
            return mp;
        }

        public static void WriteAfterAll(this MsgPack mp, bool temp, string file, bool json = false)
        { if (temp) MsgPack.New.Add(mp).WriteAfterAll(file, json).Dispose();
          else                      mp .WriteAfterAll(file, json); }

        public static MsgPack WriteAfterAll(this MsgPack mp, string file, bool json = false)
        {
            byte[] data = null;
            if (json) using (JSON _IO = new JSON(File.OpenWriter())) { _IO.W(mp, true); data = _IO.ToArray(); }
            else      using (  MP _IO = new   MP(File.OpenWriter())) { _IO.W(mp      ); data = _IO.ToArray(); }
            File.WriteAllBytes(file + (json ? ".json" : ".mp"), data);
            return mp;
        }

        public static void ToJSON   (this string file) =>
            file.ReadMP(    ).Write(file, true).Dispose();

        public static void ToMsgPack(this string file) =>
            file.ReadMP(true).Write(file      ).Dispose();
    }

    public static class IKFExt
    {
        public static IKF Round(this IKF kf, int d)
        {
                 if (kf is KFT0 kft0) { kft0.F  = kft0.F .Round(d);                             return kft0; }
            else if (kf is KFT1 kft1) { kft1.F  = kft1.F .Round(d); kft1.V  = kft1.V .Round(d); return kft1; }
            else if (kf is KFT2 kft2) { kft2.F  = kft2.F .Round(d); kft2.V  = kft2.V .Round(d);
                                        kft2.T  = kft2.T .Round(d);                             return kft2; }
            else if (kf is KFT3 kft3) { kft3.F  = kft3.F .Round(d); kft3.V  = kft3.V .Round(d);
                                        kft3.T1 = kft3.T1.Round(d); kft3.T2 = kft3.T2.Round(d); return kft3; }
            return   kf;
        }
    }

    public static class LibDeflate
    {
        public static void Extract()
        {
            if (!File.Exists("libdeflate.dll"))
                File.WriteAllBytes("libdeflate.dll", Properties.Resources.libdeflate);
        }
    }
}