// cloudquota.go — 雲端資料庫（Cloudflare D1）每日免費額度用完時，講明白、且不再重打
// （`inkstone/arcrun-rag#197`，母票 `inkstone/InkStoneCo#132`）。
//
// 病（2026-09-13 實撞，youlin 與 geek6688 兩台免費帳號同一天）：
// D1 當日讀取額度用完，雲端每一件事都失敗。使用者在小幫手上看到的是
// 「缺少 credential: kbdb_internal_token」（假的，`inkstone/Arcrun#216` c7066 已查證）
// 與「連續失敗 N 次，已暫停自動重試」——**沒有任何一處說「額度用完了」**。
//
// 而雲端其實早就知道：cypher `/health` 的 `data_layer.probe_error` 當下寫著
//
//	D1_ERROR: Your account has exceeded D1's free tier daily row read limit.
//	Upgrade to a paid plan or wait until tomorrow (midnight UTC) to continue.
//
// 小幫手每分鐘本來就在打這支 `/health`（cloudcheck.go），只是只讀了版本號。
// ⇒ 本檔把「知道的地方」接到「用戶看得到的地方」，並且在額度恢復前**整個帳號不再打雲端寫入**
// （恢復前每一發都只會拿到同一個錯，還會讓免費帳號的其他額度一起被燒）。
//
// 🔴 用量數字：Cloudflare 只在帳號層的 GraphQL 分析 API 給得出「今天用了幾列」，
// 而小幫手身上沒有任何 Cloudflare 金鑰，安裝器要的授權範圍也沒有分析讀取權限
// （installer/oauth-prototype/worker.js `OAUTH_SCOPES`）。所以這一版**不假裝查得到**：
// 上限是官方公開的固定值（developers.cloudflare.com/d1/platform/pricing/，
// Workers Free：每天讀 500 萬列、寫 10 萬列，00:00 UTC 重置），
// 而「已經用到上限」是錯誤訊息本身證明的事實——兩句都是真的，不必查。
package collector

import (
	"encoding/json"
	"fmt"
	"strings"
	"sync"
	"time"
)

// D1 免費方案每日上限（Cloudflare 官方定價頁，2026-09-13 查）。
const (
	d1FreeRowsReadPerDay    = 5_000_000
	d1FreeRowsWrittenPerDay = 100_000
)

// QuotaNotice.Kind 的值。空字串＝舊的 Workers AI 額度（向後相容 status.json）。
const (
	QuotaKindD1Read  = "d1_read"
	QuotaKindD1Write = "d1_write"
)

// d1QuotaKind 從任何一段上游文字認出「D1 每日免費額度用完」，回 read／write／""。
// 只認 Cloudflare 錯誤原文裡的固定片段——「D1_ERROR」單獨出現不算（別的 D1 錯誤也帶它）。
func d1QuotaKind(text string) string {
	t := strings.ToLower(text)
	if !strings.Contains(t, "free tier") && !strings.Contains(t, "daily") {
		return ""
	}
	switch {
	case strings.Contains(t, "row read limit"), strings.Contains(t, "rows read"):
		return QuotaKindD1Read
	case strings.Contains(t, "row write limit"), strings.Contains(t, "rows written"), strings.Contains(t, "row written"):
		return QuotaKindD1Write
	}
	return ""
}

// d1QuotaFromHealth 讀 cypher `/health` 的回應，回它回報的 D1 額度狀況（""＝沒有）。
// 舊版雲端（沒有 data_layer 區塊，例如 1.4.46）一律回 ""——認不出來就不編故事。
func d1QuotaFromHealth(body []byte) string {
	var payload struct {
		DataLayer *struct {
			OK         bool   `json:"ok"`
			ProbeError string `json:"probe_error"`
			Summary    string `json:"summary"`
		} `json:"data_layer"`
	}
	if err := json.Unmarshal(body, &payload); err != nil || payload.DataLayer == nil || payload.DataLayer.OK {
		return ""
	}
	if k := d1QuotaKind(payload.DataLayer.ProbeError); k != "" {
		return k
	}
	return d1QuotaKind(payload.DataLayer.Summary)
}

