From 513cacc2223937fd0c1dcefb2f653358344ed076 Mon Sep 17 00:00:00 2001
From: KillzXGaming <millernathan35@gmail.com>
Date: Tue, 12 Oct 2021 19:51:48 -0400
Subject: [PATCH] PKG : use names in the file data and organize the file types.

Will support proper file paths when the hashing method is figured out.
---
 .../FileFormats/Archives/PKG.cs               | 96 ++++++++++++++++---
 1 file changed, 81 insertions(+), 15 deletions(-)

diff --git a/File_Format_Library/FileFormats/Archives/PKG.cs b/File_Format_Library/FileFormats/Archives/PKG.cs
index e63a664f..8fed4dd8 100644
--- a/File_Format_Library/FileFormats/Archives/PKG.cs
+++ b/File_Format_Library/FileFormats/Archives/PKG.cs
@@ -14,17 +14,17 @@ namespace FirstPlugin
     {
         public FileType FileType { get; set; } = FileType.Archive;
 
-        public bool CanSave { get; set; }
+        public bool CanSave { get; set; } = false;
         public string[] Description { get; set; } = new string[] { "PKG" };
         public string[] Extension { get; set; } = new string[] { "*.pkg" };
         public string FileName { get; set; }
         public string FilePath { get; set; }
         public IFileInfo IFileInfo { get; set; }
 
-        public bool CanAddFiles { get; set; }
-        public bool CanRenameFiles { get; set; }
-        public bool CanReplaceFiles { get; set; }
-        public bool CanDeleteFiles { get; set; }
+        public bool CanAddFiles { get; set; } = true;
+        public bool CanRenameFiles { get; set; } = true;
+        public bool CanReplaceFiles { get; set; } = true;
+        public bool CanDeleteFiles { get; set; } = true;
 
         public bool Identify(System.IO.Stream stream)
         {
@@ -76,13 +76,48 @@ namespace FirstPlugin
                         using (reader.TemporarySeek(fileStartOffset, SeekOrigin.Begin)) {
                             string magic = reader.ReadString(4);
                             if (magic == "FWAV") ext = ".bfwav";
-                            if (magic == "MTXT") ext = ".bctext";
-                            if (magic == "MCAN") ext = ".mcamera";
-                            if (magic == "MANM") ext = ".manim";
-                            if (magic == "MSAS") ext = ".msas";
-                            if (magic == "MMDL") ext = ".mmodel";
-                            if (magic == "MSUR") ext = ".mmaterial";
-                            if (magic == "MNAV") ext = ".mnavigation";
+                            if (magic == "MTXT") ext = ".bctex";
+                            if (magic == "MCAN") ext = ".bccam";
+                            if (magic == "MANM") ext = ".bcskla";
+                            if (magic == "MSAS") ext = ".bmsas";
+                            if (magic == "MMDL") ext = ".mmdl"; //Original extension is bcmdl but use mmdl for noesis script
+                            if (magic == "MSUR") ext = ".bsmat";
+                            if (magic == "MNAV") ext = ".bmscd";
+
+                            if (magic == "MSUR")
+                            {
+                                reader.ReadUInt32();
+                                file.FileName = $"imats/" + reader.ReadZeroTerminatedString();
+                            }
+                            else if (magic == "MSAS")
+                            {
+                                reader.ReadUInt32();
+                                ushort length = reader.ReadUInt16();
+                                file.FileName = $"msas/" + reader.ReadString(length);
+                            }
+                            else if (magic == "MSCD")
+                            {
+                                reader.ReadUInt32();
+                                reader.ReadUInt32();
+                                file.FileName = $"mscd/" + reader.ReadZeroTerminatedString();
+                            }
+                            else if (magic == "MSAD")
+                            {
+                                reader.ReadUInt32();
+                                file.FileName = $"msad/" + reader.ReadZeroTerminatedString();
+                            }
+                            else if (magic == "MMDL")
+                                file.FileName = $"models/" + file.FileName;
+                            else if (magic == "MCAN")
+                                file.FileName = $"cameras/" + file.FileName;
+                            else if (magic == "MANM")
+                                file.FileName = $"anims/" + file.FileName;
+                            else if (magic == "MNAV")
+                                file.FileName = $"collision/" + file.FileName;
+                            else if (magic == "bfwav")
+                                file.FileName = $"audio/" + file.FileName;
+                            else
+                                file.FileName = $"other/" + file.FileName;
                         }
                     }
 
@@ -95,21 +130,52 @@ namespace FirstPlugin
 
         public void Unload()
         {
-
+            _stream?.Dispose();
         }
 
         public void Save(System.IO.Stream stream)
         {
+            using (var writer = new FileWriter(stream))
+            {
+                writer.Write(uint.MaxValue); //header size
+                writer.Write(uint.MaxValue); //file size
+                writer.Write(files.Count);
+                for (int i = 0; i < files.Count; i++)
+                {
+                    ulong hash = ulong.Parse(Path.GetFileNameWithoutExtension(files[i].FileName));
+                    writer.Write(hash);
+                    writer.Write(uint.MaxValue); //start offset
+                    writer.Write(uint.MaxValue); //end offset
+                }
+                writer.Align(128);
+
+                writer.WriteUint32Offset(0, 4); //Size of header - 4
+                for (int i = 0; i < files.Count; i++)
+                {
+                    writer.WriteUint32Offset(20 + (i * 16)); //start offset
+                    files[i].FileDataStream.CopyTo(writer.BaseStream);
+                    writer.WriteUint32Offset(24 + (i * 16)); //end offset
+                }
+                using (writer.TemporarySeek(4, SeekOrigin.Begin)) {
+                    writer.Write((int)writer.BaseStream.Length);
+                }
+            }
         }
 
         public bool AddFile(ArchiveFileInfo archiveFileInfo)
         {
-            return false;
+            files.Add(new FileEntry()
+            {
+                FileDataStream = archiveFileInfo.FileDataStream,
+                FileName = archiveFileInfo.FileName,
+            });
+            return true;
         }
 
         public bool DeleteFile(ArchiveFileInfo archiveFileInfo)
         {
-            return false;
+            files.Remove((FileEntry)archiveFileInfo);
+            return true;
         }
 
         public class FileEntry : ArchiveFileInfo