osm_sdk/osm_sdk.go

package osm_sdk

import (
	"bytes"
	"encoding/json"
	"git.sxidc.com/go-framework/baize/framework/core/api/response"
	"git.sxidc.com/go-tools/utils/strutils"
	"git.sxidc.com/service-supports/dapr_api/invoke"
	"github.com/pkg/errors"
	"io"
	"mime/multipart"
	"time"
)

const TimeoutSec = 30
const servicePrefix = "/osm/api/v1"

var prefix string
var namespace string
var name string
var invokeAPI *invoke.API

type (
	ConfigKey struct {
		Namespace string `json:"namespace" form:"namespace" binding:"required"`
		Name      string `json:"name" form:"name" binding:"required"`
	}

	DeleteMultiObjJsonBody struct {
		ConfigKey
		ObjNames []string `json:"objNames" binding:"required"`
		Prefix   string   `json:"prefix"`
	}
	CopyJsonBody struct {
		ConfigKey
		SrcObjName   string   `json:"srcObjName" binding:"required"`
		DstObjName   string   `json:"dstObjName" binding:"required"`
		RequireInfos []string `json:"requireInfos"`
		Prefix       string   `json:"prefix"`
	}
	MoveJsonBody struct {
		ConfigKey
		SrcObjName   string   `json:"srcObjName" binding:"required"`
		DstObjName   string   `json:"dstObjName" binding:"required"`
		RequireInfos []string `json:"requireInfos"`
		Prefix       string   `json:"prefix"`
	}
	ZoomUrlJsonBody struct {
		ConfigKey
		ObjName      string   `json:"srcObjName" binding:"required"`
		Process      string   `json:"dstObjName" binding:"required"`
		RequireInfos []string `json:"requireInfos"`
		Prefix       string   `json:"prefix"`
	}
)

func Destroy() {
	if invokeAPI == nil {
		return
	}
	invoke.DestroyAPI(invokeAPI)
	invokeAPI = nil
}

func Init(configPrefix, configNamespace, configName, baseUrl string) error {
	prefix = configPrefix
	namespace = configNamespace
	name = configName

	if invokeAPI == nil {
		invokeAPI = invoke.NewAPI(baseUrl, TimeoutSec*time.Second)
	}

	return nil
}

//func Touch(reader io.Reader, to string) error {
//	return TouchPrefix(reader, to, true)
//}

//func TouchPrefix(reader io.Reader, to string, addPfx bool) error {
//	if addPfx {
//		return Bkt.PutObject(prefix+to, reader)
//	}
//	return Bkt.PutObject(to, reader)
//}

func GetUrl(objName string) (string, error) {
	return getUrl(objName)
}

func GetUrlWithPrefix(objName string) (string, error) {
	return getUrl(prefix + objName)
}

func getUrl(objName string) (string, error) {
	if strutils.IsStringEmpty(objName) {
		return "", nil
	}

	responseBytes, err := invokeAPI.GetWithoutHeaders(servicePrefix+"/operation/url/get", map[string]string{
		"namespace": namespace,
		"name":      name,
		"objName":   objName,
	})
	if err != nil {
		return "", err
	}

	resp := new(response.InfoResponse[string])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return "", err
	}

	if !resp.Success {
		return "", errors.New(resp.Msg)
	}

	return resp.Info, nil
}

//func ZoomUrl(obj, s string) (string, error) {
//	if strutils.HasBlank(obj) {
//		log.Println("oss发现违规参数")
//		return "", nil
//	}
//	return Bkt.SignURL(prefix+obj, oss.HTTPGet, 60, oss.Process(s))
//}
//
//func ZoomUrlPfx(obj, s string, addPfx bool) (string, error) {
//	if strutils.HasBlank(obj) {
//		log.Println("oss发现违规参数")
//		return "", nil
//	}
//
//	if addPfx {
//		return Bkt.SignURL(prefix+obj, oss.HTTPGet, 60, oss.Process(s))
//	}
//
//	return Bkt.SignURL(obj, oss.HTTPGet, 60, oss.Process(s))
//
//}

