cli.py

"""Command-line interface; run with --help for options."""
from __future__ import annotations

import argparse
import os
import sys

from jobpilot import history
from jobpilot.agent import Agent
from jobpilot.export import export
from jobpilot.llm import LLMError, OllamaClient
from jobpilot.pipeline import ats_check, generate_cover_letter, generate_rewrite, rank_jobs


def main() -> int:
    ap = argparse.ArgumentParser(description="JobPilot: local AI job-application agent")
    ap.add_argument("--resume", help="Path to resume (PDF/DOCX/MD/TXT)")
    ap.add_argument("--job-url", default="", help="URL of the job posting")
    ap.add_argument("--job-file", default="", help="Text file containing the job description")
    ap.add_argument("--ats-only", action="store_true",
                    help="Standalone ATS test, no agent (job optional)")
    ap.add_argument("--rank", nargs="+", metavar="JOB_FILE",
                    help="Rank the resume against several posting files")
    ap.add_argument("--rewrite", action="store_true",
                    help="Generate the rewritten resume without asking")
    ap.add_argument("--cover-letter", action="store_true",
                    help="Generate a cover letter without asking (needs a job)")
    ap.add_argument("--history", action="store_true", help="Show past runs and score changes")
    args = ap.parse_args()

    if args.history:
        return _print_history()
    if not args.resume:
        ap.error("--resume is required (except with --history)")
    if args.rank:
        return _rank(args)

    job_text = ""
    if args.job_file:
        with open(args.job_file, "r", encoding="utf-8") as f:
            job_text = f.read()
    if not args.ats_only and not (job_text or args.job_url):
        ap.error("full agent mode needs --job-url or --job-file (or use --ats-only)")

    if args.ats_only:
        try:
            ctx = ats_check(args.resume, job_text=job_text, job_url=args.job_url)
        except (ValueError, LLMError) as exc:
            print(exc, file=sys.stderr)
            return 1
        _print_ats(ctx.ats_report)
        if ctx.extras.get("resume_review"):
            print("\n" + "=" * 60 + "\n" + ctx.extras["resume_review"])
    else:
        def show(step):
            if step["type"] == "tool_call":
                print(f"→ [{step['t']}s] calling {step['tool']}({step['args'] or ''})")
            elif step["type"] == "tool_result":
                print(f"  {'✓' if step['ok'] else '✗'} {step['detail'][:160]}")
            elif step["type"] == "error":
                print(f"  ! {step['detail']}", file=sys.stderr)

        result = Agent(OllamaClient()).run(
            resume_path=args.resume, job_url=args.job_url, job_text=job_text, on_step=show
        )
        ctx = result["context"]
        print("\n" + "=" * 60)
        print(result["answer"])
        if ctx.report_path:
            print(f"\nFull report: {ctx.report_path}")

    _offer_rewrite(ctx, args)
    _offer_cover_letter(ctx, args)
    return 0


def _offer_rewrite(ctx, args) -> None:
    question = (
        "Want to see your resume rewritten for this job?"
        if ctx.job_info
        else "Want a modified resume with an improved ATS score?"
    )
    wants = args.rewrite or (sys.stdin.isatty() and _ask(question))
    if not wants:
        return
    print("\nRewriting resume (never invents experience)...")
    md = generate_rewrite(ctx)
    print("=" * 60 + "\n" + md)
    base = os.path.splitext(ctx.extras["rewritten_resume_path"])[0]
    for fmt in ("pdf", "docx"):
        data, _ = export(md, fmt)
        with open(f"{base}.{fmt}", "wb") as f:
            f.write(data)
    print(f"\nSaved: {base}.md / .pdf / .docx")


def _offer_cover_letter(ctx, args) -> None:
    if not ctx.job_info:
        return
    wants = args.cover_letter or (sys.stdin.isatty() and _ask("Also want a cover letter for it?"))
    if not wants:
        return
    print("\nWriting cover letter (grounded in your resume)...")
    md = generate_cover_letter(ctx)
    print("=" * 60 + "\n" + md)
    path = os.path.join("reports", "cover-letter")
    os.makedirs("reports", exist_ok=True)
    for fmt in ("md", "pdf", "docx"):
        data, _ = export(md, fmt)
        with open(f"{path}.{fmt}", "wb") as f:
            f.write(data)
    print(f"\nSaved: {path}.md / .pdf / .docx")


def _ask(question: str) -> bool:
    return input(f"\n{question} [y/N] ").strip().lower() in ("y", "yes")


def _rank(args) -> int:
    texts = []
    for path in args.rank:
        with open(path, "r", encoding="utf-8") as f:
            texts.append(f.read())

    def progress(e):
        print(f"  scored {e['index']}/{e['total']}: {e['title']} — {e['overall']}/100")

    try:
        ranked = rank_jobs(args.resume, texts, on_progress=progress)
    except (ValueError, LLMError) as exc:
        print(exc, file=sys.stderr)
        return 1
    print(f"\n{'#':<3}{'score':<8}{'grade':<12}job")
    print("-" * 60)
    for i, e in enumerate(ranked, 1):
        print(f"{i:<3}{e['overall']:<8}{e['grade']:<12}{e['title']} @ {e['company']}")
        if e["missing_required"]:
            print(f"{'':>23}missing: {', '.join(e['missing_required'][:6])}")
    return 0


def _print_history() -> int:
    entries = history.load()
    if not entries:
        print("No runs recorded yet.")
        return 0
    print(f"{'when':<20}{'score':<8}{'Δ':<7}{'resume':<28}job")
    print("-" * 90)
    for e in entries[:25]:
        delta = "" if e["delta"] is None else f"{e['delta']:+.1f}"
        job = f"{e['job_title']} @ {e['company']}" if e["job_title"] else "(health check)"
        print(f"{e['ts']:<20}{e['overall']:<8}{delta:<7}{e['resume'][:26]:<28}{job}")
    return 0


def _print_ats(ats) -> None:
    print(f"\nATS score: {ats['overall']}/100 ({ats['grade']})")
    for name, res in ats["breakdown"].items():
        score = res.get("score")
        print(f"  {name:10s} {'skipped' if score is None else score}")
    kw = ats["breakdown"]["keywords"]
    if kw.get("missing_required"):
        print("  missing required: " + ", ".join(kw["missing_required"]))
    if kw.get("missing_keywords"):
        print("  missing keywords: " + ", ".join(kw["missing_keywords"][:10]))
    for issue in ats["breakdown"]["quality"]["issues"]:
        print(f"  issue: {issue}")


if __name__ == "__main__":
    sys.exit(main())