"""aubin-mini command line.

    aubin-mini decide request.json            # {"state": ..., "questions": {...}} → JSON result
    aubin-mini serve --port 8009              # POST /decide with the same JSON, GET /health
"""
import argparse
import json
import sys
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer

from . import AubinMini, __version__


def main(argv=None):
    ap = argparse.ArgumentParser(prog="aubin-mini")
    ap.add_argument("--model", default="emrevrg/AUBIN-Mini", help="HF repo or local directory")
    ap.add_argument("--variant", default="phone", choices=["phone", "nb4"])
    ap.add_argument("--threads", type=int, default=None)
    sub = ap.add_subparsers(dest="cmd", required=True)
    d = sub.add_parser("decide"); d.add_argument("request", help="JSON file, or - for stdin")
    s = sub.add_parser("serve"); s.add_argument("--host", default="127.0.0.1"); s.add_argument("--port", type=int, default=8009)
    a = ap.parse_args(argv)
    m = AubinMini.from_pretrained(a.model, a.variant, threads=a.threads)

    if a.cmd == "decide":
        req = json.load(sys.stdin if a.request == "-" else open(a.request, encoding="utf-8"))
        print(json.dumps(m.decide(req.get("state", ""), req["questions"]), ensure_ascii=False, indent=2))
        return

    class H(BaseHTTPRequestHandler):
        def _send(self, code, obj):
            b = json.dumps(obj, ensure_ascii=False).encode("utf-8")
            self.send_response(code); self.send_header("Content-Type", "application/json; charset=utf-8")
            self.send_header("Content-Length", str(len(b))); self.end_headers(); self.wfile.write(b)

        def do_GET(self):
            if self.path == "/health":
                self._send(200, {"ok": True, "model": "AUBIN Mini", "version": __version__, "variant": m.variant})
            else:
                self._send(404, {"error": "not found"})

        def do_POST(self):
            if self.path != "/decide":
                return self._send(404, {"error": "not found"})
            try:
                req = json.loads(self.rfile.read(int(self.headers.get("Content-Length", 0))) or b"{}")
                self._send(200, m.decide(req.get("state", ""), req["questions"]))
            except KeyError as e:
                self._send(400, {"error": f"missing field: {e.args[0]}"})
            except (ValueError, TypeError, AttributeError) as e:
                self._send(400, {"error": str(e)})

        def log_message(self, *args):
            pass

    print(f"AUBIN Mini {__version__} ({m.variant}) on http://{a.host}:{a.port}  POST /decide", flush=True)
    ThreadingHTTPServer((a.host, a.port), H).serve_forever()


if __name__ == "__main__":
    main()