Add thumbnails store

server/thumb/thumb.go

@@ -0,0 +1,61 @@
+package thumb
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"path"
+)
+
+type Thumbs struct {
+	path string
+}
+
+func NewThumbs(p string) (*Thumbs, error) {
+	f, err := os.Stat(p)
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			err = os.MkdirAll(p, 0700)
+			if err != nil {
+				return nil, fmt.Errorf("could not create directory: %w", err)
+			}
+			f, err = os.Stat(p)
+			if err != nil {
+				return nil, fmt.Errorf("could not stat thumbs: %w", err)
+			}
+		} else {
+			return nil, fmt.Errorf("thumbs stat error: %w", err)
+		}
+	}
+
+	if !f.IsDir() {
+		return nil, fmt.Errorf("thumbs path is not a directory")
+	}
+
+	return &Thumbs{path: p}, nil
+}
+
+func (t Thumbs) Path() string {
+	return t.path
+}
+
+func (t Thumbs) Save(n string, d []byte) error {
+	dir, n := path.Split(n)
+	if dir != "" {
+		return fmt.Errorf("malformed path")
+	}
+
+	p := path.Join(t.path, n)
+	f, err := os.Create(p)
+	if err != nil {
+		return fmt.Errorf("could not create thumb %s: %w", p, err)
+	}
+	defer f.Close()
+
+	_, err = f.Write(d)
+	if err != nil {
+		return fmt.Errorf("could not write thumb: %w", err)
+	}
+
+	return nil
+}

server/thumb/thumb_test.go

@@ -0,0 +1,31 @@
+package thumb
+
+import (
+	"io/ioutil"
+	"os"
+	"path"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestThumbs(t *testing.T) {
+	d, err := ioutil.TempDir("", "glsl")
+	require.NoError(t, err)
+	defer os.RemoveAll(d)
+
+	thumbs, err := NewThumbs(d)
+	require.NoError(t, err)
+	require.Equal(t, d, thumbs.Path())
+
+	err = thumbs.Save("../secret", []byte{0})
+	require.Error(t, err)
+
+	err = thumbs.Save("1.png", []byte("data"))
+	require.NoError(t, err)
+
+	data, err := ioutil.ReadFile(path.Join(d, "1.png"))
+	require.NoError(t, err)
+
+	require.Equal(t, []byte("data"), data)
+}