1.4.5.2

EdgeCommon/pkg/serverconfigs/ossconfigs/oss_config_plus.go

@@ -0,0 +1,142 @@
+// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package ossconfigs
+
+import (
+	"crypto/md5"
+	"encoding/json"
+	"fmt"
+	"net"
+	"net/http"
+	"path/filepath"
+	"reflect"
+	"strings"
+)
+
+type OSSConfig struct {
+	Type          OSSType        `yaml:"oss" json:"type"`
+	BucketParam   OSSBucketParam `yaml:"bucketParam" json:"bucketParam"`
+	BucketArgName string         `yaml:"bucketArgName" json:"bucketArgName"`
+	BucketName    string         `yaml:"bucketName" json:"bucketName"`
+	Options       any            `yaml:"options" json:"options"`
+
+	optionsUniqueId string
+}
+
+func NewOSSConfig() *OSSConfig {
+	return &OSSConfig{}
+}
+
+func (this *OSSConfig) Init() error {
+	if this.Options != nil {
+		// decode options
+		if reflect.TypeOf(this.Options).Kind() == reflect.Map {
+			optionsJSON, err := json.Marshal(this.Options)
+			if err != nil {
+				return err
+			}
+
+			newOptions, decodeErr := DecodeOSSOptions(this.Type, optionsJSON)
+			if decodeErr != nil {
+				return decodeErr
+			}
+			if newOptions != nil {
+				this.Options = newOptions
+			}
+		}
+
+		options, ok := this.Options.(OSSOptions)
+		if ok {
+			err := options.Init()
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	configJSON, err := json.Marshal(this)
+	if err != nil {
+		return err
+	}
+	this.optionsUniqueId = fmt.Sprintf("%x", md5.Sum(configJSON))
+
+	return nil
+}
+
+func (this *OSSConfig) Summary() string {
+	var def = FindOSSType(this.Type)
+	if def == nil {
+		return ""
+	}
+	var name = def.Name
+
+	switch this.BucketParam {
+	case OSSBucketParamInput:
+		if len(this.BucketName) > 0 {
+			return name + " - " + this.BucketName
+		}
+	case OSSBucketParamPrefix:
+		return name + " - /BUCKET-NAME/filename"
+	case OSSBucketParamArg:
+		return name + " - /filename?" + this.BucketArgName + "=BUCKET-NAME"
+	case OSSBucketParamSubDomain:
+		return name + " - BUCKET-NAME.HOST"
+	}
+	return name
+}
+
+func (this *OSSConfig) ParseRequest(req *http.Request, host string) (bucketName string, key string, uniqueId string) {
+	var def = FindOSSType(this.Type)
+	if def == nil {
+		return
+	}
+
+	var path = req.URL.Path
+	if len(path) > 0 {
+		path = strings.Trim(filepath.Clean(path), "/")
+	}
+
+	switch this.BucketParam {
+	case OSSBucketParamInput:
+		bucketName = this.BucketName
+		key = path
+	case OSSBucketParamPrefix:
+		var index = strings.Index(path, "/")
+		if index >= 0 {
+			bucketName = path[:index]
+			key = path[index+1:]
+		} else {
+			bucketName = path
+			key = ""
+		}
+	case OSSBucketParamArg:
+		bucketName = req.URL.Query().Get(this.BucketArgName)
+		key = path
+	case OSSBucketParamSubDomain:
+		if len(host) == 0 {
+			host, _, _ = net.SplitHostPort(req.URL.Host)
+		}
+		if len(host) > 0 {
+			var index = strings.Index(host, ".")
+			if index >= 0 {
+				bucketName = host[:index]
+			} else {
+				bucketName = host
+			}
+		}
+		key = path
+	default: // same as 'input'
+		bucketName = this.BucketName
+		key = path
+	}
+
+	// 需要防止外部传入无限个bucket名称造成内存泄露
+	if def.BucketIgnored {
+		bucketName = "default"
+	}
+	uniqueId = this.Type + "$" + this.optionsUniqueId + "$" + bucketName
+	key = strings.TrimLeft(strings.TrimSpace(key), "/")
+
+	return
+}