package storage

import (
	"bufio"
	"io"
	"os"
	"path/filepath"
)

type LocalStorage struct {
	pathPrefix string
}

type LocalStorageOption func(*LocalStorage)

func WithLocalPathPrefix(prefix string) LocalStorageOption {
	return func(storage *LocalStorage) {
		storage.pathPrefix = prefix
	}
}

func NewLocal(opts ...LocalStorageOption) Storage {
	l := &LocalStorage{}
	for _, o := range opts {
		o(l)
	}
	return l
}

func (l *LocalStorage) Save(fileName string, data []byte) (filePath string, err error) {
	f, err := os.OpenFile(filepath.Join(l.pathPrefix, fileName), os.O_CREATE|os.O_RDWR, os.ModePerm)
	if err != nil {
		return "", err
	}
	defer f.Close()
	w := bufio.NewWriter(f)
	w.Write(data)
	w.Flush()
	return fileName, nil
}

func (l *LocalStorage) Download(fileName string) ([]byte, error) {
	f, err := os.OpenFile(filepath.Join(l.pathPrefix, fileName), os.O_CREATE|os.O_RDWR, os.ModePerm)
	if err != nil {
		return nil, err
	}
	defer f.Close()
	return io.ReadAll(f)
}

func (l *LocalStorage) Close() {}