diff --git a/test/auto_apply_plugin_metadata.py b/test/auto_apply_plugin_metadata.py index 7c96352..7cc66f1 100644 --- a/test/auto_apply_plugin_metadata.py +++ b/test/auto_apply_plugin_metadata.py @@ -1,59 +1,89 @@ import ast import json import sys +from ast import Call, Dict, Name, Constant, keyword def update_plugin_json(plugin_info, category): - name = next(iter(plugin_info)) - details = plugin_info[name] + name = plugin_info["plugin_name"] - with open(f"plugins/{category}/{category}.json", 'r+') as file: + with open(f"plugins/{category}/{category}.json", "r+") as file: data = json.load(file) try: # Check if plugin is already in the json - plugin = data['plugins'][name] - plugman_version = int(next(iter(data["plugins"][name]["versions"])).replace('.', '')) - current_version = int(next(iter(details["versions"])).replace('.', '')) + plugin = data["plugins"][name] + plugman_version = int(plugin["version"].replace(".", "")) + current_version = int(plugin_info["version"].replace(".", "")) # `or` In case another change was made on the plugin while still on pr if current_version > plugman_version or current_version == plugman_version: - plugin["versions"][next(iter(details["versions"]))] = None + plugin["versions"][plugin_info["version"]] = None # Ensure latest version appears first - plugin["versions"] = dict(sorted(plugin["versions"].items(), reverse=True)) - plugin["description"] = details["description"] - plugin["external_url"] = details["external_url"] - plugin["authors"] = details["authors"] + plugin["versions"] = dict( + sorted(plugin["versions"].items(), reverse=True) + ) + plugin["description"] = plugin_info["description"] + plugin["external_url"] = plugin_info["external_url"] + plugin["authors"] = [ + { + "name": ", ".join(plugin_info["author"]), + "discord": plugin_info["discord"], + "email": plugin_info["email"], + } + ] elif current_version < plugman_version: - raise Exception('Version cant be lower than the previous') + raise Exception("Version cant be lower than the previous") except KeyError: - data["plugins"][name] = details + data["plugins"][name] = { + "description": plugin_info["description"], + "external_url": plugin_info["external_url"], + "authors": [ + { + "name": ", ".join(plugin_info["author"]), + "discord": plugin_info["discord"], + "email": plugin_info["email"], + } + ], + "versions": {plugin_info["version"]: None}, + } + file.seek(0) json.dump(data, file, indent=2, ensure_ascii=False) # Ensure old content is removed file.truncate() -def extract_ba_plugman(plugins: str) -> dict | list: +def extract_plugman(plugins: str) -> dict: for plugin in plugins: if "plugins/" in plugin: # Split the path and get the part after 'plugins/' parts = plugin.split("plugins/")[1].split("/") category = parts[0] # First part after plugins/ + with open(plugin, "r") as f: + tree = ast.parse(f.read()) - with open(plugin, "r") as f: - tree = ast.parse(f.read()) + for node in ast.walk(tree): + if isinstance(node, ast.Assign) and len(node.targets) == 1: + target = node.targets[0] + if isinstance(target, ast.Name) and target.id == "plugman": + if isinstance(node.value, ast.Dict): + # Standard dictionary format {key: value} + return ast.literal_eval(node.value) + elif ( + isinstance(node.value, ast.Call) + and isinstance(node.value.func, ast.Name) + and node.value.func.id == "dict" + ): + # dict() constructor format + result = {} + for kw in node.value.keywords: + result[kw.arg] = ast.literal_eval(kw.value) + return result, category - for node in ast.walk(tree): - if isinstance(node, ast.Assign): - # Check if the assignment matches our variable name - if len(node.targets) == 1 and getattr(node.targets[0], "id", None) == 'ba_plugman': - # Convert AST node to a Python object safely - # return ast.literal_eval(node.value) - update_plugin_json(ast.literal_eval(node.value), category) - break - else: - raise ValueError(f"Variable ba_plugman not found.") + raise ValueError( + "Variable plugman not found in the file or has unsupported format." + ) if __name__ == "__main__": plugins = sys.argv - extract_ba_plugman(plugins) + update_plugin_json(extract_plugman(plugins))