package main

import (
	"crypto/hmac"
	"crypto/sha1"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"path/filepath"
	"strings"
	"time"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
	"github.com/gin-gonic/gin"
)

var (
	// 这里替换为你的多吉云永久 AccessKey 和 SecretKey，可在用户中心 - 密钥管理中查看
	// 请勿在客户端暴露 AccessKey 和 SecretKey，那样恶意用户将获得账号完全控制权
	AccessKey = "xxxxxx"
	SecretKey = "xxxxxxxxxxxxxxxxxxxx"
)

func main() {
	// 创建gin实例
	r := gin.Default()

	// 设置路由和处理函数
	r.POST("/upload", func(c *gin.Context) {
		file, err := c.FormFile("source")
		if err != nil {
			c.JSON(400, gin.H{
				"error": err.Error(),
			})
			return
		}
		src, err := file.Open()
		if err != nil {
			c.JSON(500, gin.H{
				"error": err.Error(),
			})
		}
		defer src.Close()
		fpath := getFilePath(file.Filename)
		_, err = GetS3Client().PutObject(&s3.PutObjectInput{
			Bucket: aws.String("s-sh-5485-wiki-images-1258813047"),
			Key:    aws.String(fpath),
			Body:   src,
		})
		if err != nil {
			fmt.Printf("文件上传出错，请检查: %v\n", err)
			c.JSON(500, gin.H{
				"error": err.Error(),
			})
		}

		c.JSON(200, gin.H{
			"code":    200,
			"url":     fpath,
			"message": "File uploaded successfully",
		})
	})

	// 启动服务器
	r.Run(":666")
}

func getFilePath(name string) string {
	formattedTime := time.Unix(time.Now().Unix(), 0).Format("2006/01")
	return fmt.Sprintf("imgs/%s/%d%s", formattedTime, time.Now().UnixMilli(), filepath.Ext(name))
}

func DogeCloudAPI(apiPath string, data map[string]interface{}, jsonMode bool) (ret map[string]interface{}) {
	body := ""
	mime := ""
	if jsonMode {
		_body, err := json.Marshal(data)
		if err != nil {
			log.Fatalln(err)
		}
		body = string(_body)
		mime = "application/json"
	} else {
		values := url.Values{}
		for k, v := range data {
			values.Set(k, v.(string))
		}
		body = values.Encode()
		mime = "application/x-www-form-urlencoded"
	}

	signStr := apiPath + "\n" + body
	hmacObj := hmac.New(sha1.New, []byte(SecretKey))
	hmacObj.Write([]byte(signStr))
	sign := hex.EncodeToString(hmacObj.Sum(nil))
	Authorization := "TOKEN " + AccessKey + ":" + sign

	req, err := http.NewRequest("POST", "https://api.dogecloud.com"+apiPath, strings.NewReader(body))
	if err != nil {
		fmt.Printf("http request failed : %v\n", err)
	}
	req.Header.Add("Content-Type", mime)
	req.Header.Add("Authorization", Authorization)
	client := http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		log.Fatalln(err)
	} // 网络错误
	defer resp.Body.Close()
	r, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("read body failed : %v\n", err)
	}
	json.Unmarshal([]byte(r), &ret)

	return
}

func GetS3Client() *s3.S3 {
	prof := make(map[string]interface{})
	prof["channel"] = "OSS_FULL"
	prof["scopes"] = [1]string{"*"}
	r := DogeCloudAPI("/auth/tmp_token.json", prof, true)
	data := r["data"].(map[string]interface{})
	creds := data["Credentials"].(map[string]interface{})
	s3Config := &aws.Config{
		Credentials: credentials.NewStaticCredentials(creds["accessKeyId"].(string), creds["secretAccessKey"].(string), creds["sessionToken"].(string)),
		Region:      aws.String("automatic"),
		Endpoint:    aws.String("https://cos.ap-shanghai.myqcloud.com"), // 修改为多吉云控制台存储空间 SDK 参数中的 s3Endpoint
	}
	newSession, err := session.NewSession(s3Config)
	if err != nil {
		panic(fmt.Errorf("初始化s3客户端失败: %v\n", err))
	}
	return s3.New(newSession)
}

docker run -itd --name doge -p 666:666 doge