JCS-pub/client/internal/http/server.go

package http

import (
	"github.com/gin-gonic/gin"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/common/sdks/storage/cdsapi"
	"gitlink.org.cn/cloudream/storage/client/internal/services"
)

type Server struct {
	engine     *gin.Engine
	listenAddr string
	svc        *services.Service
	awsAuth    *AWSAuth
}

func NewServer(listenAddr string, svc *services.Service, awsAuth *AWSAuth) (*Server, error) {
	engine := gin.New()

	return &Server{
		engine:     engine,
		listenAddr: listenAddr,
		svc:        svc,
		awsAuth:    awsAuth,
	}, nil
}

func (s *Server) Serve() error {
	s.initRouters()

	logger.Infof("start serving http at: %s", s.listenAddr)
	err := s.engine.Run(s.listenAddr)

	if err != nil {
		logger.Infof("http stopped with error: %s", err.Error())
		return err
	}

	logger.Infof("http stopped")
	return nil
}

func (s *Server) initRouters() {
	rt := s.engine.Use()

	initTemp(rt, s)

	s.routeV1(s.engine, rt)

	rt.GET(cdsapi.ObjectListPathByPath, s.Object().ListByPath)
	rt.POST(cdsapi.ObjectListByIDsPath, s.Object().ListByIDs)
	rt.GET(cdsapi.ObjectDownloadPath, s.Object().Download)
	rt.GET(cdsapi.ObjectDownloadByPathPath, s.Object().DownloadByPath)
	rt.POST(cdsapi.ObjectUploadPath, s.Object().Upload)
	rt.GET(cdsapi.ObjectGetPackageObjectsPath, s.Object().GetPackageObjects)
	rt.POST(cdsapi.ObjectUpdateInfoPath, s.Object().UpdateInfo)
	rt.POST(cdsapi.ObjectUpdateInfoByPathPath, s.Object().UpdateInfoByPath)
	rt.POST(cdsapi.ObjectMovePath, s.Object().Move)
	rt.POST(cdsapi.ObjectDeletePath, s.Object().Delete)
	rt.POST(cdsapi.ObjectDeleteByPathPath, s.Object().DeleteByPath)
	rt.POST(cdsapi.ObjectClonePath, s.Object().Clone)

	rt.GET(cdsapi.PackageGetPath, s.Package().Get)
	rt.GET(cdsapi.PackageGetByFullNamePath, s.Package().GetByFullName)
	rt.POST(cdsapi.PackageCreatePath, s.Package().Create)
	rt.POST(cdsapi.PackageCreateLoadPath, s.Package().CreateLoad)
	rt.POST(cdsapi.PackageDeletePath, s.Package().Delete)
	rt.POST(cdsapi.PackageClonePath, s.Package().Clone)
	rt.GET(cdsapi.PackageListBucketPackagesPath, s.Package().ListBucketPackages)
	rt.GET(cdsapi.PackageGetCachedStoragesPath, s.Package().GetCachedStorages)

	rt.POST(cdsapi.StorageLoadPackagePath, s.Storage().LoadPackage)
	rt.POST(cdsapi.StorageCreatePackagePath, s.Storage().CreatePackage)
	rt.GET(cdsapi.StorageGetPath, s.Storage().Get)

	rt.POST(cdsapi.CacheMovePackagePath, s.Cache().MovePackage)

	rt.GET(cdsapi.BucketGetByNamePath, s.Bucket().GetByName)
	rt.POST(cdsapi.BucketCreatePath, s.Bucket().Create)
	rt.POST(cdsapi.BucketDeletePath, s.Bucket().Delete)
	rt.GET(cdsapi.BucketListUserBucketsPath, s.Bucket().ListUserBuckets)
}

