package notify

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"
	"strings"
	"sync"
	"time"
)

const (
	// 获取 access_token 的 API 地址
	getTokenURL = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
	// 发送应用消息的 API 地址
	sendMessageURL = "https://qyapi.weixin.qq.com/cgi-bin/message/send"
)

// wechatNotifier 实现了 Notifier 接口，用于通过企业微信自建应用发送私聊消息。
type wechatNotifier struct {
	corpID  string // 企业ID (CorpID)
	agentID string // 应用ID (AgentID)
	secret  string // 应用密钥 (Secret)

	// 用于线程安全地管理 access_token
	mu             sync.Mutex
	accessToken    string
	tokenExpiresAt time.Time
}

// NewWechatNotifier 创建一个新的 wechatNotifier 实例。
// 调用者需要注入企业微信应用的 CorpID, AgentID 和 Secret。
func NewWechatNotifier(corpID, agentID, secret string) Notifier {
	return &wechatNotifier{
		corpID:  corpID,
		agentID: agentID,
		secret:  secret,
	}
}

// Send 向指定用户发送一条 markdown 格式的私聊消息。
// toAddr 参数是接收者的 UserID 列表，用逗号或竖线分隔。
func (w *wechatNotifier) Send(content AlarmContent, toAddr string) error {
	// 1. 获取有效的 access_token
	token, err := w.getAccessToken()
	if err != nil {
		return err
	}

	// 2. 构建 markdown 内容
	markdownContent := fmt.Sprintf("## %s\n> 级别: <font color=\"warning\">%s</font>\n> 时间: %s\n\n%s",
		content.Title,
		content.Level.String(),
		content.Timestamp.Format(DefaultTimeFormat),
		content.Message,
	)

	// 3. 构建请求的 JSON Body
	// 将逗号分隔的 toAddr 转换为竖线分隔，以符合 API 要求
	userList := strings.ReplaceAll(toAddr, ",", "|")
	payload := wechatMessagePayload{
		ToUser:  userList,
		MsgType: "markdown",
		AgentID: w.agentID,
		Markdown: struct {
			Content string `json:"content"`
		}{
			Content: markdownContent,
		},
	}

	jsonBytes, err := json.Marshal(payload)
	if err != nil {
		return fmt.Errorf("序列化企业微信消息失败: %w", err)
	}

	// 4. 发送 HTTP POST 请求
	url := fmt.Sprintf("%s?access_token=%s", sendMessageURL, token)
	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBytes))
	if err != nil {
		return fmt.Errorf("创建企业微信请求失败: %w", err)
	}
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return fmt.Errorf("发送企业微信通知失败: %w", err)
	}
	defer resp.Body.Close()

	// 5. 检查响应
	var response wechatResponse
	if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
		return fmt.Errorf("解析企业微信响应失败: %w", err)
	}

	if response.ErrCode != 0 {
		return fmt.Errorf("企业微信API返回错误: code=%d, msg=%s", response.ErrCode, response.ErrMsg)
	}

	return nil
}

// getAccessToken 获取并缓存 access_token，处理了线程安全和自动刷新。
func (w *wechatNotifier) getAccessToken() (string, error) {
	w.mu.Lock()
	defer w.mu.Unlock()

	// 如果 token 存在且有效期还有5分钟以上，则直接返回缓存的 token
	if w.accessToken != "" && time.Now().Before(w.tokenExpiresAt.Add(-5*time.Minute)) {
		return w.accessToken, nil
	}

	// 否则，重新获取 token
	url := fmt.Sprintf("%s?corpid=%s&corpsecret=%s", getTokenURL, w.corpID, w.secret)
	resp, err := http.Get(url)
	if err != nil {
		return "", fmt.Errorf("获取 access_token 请求失败: %w", err)
	}
	defer resp.Body.Close()

	var tokenResp tokenResponse
	if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
		return "", fmt.Errorf("解析 access_token 响应失败: %w", err)
	}

	if tokenResp.ErrCode != 0 {
		return "", fmt.Errorf("获取 access_token API 返回错误: code=%d, msg=%s", tokenResp.ErrCode, tokenResp.ErrMsg)
	}

	// 缓存新的 token 和过期时间
	w.accessToken = tokenResp.AccessToken
	w.tokenExpiresAt = time.Now().Add(time.Duration(tokenResp.ExpiresIn) * time.Second)

	return w.accessToken, nil
}

// Type 返回通知器的类型
func (w *wechatNotifier) Type() NotifierType {
	return NotifierTypeWeChat
}

// --- API 数据结构 ---

// tokenResponse 是获取 access_token API 的响应结构体
type tokenResponse struct {
	ErrCode     int    `json:"errcode"`
	ErrMsg      string `json:"errmsg"`
	AccessToken string `json:"access_token"`
	ExpiresIn   int    `json:"expires_in"`
}

// wechatMessagePayload 是发送应用消息 API 的请求体结构
type wechatMessagePayload struct {
	ToUser   string `json:"touser"`
	MsgType  string `json:"msgtype"`
	AgentID  string `json:"agentid"`
	Markdown struct {
		Content string `json:"content"`
	} `json:"markdown"`
}

// wechatResponse 是企业微信 API 的通用响应结构体
type wechatResponse struct {
	ErrCode int    `json:"errcode"`
	ErrMsg  string `json:"errmsg"`
}