#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import sys
from http.server import BaseHTTPRequestHandler, HTTPServer

import dummy_killer
import dummy_pidfile


def make_embedding(text: str, dim: int = 32):
    # Deterministic: if text contains 'SPAM' (case-insensitive) -> ones; else zeros
    if 'spam' in text.lower():
        return [1.0] * dim
    return [0.0] * dim


class EmbeddingHandler(BaseHTTPRequestHandler):
    # OpenAI-like embeddings API
    def do_POST(self):
        length = int(self.headers.get('Content-Length', '0'))
        raw = self.rfile.read(length) if length > 0 else b''
        try:
            data = json.loads(raw.decode('utf-8') or '{}')
        except Exception:
            data = {}

        # Support both OpenAI ({input, model}) and Ollama ({prompt, model}) shapes
        text = data.get('input') or data.get('prompt') or ''
        # Optional dimension override for tests
        dim = int(data.get('dim') or 32)
        emb = make_embedding(text, dim)

        if 'openai' in (self.headers.get('User-Agent') or '').lower() or True:
            # Always reply in OpenAI-like format expected by neural provider
            body = {
                "data": [
                    {"embedding": emb}
                ]
            }
        else:
            body = {"embedding": emb}

        reply = json.dumps(body).encode('utf-8')
        self.send_response(200)
        self.send_header('Content-Type', 'application/json')
        self.send_header('Content-Length', str(len(reply)))
        self.end_headers()
        self.wfile.write(reply)

    def log_message(self, fmt, *args):
        # Keep test output quiet
        return


if __name__ == "__main__":
    import traceback
    try:
        alen = len(sys.argv)
        if alen > 1:
            port = int(sys.argv[1])
        else:
            port = 18080
        pid_path = sys.argv[2] if alen > 2 else dummy_pidfile.pid_path('llm', port)
        print(f"dummy_llm.py: Starting server on 127.0.0.1:{port}", file=sys.stderr)
        server = HTTPServer(("127.0.0.1", port), EmbeddingHandler)
        dummy_killer.write_pid(pid_path)
        print(f"dummy_llm.py: PID file written to {pid_path}", file=sys.stderr)
        print(f"dummy_llm.py: Server started successfully", file=sys.stderr)
        server.serve_forever()
    except KeyboardInterrupt:
        pass
    except Exception as e:
        print(f"dummy_llm.py: FATAL ERROR: {type(e).__name__}: {e}", file=sys.stderr)
        traceback.print_exc(file=sys.stderr)
        sys.exit(1)
    finally:
        if 'server' in dir():
            server.server_close()
