From 07779883305ec4992e443a8c1fec227b9e0d0480 Mon Sep 17 00:00:00 2001
From: Georgii Korogodov <korogodoff@yandex.ru>
Date: Sat, 25 Jun 2022 22:58:28 +0400
Subject: [PATCH] additional data for books

---
 NbtScrapper/NbtScrapper/MainForm.cs        | 80 +++++++++++++++++-----
 NbtScrapper/NbtScrapper/NbtScrapper.csproj |  2 +
 2 files changed, 63 insertions(+), 19 deletions(-)

diff --git a/NbtScrapper/NbtScrapper/MainForm.cs b/NbtScrapper/NbtScrapper/MainForm.cs
index 5787c63..e94f431 100644
--- a/NbtScrapper/NbtScrapper/MainForm.cs
+++ b/NbtScrapper/NbtScrapper/MainForm.cs
@@ -41,25 +41,37 @@ private void extractButton_Click(object sender, EventArgs e)
         extractButton.Enabled = false;
         var regionFiles = new List<string>();
         var playerFiles = new List<string>();
-        var regionPath = Path.Combine(inputPathTextBox.Text, "region");
-        if (Directory.Exists(regionPath))
-        {
-            regionFiles.AddRange(Directory.GetFiles(regionPath));
-        }
+        var path = inputPathTextBox.Text;
 
-        var entitiesPath = Path.Combine(inputPathTextBox.Text, "entities");
-        if (Directory.Exists(entitiesPath))
-        {
-            regionFiles.AddRange(Directory.GetFiles(entitiesPath));
-        }
+        RecursiveFolderSearch(path);
 
-        var playerDataPath = Path.Combine(inputPathTextBox.Text, "playerdata");
-        if (Directory.Exists(playerDataPath))
+        Task.Run(() => FindBooksAndSigns(regionFiles, playerFiles));
+
+        void RecursiveFolderSearch(string path)
         {
-            playerFiles.AddRange(Directory.GetFiles(playerDataPath));
-        }
+            var regionPath = Path.Combine(path, "region");
+            if (Directory.Exists(regionPath))
+            {
+                regionFiles.AddRange(Directory.GetFiles(regionPath));
+            }
 
-        Task.Run(() => FindBooksAndSigns(regionFiles, playerFiles));
+            var entitiesPath = Path.Combine(path, "entities");
+            if (Directory.Exists(entitiesPath))
+            {
+                regionFiles.AddRange(Directory.GetFiles(entitiesPath));
+            }
+
+            var playerDataPath = Path.Combine(path, "playerdata");
+            if (Directory.Exists(playerDataPath))
+            {
+                playerFiles.AddRange(Directory.GetFiles(playerDataPath));
+            }
+
+            foreach (var dir in Directory.GetDirectories(path))
+            {
+                RecursiveFolderSearch(dir);
+            }
+        }
     }
 
     private void FindBooksAndSigns(ICollection<string> regionFiles, ICollection<string> playerFiles)
@@ -110,6 +122,7 @@ private void FindBooksAndSigns(ICollection<string> regionFiles, ICollection<stri
                 var region = new RegionFile(file);
                 var j = 0;
                 var count = region.AllChunks.Count();
+
                 foreach (var chunk in region.AllChunks)
                 {
                     j++;
@@ -125,7 +138,7 @@ private void FindBooksAndSigns(ICollection<string> regionFiles, ICollection<stri
                         {
                             continue;
                         }
-                        
+
                         chunk.Load();
                         CyclicSearch(chunk.Data, SearchMethod);
                         chunk.Remove();
@@ -165,7 +178,7 @@ void SearchMethod(NbtContainerTag container, NbtTag tag)
 
                     if (pages != null && pages.Count != 0)
                     {
-                        var text = "";
+                        var text = string.Empty;
                         foreach (var page in pages)
                         {
                             var isSnbt = SnbtParser.ClassicTryParse(page.StringValue, false, out var result);
@@ -180,9 +193,39 @@ void SearchMethod(NbtContainerTag container, NbtTag tag)
 
                         hashes.Add(bookHash);
 
+                        NbtContainerTag inventory = null;
+                        NbtTag id = null;
+                        if (container.Parent is NbtCompound)
+                        {
+                            id = container.Parent?["id"];
+                            if (id != null)
+                            {
+                                inventory = container.Parent;
+                            }
+                            else if (container.Parent?.Parent is NbtCompound)
+                            {
+                                id = container.Parent?.Parent?["id"];
+                                if (id != null)
+                                {
+                                    inventory = container.Parent.Parent;
+                                }
+                            }
+                        }
+
+
+                        var coords = string.Empty;
+                        if (inventory?["x"] != null)
+                        {
+                            coords =
+                                $"at x:{inventory["x"].IntValue}, y:{inventory["y"].IntValue}, z:{inventory["z"].IntValue}";
+                        }
+
                         bookFileWriter.WriteLine("=====================================");
                         bookFileWriter.WriteLine(
                             $@"Book ""{(book["title"] == null ? "No name" : book["title"].StringValue)}"" by {(book["author"] == null ? "Unknown" : book["author"].StringValue)}");
+                        bookFileWriter.WriteLine($"(content hash = {Convert.ToHexString(bookHash)})");
+                        bookFileWriter.WriteLine(
+                            $"Found {container["Count"].ByteValue}x {tag.StringValue} in slot {(container["Slot"] == null ? "None" : container["Slot"].ByteValue)} {(id == null ? string.Empty : "of block entity " + id.StringValue)} {coords}");
                         bookFileWriter.WriteLine();
                         var i = 0;
                         foreach (var page in pages)
@@ -252,7 +295,7 @@ private void RecursiveSearch(NbtContainerTag container, Action<NbtContainerTag,
 
     private void CyclicSearch(NbtContainerTag rootTag, Action<NbtContainerTag, NbtTag> action)
     {
-        var stack = new ConcurrentStack<NbtContainerTag>();
+        var stack = new Stack<NbtContainerTag>();
         stack.Push(rootTag);
         while (stack.TryPop(out var container))
         {
@@ -267,7 +310,6 @@ private void CyclicSearch(NbtContainerTag rootTag, Action<NbtContainerTag, NbtTa
                     }
                 }
             }
-                
         }
     }
 
diff --git a/NbtScrapper/NbtScrapper/NbtScrapper.csproj b/NbtScrapper/NbtScrapper/NbtScrapper.csproj
index fdebf98..66953e2 100644
--- a/NbtScrapper/NbtScrapper/NbtScrapper.csproj
+++ b/NbtScrapper/NbtScrapper/NbtScrapper.csproj
@@ -5,6 +5,8 @@
     <TargetFramework>net6.0-windows</TargetFramework>
     <Nullable>enable</Nullable>
     <UseWindowsForms>true</UseWindowsForms>
+	<PublishSingleFile>true</PublishSingleFile>
+    <SelfContained>false</SelfContained>
     <ImplicitUsings>enable</ImplicitUsings>
   </PropertyGroup>