#!/usr/bin/env python3
"""
从听记中提取所有待办事项并汇总

用法:
    python minutes_extract_todos.py              # 最近 5 条听记
    python minutes_extract_todos.py --max 10     # 最近 10 条
    python minutes_extract_todos.py --id <uuid>  # 指定听记
    python minutes_extract_todos.py --dry-run
"""

import sys
import json
import subprocess
import argparse
from pathlib import Path
from typing import List, Any, Optional

_scripts_dir = Path(__file__).resolve().parent
if str(_scripts_dir) not in sys.path:
    sys.path.insert(0, str(_scripts_dir))

from minutes_list_parse import uuid_title_pairs_from_payload


def run_dws(
    args: List[str], dry_run: bool = False,
) -> Optional[Any]:
    cmd = ['dws'] + args
    if dry_run:
        print(f"[dry-run] {' '.join(cmd)}")
        return None
    try:
        result = subprocess.run(
            cmd, capture_output=True, text=True, timeout=60
        )
        if result.returncode != 0:
            print(f"错误：{result.stderr.strip()}", file=sys.stderr)
            return None
        return json.loads(result.stdout)
    except (subprocess.TimeoutExpired, json.JSONDecodeError,
            FileNotFoundError) as e:
        print(f"错误：{e}", file=sys.stderr)
        return None


def todos_from_payload(payload: Any) -> List[dict]:
    """解析 `minutes get todos` 返回为待办列表。

    返回结构: result.dingtalkTodoList(对象数组, 含 title 等)与
    result.actions(JSON 字符串数组, 每条形如
    {"mark":[],"value":"..."}). 二者无 todos 键。
    优先取 dingtalkTodoList; 为空时回退解析 actions。
    """
    if isinstance(payload, dict):
        inner = payload.get('result', payload)
    else:
        inner = payload
    out: List[dict] = []
    if isinstance(inner, dict):
        ding_list = inner.get('dingtalkTodoList')
        if isinstance(ding_list, list) and ding_list:
            for t in ding_list:
                if isinstance(t, dict):
                    content = t.get('title') or t.get('content') or ''
                    if content:
                        out.append({'content': str(content), '_raw': t})
            if out:
                return out
        actions = inner.get('actions')
        if isinstance(actions, list):
            for a in actions:
                content = ''
                if isinstance(a, str):
                    text = a.strip()
                    if text.startswith('{'):
                        try:
                            parsed = json.loads(text)
                            content = parsed.get('value') or ''
                        except json.JSONDecodeError:
                            content = text
                    else:
                        content = text
                elif isinstance(a, dict):
                    content = (a.get('value') or a.get('content')
                               or a.get('title') or '')
                if content:
                    out.append({'content': str(content)})
    elif isinstance(inner, list):
        for t in inner:
            if isinstance(t, dict):
                content = (t.get('content') or t.get('text')
                           or t.get('title') or t.get('value') or '')
                if content:
                    out.append({'content': str(content), '_raw': t})
    return out


def main():
    parser = argparse.ArgumentParser(
        description='从听记中提取待办事项'
    )
    parser.add_argument('--max', type=int, default=5)
    parser.add_argument('--id', default='', help='指定听记 UUID')
    parser.add_argument('--dry-run', action='store_true')
    args = parser.parse_args()

    uuids_with_titles = []
    if args.id:
        uuids_with_titles = [(args.id, args.id)]
    else:
        print('🎙️ 获取听记列表...')
        data = run_dws([
            'minutes', 'list', 'mine',
            '--max', str(args.max),
            '--format', 'json',
        ], dry_run=args.dry_run)
        if args.dry_run:
            run_dws([
                'minutes', 'get', 'todos',
                '--id', '<TASK_UUID>', '--format', 'json',
            ], dry_run=True)
            return
        if not data:
            return
        uuids_with_titles = uuid_title_pairs_from_payload(data)

    all_todos = []
    for uuid, title in uuids_with_titles:
        print(f"  提取待办: {title}")
        todos_data = run_dws([
            'minutes', 'get', 'todos',
            '--id', uuid, '--format', 'json',
        ])
        if not todos_data:
            continue
        items = todos_from_payload(todos_data)
        for t in items:
            t['_source'] = title
        all_todos.extend(items)

    print(f"\n📋 听记待办汇总")
    print('=' * 50)

    if not all_todos:
        print('  ✅ 暂无待办事项')
        return

    for t in all_todos:
        if not isinstance(t, dict):
            print(f"  • {t!r}")
            continue
        content = (t.get('content') or t.get('text')
                   or t.get('title', ''))
        source = t.get('_source', '')
        print(f"  • {content}")
        if source:
            print(f"    来自: {source}")

    print(f"\n合计: {len(all_todos)} 条待办")


if __name__ == '__main__':
    main()