journals.py

"""Archivist API — Journal entry examples."""

import os
import requests

BASE_URL = "https://api.myarchivist.ai"
API_KEY = os.environ["ARCHIVIST_API_KEY"]
HEADERS = {"x-api-key": API_KEY, "Content-Type": "application/json"}


def list_journal_entries(campaign_id: str, page: int = 1, size: int = 20, fields: str | None = None):
    """List journal entries with pagination. Use fields='card' for lightweight previews."""
    params = {"campaign_id": campaign_id, "page": page, "size": size}
    if fields:
        params["fields"] = fields
    resp = requests.get(
        f"{BASE_URL}/v1/journals",
        headers=HEADERS,
        params=params,
    )
    resp.raise_for_status()
    data = resp.json()

    print(f"Journal entries ({data['total']} total, page {data['page']}/{data['pages']}):\n")
    for entry in data["data"]:
        status_badge = f"[{entry.get('status', 'draft')}]"
        print(f"  {status_badge} {entry['title']} (tokens: {entry.get('token_count', 0)})")

    return data["data"]


def list_journal_cards(campaign_id: str, page: int = 1, size: int = 20):
    """List journal entries using fields=card for picker-style previews."""
    params = {"campaign_id": campaign_id, "page": page, "size": size, "fields": "card"}
    resp = requests.get(f"{BASE_URL}/v1/journals", headers=HEADERS, params=params)
    resp.raise_for_status()
    data = resp.json()

    print(f"Journal cards ({data['total']} total, page {data['page']}/{data['pages']}):\n")
    for entry in data["data"]:
        print(f"  {entry['title']}")
    return data["data"]


def get_journal_entry(entry_id: str):
    """Get a full journal entry including content."""
    resp = requests.get(f"{BASE_URL}/v1/journals/{entry_id}", headers=HEADERS)
    resp.raise_for_status()
    entry = resp.json()

    print(f"\n{'='*60}")
    print(f"Title: {entry['title']}")
    print(f"Status: {entry.get('status', 'draft')}")
    print(f"Tags: {', '.join(entry.get('tags', []))}")
    print(f"Permission: {entry.get('permission_level', 'N/A')}")
    print(f"{'='*60}")
    print(f"\n{entry.get('content', '(no content)')[:500]}")


def create_journal_entry(campaign_id: str):
    """Create a new journal entry."""
    payload = {
        "campaign_id": campaign_id,
        "title": "Session 12 — The Battle of Thornwall",
        "content": "The party finally reached Thornwall at dusk...",
        "tags": ["session-recap", "battle", "thornwall"],
        "is_public": False,
        "status": "draft",
    }

    resp = requests.post(f"{BASE_URL}/v1/journals", headers=HEADERS, json=payload)
    resp.raise_for_status()
    result = resp.json()

    print(f"\nCreated journal entry: {result['id']}")
    return result


def get_journal_markdown(entry_id: str):
    """Get a journal entry's content as Markdown (format=markdown).

    Headings, lists, code, diagrams and tables come back as Markdown instead of flattened text.
    Tables are GFM pipe tables, so the result can be sent back with content_format="markdown".
    """
    resp = requests.get(
        f"{BASE_URL}/v1/journals/{entry_id}",
        headers=HEADERS,
        params={"format": "markdown"},
    )
    resp.raise_for_status()
    return resp.json()["content"]


def create_markdown_journal_entry(campaign_id: str):
    """Create a journal entry from Markdown (content_format="markdown")."""
    content = """# Session 12 recap

| NPC | Attitude |
| --- | ---: |
| Mira | friendly |

```mermaid
graph TD
  Thornwall --> Harbor
```
"""
    payload = {
        "campaign_id": campaign_id,
        "title": "Session 12 recap (Markdown)",
        "content": content,
        "content_format": "markdown",  # default is "plain"
        "status": "draft",
    }
    resp = requests.post(f"{BASE_URL}/v1/journals", headers=HEADERS, json=payload)
    resp.raise_for_status()
    result = resp.json()
    print(f"\nCreated Markdown journal entry: {result['id']}")
    return result


def list_journal_folders(campaign_id: str):
    """List journal folders for organizing entries."""
    resp = requests.get(
        f"{BASE_URL}/v1/journal-folders",
        headers=HEADERS,
        params={"campaign_id": campaign_id},
    )
    resp.raise_for_status()
    data = resp.json()

    print(f"\nJournal folders ({len(data['data'])} total):\n")
    for folder in data["data"]:
        indent = "  " * folder["path"].count("/")
        print(f"  {indent}{folder['name']} (path: {folder['path']})")


if __name__ == "__main__":
    campaign_id = os.environ.get("ARCHIVIST_CAMPAIGN_ID")
    if not campaign_id:
        print("Set ARCHIVIST_CAMPAIGN_ID to run this example")
        exit(1)

    entries = list_journal_entries(campaign_id)
    if entries:
        get_journal_entry(entries[0]["id"])
    list_journal_folders(campaign_id)