# 任务：实现 Claude 支付检测工具

## 需求

自动扫描邮箱收件箱，检测 Anthropic 的支付/退款邮件，自动给账号打标签并记录时间。

**匹配规则：**
- 发件人：`invoice+statements@mail.anthropic.com`
- 标题含 `Your receipt from Anthropic` → 标记 **已支付Claude**（status=`paid`），记录支付时间
- 标题含 `Your refund from Anthropic` → 标记 **已退款**（status=`refunded`），记录退款时间
- 时间取邮件的 `receivedDateTime`
- 如果同时存在支付和退款邮件，比较时间取最新的作为当前状态

---

## 实现方案（共 5 个改动点）

### 1. `database.py` — 新增表 + 4 个方法

在 `init_database()` 中创建新表：

```sql
CREATE TABLE IF NOT EXISTS claude_payment_status (
    email TEXT PRIMARY KEY,
    status TEXT DEFAULT 'unknown',   -- paid / refunded / unknown / error
    payment_time TEXT,
    refund_time TEXT,
    checked_at TEXT
)
```

新增 3 个 async 方法（参考现有 `get_account_tags` / `set_account_tags` 的写法）：
- `set_claude_payment_status(email, status, payment_time=None, refund_time=None)` — INSERT OR REPLACE，checked_at 取当前时间
- `get_claude_payment_status(email)` → Optional[Dict]
- `get_all_claude_payment_statuses()` → Dict[str, Dict]

修改 `delete_account()`：在删除账户时联动删除 `claude_payment_status` 表中对应记录。

### 2. `mail_api.py` — 检测逻辑 + 2 个新端点

添加 `StreamingResponse` 到 import：
```python
from fastapi.responses import FileResponse, JSONResponse, PlainTextResponse, StreamingResponse
```

在"命令行入口"注释之前，添加以下内容：

**常量：**
```python
ANTHROPIC_SENDER = "invoice+statements@mail.anthropic.com"
RECEIPT_KEYWORD = "Your receipt from Anthropic"
REFUND_KEYWORD = "Your refund from Anthropic"
```

**核心检测函数** `_check_claude_payment_for_account(email_addr)`：
1. 调用 `email_manager.get_client(email_addr)` 获取客户端
2. 调用 `client.get_messages_with_content(top=30)` 获取最近 30 封 INBOX 邮件
3. 遍历邮件，匹配 `sender.emailAddress.address` == ANTHROPIC_SENDER（不区分大小写）
4. 匹配标题关键词，记录最新的 payment_time 和 refund_time
5. 确定状态：两者都有则比较时间取最新；只有支付=paid；只有退款=refunded；都没有=unknown
6. 调用 `db_manager.set_claude_payment_status()` 写入数据库
7. 同时双写标签：获取现有标签，移除旧的"已支付Claude"/"已退款"标签，根据状态添加新标签
8. 返回 `{email, status, payment_time, refund_time}`
9. 出错时记录 error 状态并返回 `{email, status: "error", message: str(e)}`

**端点 1：** `POST /api/tools/check-claude-payment` — SSE 流式扫描
- 获取所有账户 `load_accounts_config()`
- 逐个调用 `_check_claude_payment_for_account()`
- SSE 事件格式（每行 `data: {JSON}\n\n`）：
  - `{type: "start", total: N}`
  - `{type: "progress", current: i, total: N, email: "..."}`（扫描前发送）
  - `{type: "result", current: i, total: N, email, status, payment_time, refund_time}`（扫描后发送）
  - `{type: "done", total: N}`
- 账户之间 `await asyncio.sleep(0.5)` 避免限速
- 返回 `StreamingResponse(event_generator(), media_type="text/event-stream")`

**端点 2：** `GET /api/tools/claude-payment-status` — 获取缓存状态
- 调用 `db_manager.get_all_claude_payment_statuses()`
- 返回 `ApiResponse(success=True, data=statuses)`

### 3. `static/index.html` — 按钮 + 表格列

**导航栏**：在 `nav-actions` 的"导出"按钮之后、"刷新"按钮之前，添加：
```html
<button class="btn btn-primary" id="claudePaymentBtn">
    <i class="bi bi-credit-card"></i>
    <span>Claude检测</span>
</button>
```

**表格表头**：在"令牌"和"操作"之间，添加新列：
```html
<th>Claude状态</th>
```
同时把所有 `colspan="6"` 改为 `colspan="7"`（空状态行）。

### 4. `static/script.js` — SSE 扫描 + 状态渲染

**constructor** 中新增：
```javascript
this.claudePaymentStatuses = {};
```

**bindAccountEvents()** 中新增按钮绑定：
```javascript
document.getElementById('claudePaymentBtn').addEventListener('click', () => this.startClaudePaymentCheck());
```

**loadAccounts()** 中：在 `renderAccounts()` 调用前，先 `await this.loadClaudePaymentStatuses()`。同时把所有 `colspan="6"` 改为 `colspan="7"`。

**renderAccounts()** 中：在令牌列 `</td>` 和操作列 `<td>` 之间，插入新列：
```javascript
<td>${this.renderClaudeStatusBadge(email)}</td>
```