func ZoomUrl(objName, profess string) (string, error) {
	return zoomUrl(objName, profess)
}

func ZoomUrlWithPrefix(objName, profess string) (string, error) {
	return zoomUrl(prefix+objName, profess)
}

func zoomUrl(objName, process string) (string, error) {
	if strutils.IsStringEmpty(objName) {
		return "", nil
	}

	requestJson, err := json.Marshal(ZoomUrlJsonBody{
		ConfigKey: ConfigKey{
			Namespace: namespace,
			Name:      name,
		},
		ObjName:      objName,
		Process:      process,
		RequireInfos: nil,
	})
	if err != nil {
		return "", err
	}

	responseBytes, err := invokeAPI.PostJSON(servicePrefix+"/operation/obj/zoomUrl", requestJson, nil)
	if err != nil {
		return "", err
	}

	resp := new(response.InfoResponse[string])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return "", err
	}

	if !resp.Success {
		return "", errors.New(resp.Msg)
	}

	return resp.Info, nil

}

//func Exist(obj string) (bool, error) {
//	if strutils.HasBlank(obj) {
//		return false, nil
//	}
//
//	return Bkt.IsObjectExist(prefix + obj)
//}

func Exist(objName string) (bool, error) {
	return exist(objName)
}

func ExistWithPrefix(objName string) (bool, error) {
	return exist(prefix + objName)
}

func exist(objName string) (bool, error) {
	if strutils.IsStringEmpty(objName) {
		return false, nil
	}
	responseBytes, err := invokeAPI.GetWithoutHeaders(servicePrefix+"/operation/obj/check", map[string]string{
		"namespace": namespace,
		"name":      name,
		"objName":   objName,
	})
	if err != nil {
		return false, err
	}

	resp := new(response.InfoResponse[bool])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return false, err
	}

	if !resp.Success {
		return false, errors.New(resp.Msg)
	}

	return true, nil
}

func DeleteMulti(objNames ...string) error {
	return deleteMulti(objNames...)
}

func DeleteMultiWithPrefix(userId string, objNames ...string) error {
	for _, objName := range objNames {
		objName = prefix + objName
	}

	return deleteMulti(objNames...)
}

func deleteMulti(objNames ...string) error {
	if len(objNames) == 0 {
		return nil
	}

	requestJson, err := json.Marshal(DeleteMultiObjJsonBody{
		ConfigKey: ConfigKey{
			Namespace: namespace,
			Name:      name,
		},
		Prefix:   prefix,
		ObjNames: objNames,
	})
	if err != nil {
		return err
	}

	responseBytes, err := invokeAPI.PostJSON(servicePrefix+"/operation/obj/deleteMulti", requestJson, nil)
	if err != nil {
		return err
	}

	resp := new(response.MsgResponse)
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return err
	}

	if !resp.Success {
		return errors.New(resp.Msg)
	}

	return nil
}

func ListDir(dirPath string) ([]string, error) {
	return listDir(dirPath)
}

func ListDirWithPrefix(dirPath string) ([]string, error) {
	return listDir(prefix + dirPath)
}

func listDir(dirPath string) ([]string, error) {
	if strutils.IsStringEmpty(dirPath) {
		return nil, nil
	}

	responseBytes, err := invokeAPI.GetWithoutHeaders(servicePrefix+"/operation/objPath/list", map[string]string{
		"namespace": namespace,
		"name":      name,
		"dirPath":   dirPath,
	})
	if err != nil {
		return nil, err
	}

	resp := new(response.InfoResponse[[]string])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return nil, err
	}

	if !resp.Success {
		return nil, errors.New(resp.Msg)
	}

	return resp.Info, nil
}

func LsDetails(dirPath string) ([]ObjectInfo, error) {
	return lsDetails(dirPath)
}

func LsDetailsWithPrefix(dirPath string) ([]ObjectInfo, error) {
	return lsDetails(prefix + dirPath)
}

type ObjectInfo struct {
	Key  string
	Type string
	Size int64
}

