Ensure config files/directories for sessions & KMS (#6)

Author: swift1337

.gitignore

@@ -2,6 +2,7 @@
 .vscode
 !.vscode/extensions.json
 bin/
+tmp/*
 config/*.yml
 !config/*example.yml
 web/redoc/**/*.json

internal/db/connection/bolt/bolt.go

@@ -1,6 +1,7 @@
 package bolt
 
 import (
+	"github.com/oxygenpay/oxygen/internal/util"
 	"github.com/pkg/errors"
 	"github.com/rs/zerolog"
 	"go.etcd.io/bbolt"
@@ -17,9 +18,15 @@ type Connection struct {
 
 const WalletsBucket = "wallets"
 
+const chmodReadWrite = 0660
+
 func Open(cfg Config, logger *zerolog.Logger) (*Connection, error) {
 	log := logger.With().Str("channel", "bolt").Logger()
 
+	if err := util.EnsureFile(cfg.DataSource, chmodReadWrite); err != nil {
+		return nil, errors.Wrap(err, "unable to ensure KMS database file")
+	}
+
 	log.Info().Str("db_path", cfg.DataSource).Msg("connecting to bolt")
 	db, err := bbolt.Open(cfg.DataSource, 0660, nil)
 	if err != nil {

internal/server/http/middleware/middleware.go

@@ -1,6 +1,7 @@
 package middleware
 
 import (
+	"fmt"
 	"net/http"
 
 	"github.com/gorilla/sessions"
@@ -9,6 +10,7 @@ import (
 	mw "github.com/labstack/echo/v4/middleware"
 	"github.com/labstack/gommon/log"
 	"github.com/oxygenpay/oxygen/internal/server/http/common"
+	"github.com/oxygenpay/oxygen/internal/util"
 	"github.com/rs/zerolog"
 )
 
@@ -25,7 +27,13 @@ type SessionConfig struct {
 
 const sessionOptionsKey = "session_options"
 
+const chmodReadWrite = 0660
+
 func Session(cfg SessionConfig) echo.MiddlewareFunc {
+	if err := util.EnsureDirectory(cfg.FilesystemPath, chmodReadWrite); err != nil {
+		panic(fmt.Sprintf("unable to ensure session directory %q", cfg.FilesystemPath))
+	}
+
 	sessionOptions := sessions.Options{
 		Path:     cfg.CookiePath,
 		Domain:   cfg.CookieDomain,

internal/util/fs.go

@@ -0,0 +1,50 @@
+package util
+
+import (
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+
+	"github.com/pkg/errors"
+)
+
+// EnsureFile checks that file exists. If not, creates an empty file.
+func EnsureFile(path string, mode fs.FileMode) error {
+	stat, err := os.Stat(path)
+
+	switch {
+	case os.IsNotExist(err):
+		if errDir := EnsureDirectory(filepath.Dir(path), mode); errDir != nil {
+			return errDir
+		}
+
+		return createFile(path, mode)
+	case err != nil:
+		return err
+	case stat.IsDir():
+		return fmt.Errorf("path %q is a directory", path)
+	}
+
+	return nil
+}
+
+// EnsureDirectory checks that file exists. If not, creates an empty directory.
+func EnsureDirectory(path string, mode fs.FileMode) error {
+	if err := os.MkdirAll(path, mode); err != nil {
+		return errors.Wrapf(err, "unable to ensure directory %q", path)
+	}
+
+	return nil
+}
+
+func createFile(path string, mode fs.FileMode) error {
+	file, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, mode)
+	if err != nil {
+		return err
+	}
+
+	_ = file.Close()
+
+	return nil
+}

internal/util/fs_test.go

@@ -0,0 +1,65 @@
+package util
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFS(t *testing.T) {
+	// RW
+	const mode = 0770
+
+	tempRoot := t.TempDir()
+	wrap := func(path string) string {
+		return tempRoot + path
+	}
+
+	for _, tt := range []struct {
+		name   string
+		path   string
+		isFile bool
+	}{
+		{
+			name:   "ensures env file",
+			path:   "/app/oxygen.env",
+			isFile: true,
+		},
+		{
+			name:   "ensures env file one more time",
+			path:   "/app/oxygen.env",
+			isFile: true,
+		},
+		{
+			name: "creates sessions directory",
+			path: "/app/sessions",
+		},
+		{
+			name:   "creates bolt.db file",
+			path:   "/app/kms/kms.db",
+			isFile: true,
+		},
+	} {
+		t.Run(tt.name, func(t *testing.T) {
+			// ARRANGE
+			p := wrap(tt.path)
+
+			// ACT
+			var err error
+			if tt.isFile {
+				err = EnsureFile(p, mode)
+			} else {
+				err = EnsureDirectory(p, mode)
+			}
+
+			// ASSERT
+			assert.NoError(t, err)
+
+			if tt.isFile {
+				assert.FileExists(t, p)
+			} else {
+				assert.DirExists(t, p)
+			}
+		})
+	}
+}