package main

import (
	"bytes"
	"crypto/md5"
	"crypto/tls"
	"encoding/base64"
	"encoding/json"
	"errors"
	"fmt"
	"math/rand"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"
)

// 全局配置（与PHP代码保持一致）
const (
	// 获取上传地址的接口URL
	cdn58GetUploadURL = "https://im.58.com/msg/get_pic_upload_url"
	// 固定请求头配置
	cdn58Referer = "https://ai.58.com/pc/"
	cdn58UA      = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.6261.95 Safari/537.36"
	cdn58Origin  = "https://ai.58.com"
	// 请求超时时间
	cdn58Timeout = 30 * time.Second
	// 最终图片URL前缀
	cdn58ImgPrefix = "https://pic%d.58cdn.com.cn/nowater/im/"
)

// 允许的文件后缀（与PHP代码保持一致）
var allowedFileExts = map[string]bool{
	"jpg":  true,
	"jpeg": true,
	"png":  true,
	"gif":  true,
	"bmp":  true,
}

// Cdn58UploadInfo 上传信息结构体（对应PHP响应中的upload_info）
type Cdn58UploadInfo struct {
	URL string `json:"url"` // 上传地址
}

// Cdn58Data 响应数据结构体（对应PHP响应中的data）
type Cdn58Data struct {
	UploadInfo []Cdn58UploadInfo `json:"upload_info"`
}

// Cdn58GetUploadURLResponse 获取上传地址的响应结构体
type Cdn58GetUploadURLResponse struct {
	ErrorCode int       `json:"error_code"`
	ErrorMsg  string    `json:"error_msg"`
	Data      Cdn58Data `json:"data"`
}

// -------------------------- 对应PHP的私有方法实现 --------------------------

// guid 生成与PHP guid()方法一致的UUID格式字符串
func guid() string {
	// 模拟PHP: md5(uniqid(mt_rand(), true))
	rand.Seed(time.Now().UnixNano())
	uniqID := fmt.Sprintf("%d%d", rand.Int63(), time.Now().UnixNano())
	md5Hash := md5.Sum([]byte(uniqID))
	guidStr := fmt.Sprintf("%x", md5Hash)

	// 拼接为UUID格式：8-4-4-4-12
	return fmt.Sprintf("%s-%s-4%s-%s-%s",
		guidStr[:8],
		guidStr[8:12],
		guidStr[13:16], // 固定第13位为4（符合UUID v4规范）
		guidStr[16:20],
		guidStr[20:32],
	)
}

// encrypt 实现与PHP一致的加密逻辑
func encrypt(data string) string {
	// 1. base64编码
	base64Str := base64.StdEncoding.EncodeToString([]byte(data))

	// 2. 统计等号数量
	equalCount := strings.Count(base64Str, "=")

	// 3. 替换字符并移除等号，拼接等号数量
	encodedStr := strings.ReplaceAll(base64Str, "+", "-")
	encodedStr = strings.ReplaceAll(encodedStr, "/", "_")
	encodedStr = strings.ReplaceAll(encodedStr, "=", "")
	encodedStr += strconv.Itoa(equalCount)

	// 4. 分成两半，交换前后部分
	halfLen := len(encodedStr) / 2
	firstHalf := encodedStr[:halfLen]
	secondHalf := encodedStr[halfLen:]

	return secondHalf + firstHalf
}

// decrypt 实现与PHP一致的解密逻辑（虽然核心流程未使用，保持完整性）
func decrypt(data string) string {
	// 1. 分成两半，交换前后部分
	halfLen := (len(data) + 1) / 2
	firstHalf := data[:halfLen]
	secondHalf := data[halfLen:]
	data = secondHalf + firstHalf

	// 2. 提取末尾的等号数量
	equalCount, _ := strconv.Atoi(string(data[len(data)-1]))
	data = data[:len(data)-1]

	// 3. 还原字符
	decodedStr := strings.ReplaceAll(data, "-", "+")
	decodedStr = strings.ReplaceAll(data, "_", "/")

	// 4. 补充等号
	decodedStr += strings.Repeat("=", equalCount)

	// 5. base64解码
	raw, _ := base64.StdEncoding.DecodeString(decodedStr)
	return string(raw)
}

