user authentication structure in place

db/db_cold/authTokens.go

@@ -1,11 +1,15 @@
 package db_cold
 
 import (
+	"log"
+	"reflect"
 	"time"
 
 	"gorm.io/gorm"
 )
 
+const EXPIRATION_BUFFER = 5 * time.Minute
+
 type UserAuth struct {
 	gorm.Model
 	UserID       string `gorm:"primarykey"`
@@ -31,12 +35,12 @@ func (db *DBColdConn) GetAllUserAuth() ([]UserAuth, error) {
 }
 
 // add or update user auth, based on ID
-func (db *DBColdConn) UpdateUserAuth(userID, userName, userLogin, accessToken, refreshToken string, tokenExpires time.Time) error {
+func (db *DBColdConn) updateUserAuthTime(userID, userName, userLogin, accessToken, refreshToken string, tokenExpires time.Time) error {
 	userAuth := UserAuth{
-		UserID: userID,
-		UserName: userName,
-		UserLogin: userLogin,
-		AccessToken: accessToken,
+		UserID:       userID,
+		UserName:     userName,
+		UserLogin:    userLogin,
+		AccessToken:  accessToken,
 		RefreshToken: refreshToken,
 		TokenExpires: tokenExpires,
 	}
@@ -62,3 +66,36 @@ func (db *DBColdConn) UpdateUserAuth(userID, userName, userLogin, accessToken, r
 
 	return nil
 }
+
+func (db *DBColdConn) updateUserAuthDuration(userID, userName, userLogin, accessToken, refreshToken string, tokenExpires time.Duration) error {
+	return db.updateUserAuthTime(userID, userName, userLogin, accessToken, refreshToken, time.Now().Add(tokenExpires-EXPIRATION_BUFFER))
+}
+
+func (db *DBColdConn) updateUserAuthInt(userID, userName, userLogin, accessToken, refreshToken string, tokenExpires int) error {
+	return db.updateUserAuthDuration(userID, userName, userLogin, accessToken, refreshToken, time.Duration(tokenExpires)*time.Second)
+}
+
+/** tokenExpires: time.Time | time.Duration | int */
+func (db *DBColdConn) UpdateUserAuth(userID, userName, userLogin, accessToken, refreshToken string, tokenExpires any) error {
+	/*
+		expires can be:
+		- time.Time
+			- passed directly
+		- time.Duration
+			- converted to time.Time from now
+		- int
+			- converted to time.Duration in seconds then time.Time from now
+	*/
+
+	switch exp := tokenExpires.(type) {
+	case time.Time:
+		return db.updateUserAuthTime(userID, userName, userLogin, accessToken, refreshToken, exp)
+	case time.Duration:
+		return db.updateUserAuthDuration(userID, userName, userLogin, accessToken, refreshToken, exp)
+	case int:
+		return db.updateUserAuthInt(userID, userName, userLogin, accessToken, refreshToken, exp)
+	default:
+		log.Panicf("invalid type passed to any field in TwitchAuth.updateUserDetails(): %s, value: %v", reflect.TypeOf(tokenExpires), tokenExpires)
+		panic("we've already panicked")
+	}
+}

db/db_cold/main.go

@@ -16,8 +16,8 @@ func InitDBColdConn(conf *util.Config) (*DBColdConn, error) {
 		return nil, err
 	}
 	ctx := context.Background()
-	
-	cold := &DBColdConn{ db, ctx }
+
+	cold := &DBColdConn{db, ctx}
 	cold.initUserAuth()
 
 	return cold, nil
@@ -25,5 +25,5 @@ func InitDBColdConn(conf *util.Config) (*DBColdConn, error) {
 
 type DBColdConn struct {
 	Gorm *gorm.DB
-	Ctx context.Context
+	Ctx  context.Context
 }

db/main.go

@@ -15,10 +15,10 @@ func InitDBConn(conf *util.Config) (*DBConn, error) {
 	if err != nil {
 		return nil, err
 	}
-	return &DBConn{ hot, cold }, nil
+	return &DBConn{hot, cold}, nil
 }
 
 type DBConn struct {
-	Hot *db_hot.DBHotConn
+	Hot  *db_hot.DBHotConn
 	Cold *db_cold.DBColdConn
 }