save-plugin.py

#!/usr/bin/env python3
"""
Simple file saving API for plugin development
Handles POST requests to save plugin files to disk
"""

import json
import os
import sys
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import parse_qs, urlparse

class FileSaveHandler(BaseHTTPRequestHandler):
    def do_OPTIONS(self):
        """Handle CORS preflight requests"""
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'POST, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
        self.end_headers()

    def do_POST(self):
        """Handle plugin file save requests"""
        try:
            # CORS headers
            self.send_response(200)
            self.send_header('Access-Control-Allow-Origin', '*')
            self.send_header('Content-Type', 'application/json')
            self.end_headers()

            # Parse request body
            content_length = int(self.headers['Content-Length'])
            post_data = self.rfile.read(content_length)
            data = json.loads(post_data.decode('utf-8'))

            # Extract parameters
            file_path = data.get('filePath')
            content = data.get('content')
            backup = data.get('backup', True)

            if not file_path or content is None:
                raise ValueError("Missing filePath or content")

            # Security check: ensure file is within project directory
            project_root = os.path.abspath('.')
            full_path = os.path.abspath(file_path)
            
            if not full_path.startswith(project_root):
                raise ValueError("File path outside project directory")

            # Create backup if requested
            if backup and os.path.exists(full_path):
                backup_path = f"{full_path}.backup"
                with open(full_path, 'r', encoding='utf-8') as f:
                    backup_content = f.read()
                with open(backup_path, 'w', encoding='utf-8') as f:
                    f.write(backup_content)

            # Ensure directory exists
            os.makedirs(os.path.dirname(full_path), exist_ok=True)

            # Write file
            with open(full_path, 'w', encoding='utf-8') as f:
                f.write(content)

            # Success response
            response = {
                'success': True,
                'message': f'File saved successfully: {file_path}',
                'filePath': file_path,
                'size': len(content)
            }
            
            self.wfile.write(json.dumps(response).encode('utf-8'))
            print(f"✅ Saved file: {file_path} ({len(content)} bytes)")

        except Exception as e:
            # Error response
            response = {
                'success': False,
                'error': str(e),
                'message': f'Failed to save file: {str(e)}'
            }
            
            self.wfile.write(json.dumps(response).encode('utf-8'))
            print(f"❌ Error saving file: {str(e)}")

    def log_message(self, format, *args):
        """Suppress default logging"""
        pass

if __name__ == '__main__':
    port = 10001  # Different port from main server
    server = HTTPServer(('', port), FileSaveHandler)
    print(f"🚀 File saving API server running on port {port}")
    print(f"💾 Ready to save plugin files via POST requests")
    try:
        server.serve_forever()
    except KeyboardInterrupt:
        print(f"\n🛑 File saving API server stopped")