func lsDetails(dirPath string) ([]ObjectInfo, error) {
	if strutils.IsStringEmpty(dirPath) {
		return []ObjectInfo{}, nil
	}

	responseBytes, err := invokeAPI.GetWithoutHeaders(servicePrefix+"/operation/objPath/listDetail", map[string]string{
		"namespace": namespace,
		"name":      name,
		"dirPath":   dirPath,
	})
	if err != nil {
		return nil, err
	}

	resp := new(response.InfoResponse[[]ObjectInfo])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return nil, err
	}

	if !resp.Success {
		return nil, errors.New(resp.Msg)
	}

	return resp.Info, nil

}

//func Mv(f, t string) error {
//	if err := Cp(f, t); err != nil {
//		return err
//	}
//
//	return Bkt.DeleteObject(prefix + f)
//}

func Move(srcObjName, dstObjName string) error {
	return move(srcObjName, dstObjName, nil)
}

func MoveWithPrefix(srcObjName, dstObjName string) error {
	return move(prefix+srcObjName, prefix+dstObjName, nil)
}

func move(srcObjName, dstObjName string, requireInfos []string) error {
	if strutils.IsStringEmpty(srcObjName) {
		return nil
	}
	if strutils.IsStringEmpty(dstObjName) {
		return nil
	}
	requestJson, err := json.Marshal(MoveJsonBody{
		ConfigKey: ConfigKey{
			Namespace: namespace,
			Name:      name,
		},
		SrcObjName:   srcObjName,
		DstObjName:   dstObjName,
		RequireInfos: requireInfos,
		Prefix:       prefix,
	})
	if err != nil {
		return err
	}
	responseBytes, err := invokeAPI.PutJSON(servicePrefix+"/operation/obj/move", requestJson, nil)
	if err != nil {
		return err
	}
	resp := new(response.InfoResponse[any])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return err
	}
	if !resp.Success {
		return errors.New(resp.Msg)
	}

	return nil
}

//func CpOrigin(f, t string) error {
//	_, err := Bkt.CopyObject(f, t)
//	return err
//}

func Copy(srcObjName, dstObjName string) error {
	return copying(srcObjName, dstObjName, nil)
}

func CopyWithPrefix(srcObjName, dstObjName string) error {
	return copying(prefix+srcObjName, prefix+dstObjName, nil)
}

func copying(srcObjName, dstObjName string, requireInfos []string) error {
	if strutils.IsStringEmpty(srcObjName) {
		return nil
	}
	if strutils.IsStringEmpty(dstObjName) {
		return nil
	}

	requestJson, err := json.Marshal(CopyJsonBody{
		ConfigKey: ConfigKey{
			Namespace: namespace,
			Name:      name,
		},
		SrcObjName:   srcObjName,
		DstObjName:   dstObjName,
		RequireInfos: requireInfos,
		Prefix:       prefix,
	})
	if err != nil {
		return err
	}

	responseBytes, err := invokeAPI.PostJSON(servicePrefix+"/operation/obj/copy", requestJson, nil)
	if err != nil {
		return err
	}

	resp := new(response.InfoResponse[any])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return err
	}

	if !resp.Success {
		return errors.New(resp.Msg)
	}

	return nil
}

func CopyPublic(srcObjName, dstObjName string) error {
	return copyPublic(srcObjName, dstObjName, nil)
}

func CopyPublicWithPrefix(srcObjName, dstObjName string) error {
	return copyPublic(prefix+srcObjName, prefix+dstObjName, nil)
}

func copyPublic(srcObjName, dstObjName string, requireInfos []string) error {
	if strutils.IsStringEmpty(srcObjName) {
		return nil
	}
	if strutils.IsStringEmpty(dstObjName) {
		return nil
	}
	requestJson, err := json.Marshal(CopyJsonBody{
		ConfigKey: ConfigKey{
			Namespace: namespace,
			Name:      name,
		},
		SrcObjName:   srcObjName,
		DstObjName:   dstObjName,
		RequireInfos: requireInfos,
		Prefix:       prefix,
	})
	if err != nil {
		return err
	}

	responseBytes, err := invokeAPI.PostJSON(servicePrefix+"/operation/obj/copyPublic", requestJson, nil)
	if err != nil {
		return err
	}

	resp := new(response.InfoResponse[any])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return err
	}

	if !resp.Success {
		return errors.New(resp.Msg)
	}

	return nil

}

