Files

103 lines
2.8 KiB
Go

package disk
import (
"context"
"time"
"bsm/full/module/base/cloud/internal/impl"
"bsm/full/module/base/cloud/internal/models"
pb "bsm/full/module/base/cloud/pb"
"git.apinb.com/bsm-sdk/core/errcode"
"git.apinb.com/bsm-sdk/core/printer"
"git.apinb.com/bsm-sdk/core/service"
)
// 获取目录详情
func GetDir(ctx context.Context, in *pb.IdentRequest) (reply *pb.CloudDiskDirItem, err error) {
// parse authorization meta.
auth, err := service.ParseMetaCtx(ctx, nil)
if err != nil {
return nil, err
}
// validate request id,identity.
if in.Id == 0 && in.Identity == "" {
return nil, errcode.ErrInvalidArgument
}
// logic code
var dir models.CloudDiskDir
query := impl.DBService.Where("passport_id = ?", auth.ID)
if in.Id > 0 {
query = query.Where("id = ?", in.Id)
} else {
query = query.Where("identity = ?", in.Identity)
}
if err := query.Preload("Parent").Preload("Subdirectories").Preload("Files").First(&dir).Error; err != nil {
printer.Error("Directory not found: %v", err)
return nil, errcode.ErrInvalidArgument
}
// 转换子目录数据
var subdirs []*pb.CloudDiskDirItem
for _, subdir := range dir.Subdirectories {
subdirs = append(subdirs, &pb.CloudDiskDirItem{
Id: uint64(subdir.ID),
Identity: subdir.Identity,
ParentId: uint64(*subdir.ParentID),
Name: subdir.Name,
Path: subdir.Path,
CreatedAt: subdir.CreatedAt.Format(time.RFC3339),
UpdatedAt: subdir.UpdatedAt.Format(time.RFC3339),
})
}
// 转换文件数据
var files []*pb.CloudDiskFileItem
for _, file := range dir.Files {
files = append(files, &pb.CloudDiskFileItem{
Id: uint64(file.ID),
Identity: file.Identity,
DirectoryId: uint64(*file.DirectoryID),
Name: file.Name,
OriginalName: file.OriginalName,
Size: file.Size,
MimeType: file.MimeType,
StoragePath: file.StoragePath,
Hash: file.Hash,
CreatedAt: file.CreatedAt.Format(time.RFC3339),
UpdatedAt: file.UpdatedAt.Format(time.RFC3339),
})
}
// 构建父目录信息
var parent *pb.CloudDiskDirItem
if dir.Parent != nil {
parent = &pb.CloudDiskDirItem{
Id: uint64(dir.Parent.ID),
Identity: dir.Parent.Identity,
Name: dir.Parent.Name,
Path: dir.Parent.Path,
CreatedAt: dir.Parent.CreatedAt.Format(time.RFC3339),
UpdatedAt: dir.Parent.UpdatedAt.Format(time.RFC3339),
}
}
reply = &pb.CloudDiskDirItem{
Id: uint64(dir.ID),
Identity: dir.Identity,
ParentId: uint64(*dir.ParentID),
Name: dir.Name,
Path: dir.Path,
CreatedAt: dir.CreatedAt.Format(time.RFC3339),
UpdatedAt: dir.UpdatedAt.Format(time.RFC3339),
Parent: parent,
Subdirectories: subdirs,
Files: files,
}
return reply, nil
}