112 lines
2.2 KiB
Go
112 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"errors"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Storage interface {
|
|
ListFiles() ([]string, error)
|
|
ReadFile(filename string) ([]byte, error)
|
|
WriteFile(filename string, content []byte) error
|
|
DeleteFile(filename string) error
|
|
FileExists(filename string) bool
|
|
}
|
|
|
|
type FileStorage struct {
|
|
dataDir string
|
|
logger *logrus.Logger
|
|
}
|
|
|
|
func NewFileStorage(dataDir string) (*FileStorage, error) {
|
|
// Create data directory if it doesn't exist
|
|
if err := os.MkdirAll(dataDir, 0755); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log := logrus.New()
|
|
log.SetOutput(os.Stdout)
|
|
log.SetFormatter(&logrus.TextFormatter{
|
|
ForceColors: true,
|
|
FullTimestamp: true,
|
|
})
|
|
log.SetLevel(logrus.InfoLevel)
|
|
|
|
return &FileStorage{
|
|
dataDir: dataDir,
|
|
logger: log,
|
|
}, nil
|
|
}
|
|
|
|
func (s *FileStorage) ListFiles() ([]string, error) {
|
|
files, err := os.ReadDir(s.dataDir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var markdownFiles []string
|
|
for _, file := range files {
|
|
if !file.IsDir() && filepath.Ext(file.Name()) == ".md" {
|
|
markdownFiles = append(markdownFiles, file.Name())
|
|
}
|
|
}
|
|
|
|
return markdownFiles, nil
|
|
}
|
|
|
|
func (s *FileStorage) ReadFile(filename string) ([]byte, error) {
|
|
if !s.FileExists(filename) {
|
|
return nil, errors.New("file not found")
|
|
}
|
|
|
|
content, err := os.ReadFile(filepath.Join(s.dataDir, filename))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return content, nil
|
|
}
|
|
|
|
func (s *FileStorage) WriteFile(filename string, content []byte) error {
|
|
// Ensure the file has .md extension
|
|
if filepath.Ext(filename) != ".md" {
|
|
filename += ".md"
|
|
}
|
|
|
|
filepath := filepath.Join(s.dataDir, filename)
|
|
err := os.WriteFile(filepath, content, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s.logger.Infof("Saved file: %s", filename)
|
|
return nil
|
|
}
|
|
|
|
func (s *FileStorage) DeleteFile(filename string) error {
|
|
filepath := filepath.Join(s.dataDir, filename)
|
|
err := os.Remove(filepath)
|
|
if err != nil {
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
return errors.New("file not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
s.logger.Infof("Deleted file: %s", filename)
|
|
return nil
|
|
}
|
|
|
|
func (s *FileStorage) FileExists(filename string) bool {
|
|
filepath := filepath.Join(s.dataDir, filename)
|
|
info, err := os.Stat(filepath)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return !info.IsDir()
|
|
}
|