Initial commit

store/memory.go

@@ -0,0 +1,53 @@
+package store
+
+import (
+	"sync"
+
+	"gitea.benny.dog/torjus/ezshare/pb"
+	"github.com/google/uuid"
+)
+
+var _ FileStore = &MemoryFileStore{}
+
+type MemoryFileStore struct {
+	filesLock sync.RWMutex
+	files     map[string]*pb.File
+}
+
+func NewMemoryFileStore() *MemoryFileStore {
+	return &MemoryFileStore{files: make(map[string]*pb.File)}
+}
+
+func (s *MemoryFileStore) GetFile(id string) (*pb.File, error) {
+	s.filesLock.RLock()
+	defer s.filesLock.RUnlock()
+
+	if file, ok := s.files[id]; ok {
+		return file, nil
+	}
+
+	return nil, ErrNoSuchFile
+}
+
+func (s *MemoryFileStore) StoreFile(file *pb.File) (string, error) {
+	s.filesLock.Lock()
+	defer s.filesLock.Unlock()
+
+	id := uuid.Must(uuid.NewRandom()).String()
+	file.FileId = id
+
+	s.files[id] = file
+
+	return id, nil
+}
+
+func (s *MemoryFileStore) DeleteFile(id string) error {
+	s.filesLock.Lock()
+	defer s.filesLock.Unlock()
+	if _, ok := s.files[id]; !ok {
+		return ErrNoSuchFile
+	}
+
+	delete(s.files, id)
+	return nil
+}

store/store.go

@@ -0,0 +1,15 @@
+package store
+
+import (
+	"fmt"
+
+	"gitea.benny.dog/torjus/ezshare/pb"
+)
+
+var ErrNoSuchFile = fmt.Errorf("no such file")
+
+type FileStore interface {
+	GetFile(id string) (*pb.File, error)
+	StoreFile(file *pb.File) (string, error)
+	DeleteFile(id string) error
+}