func CreateImage(reader io.Reader, objName string) (string, error) {
	return createImage(reader, objName)
}

func CreateImageWithPrefix(reader io.Reader, objName string) (string, error) {
	return createImage(reader, prefix+objName)
}

func createImage(reader io.Reader, objName string) (string, error) {
	if strutils.IsStringEmpty(objName) {
		return "", nil
	}
	bodyBuffer := &bytes.Buffer{}
	multipartWriter := multipart.NewWriter(bodyBuffer)
	// 添加字段
	fieldWriter, err := multipartWriter.CreateFormField("objName")
	if err != nil {
		return "", err
	}
	_, err = fieldWriter.Write([]byte(objName))
	if err != nil {
		return "", err
	}

	// 添加文件
	fileWriter, err := multipartWriter.CreateFormFile("file", objName)
	if err != nil {
		return "", err
	}
	_, err = io.Copy(fileWriter, reader)
	if err != nil {
		return "", err
	}

	bodyByte, err := io.ReadAll(bodyBuffer)
	if err != nil {
		return "", err
	}
	responseBytes, err := invokeAPI.Post(servicePrefix+"/operation/obj/create/image", map[string]string{
		"Content-Type": multipartWriter.FormDataContentType()}, bodyByte)
	if err != nil {
		return "", err
	}

	resp := new(response.InfoResponse[string])
	err = json.Unmarshal(responseBytes, resp)
	if err != nil {
		return "", err
	}

	if !resp.Success {
		return "", errors.New(resp.Msg)
	}

	return resp.Info, nil
}

//func Cp(f, t string) error {
//	return CpOrigin(prefix+f, prefix+t)
//}
//
//func TouchFormatImg(reader io.Reader, to string) error {
//	if reader == nil || strutils.HasBlank(to) {
//		log.Println("oss上传发现违规参数")
//		return nil
//	}
//
//	return Bkt.PutObject(prefix+to, reader, oss.ContentType("image/jpg"))
//}

//func GetObjectBytes(obj string) ([]byte, error) {
//	if strutils.HasBlank(obj) {
//		return nil, errors.New("参数错误")
//	}
//	buf := new(bytes.Buffer)
//	body, err := Bkt.GetObject(prefix + obj)
//	if err != nil {
//		return nil, err
//	}
//	_, err = io.Copy(buf, body)
//	if err != nil {
//		return nil, err
//	}
//	err = body.Close()
//	if err != nil {
//		return nil, err
//	}
//
//	return buf.Bytes(), nil
//}

//func getObjectBytes(objName, userId string) ([]byte, error) {
//	if strutils.IsStringEmpty(objName) {
//		return nil, nil
//	}
//	if strutils.IsStringEmpty(objName) {
//		return nil, nil
//	}
//	buf := new(bytes.Buffer)
//
//	responseBytes, err := invokeAPI.GetWithoutHeaders(servicePrefix+"/operation/obj/getContent", map[string]string{
//		"namespace": namespace,
//		"name":      name,
//		"objName":   objName,
//		"userId":    userId,
//	})
//	if err != nil {
//		return nil, err
//	}
//
//	resp := new(response.InfoResponse[[]byte])
//	err = json.Unmarshal(responseBytes, resp)
//	if err != nil {
//		return nil, err
//	}
//
//	if !resp.Success {
//		return nil, errors.New(resp.Msg)
//	}
//
//	//return resp.Info, nil
//	_, err = io.Copy(buf, resp.Info)
//	if err != nil {
//		return nil, err
//	}
//	err = body.Close()
//	if err != nil {
//		return nil, err
//	}
//
//	return buf.Bytes(), nil
//}

//func MvWithoutPrefix(f, t string) error {
//	if _, err := Bkt.CopyObject(f, t); err != nil {
//		return err
//	}
//
//	return Bkt.DeleteObject(f)
//}