From 3939bcb541c95c86428cbd1c0af170bab0ee914f Mon Sep 17 00:00:00 2001
From: Cory Solovewicz <cory@solovewicz.com>
Date: Thu, 18 May 2023 09:35:09 -0700
Subject: [PATCH] Create update_amiibo_tools_lua.py

This is a python script to automate what the updating of the amiibo_tools.lua
file which holds a lua table of all known amiibos. Previously updating the
amiibo_tools.lua was a very manual process and prone to errors. This simplifies the process.

Signed-off-by: Cory Solovewicz <cory@solovewicz.com>
---
 client/update_amiibo_tools_lua.py | 114 ++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 client/update_amiibo_tools_lua.py

diff --git a/client/update_amiibo_tools_lua.py b/client/update_amiibo_tools_lua.py
new file mode 100644
index 000000000..873a6e18d
--- /dev/null
+++ b/client/update_amiibo_tools_lua.py
@@ -0,0 +1,114 @@
+#!/usr/bin/env python3
+
+"""
+-----------------------------------------------------------------------------
+Name: update_amiibo_tools_lua.py
+
+Author: Cory Solovewicz
+
+Description:
+This is a python script to automate what the updating of the amiibo_tools.lua
+file which holds a lua table of all known amiibos. Previously updating the 
+amiibo_tools.lua was a very manual process.
+
+This script automates the following original command: 
+curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json | jq 'del(.amiibos[].release)' | jq 'del(.characters)' | pbcopy --> transform to table
+And outputs the formatted file as amiibo_tools.lua 
+If everything goes well, this should be an updated copy of amiibo_tools.lua 
+which can then be placed in the /lualibs/ directory. 
+The temporary amiibo.json file is then deleted 
+
+Dependencies:
+python3 -m pip install jq
+
+How to run:
+python update_amiibo_tools_lua.py
+The script will create the file amiibo_tools.lua
+
+After running, manually backup the original /lualibs/amiibo_tools.lua and move the 
+updated amiibo_tools.lua to the /lualibs/ directory.
+-----------------------------------------------------------------------------
+"""
+
+import os
+import re
+import subprocess
+import json
+from jq import jq
+
+def fetch_data():
+    print("Fetching amiibo.json")
+    # Perform the curl command
+    curl_command = "curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json"
+    curl_process = subprocess.Popen(curl_command.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    output, error = curl_process.communicate()
+
+    if curl_process.returncode != 0:
+        print("Error fetching data: ", error.decode())
+        return None
+    return output
+
+def filter_data(data):
+    print("Filtering downloaded data")
+    # Convert the output to JSON and use jq to filter data
+    data_json = json.loads(data)
+    filtered_data_json = jq('del(.amiibos[].release) | del(.characters)').transform(data_json)
+    # Convert the filtered JSON data back to a string, preserving Unicode characters
+    filtered_data = json.dumps(filtered_data_json, indent=2, ensure_ascii=False)
+    return filtered_data
+
+def save_data(filtered_data, filename):
+    # Save filtered data to file
+    with open(filename, 'w', encoding='utf-8') as file:
+        file.write(filtered_data)
+    print(f"Data saved to {filename}")
+
+def process_file(filename):
+    # Open the file
+    with open(filename, 'r', encoding='utf-8') as file:
+        data = file.read()
+
+    # Perform the replacements
+    data = data.replace('"name"', 'name')
+    data = data.replace('"amiibo_series"', 'amiibo_series')
+    data = data.replace('"amiibos"', 'amiibos')
+    data = data.replace('"game_series"', 'game_series')
+    data = data.replace('"types"', 'types')
+    data = data.replace(':', ' =')
+    data = re.sub('"0x', '["0x', data)
+    data = re.sub('" =', '"] =', data)
+
+    # Prepend the text
+    prepend_text = 'local amiibo_tools = {}\n\n-- curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json | jq \'del(.amiibos[].release)\' | jq \'del(.characters)\' | pbcopy --> transform to table\namiibo_tools.db =\n'
+    data = prepend_text + data
+
+    # Append the text
+    append_text = '\n\nreturn amiibo_tools\n'
+    data = data + append_text
+
+    return data
+
+def write_to_file(data, filename):
+    # Write the output
+    with open(filename, 'w', encoding='utf-8') as file:
+        file.write(data)
+    print(f"Output written to {filename}")
+
+def delete_file(filename):
+    try:
+        os.remove(filename)
+        print(f"Temporary file {filename} deleted")
+    except OSError as e:
+        print(f"Error deleting file {filename}: ", e)
+
+def main():
+    data = fetch_data()
+    if data:
+        filtered_data = filter_data(data)
+        save_data(filtered_data, 'amiibo.json')
+        processed_data = process_file('amiibo.json')
+        write_to_file(processed_data, 'amiibo_tools.lua')
+        delete_file('amiibo.json')
+
+if __name__ == "__main__":
+    main()