Create users package

auth.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"time"
 
+	"git.t-juice.club/torjus/gpaste/users"
 	"github.com/golang-jwt/jwt"
 	"github.com/google/uuid"
 )
@@ -17,11 +18,11 @@ const (
 )
 
 type AuthService struct {
-	users      UserStore
+	users      users.UserStore
 	hmacSecret []byte
 }
 
-func NewAuthService(store UserStore, signingSecret []byte) *AuthService {
+func NewAuthService(store users.UserStore, signingSecret []byte) *AuthService {
 	return &AuthService{users: store, hmacSecret: signingSecret}
 }
 

auth_test.go

@@ -1,21 +1,23 @@
 package gpaste_test
 
 import (
+	"math/rand"
 	"testing"
 
 	"git.t-juice.club/torjus/gpaste"
+	"git.t-juice.club/torjus/gpaste/users"
 )
 
 func TestAuth(t *testing.T) {
 	t.Run("Token", func(t *testing.T) {
-		us := gpaste.NewMemoryUserStore()
+		us := users.NewMemoryUserStore()
 		secret := []byte(randomString(16))
 		as := gpaste.NewAuthService(us, secret)
 
 		username := randomString(8)
 		password := randomString(16)
 
-		user := &gpaste.User{Username: username}
+		user := &users.User{Username: username}
 		if err := user.SetPassword(password); err != nil {
 			t.Fatalf("error setting user password: %s", err)
 		}
@@ -37,3 +39,13 @@ func TestAuth(t *testing.T) {
 		}
 	})
 }
+
+func randomString(length int) string {
+	const charset = "abcdefghijklmnopqrstabcdefghijklmnopqrstuvwxyz" +
+		"ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+	b := make([]byte, length)
+	for i := range b {
+		b[i] = charset[rand.Intn(len(charset))]
+	}
+	return string(b)
+}

http.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"strings"
 
+	"git.t-juice.club/torjus/gpaste/users"
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/google/uuid"
@@ -14,7 +15,7 @@ import (
 
 type HTTPServer struct {
 	Files        FileStore
-	Users        UserStore
+	Users        users.UserStore
 	Auth         *AuthService
 	config       *ServerConfig
 	Logger       *zap.SugaredLogger
@@ -29,12 +30,12 @@ func NewHTTPServer(cfg *ServerConfig) *HTTPServer {
 		AccessLogger: zap.NewNop().Sugar(),
 	}
 	srv.Files = NewMemoryFileStore()
-	srv.Users = NewMemoryUserStore()
+	srv.Users = users.NewMemoryUserStore()
 	srv.Auth = NewAuthService(srv.Users, []byte(srv.config.SigningSecret))
 
 	// Create initial user
 	// TODO: Do properly
-	user := &User{Username: "admin"}
+	user := &users.User{Username: "admin"}
 	user.SetPassword("admin")
 	srv.Users.Store(user)
 
@@ -214,7 +215,7 @@ func (s *HTTPServer) HandlerAPIUserCreate(w http.ResponseWriter, r *http.Request
 	}
 
 	// TODO: Ensure user does not already exist
-	user := &User{Username: req.Username}
+	user := &users.User{Username: req.Username}
 	if err := user.SetPassword(req.Password); err != nil {
 		s.Logger.Warnw("Error setting user password.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
 		w.WriteHeader(http.StatusBadRequest)

http_test.go

@@ -11,6 +11,7 @@ import (
 	"testing"
 
 	"git.t-juice.club/torjus/gpaste"
+	"git.t-juice.club/torjus/gpaste/users"
 )
 
 func TestHandlers(t *testing.T) {
@@ -101,7 +102,7 @@ func TestHandlers(t *testing.T) {
 		// TODO: Add test
 		username := "admin"
 		password := "admin"
-		user := &gpaste.User{Username: username}
+		user := &users.User{Username: username}
 		if err := user.SetPassword(password); err != nil {
 			t.Fatalf("Error setting user password: %s", err)
 		}

users/user.go

@@ -1,4 +1,4 @@
-package gpaste
+package users
 
 import "golang.org/x/crypto/bcrypt"
 

users/user_test.go

@@ -1,10 +1,10 @@
-package gpaste_test
+package users_test
 
 import (
 	"math/rand"
 	"testing"
 
-	"git.t-juice.club/torjus/gpaste"
+	"git.t-juice.club/torjus/gpaste/users"
 )
 
 func TestUser(t *testing.T) {
@@ -15,7 +15,7 @@ func TestUser(t *testing.T) {
 		}
 
 		for username, password := range userMap {
-			user := &gpaste.User{Username: username}
+			user := &users.User{Username: username}
 			if err := user.SetPassword(password); err != nil {
 				t.Fatalf("Error setting password: %s", err)
 			}

users/userstore_bolt.go

@@ -1,4 +1,4 @@
-package gpaste
+package users
 
 import (
 	"encoding/json"

users/userstore_bolt_test.go

@@ -1,18 +1,18 @@
-package gpaste_test
+package users_test
 
 import (
 	"path/filepath"
 	"testing"
 
-	"git.t-juice.club/torjus/gpaste"
+	"git.t-juice.club/torjus/gpaste/users"
 )
 
 func TestBoltUserStore(t *testing.T) {
 	tmpDir := t.TempDir()
-	newFunc := func() (func(), gpaste.UserStore) {
+	newFunc := func() (func(), users.UserStore) {
 		tmpFile := filepath.Join(tmpDir, randomString(8))
 
-		store, err := gpaste.NewBoltUserStore(tmpFile)
+		store, err := users.NewBoltUserStore(tmpFile)
 		if err != nil {
 			t.Fatalf("Error creating store: %s", err)
 		}

users/userstore_memory.go

@@ -1,4 +1,4 @@
-package gpaste
+package users
 
 import (
 	"fmt"

users/userstore_memory_test.go

@@ -0,0 +1,15 @@
+package users_test
+
+import (
+	"testing"
+
+	"git.t-juice.club/torjus/gpaste/users"
+)
+
+func TestMemoryUserStore(t *testing.T) {
+	newFunc := func() (func(), users.UserStore) {
+		return func() {}, users.NewMemoryUserStore()
+	}
+
+	RunUserStoreTest(newFunc, t)
+}

users/userstore_test.go

@@ -1,26 +1,26 @@
-package gpaste_test
+package users_test
 
 import (
 	"testing"
 
-	"git.t-juice.club/torjus/gpaste"
+	"git.t-juice.club/torjus/gpaste/users"
 	"github.com/google/go-cmp/cmp"
 )
 
-func RunUserStoreTest(newFunc func() (func(), gpaste.UserStore), t *testing.T) {
+func RunUserStoreTest(newFunc func() (func(), users.UserStore), t *testing.T) {
 	t.Run("Basics", func(t *testing.T) {
 		cleanup, s := newFunc()
 		t.Cleanup(cleanup)
 
-		userMap := make(map[string]*gpaste.User)
+		userMap := make(map[string]*users.User)
 		passwordMap := make(map[string]string)
 		for i := 0; i < 10; i++ {
 			username := randomString(8)
 			password := randomString(16)
 			passwordMap[username] = password
-			user := &gpaste.User{
+			user := &users.User{
 				Username: username,
-				Roles:    []gpaste.Role{gpaste.RoleAdmin},
+				Roles:    []users.Role{users.RoleAdmin},
 			}
 			if err := user.SetPassword(password); err != nil {
 				t.Fatalf("Error setting password: %s", err)

userstore_memory_test.go

@@ -1,15 +0,0 @@
-package gpaste_test
-
-import (
-	"testing"
-
-	"git.t-juice.club/torjus/gpaste"
-)
-
-func TestMemoryUserStore(t *testing.T) {
-	newFunc := func() (func(), gpaste.UserStore) {
-		return func() {}, gpaste.NewMemoryUserStore()
-	}
-
-	RunUserStoreTest(newFunc, t)
-}