Init V4 community edition (#2265)

* Init V4 community edition

* Init V4 community edition

pkg/filemanager/manager/fs.go

@@ -0,0 +1,114 @@
+package manager
+
+import (
+	"context"
+
+	"github.com/cloudreve/Cloudreve/v4/ent"
+	"github.com/cloudreve/Cloudreve/v4/inventory/types"
+	"github.com/cloudreve/Cloudreve/v4/pkg/cluster"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/cos"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/local"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/obs"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/onedrive"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/oss"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/qiniu"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/remote"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/s3"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/upyun"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/fs"
+	"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
+)
+
+func (m *manager) LocalDriver(policy *ent.StoragePolicy) driver.Handler {
+	if policy == nil {
+		policy = &ent.StoragePolicy{Type: types.PolicyTypeLocal, Settings: &types.PolicySetting{}}
+	}
+	return local.New(policy, m.l, m.config)
+}
+
+func (m *manager) CastStoragePolicyOnSlave(ctx context.Context, policy *ent.StoragePolicy) *ent.StoragePolicy {
+	if !m.stateless {
+		return policy
+	}
+
+	nodeId := cluster.NodeIdFromContext(ctx)
+	if policy.Type == types.PolicyTypeRemote {
+		if nodeId != policy.NodeID {
+			return policy
+		}
+
+		policyCopy := *policy
+		policyCopy.Type = types.PolicyTypeLocal
+		return &policyCopy
+	} else if policy.Type == types.PolicyTypeLocal {
+		policyCopy := *policy
+		policyCopy.NodeID = nodeId
+		policyCopy.Type = types.PolicyTypeRemote
+		policyCopy.SetNode(&ent.Node{
+			ID:       nodeId,
+			Server:   cluster.MasterSiteUrlFromContext(ctx),
+			SlaveKey: m.config.Slave().Secret,
+		})
+		return &policyCopy
+	} else if policy.Type == types.PolicyTypeOss {
+		policyCopy := *policy
+		if policyCopy.Settings != nil {
+			policyCopy.Settings.ServerSideEndpoint = ""
+		}
+	}
+
+	return policy
+}
+
+func (m *manager) GetStorageDriver(ctx context.Context, policy *ent.StoragePolicy) (driver.Handler, error) {
+	switch policy.Type {
+	case types.PolicyTypeLocal:
+		return local.New(policy, m.l, m.config), nil
+	case types.PolicyTypeRemote:
+		return remote.New(ctx, policy, m.settings, m.config, m.l)
+	case types.PolicyTypeOss:
+		return oss.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeCos:
+		return cos.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeS3:
+		return s3.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeObs:
+		return obs.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeQiniu:
+		return qiniu.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeUpyun:
+		return upyun.New(ctx, policy, m.settings, m.config, m.l, m.dep.MimeDetector(ctx))
+	case types.PolicyTypeOd:
+		return onedrive.New(ctx, policy, m.settings, m.config, m.l, m.dep.CredManager())
+	default:
+		return nil, ErrUnknownPolicyType
+	}
+}
+
+func (m *manager) getEntityPolicyDriver(cxt context.Context, e fs.Entity, policyOverwrite *ent.StoragePolicy) (*ent.StoragePolicy, driver.Handler, error) {
+	policyID := e.PolicyID()
+	var (
+		policy *ent.StoragePolicy
+		err    error
+	)
+	if policyID == 0 {
+		policy = &ent.StoragePolicy{Type: types.PolicyTypeLocal, Settings: &types.PolicySetting{}}
+	} else {
+		if policyOverwrite != nil && policyOverwrite.ID == policyID {
+			policy = policyOverwrite
+		} else {
+			policy, err = m.policyClient.GetPolicyByID(cxt, e.PolicyID())
+			if err != nil {
+				return nil, nil, serializer.NewError(serializer.CodeDBError, "failed to get policy", err)
+			}
+		}
+	}
+
+	d, err := m.GetStorageDriver(cxt, policy)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	return policy, d, nil
+}