//Written for PixelJunk Shooter. https://store.steampowered.com/app/255870/
using System.IO;
using System.IO.Compression;

namespace PixelJunk_Extractor
{
    class Program
    {
        static void Main(string[] args)
        {
            BinaryReader br = new(File.OpenRead(Path.GetDirectoryName(args[0]) + "//" + Path.GetFileNameWithoutExtension(args[0]) + ".pkiwin"));
            string path = Path.GetDirectoryName(args[0]);
            Directory.CreateDirectory(path + "//" + Path.GetFileNameWithoutExtension(args[0]));
            br.BaseStream.Position = 8;
            SUBFILE[] subfiles = new SUBFILE[br.ReadInt32()];
            for (int i = 0; i < subfiles.Length; i++)
            {
                subfiles[i].sizeUncompressed = br.ReadInt32();
                subfiles[i].sizeCompressed = br.ReadInt32();
                subfiles[i].offset = br.ReadInt32();
                br.ReadInt32();//unknown
            }
            int n = 0;
            using FileStream pkdwin = File.OpenRead(path + "//" + Path.GetFileNameWithoutExtension(args[0]) + ".pkdwin");
            foreach (SUBFILE sub in subfiles)
            {
                br = new(pkdwin);
                br.BaseStream.Position = sub.offset;
                using FileStream FS = File.Create(path + "//" + Path.GetFileNameWithoutExtension(args[0]) + "//" + n);
                BinaryWriter bw = new(FS);

                MemoryStream ms = new();
                br.ReadInt16();
                using (var ds = new DeflateStream(new MemoryStream(br.ReadBytes(sub.sizeCompressed)), CompressionMode.Decompress))
                    ds.CopyTo(ms);
                br = new(ms);
                br.BaseStream.Position = 0;
                bw.Write(br.ReadBytes(sub.sizeUncompressed));
                bw.Close();
                n++;
            }
            br.Close();
        }
    }

    struct SUBFILE
    {
        public int sizeUncompressed;
        public int sizeCompressed;
        public int offset;
    }
}