|
|
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
|
|
|
}
|
