feat: persist sessions and auto redirect to app

internal/hooks/hooks.go

@@ -1,12 +1,14 @@
 package hooks
 
 import (
+	"strings"
 	"tinyauth/internal/auth"
 	"tinyauth/internal/providers"
 	"tinyauth/internal/types"
 
 	"github.com/gin-contrib/sessions"
 	"github.com/gin-gonic/gin"
+	"golang.org/x/oauth2"
 )
 
 func NewHooks(auth *auth.Auth, providers *providers.Providers) *Hooks {
@@ -24,7 +26,6 @@ type Hooks struct {
 func (hooks *Hooks) UseUserContext(c *gin.Context) (types.UserContext, error) {
 	session := sessions.Default(c)
 	sessionCookie := session.Get("tinyauth_sid")
-	oauthProviderCookie := session.Get("tinyauth_oauth_provider")
 
 	if sessionCookie == nil {
 		return types.UserContext{
@@ -35,19 +36,33 @@ func (hooks *Hooks) UseUserContext(c *gin.Context) (types.UserContext, error) {
 		}, nil
 	}
 
-	email, emailOk := sessionCookie.(string)
-	provider, providerOk := oauthProviderCookie.(string)
+	data, dataOk := sessionCookie.(string)
 
-	if provider == "" || !providerOk {
-		if !emailOk {
-			return types.UserContext{
-				Email:      "",
-				IsLoggedIn: false,
-				OAuth:      false,
-				Provider:   "",
-			}, nil
-		}
-		user := hooks.Auth.GetUser(email)
+	if !dataOk {
+		return types.UserContext{
+			Email:      "",
+			IsLoggedIn: false,
+			OAuth:      false,
+			Provider:   "",
+		}, nil
+	}
+
+	split := strings.Split(data, ":")
+
+	if len(split) != 2 {
+		return types.UserContext{
+			Email:      "",
+			IsLoggedIn: false,
+			OAuth:      false,
+			Provider:   "",
+		}, nil
+	}
+
+	sessionType := split[0]
+	sessionValue := split[1]
+
+	if sessionType == "email" {
+		user := hooks.Auth.GetUser(sessionValue)
 		if user == nil {
 			return types.UserContext{
 				Email:      "",
@@ -57,16 +72,31 @@ func (hooks *Hooks) UseUserContext(c *gin.Context) (types.UserContext, error) {
 			}, nil
 		}
 		return types.UserContext{
-			Email:      email,
+			Email:      sessionValue,
 			IsLoggedIn: true,
 			OAuth:      false,
 			Provider:   "",
 		}, nil
 	}
 
-	oauthEmail, oauthEmailErr := hooks.Providers.GetUser(provider)
+	provider := hooks.Providers.GetProvider(sessionType)
 
-	if oauthEmailErr != nil {
+	if provider == nil {
+		return types.UserContext{
+			Email:      "",
+			IsLoggedIn: false,
+			OAuth:      false,
+			Provider:   "",
+		}, nil
+	}
+
+	provider.Token = &oauth2.Token{
+		AccessToken: sessionValue,
+	}
+
+	email, emailErr := hooks.Providers.GetUser(sessionType)
+
+	if emailErr != nil {
 		return types.UserContext{
 			Email:      "",
 			IsLoggedIn: false,
@@ -76,9 +106,9 @@ func (hooks *Hooks) UseUserContext(c *gin.Context) (types.UserContext, error) {
 	}
 
 	return types.UserContext{
-		Email:      oauthEmail,
+		Email:      email,
 		IsLoggedIn: true,
 		OAuth:      true,
-		Provider:   provider,
+		Provider:   sessionType,
 	}, nil
 }