#!/usr/bin/env python3
# /// script
# requires-python = ">=3.11"
# dependencies = ["pymongo"]
# ///
"""Log entries to the notes.brag_sheet MongoDB collection.

Usage:
    brag_sheet.py add "<note>" "<context>" [--tags tag1,tag2]
    brag_sheet.py update <doc_id> [--note "..."] [--context "..."] [--tags tag1,tag2]
"""

import sys
from datetime import datetime
from bson import ObjectId
from pymongo import MongoClient

MONGO_URI = "mongodb://root:3wwfoUjyk2E2zWELXFlLuHqfw1ALlOp4pb2H5Vq3TImbMIHL2h1u8Jej2mjzCPl@docdb.connorrhodes.com:35563"


def parse_tags(args, i):
    raw = args[i + 1]
    return [t.strip() for t in raw.split(",") if t.strip()], i + 2


def handle_add(args):
    if len(args) < 2:
        print("Usage: brag_sheet.py add \"<note>\" \"<context>\" [--tags tag1,tag2]")
        sys.exit(1)

    note = args[0]
    context = args[1]
    tags = ["work"]

    i = 2
    while i < len(args):
        if args[i] == "--tags" and i + 1 < len(args):
            tags, i = parse_tags(args, i)
        else:
            print(f"Error: unknown argument '{args[i]}'")
            sys.exit(1)

    client = MongoClient(MONGO_URI)
    result = client["notes"]["brag_sheet"].insert_one({
        "date": datetime.now(),
        "note": note,
        "context": context,
        "tags": tags,
    })
    print(f"Inserted: {result.inserted_id}")
    client.close()


def handle_update(args):
    if not args:
        print("Usage: brag_sheet.py update <doc_id> [--note ...] [--context ...] [--tags tag1,tag2]")
        sys.exit(1)

    doc_id = args.pop(0)
    updates = {}
    i = 0
    while i < len(args):
        if args[i] == "--note" and i + 1 < len(args):
            updates["note"] = args[i + 1]; i += 2
        elif args[i] == "--context" and i + 1 < len(args):
            updates["context"] = args[i + 1]; i += 2
        elif args[i] == "--tags" and i + 1 < len(args):
            updates["tags"], i = parse_tags(args, i)
        else:
            print(f"Error: unknown argument '{args[i]}'")
            sys.exit(1)

    if not updates:
        print("Error: no fields to update")
        sys.exit(1)

    client = MongoClient(MONGO_URI)
    result = client["notes"]["brag_sheet"].update_one(
        {"_id": ObjectId(doc_id)},
        {"$set": updates}
    )
    if result.matched_count:
        print(f"Updated {doc_id}: {list(updates.keys())}")
    else:
        print(f"Error: document {doc_id} not found")
    client.close()


def main():
    args = sys.argv[1:]
    if not args:
        print(__doc__)
        sys.exit(1)

    cmd = args.pop(0)
    if cmd == "add":
        handle_add(args)
    elif cmd == "update":
        handle_update(args)
    else:
        print(f"Error: unknown command '{cmd}'. Use 'add' or 'update'.")
        sys.exit(1)


if __name__ == "__main__":
    main()