file

internal/pkg/storage/local/local.go

@@ -0,0 +1,90 @@
+// Package local 提供基于本地文件系统的 Storage 实现。
+package local
+
+import (
+	"context"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"portal-mini-go/internal/pkg/errors"
+)
+
+// Local 是基于本地磁盘的存储实现。
+type Local struct {
+	root    string
+	baseURL string
+}
+
+// New 构造本地存储，root 为落盘根目录，baseURL 为对外访问基础路径（不含末尾斜杠）。
+func New(root, baseURL string) *Local {
+	return &Local{root: root, baseURL: strings.TrimRight(baseURL, "/")}
+}
+
+// Put 将内容写入 root/key，自动创建父目录。
+func (l *Local) Put(_ context.Context, key string, r io.Reader, _ int64) error {
+	p, err := l.safePath(key)
+	if err != nil {
+		return err
+	}
+	if err := os.MkdirAll(filepath.Dir(p), 0o755); err != nil {
+		return errors.Wrap(50000, "mkdir upload dir", err)
+	}
+	f, err := os.Create(p)
+	if err != nil {
+		return errors.Wrap(50000, "create upload file", err)
+	}
+	defer f.Close()
+	if _, err := io.Copy(f, r); err != nil {
+		return errors.Wrap(50000, "write upload file", err)
+	}
+	return nil
+}
+
+// Get 打开文件并返回读取流。
+func (l *Local) Get(_ context.Context, key string) (io.ReadCloser, error) {
+	p, err := l.safePath(key)
+	if err != nil {
+		return nil, err
+	}
+	f, err := os.Open(p)
+	if err != nil {
+		if os.IsNotExist(err) {
+			return nil, errors.ErrNotFound
+		}
+		return nil, errors.Wrap(50000, "open upload file", err)
+	}
+	return f, nil
+}
+
+// Delete 删除文件；不存在视为成功。
+func (l *Local) Delete(_ context.Context, key string) error {
+	p, err := l.safePath(key)
+	if err != nil {
+		return err
+	}
+	if err := os.Remove(p); err != nil && !os.IsNotExist(err) {
+		return errors.Wrap(50000, "delete upload file", err)
+	}
+	return nil
+}
+
+// URL 返回 /baseURL/key 形式的访问地址。
+func (l *Local) URL(key string) string {
+	return l.baseURL + "/" + strings.TrimLeft(key, "/")
+}
+
+// safePath 将 key 解析为 root 内的绝对路径，并禁止任何目录穿越。
+func (l *Local) safePath(key string) (string, error) {
+	clean := filepath.Clean("/" + key) // 以 / 开头，消除 . 与 ..
+	if strings.Contains(clean, "..") {
+		return "", errors.New(40000, "非法的文件 key")
+	}
+	p := filepath.Join(l.root, clean)
+	// 二次校验，确保最终路径没有逃出 root。
+	if rel, err := filepath.Rel(l.root, p); err != nil || rel == ".." || strings.HasPrefix(rel, ".."+string(filepath.Separator)) {
+		return "", errors.New(40000, "非法的文件 key")
+	}
+	return p, nil
+}