chore: 初始化 Cloudreve 源码仓库

2026-09-13 11:28:59 +08:00
commit 2a7b03ab1d
489 changed files with 163867 additions and 0 deletions
@@ -0,0 +1,161 @@
+package admin
+
+import (
+	"context"
+	"strconv"
+	"strings"
+
+	"github.com/cloudreve/Cloudreve/v4/application/dependency"
+	"github.com/cloudreve/Cloudreve/v4/ent"
+	"github.com/cloudreve/Cloudreve/v4/inventory"
+	"github.com/cloudreve/Cloudreve/v4/pkg/cluster/routes"
+	"github.com/cloudreve/Cloudreve/v4/pkg/hashid"
+	"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
+	"github.com/gin-gonic/gin"
+	"github.com/samber/lo"
+)
+
+const (
+	shareUserIDCondition = "share_user_id"
+	shareFileIDCondition = "share_file_id"
+	shareIDCondition     = "share_id"
+)
+
+func (s *AdminListService) Shares(c *gin.Context) (*ListShareResponse, error) {
+	dep := dependency.FromContext(c)
+	shareClient := dep.ShareClient()
+	hasher := dep.HashIDEncoder()
+
+	var (
+		err      error
+		userID   int
+		fileID   int
+		shareIDs []int
+	)
+
+	if s.Conditions[shareUserIDCondition] != "" {
+		userID, err = strconv.Atoi(s.Conditions[shareUserIDCondition])
+		if err != nil {
+			return nil, serializer.NewError(serializer.CodeParamErr, "Invalid share user ID", err)
+		}
+	}
+
+	if s.Conditions[shareFileIDCondition] != "" {
+		fileID, err = strconv.Atoi(s.Conditions[shareFileIDCondition])
+		if err != nil {
+			return nil, serializer.NewError(serializer.CodeParamErr, "Invalid share file ID", err)
+		}
+	}
+
+	if s.Conditions[shareIDCondition] != "" {
+		shareIdStrs := strings.Split(s.Conditions[shareIDCondition], ",")
+		for _, shareIdStr := range shareIdStrs {
+			shareID, err := strconv.Atoi(shareIdStr)
+			if err != nil {
+				return nil, serializer.NewError(serializer.CodeParamErr, "Invalid share ID", err)
+			}
+
+			shareIDs = append(shareIDs, shareID)
+		}
+	}
+
+	ctx := context.WithValue(c, inventory.LoadShareFile{}, true)
+	ctx = context.WithValue(ctx, inventory.LoadShareUser{}, true)
+
+	res, err := shareClient.List(ctx, &inventory.ListShareArgs{
+		PaginationArgs: &inventory.PaginationArgs{
+			Page:     s.Page - 1,
+			PageSize: s.PageSize,
+			OrderBy:  s.OrderBy,
+			Order:    inventory.OrderDirection(s.OrderDirection),
+		},
+		UserID:   userID,
+		FileID:   fileID,
+		ShareIDs: shareIDs,
+	})
+
+	if err != nil {
+		return nil, serializer.NewError(serializer.CodeDBError, "Failed to list shares", err)
+	}
+
+	siteUrl := dep.SettingProvider().SiteURL(c)
+
+	return &ListShareResponse{
+		Pagination: res.PaginationResults,
+		Shares: lo.Map(res.Shares, func(share *ent.Share, _ int) GetShareResponse {
+			var (
+				uid       string
+				shareLink string
+			)
+
+			if share.Edges.User != nil {
+				uid = hashid.EncodeUserID(hasher, share.Edges.User.ID)
+			}
+
+			shareLink = routes.MasterShareUrl(siteUrl, hashid.EncodeShareID(hasher, share.ID), share.Password).String()
+
+			return GetShareResponse{
+				Share:      share,
+				UserHashID: uid,
+				ShareLink:  shareLink,
+			}
+		}),
+	}, nil
+
+}
+
+type (
+	SingleShareService struct {
+		ShareID int `uri:"id" binding:"required"`
+	}
+	SingleShareParamCtx struct{}
+)
+
+func (s *SingleShareService) Get(c *gin.Context) (*GetShareResponse, error) {
+	dep := dependency.FromContext(c)
+	shareClient := dep.ShareClient()
+	hasher := dep.HashIDEncoder()
+
+	ctx := context.WithValue(c, inventory.LoadShareFile{}, true)
+	ctx = context.WithValue(ctx, inventory.LoadShareUser{}, true)
+	share, err := shareClient.GetByID(ctx, s.ShareID)
+	if err != nil {
+		return nil, serializer.NewError(serializer.CodeDBError, "Failed to get share", err)
+	}
+
+	var (
+		uid       string
+		shareLink string
+	)
+
+	if share.Edges.User != nil {
+		uid = hashid.EncodeUserID(hasher, share.Edges.User.ID)
+	}
+
+	siteUrl := dep.SettingProvider().SiteURL(c)
+	shareLink = routes.MasterShareUrl(siteUrl, hashid.EncodeShareID(hasher, share.ID), share.Password).String()
+
+	return &GetShareResponse{
+		Share:      share,
+		UserHashID: uid,
+		ShareLink:  shareLink,
+	}, nil
+}
+
+type (
+	BatchShareService struct {
+		ShareIDs []int `json:"ids" binding:"required"`
+	}
+	BatchShareParamCtx struct{}
+)
+
+func (s *BatchShareService) Delete(c *gin.Context) error {
+	dep := dependency.FromContext(c)
+	shareClient := dep.ShareClient()
+
+	if err := shareClient.DeleteBatch(c, s.ShareIDs); err != nil {
+		return serializer.NewError(serializer.CodeDBError, "Failed to delete shares", err)
+	}
+
+	return nil
+}