sing/service/filemanager/manager.go

package filemanager

import (
	"context"
	"os"

	"github.com/sagernet/sing/service"
)

type Manager interface {
	BasePath(name string) string
	OpenFile(name string, flag int, perm os.FileMode) (*os.File, error)
	Create(name string) (*os.File, error)
	CreateTemp(pattern string) (*os.File, error)
	Mkdir(path string, perm os.FileMode) error
	MkdirAll(path string, perm os.FileMode) error
}

func BasePath(ctx context.Context, name string) string {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return name
	}
	return manager.BasePath(name)
}

func OpenFile(ctx context.Context, name string, flag int, perm os.FileMode) (*os.File, error) {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.OpenFile(name, flag, perm)
	}
	return manager.OpenFile(name, flag, perm)
}

func Create(ctx context.Context, name string) (*os.File, error) {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.Create(name)
	}
	return manager.Create(name)
}

func CreateTemp(ctx context.Context, pattern string) (*os.File, error) {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.CreateTemp("", pattern)
	}
	return manager.CreateTemp(pattern)
}

func Mkdir(ctx context.Context, path string, perm os.FileMode) error {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.Mkdir(path, perm)
	}
	return manager.Mkdir(path, perm)
}

func MkdirAll(ctx context.Context, path string, perm os.FileMode) error {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.MkdirAll(path, perm)
	}
	return manager.MkdirAll(path, perm)
}

func WriteFile(ctx context.Context, name string, data []byte, perm os.FileMode) error {
	manager := service.FromContext[Manager](ctx)
	if manager == nil {
		return os.WriteFile(name, data, perm)
	}
	file, err := manager.OpenFile(name, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
	if err != nil {
		return err
	}
	if err != nil {
		return err
	}
	_, err = file.Write(data)
	if err1 := file.Close(); err1 != nil && err == nil {
		err = err1
	}
	return err
}