Pg to bun (#148)

* start moving to bun

* changing more stuff

* more

* and yet more

* tests passing

* seems stable now

* more big changes

* small fix

* little fixes

2 files changed, 10 insertions, 54 deletions

diff --git a/internal/router/router.go b/internal/router/router.go
index c5f105448..621d93ff5 100644
--- a/internal/router/router.go
+++ b/internal/router/router.go
@@ -103,7 +103,7 @@ func (r *router) Stop(ctx context.Context) error {
 //
 // The given DB is only used in the New function for parsing config values, and is not otherwise
 // pinned to the router.
-func New(cfg *config.Config, db db.DB, logger *logrus.Logger) (Router, error) {
+func New(ctx context.Context, cfg *config.Config, db db.DB, logger *logrus.Logger) (Router, error) {
 
 	// gin has different log modes; for convenience, we match the gin log mode to
 	// whatever log mode has been set for logrus
@@ -141,7 +141,7 @@ func New(cfg *config.Config, db db.DB, logger *logrus.Logger) (Router, error) {
 	}
 
 	// enable session store middleware on the engine
-	if err := useSession(cfg, db, engine); err != nil {
+	if err := useSession(ctx, cfg, db, engine); err != nil {
 		return nil, err
 	}
 
diff --git a/internal/router/session.go b/internal/router/session.go
index 38810572f..4359a8a60 100644
--- a/internal/router/session.go
+++ b/internal/router/session.go
@@ -19,7 +19,7 @@
 package router
 
 import (
-	"crypto/rand"
+	"context"
 	"errors"
 	"fmt"
 	"net/http"
@@ -29,8 +29,6 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/superseriousbusiness/gotosocial/internal/config"
 	"github.com/superseriousbusiness/gotosocial/internal/db"
-	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
-	"github.com/superseriousbusiness/gotosocial/internal/id"
 )
 
 // SessionOptions returns the standard set of options to use for each session.
@@ -45,34 +43,23 @@ func SessionOptions(cfg *config.Config) sessions.Options {
 	}
 }
 
-func useSession(cfg *config.Config, dbService db.DB, engine *gin.Engine) error {
+func useSession(ctx context.Context, cfg *config.Config, sessionDB db.Session, engine *gin.Engine) error {
 	// check if we have a saved router session already
-	routerSessions := []*gtsmodel.RouterSession{}
-	if err := dbService.GetAll(&routerSessions); err != nil {
+	rs, err := sessionDB.GetSession(ctx)
+	if err != nil {
 		if err != db.ErrNoEntries {
 			// proper error occurred
 			return err
 		}
-	}
-
-	var rs *gtsmodel.RouterSession
-	if len(routerSessions) == 1 {
-		// we have a router session stored
-		rs = routerSessions[0]
-	} else if len(routerSessions) == 0 {
-		// we have no router sessions so we need to create a new one
-		var err error
-		rs, err = routerSession(dbService)
+		// no session saved so create a new one
+		rs, err = sessionDB.CreateSession(ctx)
 		if err != nil {
-			return fmt.Errorf("error creating new router session: %s", err)
+			return err
 		}
-	} else {
-		// we should only have one router session stored ever
-		return errors.New("we had more than one router session in the db")
 	}
 
 	if rs == nil {
-		return errors.New("error getting or creating router session: router session was nil")
+		return errors.New("router session was nil")
 	}
 
 	store := memstore.NewStore(rs.Auth, rs.Crypt)
@@ -81,34 +68,3 @@ func useSession(cfg *config.Config, dbService db.DB, engine *gin.Engine) error {
 	engine.Use(sessions.Sessions(sessionName, store))
 	return nil
 }
-
-// routerSession generates a new router session with random auth and crypt bytes,
-// puts it in the database for persistence, and returns it for use.
-func routerSession(dbService db.DB) (*gtsmodel.RouterSession, error) {
-	auth := make([]byte, 32)
-	crypt := make([]byte, 32)
-
-	if _, err := rand.Read(auth); err != nil {
-		return nil, err
-	}
-	if _, err := rand.Read(crypt); err != nil {
-		return nil, err
-	}
-
-	rid, err := id.NewULID()
-	if err != nil {
-		return nil, err
-	}
-
-	rs := &gtsmodel.RouterSession{
-		ID:    rid,
-		Auth:  auth,
-		Crypt: crypt,
-	}
-
-	if err := dbService.Put(rs); err != nil {
-		return nil, err
-	}
-
-	return rs, nil
-}