package main

import (
    "context"
    "encoding/json"
    "errors"
    "fmt"
    "github.com/tencentyun/cos-go-sdk-v5"
    sts "github.com/tencentyun/qcloud-cos-sts-sdk/go"
    "math/rand"
    "net/http"
    "net/url"
    "os"
    "reflect"
    "strings"
    "time"
    "unicode"
)

type Config struct {
    filename        string
    appId           string
    SecretId        string
    SecretKey       string
    Proxy           string
    DurationSeconds int
    Bucket          string
    Region          string
    AllowActions    []string
}

type Permission struct {
    LimitExt           bool     `json:"limitExt"`
    ExtWhiteList       []string `json:"extWhiteList"`
    LimitContentType   bool     `json:"limitContentType"`
    LimitContentLength bool     `json:"limitContentLength"`
}

func generateCosKey(ext string) string {
    date := time.Now()
    m := int(date.Month()) + 1
    ymd := fmt.Sprintf("%d%02d%d", date.Year(), m, date.Day())
    r := fmt.Sprintf("%06d", rand.Intn(1000000))
    cosKey := fmt.Sprintf("file/%s/%s_%s.%s", ymd, ymd, r, ext)
    return cosKey
}

func getPermission() Permission {
    permission := Permission{
        LimitExt:           true,
        ExtWhiteList:       []string{"jpg", "jpeg", "png", "gif", "bmp"},
        LimitContentType:   false,
        LimitContentLength: false,
    }
    return permission
}

func getConfig() Config {
    config := Config{
        filename:        "test.jpg",
        appId:           "12500000000",
        SecretId:        os.Getenv("SECRETID"),  // 사용자의 SecretId, 서브 계정 키 사용을 권장하며 최소 권한 가이드에 따라 권한을 부여하여 사용 위험을 줄입니다. 서브 계정 키 획득은 https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1 참조
        SecretKey:       os.Getenv("SECRETKEY"), // 사용자의 SecretKey, 서브 계정 키 사용을 권장하며 최소 권한 가이드에 따라 권한을 부여하여 사용 위험을 줄입니다. 서브 계정 키 획득은 https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1 참조
        Proxy:           os.Getenv("Proxy"),
        DurationSeconds: 1800,
        Bucket:          "0-1253960454",
        Region:          "ap-guangzhou",
        AllowActions: []string{
            "name/cos:PutObject",
        },
    }
    return config
}

func stringInSlice(str string, list []string) bool {
    for _, v := range list {
        if v == str {
            return true
        }
    }
    return false
}

func StructToCamelMap(input interface{}) map[string]interface{} {
    v := reflect.ValueOf(input)
    if v.Kind() == reflect.Ptr {
        v = v.Elem()
    }

    result := make(map[string]interface{})
    typ := v.Type()

    for i := 0; i < v.NumField(); i++ {
        field := typ.Field(i)
        fieldValue := v.Field(i)

        // 필드 이름을 LowerCamel로 변환
        key := toLowerCamel(field.Name)

        // 중첩 구조체 처리
        if fieldValue.Kind() == reflect.Struct ||
            (fieldValue.Kind() == reflect.Ptr && fieldValue.Elem().Kind() == reflect.Struct) {

            if fieldValue.IsNil() && fieldValue.Kind() == reflect.Ptr {
                result[key] = nil
                continue
            }

            result[key] = StructToCamelMap(fieldValue.Interface())
        } else {
            // 기본 유형 처리
            result[key] = fieldValue.Interface()
        }
    }

    return result
}

// LowerCamel 형식으로 변환(첫 글자가 소문자로)
func toLowerCamel(s string) string {
    if s == "" {
        return s
    }

    // 전체 대문자 단어 처리(예: ID)
    if strings.ToUpper(s) == s {
        return strings.ToLower(s)
    }

    // 일반 LowerCamel 변환
    runes := []rune(s)
    runes[0] = unicode.ToLower(runes[0])
    return string(runes)
}

