119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
|
package cos
|
|||
|
|
|
|||
|
|
// TODO: revisit para error
|
|||
|
|
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)
|
|||
|
|
logging = logging.getLogger()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main_handler(event, context):
|
|||
|
|
logging.info("start main handler")
|
|||
|
|
for record in event['Records']:
|
|||
|
|
try:
|
|||
|
|
if "x-cos-meta-callback" not in record['cos']['cosObject']['meta']:
|
|||
|
|
logging.info("Cannot find callback URL, skiped.")
|
|||
|
|
return 'Success'
|
|||
|
|
callback = record['cos']['cosObject']['meta']['x-cos-meta-callback']
|
|||
|
|
key = record['cos']['cosObject']['key']
|
|||
|
|
logging.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
|
|||
|
|
//}
|