| # SPDX-License-Identifier: MIT | |
| """ | |
| Base64 Plugin for TrashClaw | |
| Encode or decode text and files using Base64. | |
| Plugin contract: | |
| TOOL_DEF = dict with name, description, parameters (OpenAI function schema) | |
| run(**kwargs) -> str (called with the parsed arguments) | |
| """ | |
| import base64 as _b64 | |
| import os | |
| TOOL_DEF = { | |
| "name": "base64", | |
| "description": "Encode or decode text using Base64 encoding.", | |
| "parameters": { | |
| "type": "object", | |
| "properties": { | |
| "text": { | |
| "type": "string", | |
| "description": "Text to encode or decode" | |
| }, | |
| "action": { | |
| "type": "string", | |
| "description": "Action to perform: encode or decode (default: encode)" | |
| } | |
| }, | |
| "required": ["text"] | |
| } | |
| } | |
| def run(text: str = "", action: str = "encode", **kwargs) -> str: | |
| action = action.lower().strip() | |
| if not text: | |
| return "Error: provide 'text' to encode or decode." | |
| if action == "encode": | |
| try: | |
| encoded = _b64.b64encode(text.encode("utf-8")).decode("utf-8") | |
| return f"Base64 encoded:\n{encoded}" | |
| except Exception as e: | |
| return f"Error encoding: {e}" | |
| elif action == "decode": | |
| try: | |
| # Add padding if needed | |
| padded = text + "=" * (4 - len(text) % 4) if len(text) % 4 else text | |
| decoded = _b64.b64decode(padded).decode("utf-8") | |
| return f"Base64 decoded:\n{decoded}" | |
| except Exception as e: | |
| return f"Error decoding: {e}" | |
| else: | |
| return f"Unknown action '{action}'. Use: encode, decode" | |