From 2e82e3d13ca812c3d3f5fda2359aa824067f5b7e Mon Sep 17 00:00:00 2001
From: Aaron Thompson <Aaron1178@gmail.com>
Date: Mon, 11 Jan 2021 13:36:31 +1100
Subject: [PATCH] NifExplorer - Code Redesign From Scratch based on TDD.

---
 nifexplorer.py | 86 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/nifexplorer.py b/nifexplorer.py
index 546c247..c7358c8 100644
--- a/nifexplorer.py
+++ b/nifexplorer.py
@@ -1,3 +1,6 @@
+import os
+import sys
+
 from pyffi.formats.nif import NifFormat
 
 class NifExplorer():
@@ -12,7 +15,90 @@ class NifExplorer():
     """The result path where the .nif files will be copied too. Result will be <ResultPath>/<BlockType>/"""
     ResultPath = None
 
+    """Set the BlockType, will resolve down to a string, but a valid class can be passed. e.g: NifFormat.NiNode"""
+    def SetBlockType(self, InBlockType):        
+        if isinstance(InBlockType, str):
+            self.BlockType = self.FindBlockTypeByName(InBlockType)
+
+        else:
+            BlockTypeToString = str(InBlockType)
+
+            BlockTypeToString = BlockTypeToString.split("'")[1]
+
+            self.BlockType = self.FindBlockTypeByName(BlockTypeToString)
+
+            if self.BlockType == None:
+                assert("NifExplorer.SetBlockType(): Cannot Resolve BlockType!")
+        
+    """Set the Search Path"""
+    def SetSearchPath(self, InSearchPath):
+        if not isinstance(InSearchPath, str):
+            assert "NifExplorer.SetSearchPath(): InSearchPath must be a string!"
+
+        elif self.MakeAbsolutePath(__file__, InSearchPath) != None:
+            self.SearchPath = self.MakeAbsolutePath(__file__, InSearchPath)
+
+        else:
+            assert "NifExplorer.SetSearchPath(): Cannot Resolve Search Path!"
+
+    """Set the Result Path, if the specified path doesn't exist, it will create it"""
+    def SetResultPath(self, InResultPath):
+        if not isinstance(InResultPath, str):
+            assert "NifExplorer.SetSearchPath(): InResultPath must be a string!"
+
+        elif self.MakeAbsolutePath(__file__, InResultPath) != None:
+            ResultPath = self.MakeAbsolutePath(__file__, InResultPath)
+
+            ResultPath += ("\\" + self.BlockTypeToString(self.BlockType))
+
+            if not os.path.exists(ResultPath):
+                print("Could not find Result Path: '%s', Creating Now!" % ResultPath)
+                os.makedirs(ResultPath)
+    
+            self.ResultPath = ResultPath
+
+        else:
+            assert "NifExplorer.SetResultPath(): Cannot Resolve Result Path!"
+
+    """Returns a string derived from a NifFormat BlockType"""
+    @staticmethod
+    def BlockTypeToString(BlockType):
+        s = str(BlockType)
+        strings = s.split("'")
+
+        if len(strings) > 0:
+            return strings[1]
+
+        else:
+            assert "NifExplorer.BlockTypeToString(): Could not resolve BlockType!"
+
+    """"Returns an absolute file path, where a could be __file__"""
+    @staticmethod
+    def MakeAbsolutePath(a,b):
+        str = os.path.dirname(os.path.realpath(a))
+
+        if b[0] != ("/" or "\\"):
+           b = "\\" + b
+        
+        str += b
+        str.replace("/", "\\")
+
+        if str[len(str)-1] != "\\":
+            str += "\\"
+
+        if not os.path.isabs(str):
+            assert "NifExplorer.MakeAbsolutePath(): Cou;d not make absolute path!"
 
+        return str
 
+    """Find a BlockType by Name via a string instance"""
+    def FindBlockTypeByName(self, BlockTypeName):
+        BlockTypeName = str(BlockTypeName)
 
+        if not isinstance(BlockTypeName, str):
+            assert "NifExplorer.FindBlockTypeByName(): Parameter 'BlockTypeName' Must be a string!"
 
+        for object in getattr(sys.modules["pyffi.formats.nif"], "NifFormat").__dict__.values():
+            if hasattr(object, "__name__"):
+                if object.__name__ == BlockTypeName and object != None:
+                    return object
\ No newline at end of file