func getStsCredential() (map[string]interface{}, error) {
    config := getConfig()

    permission := getPermission()

    c := sts.NewClient(
        // 환경 변수를 통해 키 획득, os.Getenv 메서드는 환경 변수 획득을 표시함
        config.SecretId,  //os.Getenv("SECRETID"),  // 사용자의 SecretId, 서브 계정 키 사용을 권장하며 최소 권한 가이드에 따라 권한을 부여하여 사용 위험을 줄입니다. 서브 계정 키 획득은 https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1 참조
        config.SecretKey, //os.Getenv("SECRETKEY"),                 // 사용자의 SecretKey, 서브 계정 키 사용을 권장하며 최소 권한 가이드에 따라 권한을 부여하여 사용 위험을 줄입니다. 서브 계정 키 획득은 https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1 참조
        nil,
        // sts.Host("sts.internal.tencentcloudapi.com"), // 도메인 설정, 기본 도메인 sts.tencentcloudapi.com
        // sts.Scheme("http"),      // 프로토콜 설정, 기본값은 https이며, 퍼블릭 클라우드 sts에서 임시 키를 가져올 때 http는 허용되지 않습니다. 특수한 시나리오에서만 http 설정이 필요합니다.
    )

    condition := make(map[string]map[string]interface{})

    segments := strings.Split(config.filename, ".")
    if len(segments) == 0 {
        //ext := ""
    }
    ext := segments[len(segments)-1]

    if permission.LimitExt {
        extInvalid := ext == "" || !stringInSlice(ext, permission.ExtWhiteList)
        if extInvalid {
            return nil, errors.New("불법 파일, 업로드 금지")
        }
    }

    if permission.LimitContentType {
        condition["string_like_if_exist"] = map[string]interface{}{
            // content-type이 이미지 유형인 경우에만 업로드 허용
            "cos:content-type": "image/*",
        }
    }

    // 3. 업로드 파일 크기 제한
    if permission.LimitContentLength {
        condition["numeric_less_than_equal"] = map[string]interface{}{
            // 업로드 크기는 최대 5MB로 제한됨(단순 업로드에만 적용됨)
            "cos:content-length": 5 * 1024 * 1024,
        }
    }

    key := generateCosKey(ext)
    // 정책 개요 https://www.tencentcloud.com/document/product/436/18023?from_cn_redirect=1
    opt := &sts.CredentialOptions{
        DurationSeconds: int64(config.DurationSeconds),
        Region:          config.Region,
        Policy: &sts.CredentialPolicy{
            Version: "2.0",
            Statement: []sts.CredentialPolicyStatement{
                {
                    // 키의 권한 목록. 단순 업로드 및 샤딩에는 다음 권한이 필요하며, 기타 권한 목록은 https://www.tencentcloud.com/document/product/436/31923?from_cn_redirect=1 참조
                    Action: config.AllowActions,
                    Effect: "allow",
                    Resource: []string{
                        // 여기서 허용할 경로 접두사를 변경합니다. 자신의 웹사이트 사용자 로그인 상태에 따라 업로드할 특정 경로를 판단할 수 있습니다. 예: a.jpg, a/* 또는 * (와일드카드 * 사용에는 중대한 보안 위험이 있으므로 신중하게 평가하여 사용하십시오).
                        // 버킷의 이름 형식은 BucketName-APPID이며, 여기에 입력한 bucket은 반드시 이 형식이어야 합니다.
                        "qcs::cos:ap-guangzhou:uid/" + config.appId + ":" + config.Bucket + "/" + key,
                    },
                    // 적용 조건 구축 시작 condition
                    // condition에 대한 상세 설정 규칙 및 COS에서 지원하는 condition 유형은 https://www.tencentcloud.com/document/product/436/71306?from_cn_redirect=1 참조
                    Condition: condition,
                },
            },
        },
    }

    // case 1 임시 키 요청
    res, err := c.GetCredential(opt)
    if err != nil {
        return nil, err
    }
    // LowerCamelMap으로 변환
    resultMap := StructToCamelMap(res)
    resultMap["bucket"] = config.Bucket
    resultMap["region"] = config.Region
    resultMap["key"] = key

    return resultMap, nil
}

func main() {
    result, err := getStsCredential()
    if err != nil {
        fmt.Printf("임시 키 요청 실패: %v\\n", err)
        return // 컨텍스트에 따라 종료 여부 결정
    }

    // credentials를 map[string]interface{}로 타입 단언
    credentials, ok := result["credentials"].(map[string]interface{})
    if !ok {
        fmt.Println("자격 증명 형식 오류")
        return
    }
    // 유형 검사가 포함된 필드 획득
    tak, tok := credentials["tmpSecretID"].(string)
    tsk, tskok := credentials["tmpSecretKey"].(string)
    token, tokenok := credentials["sessionToken"].(string)
    if !tok || !tskok || !tokenok {
        fmt.Println("임시 자격 증명 필드 누락 또는 유형 오류")
        return
    }
    host := "https://" + result["bucket"].(string) + ".cos." + result["region"].(string) + ".myqcloud.com"
    u, _ := url.Parse(host)
    b := &cos.BaseURL{BucketURL: u}
    c := cos.NewClient(b, &http.Client{})
    name := result["key"].(string)
    ctx := context.Background()
    opt := &cos.PresignedURLOptions{
        Query:  &url.Values{},
        Header: &http.Header{},
    }
    opt.Query.Add("x-cos-security-token", token)
    signature := c.Object.GetSignature(ctx, http.MethodPut, name, "tak", "tsk", time.Hour, opt, true)
    fmt.Printf("%s\\n%s\\n%s\\n%s\\n", tak, tsk, token, signature)
    data := make(map[string]string)

    // token 및 sign 추가
    data["securityToken"] = token
    data["authorization"] = signature
    data["cosHost"] = host
    data["cosKey"] = name
    // JSON으로 변환
    jsonData, err := json.MarshalIndent(data, "", "  ")
    if err != nil {
        fmt.Println("JSON 인코딩 실패:", err)
        return
    }
    fmt.Println(string(jsonData))
}