// 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
}