func (s *Server) routeV1(eg *gin.Engine, rt gin.IRoutes) {
	v1 := eg.Group("/v1")

	v1.GET(cdsapi.ObjectListPathByPath, s.awsAuth.Auth, s.Object().ListByPath)
	v1.POST(cdsapi.ObjectListByIDsPath, s.awsAuth.Auth, s.Object().ListByIDs)
	v1.GET(cdsapi.ObjectDownloadPath, s.awsAuth.Auth, s.Object().Download)
	v1.GET(cdsapi.ObjectDownloadByPathPath, s.awsAuth.Auth, s.Object().DownloadByPath)
	v1.POST(cdsapi.ObjectUploadPath, s.awsAuth.AuthWithoutBody, s.Object().Upload)
	v1.GET(cdsapi.ObjectGetPackageObjectsPath, s.awsAuth.Auth, s.Object().GetPackageObjects)
	v1.POST(cdsapi.ObjectUpdateInfoPath, s.awsAuth.Auth, s.Object().UpdateInfo)
	v1.POST(cdsapi.ObjectUpdateInfoByPathPath, s.awsAuth.Auth, s.Object().UpdateInfoByPath)
	v1.POST(cdsapi.ObjectMovePath, s.awsAuth.Auth, s.Object().Move)
	v1.POST(cdsapi.ObjectDeletePath, s.awsAuth.Auth, s.Object().Delete)
	v1.POST(cdsapi.ObjectDeleteByPathPath, s.awsAuth.Auth, s.Object().DeleteByPath)
	v1.POST(cdsapi.ObjectClonePath, s.awsAuth.Auth, s.Object().Clone)

	v1.GET(cdsapi.PackageGetPath, s.awsAuth.Auth, s.Package().Get)
	v1.GET(cdsapi.PackageGetByFullNamePath, s.awsAuth.Auth, s.Package().GetByFullName)
	v1.POST(cdsapi.PackageCreatePath, s.awsAuth.Auth, s.Package().Create)
	v1.POST(cdsapi.PackageCreateLoadPath, s.awsAuth.Auth, s.Package().CreateLoad)
	v1.POST(cdsapi.PackageDeletePath, s.awsAuth.Auth, s.Package().Delete)
	v1.POST(cdsapi.PackageClonePath, s.awsAuth.Auth, s.Package().Clone)
	v1.GET(cdsapi.PackageListBucketPackagesPath, s.awsAuth.Auth, s.Package().ListBucketPackages)
	v1.GET(cdsapi.PackageGetCachedStoragesPath, s.awsAuth.Auth, s.Package().GetCachedStorages)

	v1.POST(cdsapi.StorageLoadPackagePath, s.awsAuth.Auth, s.Storage().LoadPackage)
	v1.POST(cdsapi.StorageCreatePackagePath, s.awsAuth.Auth, s.Storage().CreatePackage)
	v1.GET(cdsapi.StorageGetPath, s.awsAuth.Auth, s.Storage().Get)

	v1.POST(cdsapi.CacheMovePackagePath, s.awsAuth.Auth, s.Cache().MovePackage)

	v1.GET(cdsapi.BucketGetByNamePath, s.awsAuth.Auth, s.Bucket().GetByName)
	v1.POST(cdsapi.BucketCreatePath, s.awsAuth.Auth, s.Bucket().Create)
	v1.POST(cdsapi.BucketDeletePath, s.awsAuth.Auth, s.Bucket().Delete)
	v1.GET(cdsapi.BucketListUserBucketsPath, s.awsAuth.Auth, s.Bucket().ListUserBuckets)

	rt.POST(cdsapi.UserCreatePath, s.User().Create)
	rt.POST(cdsapi.UserDeletePath, s.User().Delete)

	rt.POST(cdsapi.ObjectNewMultipartUploadPath, s.Object().NewMultipartUpload)
	rt.POST(cdsapi.ObjectUploadPartPath, s.Object().UploadPart)
	rt.POST(cdsapi.ObjectCompleteMultipartUploadPath, s.Object().CompleteMultipartUpload)
}