"""
Typed tool definitions for the Medication Care Agent (blueprint job B14), generated from the API's own OpenAPI
document so they can't drift from it. Served at GET /v1/tools; usage notes and example traces in docs/agent_tools.md.

Each tool: name (the blueprint's where it names one), description written for the agent (when to call it, what it
must not conclude), input_schema (JSON Schema; an uploaded file becomes an image reference the app's tool layer
resolves), output_schema ($refs into `components`), and how to call it (method, path, content type).
"""

TOOLS = [
    ("route_medication_image", "post", "/v1/check-input",
     "Before any other photo tool when you don't know what the photo shows: loose pills, blister pack, box or bottle, "
     "printed label or document, and which tools fit (next_tools). Prefer asking the user; when type is unchecked or "
     "next_tools is empty, ask for the box, label or prescription, or suggest the pharmacist."),
    ("detect_pills", "post", "/v1/detect",
     "Find each pill in a photo and say whether the count can be trusted. Call before counting pills or embedding "
     "several pills. If count_reliable is false, don't count: ask for a new photo (pills apart, plain surface)."),
    ("embed_pill", "post", "/v1/embed",
     "Turn one loose-pill photo (or a photo + box) into a 256-number embedding. Use at enrolment (store it with "
     "model_version and embedding_space_id) and before match_gallery. Never embed boxes or blister packs."),
    ("match_gallery", "post", "/v1/match",
     "Compare pill embeddings with the patient's gallery: match / unknown / ambiguous per pill. Visual evidence only: "
     "a match means 'looks like', not 'is' or 'safe to take'. ambiguous: ask for another photo. The gallery's "
     "model_version is required; on 409 re-embed the enrolment photos."),
    ("identify_pill", "post", "/v1/identify",
     "Possible catalogue matches (US products) for a pill that isn't in the patient's list. Report the verdict and "
     "confidence, never a definite identity; no_confident_match means the pill may not be a catalogue product."),
    ("search_by_embedding", "post", "/v1/search-by-embedding",
     "identify_pill for a pill you already embedded (e.g. match_gallery said unknown): send its embeddings and "
     "imprint_reads from embed_pill. Same answer, no second model run. Possible matches only, never an identity."),
    ("scan_photo", "post", "/v1/scan",
     "Diagnostics in one call: detection plus each pill's visual decision (with a gallery) or catalogue candidates. "
     "Don't use its dose check (deprecated): apply the backend's dose rules to the pills it returns."),
    ("extract_medication_document", "post", "/v1/read-label",
     "Read a Singapore pharmacy sticker (or a prescription or box; photo or PDF): per medicine the name, strength, "
     "RxNorm code, quantity, the instructions in parts (dose, times a day or every N hours, as needed, time of day, "
     "before/after food), what it is for, cautions and dates, each with its source line. Parts not printed are "
     "null; never fill them in. With the patient's medications, says same medicine / different strength / not on "
     "the list. Always have the patient confirm before anything is scheduled."),
    ("find_medication_product", "get", "/v1/pills/search",
     "Products and pictures for a medicine name (brand or generic), with strength, dose form and release as fields. "
     "Never treat products of another strength or form as the same."),
    ("screen_medication_list", "post", "/v1/safety/medications",
     "Interaction and duplicate-ingredient alerts for the whole medication list (DDInter 2.0). Read coverage: no "
     "alert does not mean safe. Advisory; defer to the pharmacist."),
    ("evaluate_dose_reference", "post", "/v1/safety/dose",
     "The reference dose rules on visual decisions (stateless). The backend's own dose policy is authoritative."),
    ("retrieve_medication_info", "get", "/v1/safety/label",
     "Warning sections of the official US label, with the label's date and DailyMed link for citation. General "
     "information, not the patient's instructions."),
    ("get_model_info", "get", "/v1/model-info",
     "The served model: version, embedding space, decision rule and known limits. Check after a 409."),
]
FILE_NOTE = "An image reference: the app's tool layer uploads the file (photo, or PDF for documents)"


def _resolve(schema, components):
    ref = schema.get("$ref", "") if isinstance(schema, dict) else ""
    return components.get(ref.rsplit("/", 1)[-1], {}) if ref else schema


def _input_schema(op, components):
    props, required = {}, []
    for p in op.get("parameters", []):
        props[p["name"]] = {**p.get("schema", {}), "description": p.get("description", "")}
        if p.get("required"):
            required.append(p["name"])
    content = op.get("requestBody", {}).get("content", {})
    for kind in ("application/json", "multipart/form-data"):
        if kind in content:
            body = _resolve(content[kind]["schema"], components)
            for name, s in body.get("properties", {}).items():
                is_file = s.get("format") == "binary" or s.get("contentMediaType") or \
                    (s.get("type") == "array" and (s.get("items", {}).get("format") == "binary"
                                                   or s.get("items", {}).get("contentMediaType")))
                props[name] = ({"type": "array", "items": {"type": "string"}, "description": FILE_NOTE}
                               if is_file and s.get("type") == "array" else
                               {"type": "string", "description": FILE_NOTE} if is_file else s)
            required += body.get("required", [])
    return {"type": "object", "properties": props, "required": sorted(set(required))}


def build_tools(openapi):
    components = openapi.get("components", {}).get("schemas", {})
    tools = []
    for name, method, path, description in TOOLS:
        op = openapi["paths"][path][method]
        ok = op.get("responses", {}).get("200", {}).get("content", {}).get("application/json", {})
        kind = next(iter(op.get("requestBody", {}).get("content", {"none": None})))
        tools.append({"name": name, "description": description, "input_schema": _input_schema(op, components),
                      "output_schema": ok.get("schema", {}),
                      "call": {"method": method.upper(), "path": path,
                               "content_type": None if kind == "none" else kind}})
    return {"tools": tools, "components": {"schemas": components},
            "notes": "Stage 2 answers are visual evidence only. The backend's safety policy decides actions; the "
                     "agent reports evidence with its limits and asks the patient to confirm."}