Initial commit.

gbtoolkit.py

@@ -0,0 +1,78 @@
+import os
+import json
+import zlib
+from prompt_toolkit import prompt
+
+def calculate_checksum(file_path):
+    """Calculate the CRC-32 checksum for a given file"""
+    with open(file_path, 'rb') as f:
+        return hex(zlib.crc32(f.read()) & 0xffffffff)
+
+def check_checksum(checksum, checksum_file):
+    """Check the given checksum against the checksums in the specified file"""
+    if not os.path.exists(checksum_file):
+        print(f"{checksum_file} does not exist.")
+        return None
+    # Open the checksum file
+    with open(checksum_file, 'r') as f:
+        data = json.load(f)
+        if checksum in data:
+            return data[checksum]
+    # Return None if no match is found
+    return None
+
+def write_json(folder_path, json_file):
+    data = {}
+    for filename in os.listdir(folder_path):
+        if filename.endswith('.gb'):
+            file_path = os.path.join(folder_path, filename)
+            checksum = calculate_checksum(file_path)
+            data[checksum] = filename
+    with open(json_file, 'w') as f:
+        json.dump(data, f)
+
+def clear(): 
+    # for windows
+    if os.name == 'nt':
+        _ = os.system('cls')
+    # for mac and linux(here, os.name is 'posix')
+    else:
+        _ = os.system('clear')
+
+def main():
+    print("GB Rom Toolkit\n")
+    print("--------------------------------------------------")
+    print("Please select an option from the following:")
+    print("1. Check a checksum")
+    print("2. Scan the roms folder")
+    print("3. Exit")
+    print("--------------------------------------------------")
+
+    folder_path = '/home/dev/code/gbchecksum/roms'
+    checksum_file = '/home/dev/code/gbchecksum/roms.json'
+
+    while True:
+        choice = prompt("Enter your choice: ")
+        if choice == "1":
+            checksum = prompt("Enter the checksum to check: ")
+            name = check_checksum(checksum, checksum_file)
+            if name:
+                print(f'The checksum belongs to the file: {name}')
+            else:
+                print("No match found for the given checksum.")
+        elif choice == "2":
+            if not os.path.exists(folder_path):
+                print(f"{folder_path} does not exist.")
+            elif not os.listdir(folder_path):
+                print(f"There are no files in {folder_path}.")
+            else:
+                write_json(folder_path, checksum_file)
+                print("File created successfully")
+        elif choice == "3":
+            break
+        else:
+            print("Invalid choice, please enter a valid number.")
+
+if __name__ == '__main__':
+    main()
+