// getMIMEType 根据文件后缀获取MIME类型（对应PHP的mime_content_type方法）
func getMIMEType(ext string) string {
	mimeMap := map[string]string{
		"png":  "image/png",
		"jpeg": "image/jpeg",
		"jpg":  "image/jpeg",
		"gif":  "image/gif",
		"bmp":  "image/bmp",
	}
	if mime, ok := mimeMap[strings.ToLower(ext)]; ok {
		return mime
	}
	return "application/octet-stream"
}

// -------------------------- 核心上传逻辑 --------------------------

// Cdn58ImgUpload cdn58图床上传函数，完全对齐PHP功能
// filepath: 本地文件路径
// filename: 上传文件名（包含后缀）
// 返回值: 最终图片URL，错误信息
func Cdn58ImgUpload(filepath1 string, filename string) (string, error) {
	// 1. 提取并校验文件后缀
	fileExt := filepath.Ext(filename)
	if fileExt != "" {
		fileExt = strings.TrimPrefix(fileExt, ".") // 移除前缀"."
	}
	fileExt = strings.ToLower(fileExt)
	if !allowedFileExts[fileExt] {
		return "", errors.New("上传失败！不支持的文件格式，仅支持jpg/jpeg/png/gif/bmp")
	}

	// 2. 生成UserID
	userID := "58Anonymous" + guid()

	// 3. 构建user_info并加密（对应PHP的user_info数组）
	userInfo := map[string]string{
		"user_id":        userID,
		"source":         "14",
		"im_token":       userID,
		"client_version": "1.0",
		"client_type":    "pcweb",
		"os_type":        "Chrome",
		"os_version":     "122.0.6261.95",
		"appid":          "10140-mcs@jitmouQrcHs",
		"extend_flag":    "0",
		"unread_index":   "1",
		"sdk_version":    "6432",
		"device_id":      userID,
		"xxzl_smartid":   "",
		"id58":           "CkwAd2e0U3tBNxbRAzQ2Ag==",
	}

	// 3.1 构建查询字符串（对应PHP的http_build_query）
	queryValues := url.Values{}
	for k, v := range userInfo {
		queryValues.Set(k, v)
	}
	userInfoQuery := queryValues.Encode()

	// 3.2 加密user_info查询字符串
	encryptedUserInfo := encrypt(userInfoQuery)

	// 4. 构建post数据并加密（对应PHP的post数组）
	postData := map[string]interface{}{
		"sender_id":     userID,
		"sender_source": 14,
		"to_id":         "10002",
		"to_source":     100,
		"file_suffixs":  []string{fileExt},
	}
	// 4.1 JSON编码（对应PHP的json_encode）
	postJSON, err := json.Marshal(postData)
	if err != nil {
		return "", fmt.Errorf("序列化post数据失败: %w", err)
	}
	// 4.2 加密post数据
	encryptedPost := encrypt(string(postJSON))

	// 5. 拼接获取上传地址的URL
	getUploadURL := fmt.Sprintf("%s?params=%s&version=j1.0",
		cdn58GetUploadURL,
		url.QueryEscape(encryptedUserInfo), // URL编码参数
	)

	// 6. 发送请求获取上传地址（对应PHP的get_curl）
	client := &http.Client{
		Timeout: cdn58Timeout,
		Transport: &http.Transport{
			TLSClientConfig: &tls.Config{
				InsecureSkipVerify: true, // 注意：字段名是 InsecureSkipVerify（非 TLSInsecureSkipVerify）
			},
		},
	}

	req, err := http.NewRequest("POST", getUploadURL, bytes.NewBufferString(encryptedPost))
	if err != nil {
		return "", fmt.Errorf("创建获取上传地址请求失败: %w", err)
	}

	// 设置请求头（与PHP保持一致）
	req.Header.Set("Content-Type", "text/plain;charset=UTF-8")
	req.Header.Set("Referer", cdn58Referer)
	req.Header.Set("User-Agent", cdn58UA)
	req.Header.Set("Origin", cdn58Origin)

	resp, err := client.Do(req)
	if err != nil {
		return "", fmt.Errorf("发送获取上传地址请求失败: %w", err)
	}
	defer resp.Body.Close()

	// 解析获取上传地址的响应
	var getUploadResp Cdn58GetUploadURLResponse
	if err := json.NewDecoder(resp.Body).Decode(&getUploadResp); err != nil {
		return "", fmt.Errorf("解析获取上传地址响应失败: %w", err)
	}

	// 校验响应结果
	if getUploadResp.ErrorCode != 0 {
		errMsg := "接口错误"
		if getUploadResp.ErrorMsg != "" {
			errMsg = getUploadResp.ErrorMsg
		}
		return "", fmt.Errorf("上传失败！%s", errMsg)
	}

	// 提取上传URL
	if len(getUploadResp.Data.UploadInfo) == 0 || getUploadResp.Data.UploadInfo[0].URL == "" {
		return "", errors.New("上传失败！未返回上传地址")
	}
	uploadURL := getUploadResp.Data.UploadInfo[0].URL

	// 7. 读取本地文件内容（对应PHP的file_get_contents）
	fileContent, err := os.ReadFile(filepath1)
	if err != nil {
		return "", fmt.Errorf("读取本地文件失败: %w, 路径: %s", err, filepath1)
	}

	// 8. 获取MIME类型
	mimeType := getMIMEType(fileExt)

	// 9. 使用PUT方法上传文件（对应PHP的curl_upload方法）
	putReq, err := http.NewRequest("PUT", uploadURL, bytes.NewBuffer(fileContent))
	if err != nil {
		return "", fmt.Errorf("创建文件上传请求失败: %w", err)
	}

	// 设置PUT请求头
	putReq.Header.Set("Content-Type", mimeType)
	putReq.Header.Set("User-Agent", cdn58UA)

	putResp, err := client.Do(putReq)
	if err != nil {
		return "", fmt.Errorf("发送文件上传请求失败: %w", err)
	}
	defer putResp.Body.Close()

	// 校验HTTP状态码
	if putResp.StatusCode != 200 {
		return "", fmt.Errorf("上传失败！httpCode=%d", putResp.StatusCode)
	}

	// 10. 截取文件名（对应PHP的getSubstr($url, '/nowater/im/', '?')）
	filenameStart := strings.Index(uploadURL, "/nowater/im/")
	if filenameStart == -1 {
		return "", errors.New("上传失败！无法从上传地址中截取文件名")
	}
	filenameStart += len("/nowater/im/")
	filenameEnd := strings.Index(uploadURL[filenameStart:], "?")
	if filenameEnd == -1 {
		filenameEnd = len(uploadURL)
	} else {
		filenameEnd += filenameStart
	}
	croppedFilename := uploadURL[filenameStart:filenameEnd]
	if croppedFilename == "" {
		return "", errors.New("上传失败！截取的文件名为空")
	}

	// 11. 拼接最终图片URL（随机生成1-8的数字）
	rand.Seed(time.Now().UnixNano())
	picNum := rand.Intn(8) + 1 // 生成1-8的随机数
	finalImgURL := fmt.Sprintf(cdn58ImgPrefix, picNum) + croppedFilename

	return finalImgURL, nil
}

// -------------------------- 示例调用 --------------------------
func main() {
	// 配置本地文件路径和上传文件名（请替换为实际值）
	localFilePath := "./test.jpg" // 本地图片路径
	uploadFilename := "test.jpg"  // 上传文件名（包含后缀）

	// 调用cdn58上传函数
	imgURL, err := Cdn58ImgUpload(localFilePath, uploadFilename)
	if err != nil {
		fmt.Printf("cdn58图床上传失败: %v\n", err)
		return
	}

	fmt.Printf("cdn58图床上传成功，最终图片URL: %s\n", imgURL)
}