Add custom session

cmd/api/main.go

@@ -2,17 +2,21 @@ package main
 
 import (
 	"fmt"
+	"github.com/joho/godotenv"
 	"gothtest/internal/server"
-  "gothtest/internal/auth"
+	"log"
 )
 
 func main() {
+	err := godotenv.Load()
+	if err != nil {
+		log.Fatal("Error loading .env file")
+	}
 
 	server := server.NewServer()
-  auth.NewAuth()
 
 	fmt.Println("Server start")
-	err := server.ListenAndServe()
+	err = server.ListenAndServe()
 	if err != nil {
 		panic(fmt.Sprintf("cannot start server: %s", err))
 	}

internal/auth/auth.go

@@ -1,40 +1,29 @@
 package auth
+
 import (
+	"github.com/gorilla/sessions"
+	"github.com/markbates/goth"
+	"github.com/markbates/goth/gothic"
+	"github.com/markbates/goth/providers/openidConnect"
 	"log"
 	"os"
-  //"github.com/gorilla/sessions"
-  "github.com/joho/godotenv"
-  "github.com/markbates/goth"
-  //"github.com/markbates/goth/gothic"
-  "github.com/markbates/goth/providers/openidConnect"
 )
 
-const (
+func NewAuth(sessionStore *sessions.Store) {
-  key = "iebdyjckwoevhdixnwgwunrvxuqobetgy"
-  MaxAge = 86400 * 30
-  IsProd = false
-)
-
-func NewAuth() {
-  err := godotenv.Load()
-  if err != nil {log.Fatal("Error loading .env file")}
-  
 	oidcId := os.Getenv("OIDC_ID")
 	oidcSec := os.Getenv("OIDC_SECRET")
 	oidcDiscUrl := os.Getenv("OIDC_DISC_URL")
 	oidcRedirectUrl := "http://localhost:3003/auth/openid-connect/callback"
-/*
-  store := sessions.NewCookieStore([]byte(key))
-  store.MaxAge(MaxAge)
-  store.Options.Path = "/"
-  store.Options.HttpOnly = true
-  store.Options.Secure = IsProd
 
-  gothic.Store = store
-*/
 	openidConnect, err := openidConnect.New(oidcId, oidcSec, oidcRedirectUrl, oidcDiscUrl)
 	if openidConnect == nil || err != nil {
 		log.Fatal("Error setting up oidc")
 	}
 	goth.UseProviders(openidConnect)
+
+	if sessionStore != nil {
+		gothic.Store = *sessionStore
+	} else {
+    log.Println("No auth session store set. Falling back to default gothic setting.")
+  }
 }

internal/server/server.go

@@ -7,15 +7,19 @@ import (
 	"strconv"
 	"time"
 
+	"github.com/gorilla/sessions"
 	_ "github.com/joho/godotenv/autoload"
 
+	"gothtest/internal/auth"
 	"gothtest/internal/database"
+	"gothtest/internal/session"
 )
 
 type Server struct {
 	port int
 
 	db    database.Service
+	store sessions.Store
 }
 
 func NewServer() *http.Server {
@@ -24,7 +28,9 @@ func NewServer() *http.Server {
 		port: port,
 
 		db:    database.New(),
+		store: session.New(),
 	}
+	auth.NewAuth(&NewServer.store)
 
 	// Declare Server config
 	server := &http.Server{

internal/session/session.go

@@ -0,0 +1,25 @@
+package session
+
+import (
+	"github.com/gorilla/sessions"
+	"os"
+)
+
+const (
+	MaxAge = 86400 * 30
+	IsProd = false
+)
+
+func New() sessions.Store {
+	key := os.Getenv("SESSION_SECRET")
+	if key == "" {
+		return nil
+	}
+	store := sessions.NewCookieStore([]byte(key))
+	store.MaxAge(MaxAge)
+	store.Options.Path = "/"
+	store.Options.HttpOnly = true
+	store.Options.Secure = IsProd
+
+	return store
+}