package main

import (
    "fmt"
    "github.com/tencentyun/qcloud-cos-sts-sdk/go"
    "math/rand"
    "os"
    "time"
)

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:           false,
       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:          "bucket-12500000000",
       Region:          "ap-guangzhou",
       AllowActions: []string{
          "name/cos:PutObject",
          "name/cos:InitiateMultipartUpload",
          "name/cos:ListMultipartUploads",
          "name/cos:ListParts",
          "name/cos:UploadPart",
          "name/cos:CompleteMultipartUpload",
       },
    }
    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)

       // 转换字段名为小驼峰
       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
}

// 转换为小驼峰格式（首字母小写）
func toLowerCamel(s string) string {
    if s == "" {
       return s
    }

    // 处理全大写单词（如 ID）
    if strings.ToUpper(s) == s {
       return strings.ToLower(s)
    }

    // 普通小驼峰转换
    runes := []rune(s)
    runes[0] = unicode.ToLower(runes[0])
    return string(runes)
}

func main() {

    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 {
          fmt.Printf("%+v\\n", "非法文件，禁止上传")
          return
       }
    }

    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 {
       panic(err)
    }
    // 转换为小驼峰 map
    resultMap := StructToCamelMap(res)
    resultMap["bucket"] = config.Bucket
    resultMap["region"] = config.Region
    resultMap["key"] = key
    // 打印结果
	jsonBytes, err := json.MarshalIndent(resultMap, "", "  ")
	if err != nil {
		panic(err)
	}
	// 转换为字符串并打印
	fmt.Println(string(jsonBytes))
}

<?php
require_once __DIR__ . '/vendor/autoload.php';

use QCloud\\COSSTS\\Sts;

// 生成要上传的 COS 文件路径文件名
function generateCosKey($ext) {
  $ymd = date('Ymd');
  $r = substr('000000' . rand(), -6);
  $cosKey = 'file/' . $ymd. '/' . $ymd . '_' . $r;
  if ($ext) {
    $cosKey = $cosKey . '.' . $ext;
  }
  return $cosKey;
};

// 获取单一文件上传权限的临时密钥
function getKeyAndCredentials($filename) {
  // 业务自行实现 用户登录态校验，比如对 token 校验
  // $canUpload = checkUserRole($userToken);
  // if (!$canUpload) {
  //   return '当前用户没有上传权限';
  // }

  // 上传文件可控制类型、大小，按需开启
  $permission = array(
    'limitExt' => false, // 限制上传文件后缀
    'extWhiteList' => ['jpg', 'jpeg', 'png', 'gif', 'bmp'], // 限制的上传后缀
    'limitContentType' => false, // 限制上传 contentType
    'limitContentLength' => false, // 限制上传文件大小
  );
  $condition = array();

  // 客户端传进原始文件名，这里根据文件后缀生成随机 Key
  $ext = pathinfo($filename, PATHINFO_EXTENSION);

  // 1. 限制上传文件后缀
  if ($permission['limitExt']) {
    if ($ext === '' || array_key_exists($ext, $permission['extWhiteList'])) {
      return '非法文件，禁止上传';
    }
  }

  // 2. 限制上传文件 content-type
  if ($permission['limitContentType']) {
    // 只允许上传 content-type 为图片类型
    $condition['string_like_if_exist'] = array('cos:content-type' => 'image/*');
  }

  // 3. 限制上传文件大小
  if ($permission['limitContentLength']) {
    // 上传大小限制不能超过5MB(只对简单上传生效)
    $condition['numeric_less_than_equal'] = array('cos:content-length' => 5 * 1024 * 1024);
  }

  $cosKey = generateCosKey($ext);
  $bucket = 'test-131234567'; // 换成你的 bucket
  $region = 'ap-guangzhou'; // 换成 bucket 所在园区
  
  $config = array(
    'url' => 'https://sts.tencentcloudapi.com/', // url 和 domain保持一致
    'domain' => 'sts.tencentcloudapi.com', // 域名，非必须，默认为 sts.tencentcloudapi.com
    'proxy' => '',
    'secretId' => getenv('GROUP_SECRET_ID'), // 固定密钥,若为明文密钥，请直接以'xxx'形式填入，不要填写到 getenv()函数中
    'secretKey' => getenv('GROUP_SECRET_KEY'), // 固定密钥,若为明文密钥，请直接以'xxx'形式填入，不要填写到 getenv()函数中
    'bucket' => $bucket, // 换成你的 bucket
    'region' => $region, // 换成 bucket 所在园区
    'durationSeconds' => 1800, // 密钥有效期
    'allowPrefix' => array($cosKey), // 只分配当前 key 的路径权限
    // 密钥的权限列表。简单上传和分片需要以下的权限，其他权限列表请看 https://www.tencentcloud.com/document/product/436/31923?from_cn_redirect=1
    'allowActions' => array (
        // 简单上传 
        'name/cos:PutObject',
        // 分片上传
        'name/cos:InitiateMultipartUpload',
        'name/cos:ListMultipartUploads',
        'name/cos:ListParts',
        'name/cos:UploadPart',
        'name/cos:CompleteMultipartUpload'
    ),
  );

  if (!empty($condition)) {
    $config['condition'] = $condition;
  }

  $sts = new Sts();
  $tempKeys = $sts->getTempKeys($config);
  $resTemp = array_merge(
     $tempKeys,
     [
       'startTime' => time(),
       'bucket' => $bucket,
       'region' => $region,
       'key' => $cosKey,
     ]
  );
  echo json_encode($resTemp, JSON_UNESCAPED_SLASHES);
  return $resTemp;
}