package main

import (
	"fmt"
	"io"
	"log"
	"net/http"
	"strings"
	"time"
)

const proxyPort = ":8680"

func main() {
	http.HandleFunc("/coze-chat", mockHandler)

	log.Printf("[MOCK] 模擬 Coze 接口已啟動")
	log.Printf("[MOCK] 監(jiān)聽地址: http://localhost%s/coze-chat", proxyPort)
	log.Printf("[MOCK] 可直接用你原來的前端代碼測試（無需 token）")
	log.Fatal(http.ListenAndServe(proxyPort, nil))
}

func mockHandler(w http.ResponseWriter, r *http.Request) {
	clientIP := r.RemoteAddr
	log.Printf("[MOCK REQUEST] %s %s from %s", r.Method, r.URL.Path, clientIP)

	// 處理 CORS 預(yù)檢
	if r.Method == http.MethodOptions {
		w.Header().Set("Access-Control-Allow-Origin", "*")
		w.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS")
		w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
		w.Header().Set("Access-Control-Max-Age", "86400")
		w.WriteHeader(http.StatusNoContent)
		return
	}

	if r.Method != http.MethodPost {
		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
		return
	}

	// 讀取請求體（僅模擬，可忽略內(nèi)容）
	_, _ = io.ReadAll(r.Body)
	defer r.Body.Close()

	// 設(shè)置 SSE 響應(yīng)頭
	w.Header().Set("Content-Type", "text/event-stream")
	w.Header().Set("Cache-Control", "no-cache")
	w.Header().Set("Connection", "keep-alive")
	w.Header().Set("Access-Control-Allow-Origin", "*")
	w.(http.Flusher).Flush()

	// 自定義回復(fù)內(nèi)容（支持 Markdown，隨時修改）
	fullResponse := `# 時尚審美評估師

**評判結(jié)果**：好看

**理由**：
- 這件帽衫的深灰色非常百搭，經(jīng)典不過時，適合各種膚色。
- 版型寬松舒適，oversize 設(shè)計很有街頭感，顯瘦又時髦。
- 帽繩和袖口細(xì)節(jié)處理精致，質(zhì)感看起來很不錯（棉感強，不廉價）。
- 前胸的小logo繡標(biāo)低調(diào)有質(zhì)感，不會太張揚。
- 整體風(fēng)格偏休閑運動風(fēng)，日常穿搭、約會、逛街都非常合適。

**搭配建議**：
- 下裝：黑色或深藍(lán)牛仔褲 + 白球鞋（經(jīng)典安全）
- 進階：卡其色工裝褲 + 馬丁靴（更有層次）
- 外搭：短款羽絨服或風(fēng)衣（冬天顯高）

總體來說，這件帽衫設(shè)計簡潔大方，性價比應(yīng)該很高，值得入手！?`

	// 逐字輸出，模擬打字機效果
	for _, char := range fullResponse {
		fmt.Fprintf(w, "event: conversation.message.delta\n")
		fmt.Fprintf(w, "data: {\"id\":\"mock123\",\"conversation_id\":\"mock\",\"bot_id\":\"mock\",\"role\":\"assistant\",\"type\":\"answer\",\"content\":\"%s\",\"content_type\":\"text\"}\n\n", escapeJSON(string(char)))
		w.(http.Flusher).Flush()

		// 隨機延遲，模擬人類打字速度
		time.Sleep(time.Millisecond * time.Duration(50+(char%100)))
	}

	// 發(fā)送結(jié)束標(biāo)志
	fmt.Fprint(w, "event: done\ndata: [DONE]\n\n")
	w.(http.Flusher).Flush()

	log.Printf("[MOCK SUCCESS] 模擬回復(fù)完成，共輸出 %d 個字符", len(fullResponse))
}

// 簡單 JSON 轉(zhuǎn)義函數(shù)
func escapeJSON(s string) string {
	s = strings.ReplaceAll(s, "\\", "\\\\")
	s = strings.ReplaceAll(s, "\"", "\\\"")
	s = strings.ReplaceAll(s, "\n", "\\n")
	s = strings.ReplaceAll(s, "\r", "\\r")
	return s
}

go run mock_coze.go

fullResponse := "這件帽衫不太行，顏色暗沉、版型臃腫，建議再看看其他款～"