133 lines
2.1 KiB
Go
133 lines
2.1 KiB
Go
package files
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type FSFileStore struct {
|
|
dir string
|
|
metadata map[string]*File
|
|
}
|
|
|
|
func NewFSFileStore(dir string) (*FSFileStore, error) {
|
|
s := &FSFileStore{
|
|
dir: dir,
|
|
metadata: make(map[string]*File),
|
|
}
|
|
|
|
err := s.readMetadata()
|
|
|
|
return s, err
|
|
}
|
|
|
|
func (s *FSFileStore) Store(f *File) error {
|
|
defer f.Body.Close()
|
|
|
|
metadata := &File{
|
|
ID: f.ID,
|
|
OriginalFilename: f.OriginalFilename,
|
|
MaxViews: f.MaxViews,
|
|
ExpiresOn: f.ExpiresOn,
|
|
}
|
|
|
|
path := filepath.Join(s.dir, f.ID)
|
|
|
|
dst, err := os.Create(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dst.Close()
|
|
|
|
n, err := io.Copy(dst, f.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s.metadata[f.ID] = metadata
|
|
s.metadata[f.ID].FileSize = n
|
|
|
|
if err := s.writeMetadata(); err != nil {
|
|
delete(s.metadata, f.ID)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *FSFileStore) Get(id string) (*File, error) {
|
|
metadata, ok := s.metadata[id]
|
|
if !ok {
|
|
return nil, fmt.Errorf("no such item")
|
|
}
|
|
|
|
path := filepath.Join(s.dir, id)
|
|
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
metadata.Body = f
|
|
|
|
return metadata, nil
|
|
}
|
|
|
|
func (s *FSFileStore) Delete(id string) error {
|
|
path := filepath.Join(s.dir, id)
|
|
if err := os.Remove(path); err != nil {
|
|
return err
|
|
}
|
|
|
|
delete(s.metadata, id)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *FSFileStore) List() ([]string, error) {
|
|
results := make([]string, 0, len(s.metadata))
|
|
for k := range s.metadata {
|
|
results = append(results, k)
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
func (s *FSFileStore) writeMetadata() error {
|
|
path := filepath.Join(s.dir, "metadata.json")
|
|
|
|
f, err := os.Create(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
encoder := json.NewEncoder(f)
|
|
if err := encoder.Encode(s.metadata); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *FSFileStore) readMetadata() error {
|
|
path := filepath.Join(s.dir, "metadata.json")
|
|
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
// TODO: Handle errors better
|
|
return nil
|
|
}
|
|
defer f.Close()
|
|
|
|
decoder := json.NewDecoder(f)
|
|
if err := decoder.Decode(&s.metadata); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|