server.py

#!/usr/bin/env python3
"""
Accessibility Inspector Server
Serves static files and handles refresh/scan commands
"""

import http.server
import socketserver
import subprocess
import json
import os

PORT = 8080
DIRECTORY = os.path.dirname(os.path.abspath(__file__))

class InspectorHandler(http.server.SimpleHTTPRequestHandler):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, directory=DIRECTORY, **kwargs)

    def end_headers(self):
        # Disable caching for all files
        self.send_header('Cache-Control', 'no-store, no-cache, must-revalidate, max-age=0')
        self.send_header('Pragma', 'no-cache')
        self.send_header('Expires', '0')
        super().end_headers()

    def do_POST(self):
        if self.path == '/api/refresh':
            self.handle_refresh()
        elif self.path == '/api/scan':
            self.handle_scan()
        else:
            self.send_error(404, 'Not Found')

    def handle_refresh(self):
        """Run the full refresh: UI dump, screenshot, parse, and scan"""
        try:
            self.log_message("Running refresh...")

            steps = []

            # 1. Capture UI dump
            result = subprocess.run(
                ['adb', 'shell', 'uiautomator', 'dump'],
                capture_output=True, text=True, timeout=30
            )
            steps.append({'step': 'UI dump', 'success': result.returncode == 0})

            # 2. Pull the dump file
            result = subprocess.run(
                ['adb', 'pull', '/sdcard/window_dump.xml', os.path.join(DIRECTORY, 'window_dump.xml')],
                capture_output=True, text=True, timeout=30
            )
            steps.append({'step': 'Pull XML', 'success': result.returncode == 0})

            # 3. Take screenshot
            with open(os.path.join(DIRECTORY, 'screen.png'), 'wb') as f:
                result = subprocess.run(
                    ['adb', 'exec-out', 'screencap', '-p'],
                    stdout=f, stderr=subprocess.PIPE, timeout=30
                )
            steps.append({'step': 'Screenshot', 'success': result.returncode == 0})

            # 4. Parse UI data
            result = subprocess.run(
                ['python3', os.path.join(DIRECTORY, 'parse_ui.py')],
                capture_output=True, text=True, timeout=30
            )
            steps.append({'step': 'Parse UI', 'success': result.returncode == 0})

            # 5. Run accessibility check
            result = subprocess.run(
                ['python3', os.path.join(DIRECTORY, 'wcag_inspector.py')],
                capture_output=True, text=True, timeout=60
            )
            steps.append({'step': 'A11y scan', 'success': result.returncode == 0})

            success = all(s['success'] for s in steps)

            self.send_response(200)
            self.send_header('Content-type', 'application/json')
            self.send_header('Access-Control-Allow-Origin', '*')
            self.end_headers()
            self.wfile.write(json.dumps({
                'success': success,
                'steps': steps
            }).encode())

        except subprocess.TimeoutExpired:
            self.send_error_json(500, 'Command timed out')
        except Exception as e:
            self.send_error_json(500, str(e))

    def handle_scan(self):
        """Run only the accessibility scan"""
        try:
            result = subprocess.run(
                ['python3', os.path.join(DIRECTORY, 'wcag_inspector.py')],
                capture_output=True, text=True, timeout=60
            )

            self.send_response(200)
            self.send_header('Content-type', 'application/json')
            self.send_header('Access-Control-Allow-Origin', '*')
            self.end_headers()
            self.wfile.write(json.dumps({
                'success': result.returncode == 0,
                'output': result.stdout
            }).encode())

        except Exception as e:
            self.send_error_json(500, str(e))

    def send_error_json(self, code, message):
        self.send_response(code)
        self.send_header('Content-type', 'application/json')
        self.send_header('Access-Control-Allow-Origin', '*')
        self.end_headers()
        self.wfile.write(json.dumps({
            'success': False,
            'error': message
        }).encode())

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


def main():
    with socketserver.TCPServer(("", PORT), InspectorHandler) as httpd:
        print(f"Accessibility Inspector Server")
        print(f"Serving at http://localhost:{PORT}")
        print(f"Inspector: http://localhost:{PORT}/ui_inspector_v2.html")
        print(f"\nPress Ctrl+C to stop")
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            print("\nServer stopped.")


if __name__ == '__main__':
    main()