package cos import ( "archive/zip" "bytes" "encoding/base64" model "github.com/HFO4/cloudreve/models" "github.com/HFO4/cloudreve/pkg/hashid" "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common" "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile" scf "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/scf/v20180416" "io" "io/ioutil" "net/url" "strconv" "strings" "time" ) const scfFunc = `# -*- coding: utf8 -*- # SCF配置COS触发,向 Cloudreve 发送回调 from qcloud_cos_v5 import CosConfig from qcloud_cos_v5 import CosS3Client from qcloud_cos_v5 import CosServiceError from qcloud_cos_v5 import CosClientError import sys import logging import requests logging.basicConfig(level=logging.INFO, stream=sys.stdout) logger = logging.getLogger() def main_handler(event, context): logger.info("start main handler") for record in event['Records']: try: if "x-cos-meta-callback" not in record['cos']['cosObject']['meta']: logger.info("Cannot find callback URL, skiped.") return 'Success' callback = record['cos']['cosObject']['meta']['x-cos-meta-callback'] key = record['cos']['cosObject']['key'] logger.info("Callback URL is " + callback) r = requests.get(callback) print(r.text) except Exception as e: print(e) print('Error getting object {} callback url. '.format(key)) raise e return "Fail" return "Success" ` // CreateSCF 创建回调云函数 func CreateSCF(policy *model.Policy, region string) error { // 初始化客户端 credential := common.NewCredential( policy.AccessKey, policy.SecretKey, ) cpf := profile.NewClientProfile() client, err := scf.NewClient(credential, region, cpf) if err != nil { return err } // 创建回调代码数据 buff := &bytes.Buffer{} bs64 := base64.NewEncoder(base64.StdEncoding, buff) zipWriter := zip.NewWriter(bs64) header := zip.FileHeader{ Name: "callback.py", Method: zip.Deflate, } writer, err := zipWriter.CreateHeader(&header) if err != nil { return err } _, err = io.Copy(writer, strings.NewReader(scfFunc)) zipWriter.Close() // 创建云函数 req := scf.NewCreateFunctionRequest() funcName := "cloudreve_" + hashid.HashID(policy.ID, hashid.PolicyID) + strconv.FormatInt(time.Now().Unix(), 10) zipFileBytes, _ := ioutil.ReadAll(buff) zipFileStr := string(zipFileBytes) codeSource := "ZipFile" handler := "callback.main_handler" desc := "Cloudreve 用回调函数" timeout := int64(60) runtime := "Python3.6" req.FunctionName = &funcName req.Code = &scf.Code{ ZipFile: &zipFileStr, } req.Handler = &handler req.Description = &desc req.Timeout = &timeout req.Runtime = &runtime req.CodeSource = &codeSource _, err = client.CreateFunction(req) if err != nil { return err } time.Sleep(time.Duration(5) * time.Second) // 创建触发器 server, _ := url.Parse(policy.Server) triggerType := "cos" triggerDesc := `{"event":"cos:ObjectCreated:Post","filter":{"Prefix":"","Suffix":""}}` enable := "OPEN" trigger := scf.NewCreateTriggerRequest() trigger.FunctionName = &funcName trigger.TriggerName = &server.Host trigger.Type = &triggerType trigger.TriggerDesc = &triggerDesc trigger.Enable = &enable _, err = client.CreateTrigger(trigger) if err != nil { return err } return nil }