"""Append one thought to today's raw file."""
import sys
from datetime import datetime
from pathlib import Path

from common import find_ustht, read_define_ini, validate_dim_name

HELP = """Usage: python write_raw.py "thought text" [--dim dimension] [--help]

Append one thought to today's #raw/ markdown file.

Arguments:
  "thought text"     Thought text to record (required)
  --dim dimension    Suggested dimension, such as rules or ui/outline
  --help             Show this help text

Behavior:
  - If today's raw file is already processed, creates a numbered file such as 2026-06-01-2.md.
  - If the day has more than five raw entries, suggests /ustht sortin.
  - If SKILL_STATUS=off, exits without writing.
"""


def count_today_raw(raw_dir: Path) -> int:
    """Count unprocessed entries across today's raw files."""
    today = datetime.now().strftime("%Y-%m-%d")
    count = 0
    for f in sorted(raw_dir.glob(f"{today}*.md")):
        content = f.read_text(encoding="utf-8")
        first_line = content.split("\n", 1)[0].strip()
        if first_line == "<!-- processed -->":
            continue
        count += sum(1 for line in content.splitlines() if line.strip().startswith("- ["))
    return count


def main():
    if "--help" in sys.argv or "-h" in sys.argv:
        print(HELP)
        sys.exit(0)

    ustht = find_ustht()
    if ustht is None:
        print("Error: .ustht/ was not found. Run /ustht init first.")
        sys.exit(1)

    cfg = read_define_ini(ustht)
    if cfg.get("SKILL_STATUS") == "off":
        print("SKILL is off; write ignored.")
        sys.exit(0)

    thought = None
    dim = None
    args = sys.argv[1:]
    i = 0
    while i < len(args):
        if args[i] == "--dim" and i + 1 < len(args):
            dim = args[i + 1]
            i += 2
        elif thought is None:
            thought = args[i]
            i += 1
        else:
            i += 1

    if not thought:
        print("Error: missing thought text.")
        print(f"Usage: {sys.argv[0]} \"thought text\" [--dim dimension]")
        sys.exit(1)

    if dim and not validate_dim_name(dim):
        print(f"Invalid dimension name: {dim}. Use lowercase letters, digits, hyphens, and optional / subdirectories.")
        sys.exit(1)

    raw_dir = ustht / "raw"
    raw_dir.mkdir(exist_ok=True)

    today = datetime.now().strftime("%Y-%m-%d")
    now = datetime.now().strftime("%H:%M")
    raw_file = raw_dir / f"{today}.md"

    if raw_file.exists():
        first_line = raw_file.read_text(encoding="utf-8").split("\n", 1)[0].strip()
        if first_line == "<!-- processed -->":
            seq = 2
            while (raw_dir / f"{today}-{seq}.md").exists():
                seq += 1
            raw_file = raw_dir / f"{today}-{seq}.md"

    thought_clean = thought.replace("\n", " ").replace("\r", "")
    suffix = f" | suggested-dim:{dim}" if dim else ""
    entry = f"- [{now}] {thought_clean}{suffix}"

    if raw_file.exists():
        content = raw_file.read_text(encoding="utf-8").rstrip()
        raw_file.write_text(f"{content}\n{entry}\n", encoding="utf-8")
    else:
        raw_file.write_text(f"{entry}\n", encoding="utf-8")

    count = count_today_raw(raw_dir)
    if count > 5:
        print(f"Today has {count} recorded thoughts. Consider running /ustht sortin.")


if __name__ == "__main__":
    main()