type d1QuotaState struct {
	kind  string
	seen  time.Time // 最後一次親眼看到額度用完的時間
	until time.Time // 下一次 00:00 UTC——過了就一定不再相信這筆紀錄
}

var (
	d1QuotaMu   sync.Mutex
	d1QuotaSeen = map[string]d1QuotaState{} // key＝cloudCheckKey(cypherURL)
)

func resetD1Quota() {
	d1QuotaMu.Lock()
	defer d1QuotaMu.Unlock()
	d1QuotaSeen = map[string]d1QuotaState{}
}

// noteD1Quota 記下「這台知識庫此刻的 D1 額度狀況」。kind==""＝雲端親口說資料層正常 ⇒ 清掉。
func noteD1Quota(cypherURL, kind string, now time.Time) {
	key := cloudCheckKey(cypherURL)
	d1QuotaMu.Lock()
	defer d1QuotaMu.Unlock()
	if kind == "" {
		delete(d1QuotaSeen, key)
		return
	}
	d1QuotaSeen[key] = d1QuotaState{kind: kind, seen: now, until: nextQuotaResetTaiwan(now)}
}

// activeD1Quota 回這台知識庫現在是否處於 D1 額度用完（過了重置時間的舊紀錄一律丟掉）。
func activeD1Quota(cypherURL string, now time.Time) (d1QuotaState, bool) {
	key := cloudCheckKey(cypherURL)
	d1QuotaMu.Lock()
	defer d1QuotaMu.Unlock()
	st, ok := d1QuotaSeen[key]
	if !ok {
		return d1QuotaState{}, false
	}
	if !now.Before(st.until) {
		delete(d1QuotaSeen, key)
		return d1QuotaState{}, false
	}
	return st, true
}

// buildD1QuotaNotice 組出給用戶看的話。欄位沿用 QuotaNotice（App 首頁那張卡、托盤狀態列都吃它）。
//
// 用戶要知道的四件事（票上的驗收條件）：哪一種額度／用了多少、上限多少／台北幾點恢復／恢復後要不要做事。
func buildD1QuotaNotice(kind string, now, resetAt time.Time) QuotaNotice {
	what, limit := "讀取", fmt.Sprintf("每天 %s列", humanRows(d1FreeRowsReadPerDay))
	if kind == QuotaKindD1Write {
		what, limit = "寫入", fmt.Sprintf("每天 %s列", humanRows(d1FreeRowsWrittenPerDay))
	}
	return QuotaNotice{
		Kind:        kind,
		Headline:    fmt.Sprintf("你的雲端知識庫今天的免費%s額度用完了", what),
		Usage:       fmt.Sprintf("Cloudflare 免費方案的資料庫%s上限是%s，今天已經用到上限（這不是小幫手或你的檔案壞掉）", what, limit),
		Achievement: fmt.Sprintf("雲端資料庫今天的免費%s額度用完了", what),
		ExitOptions: "升級 Cloudflare Workers 付費方案（每月 5 美元起）就沒有每日上限",
		Guarantee: fmt.Sprintf("台北時間%s早上 8:00 恢復，恢復後小幫手會自動接著傳，你不用做任何事",
			quotaResetDayWord(now, resetAt)),
		ResumeAt: resetAt.Format(time.RFC3339),
	}
}

func humanRows(n int) string {
	if n >= 10_000 && n%10_000 == 0 {
		return fmt.Sprintf("%d 萬", n/10_000)
	}
	return fmt.Sprintf("%d", n)
}

// d1QuotaNote＝「這一發不打，因為雲端資料庫額度用完」的一句話；空＝可以打。
//
// 🔴 結尾「會自動恢復」是 explainsWhySkipped 的識別字（sync_status.go）——被擋下的檔
// 要帶著這句話出現在畫面上，不能安靜消失。
func d1QuotaNote(cypherURL string, now time.Time) string {
	st, ok := activeD1Quota(cypherURL, now)
	if !ok {
		return ""
	}
	n := buildD1QuotaNotice(st.kind, now, st.until)
	return n.Headline + "，先不送，避免白白重打；" + n.Guarantee + "（會自動恢復）。"
}