config.go

package sdk

import (
	"fmt"
	"github.com/flyleft/gprofile"
	"os"
	"strings"
)

var appConfig ApplicationConfig
var cssConfig Configuration
var cssLoggerConfig LoggerConf

//var GatewayTimeout = 60
//var Port int
//var MaxExecTime = 300
//var IsAFSExist bool
//var DownloadDir string
//var UploadDir string
//var UploadFieldModel string
//
//const DownloadFolder = "download/"
//const UploadFolder = "upload/"

//var ContentTypeMap = make(map[string]string, 20)

type Configuration struct {
	// 链存服务API地址
	ChainStorageApiEndpoint string `profile:"chainStorageApiEndpoint" profileDefault:"http://127.0.0.1:8821" json:"chainStorageApiEndpoint"`

	// CAR文件工作目录
	CarFileWorkPath string `profile:"carFileWorkPath" profileDefault:"./tmp/carfile" json:"carFileWorkPath"`

	// CAR文件分片阈值
	CarFileShardingThreshold int `profile:"carFileShardingThreshold" profileDefault:"46137344" json:"carFileShardingThreshold"`

	// 链存服务API token
	ChainStorageApiToken string `profile:"chainStorageApiToken" profileDefault:"" json:"chainStorageApiToken"`

	// HTTP request user agent (K2请求需要)
	HttpRequestUserAgent string `profile:"httpRequestUserAgent" profileDefault:"" json:"httpRequestUserAgent"`

	// HTTP request user agent (K2请求需要)
	HttpRequestOvertime int `profile:"httpRequestOvertime" profileDefault:"30" json:"httpRequestOvertime"`

	// CAR version
	CarVersion int `profile:"carVersion" profileDefault:"1" json:"carVersion"`

	UseHTTPSProtocol bool `profile:"useHttpsProtocol" profileDefault:"true" json:"useHttpsProtocol"`
}

type LoggerConf struct {
	LogPath      string `profile:"logPath" profileDefault:"./logs" json:"logPath"`
	Mode         string `prfile:"mode" profileDefault:"release" json:"mode"`
	Level        string `prfile:"level" profileDefault:"info" json:"level"`
	IsOutPutFile bool   `profile:"isOutPutFile" profileDefault:"false" json:"isOutPutFile"`
	MaxAgeDay    int64  `profile:"maxAgeDay" profileDefault:"7" json:"maxAgeDay"`
	RotationTime int64  `profile:"rotationTime" profileDefault:"1" json:"rotationTime"`
}

type ApplicationConfig struct {
	Server Configuration `profile:"server"`
	Logger LoggerConf    `profile:"logger"`
}

func initConfig(config *ApplicationConfig) {
	cssConfig = config.Server
	cssLoggerConfig = config.Logger

	//check chain-storage-api base address
	if len(cssConfig.ChainStorageApiEndpoint) > 0 {
		chainStorageAPIEndpoint := cssConfig.ChainStorageApiEndpoint
		if !strings.HasPrefix(chainStorageAPIEndpoint, "http://") &&
			!strings.HasPrefix(chainStorageAPIEndpoint, "https://") {

			if cssConfig.UseHTTPSProtocol {
				cssConfig.ChainStorageApiEndpoint = "https://" + chainStorageAPIEndpoint
			} else {
				cssConfig.ChainStorageApiEndpoint = "http://" + chainStorageAPIEndpoint
			}

			//fmt.Println("ERROR: invalid chain-storage-api endpoint in Configuration, chain-storage-api endpoint must be a valid http/https url, exiting")
			//os.Exit(1)
		}

		if !strings.HasSuffix(chainStorageAPIEndpoint, "/") {
			cssConfig.ChainStorageApiEndpoint += "/"
		}
	} else {
		fmt.Println("ERROR: no chain-storage-api endpoint provided in Configuration, at least 1 valid http/https chain-storage-api endpoint must be given, exiting")
		os.Exit(1)
	}

	if len(cssConfig.ChainStorageApiToken) == 0 {
		fmt.Println("ERROR: invalid chain-storage-api token in Configuration, chain-storage-api token must not be empty")
		os.Exit(1)
	} else if !strings.HasPrefix(cssConfig.ChainStorageApiToken, "Bearer ") {
		cssConfig.ChainStorageApiToken = "Bearer " + cssConfig.ChainStorageApiToken
	}

	//// CAR文件分片阈值，缺省10MB
	//carFileShardingThreshold := cssConfig.CarFileShardingThreshold
	//if carFileShardingThreshold <= 0 {
	//	cssConfig.CarFileShardingThreshold = 10485760
	//}
	// CAR文件分片阈值（固定44Mb）
	cssConfig.CarFileShardingThreshold = 46137344

	// CAR文件工作目录
	carFileWorkPath := cssConfig.CarFileWorkPath
	if len(carFileWorkPath) == 0 {
		cssConfig.CarFileWorkPath = `./tmp/carfile`
	}

	// HTTP request user agent (K2请求需要)
	httpRequestUserAgent := cssConfig.HttpRequestUserAgent
	if len(httpRequestUserAgent) == 0 {
		cssConfig.HttpRequestUserAgent = `Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36`
	}

	// HTTP request overtime
	httpRequestOvertime := cssConfig.HttpRequestOvertime
	if httpRequestOvertime <= 0 {
		cssConfig.HttpRequestOvertime = 30
	}

	// CAR version
	carVersion := cssConfig.CarVersion
	if carVersion <= 0 {
		cssConfig.CarVersion = 1
	}
}