**新增 4 个方法：**

```javascript
async loadClaudePaymentStatuses() {
    try {
        const resp = await fetch('/api/tools/claude-payment-status');
        if (resp.ok) {
            const result = await resp.json();
            if (result.success) this.claudePaymentStatuses = result.data || {};
        }
    } catch (e) { console.error('加载Claude支付状态失败:', e); }
}

renderClaudeStatusBadge(email) {
    const info = this.claudePaymentStatuses[email];
    if (!info) return '<span class="claude-badge claude-unknown">未检测</span>';
    const map = {
        paid:     { cls: 'claude-paid',     label: '已支付' },
        refunded: { cls: 'claude-refunded', label: '已退款' },
        unknown:  { cls: 'claude-unknown',  label: '未知' },
        error:    { cls: 'claude-error',    label: '错误' }
    };
    const s = map[info.status] || map.unknown;
    let tips = [];
    if (info.payment_time) tips.push('支付: ' + info.payment_time);
    if (info.refund_time)  tips.push('退款: ' + info.refund_time);
    if (info.checked_at)   tips.push('检测: ' + info.checked_at);
    const title = tips.length ? ` title="${this.escapeHtml(tips.join('\n'))}"` : '';
    return `<span class="claude-badge ${s.cls}"${title}>${s.label}</span>`;
}

async startClaudePaymentCheck() {
    const btn = document.getElementById('claudePaymentBtn');
    if (!btn) return;
    btn.disabled = true;
    const origHtml = btn.innerHTML;
    btn.innerHTML = '<i class="bi bi-hourglass-split"></i><span>检测中...</span>';

    try {
        const response = await fetch('/api/tools/check-claude-payment', { method: 'POST' });
        const reader = response.body.getReader();
        const decoder = new TextDecoder();
        let buffer = '';
        while (true) {
            const { done, value } = await reader.read();
            if (done) break;
            buffer += decoder.decode(value, { stream: true });
            const lines = buffer.split('\n');
            buffer = lines.pop();
            for (const line of lines) {
                if (!line.startsWith('data: ')) continue;
                try {
                    const data = JSON.parse(line.slice(6));
                    if (data.type === 'progress') {
                        btn.innerHTML = `<i class="bi bi-hourglass-split"></i><span>${data.current}/${data.total}</span>`;
                    } else if (data.type === 'result') {
                        this.claudePaymentStatuses[data.email] = {
                            status: data.status,
                            payment_time: data.payment_time || null,
                            refund_time: data.refund_time || null,
                            checked_at: new Date().toLocaleString('zh-CN')
                        };
                        this.updateClaudeBadgeInTable(data.email);
                    } else if (data.type === 'done') {
                        btn.innerHTML = '<i class="bi bi-check-circle"></i><span>完成</span>';
                        setTimeout(() => { btn.innerHTML = origHtml; btn.disabled = false; }, 2000);
                    }
                } catch (e) {}
            }
        }
    } catch (err) {
        console.error('Claude支付检测失败:', err);
        btn.innerHTML = '<i class="bi bi-exclamation-triangle"></i><span>失败</span>';
        setTimeout(() => { btn.innerHTML = origHtml; btn.disabled = false; }, 2000);
    }
}

updateClaudeBadgeInTable(email) {
    const tbody = document.getElementById('accountTableBody');
    if (!tbody) return;
    for (const row of tbody.querySelectorAll('tr')) {
        const firstTd = row.querySelector('td');
        if (firstTd && firstTd.textContent.includes(email)) {
            // Claude状态列 = 第6个td（index 5，在令牌后、操作前）
            const cells = row.querySelectorAll('td');
            if (cells.length >= 6) {
                cells[5].innerHTML = this.renderClaudeStatusBadge(email);
            }
            break;
        }
    }
}
```

### 5. `static/style.css` — 徽章样式

在文件末尾（`@media` 之前或之后均可）添加：

```css
/* Claude支付状态徽章 */
.claude-badge {
    display: inline-block;
    padding: 3px 10px;
    border-radius: 12px;
    font-size: 12px;
    font-weight: 600;
    cursor: default;
    white-space: nowrap;
}
.claude-paid     { background: rgba(16, 185, 129, 0.12); color: #059669; }
.claude-refunded { background: rgba(245, 87, 108, 0.12); color: #e11d48; }
.claude-unknown  { background: rgba(148, 163, 184, 0.15); color: #64748b; }
.claude-error    { background: rgba(245, 158, 11, 0.15); color: #d97706; }
```

---

## 验证步骤

1. `python mail_api.py web` → 访问 `http://localhost:5001/`
2. 表格应多出一列"Claude状态"，初始都显示灰色"未检测"
3. 点击导航栏"Claude检测"按钮 → 按钮显示进度 `1/N`、`2/N`...
4. 扫描完成后，有 Anthropic 收据的账号显示绿色"已支付"，有退款的显示红色"已退款"
5. 鼠标悬停徽章可看到支付/退款时间
6. 刷新页面后状态仍然保留（数据库持久化）