package local import ( "context" "errors" "fmt" "io" "net/url" "os" "path/filepath" model "github.com/cloudreve/Cloudreve/v3/models" "github.com/cloudreve/Cloudreve/v3/pkg/auth" "github.com/cloudreve/Cloudreve/v3/pkg/cache" "github.com/cloudreve/Cloudreve/v3/pkg/filesystem/fsctx" "github.com/cloudreve/Cloudreve/v3/pkg/filesystem/response" "github.com/cloudreve/Cloudreve/v3/pkg/serializer" "github.com/cloudreve/Cloudreve/v3/pkg/util" ) const ( Perm = 0744 ) // Driver 本地策略适配器 type Driver struct { Policy *model.Policy } // List 递归列取给定物理路径下所有文件 func (handler Driver) List(ctx context.Context, path string, recursive bool) ([]response.Object, error) { var res []response.Object // 取得起始路径 root := util.RelativePath(filepath.FromSlash(path)) // 开始遍历路径下的文件、目录 err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error { // 跳过根目录 if path == root { return nil } if err != nil { util.Log().Warning("Failed to walk folder %q: %s", path, err) return filepath.SkipDir } // 将遍历对象的绝对路径转换为相对路径 rel, err := filepath.Rel(root, path) if err != nil { return err } res = append(res, response.Object{ Name: info.Name(), RelativePath: filepath.ToSlash(rel), Source: path, Size: uint64(info.Size()), IsDir: info.IsDir(), LastModify: info.ModTime(), }) // 如果非递归,则不步入目录 if !recursive && info.IsDir() { return filepath.SkipDir } return nil }) return res, err } // Get 获取文件内容 func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) { // 打开文件 file, err := os.Open(util.RelativePath(path)) if err != nil { util.Log().Debug("Failed to open file: %s", err) return nil, err } return file, nil } // Put 将文件流保存到指定目录 func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error { defer file.Close() fileInfo := file.Info() dst := util.RelativePath(filepath.FromSlash(fileInfo.SavePath)) // 如果非 Overwrite,则检查是否有重名冲突 if fileInfo.Mode&fsctx.Overwrite != fsctx.Overwrite { if util.Exists(dst) { util.Log().Warning("File with the same name existed or unavailable: %s", dst) return errors.New("file with the same name existed or unavailable") } } // 如果目标目录不存在,创建 basePath := filepath.Dir(dst) if !util.Exists(basePath) { err := os.MkdirAll(basePath, Perm) if err != nil { util.Log().Warning("Failed to create directory: %s", err) return err } } var ( out *os.File err error ) openMode := os.O_CREATE | os.O_RDWR if fileInfo.Mode&fsctx.Append == fsctx.Append { openMode |= os.O_APPEND } else { openMode |= os.O_TRUNC } out, err = os.OpenFile(dst, openMode, Perm) if err != nil { util.Log().Warning("Failed to open or create file: %s", err) return err } defer out.Close() if fileInfo.Mode&fsctx.Append == fsctx.Append { stat, err := out.Stat() if err != nil { util.Log().Warning("Failed to read file info: %s", err) return err } if uint64(stat.Size()) < fileInfo.AppendStart { return errors.New("size of unfinished uploaded chunks is not as expected") } else if uint64(stat.Size()) > fileInfo.AppendStart { out.Close() if err := handler.Truncate(ctx, dst, fileInfo.AppendStart); err != nil { return fmt.Errorf("failed to overwrite chunk: %w", err) } out, err = os.OpenFile(dst, openMode, Perm) defer out.Close() if err != nil { util.Log().Warning("Failed to create or open file: %s", err) return err } } } // 写入文件内容 _, err = io.Copy(out, file) return err } func (handler Driver) Truncate(ctx context.Context, src string, size uint64) error { util.Log().Warning("Truncate file %q to [%d].", src, size) out, err := os.OpenFile(src, os.O_WRONLY, Perm) if err != nil { util.Log().Warning("Failed to open file: %s", err) return err } defer out.Close() return out.Truncate(int64(size)) } // Delete 删除一个或多个文件, // 返回未删除的文件,及遇到的最后一个错误 func (handler Driver) Delete(ctx context.Context, files []string) ([]string, error) { deleteFailed := make([]string, 0, len(files)) var retErr error for _, value := range files { filePath := util.RelativePath(filepath.FromSlash(value)) if util.Exists(filePath) { err := os.Remove(filePath) if err != nil { util.Log().Warning("Failed to delete file: %s", err) retErr = err deleteFailed = append(deleteFailed, value) } } // 尝试删除文件的缩略图(如果有) _ = os.Remove(util.RelativePath(value + model.GetSettingByNameWithDefault("thumb_file_suffix", "._thumb"))) } return deleteFailed, retErr } // Thumb 获取文件缩略图 func (handler Driver) Thumb(ctx context.Context, path string) (*response.ContentResponse, error) { file, err := handler.Get(ctx, path+model.GetSettingByNameWithDefault("thumb_file_suffix", "._thumb")) if err != nil { return nil, err } return &response.ContentResponse{ Redirect: false, Content: file, }, nil } // Source 获取外链URL func (handler Driver) Source( ctx context.Context, path string, baseURL url.URL, ttl int64, isDownload bool, speed int, ) (string, error) { file, ok := ctx.Value(fsctx.FileModelCtx).(model.File) if !ok { return "", errors.New("failed to read file model context") } // 是否启用了CDN if handler.Policy.BaseURL != "" { cdnURL, err := url.Parse(handler.Policy.BaseURL) if err != nil { return "", err } baseURL = *cdnURL } var ( signedURI *url.URL err error ) if isDownload { // 创建下载会话,将文件信息写入缓存 downloadSessionID := util.RandStringRunes(16) err = cache.Set("download_"+downloadSessionID, file, int(ttl)) if err != nil { return "", serializer.NewError(serializer.CodeCacheOperation, "Failed to create download session", err) } // 签名生成文件记录 signedURI, err = auth.SignURI( auth.General, fmt.Sprintf("/api/v3/file/download/%s", downloadSessionID), ttl, ) } else { // 签名生成文件记录 signedURI, err = auth.SignURI( auth.General, fmt.Sprintf("/api/v3/file/get/%d/%s", file.ID, file.Name), ttl, ) } if err != nil { return "", serializer.NewError(serializer.CodeEncryptError, "Failed to sign url", err) } finalURL := baseURL.ResolveReference(signedURI).String() return finalURL, nil } // Token 获取上传策略和认证Token,本地策略直接返回空值 func (handler Driver) Token(ctx context.Context, ttl int64, uploadSession *serializer.UploadSession, file fsctx.FileHeader) (*serializer.UploadCredential, error) { if util.Exists(uploadSession.SavePath) { return nil, errors.New("placeholder file already exist") } return &serializer.UploadCredential{ SessionID: uploadSession.Key, ChunkSize: handler.Policy.OptionsSerialized.ChunkSize, }, nil } // 取消上传凭证 func (handler Driver) CancelToken(ctx context.Context, uploadSession *serializer.UploadSession) error { return nil }