func initConfigWithConfigFile(configFile string) {
	//rand.Seed(time.Now().UnixNano())
	//if len(configFile) == 0 {
	//	configFile = "./github.com/paradeum-team/chainstorage-sdk.yaml"
	//}
	if len(configFile) == 0 {
		configFile = "./chainstorage-sdk.yaml"
	}

	config, err := gprofile.Profile(&ApplicationConfig{}, configFile, true)
	if err != nil {
		fmt.Errorf("Profile execute error", err)
	}

	appConfig = *config.(*ApplicationConfig)
	cssConfig = config.(*ApplicationConfig).Server
	cssLoggerConfig = config.(*ApplicationConfig).Logger

	//dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
	//if err != nil {
	//	log.Fatal(err)
	//	os.Exit(1)
	//}

	//check chain-storage-api base address
	if len(cssConfig.ChainStorageApiEndpoint) > 0 {
		chainStorageAPIEndpoint := cssConfig.ChainStorageApiEndpoint
		if !strings.HasPrefix(chainStorageAPIEndpoint, "http://") &&
			!strings.HasPrefix(chainStorageAPIEndpoint, "https://") {

			if cssConfig.UseHTTPSProtocol {
				cssConfig.ChainStorageApiEndpoint = "https://" + chainStorageAPIEndpoint
			} else {
				cssConfig.ChainStorageApiEndpoint = "http://" + chainStorageAPIEndpoint
			}

			//fmt.Println("ERROR: invalid chain-storage-api endpoint in Configuration, chain-storage-api endpoint must be a valid http/https url, exiting")
			//os.Exit(1)
		}

		if !strings.HasSuffix(chainStorageAPIEndpoint, "/") {
			cssConfig.ChainStorageApiEndpoint += "/"
		}
	} else {
		fmt.Println("ERROR: no chain-storage-api endpoint provided in Configuration, at least 1 valid http/https chain-storage-api endpoint must be given, exiting")
		os.Exit(1)
	}

	if len(cssConfig.ChainStorageApiToken) == 0 {
		fmt.Println("ERROR: invalid chain-storage-api token in Configuration file, chain-storage-api token must not be empty")
		os.Exit(1)
	} else if !strings.HasPrefix(cssConfig.ChainStorageApiToken, "Bearer ") {
		cssConfig.ChainStorageApiToken = "Bearer " + cssConfig.ChainStorageApiToken
	}

	//if _, err := os.Stat(filepath.Join(Config.AbsAFSDir, Config.AFSProgram)); os.IsExist(err) {
	//	IsAFSExist = false
	//} else if err != nil {
	//	IsAFSExist = false
	//}
	////check port
	//if Config.Port > 1023 && Config.Port <= 65535 {
	//	Port = Config.Port
	//} else {
	//	fmt.Printf("WARNING: invalid port number(port) in Configuration file, using default:%s\n", Port)
	//}

}

func InitConfigWithDefault() {
	//rand.Seed(time.Now().UnixNano())
	config, err := gprofile.Profile(&ApplicationConfig{}, "./chainstorage-sdk.yaml", true)
	if err != nil {
		fmt.Errorf("Profile execute error", err)
	}
	appConfig = *config.(*ApplicationConfig)
	cssConfig = config.(*ApplicationConfig).Server
	cssLoggerConfig = config.(*ApplicationConfig).Logger

	//dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
	//if err != nil {
	//	log.Fatal(err)
	//	os.Exit(1)
	//}

	//check chain-storage-api base address
	if len(cssConfig.ChainStorageApiEndpoint) > 0 {
		chainStorageAPIEndpoint := cssConfig.ChainStorageApiEndpoint
		if !strings.HasPrefix(chainStorageAPIEndpoint, "http://") &&
			!strings.HasPrefix(chainStorageAPIEndpoint, "https://") {

			if cssConfig.UseHTTPSProtocol {
				cssConfig.ChainStorageApiEndpoint = "https://" + chainStorageAPIEndpoint
			} else {
				cssConfig.ChainStorageApiEndpoint = "http://" + chainStorageAPIEndpoint
			}

			//fmt.Println("ERROR: invalid chain-storage-api endpoint in Configuration, chain-storage-api endpoint must be a valid http/https url, exiting")
			//os.Exit(1)
		}

		if !strings.HasSuffix(chainStorageAPIEndpoint, "/") {
			cssConfig.ChainStorageApiEndpoint += "/"
		}
	} else {
		fmt.Println("ERROR: no chain-storage-api endpoint provided in Configuration, at least 1 valid http/https chain-storage-api endpoint must be given, exiting")
		os.Exit(1)
	}

	if len(cssConfig.ChainStorageApiToken) == 0 {
		fmt.Println("ERROR: invalid chain-storage-api token in Configuration file, chain-storage-api token must not be empty")
		os.Exit(1)
	} else if !strings.HasPrefix(cssConfig.ChainStorageApiToken, "Bearer ") {
		cssConfig.ChainStorageApiToken = "Bearer " + cssConfig.ChainStorageApiToken
	}
}