Add API token to session object

Author: AbdulrhmnGhanem

models/auth/token.go

@@ -207,6 +207,63 @@ func UpdateAccessToken(t *AccessToken) error {
 	return err
 }
 
+const KitspaceTokenName = "__kitspace"
+const KitspaceApiToken = "ApiToken"
+
+func RenewKitspaceAccessToken(t *AccessToken) (*AccessToken, error) {
+	err := DeleteKitspaceAccessToken(t.UID)
+	if err != nil {
+		return nil, err
+	}
+	err = NewAccessToken(t)
+	if err != nil {
+		return nil, err
+	}
+	return t, nil
+}
+
+func GetKitspaceAccessToken(userID int64) (*AccessToken, error) {
+	token := &AccessToken{
+		UID:  userID,
+		Name: KitspaceTokenName,
+	}
+
+	exists, err := AccessTokenByNameExists(token)
+
+	if err != nil {
+		return nil, err
+	}
+
+	if !exists {
+		return CreateKitspaceAccessToken(userID)
+	}
+
+	return RenewKitspaceAccessToken(token)
+}
+
+func CreateKitspaceAccessToken(userID int64) (*AccessToken, error) {
+	token := &AccessToken{
+		UID:  userID,
+		Name: KitspaceTokenName,
+	}
+
+	err := NewAccessToken(token)
+
+	if err != nil {
+		return nil, err
+	}
+	return token, nil
+}
+
+func DeleteKitspaceAccessToken(userID int64) error {
+	_, err := db.GetEngine(db.DefaultContext).
+		Table("access_token").
+		Where("name = ?", KitspaceTokenName).
+		And("uid = ?", userID).
+		Delete()
+	return err
+}
+
 // CountAccessTokens count access tokens belongs to given user by options
 func CountAccessTokens(opts ListAccessTokensOptions) (int64, error) {
 	sess := db.GetEngine(db.DefaultContext).Where("uid=?", opts.UserID)

routers/web/auth/auth.go

@@ -382,6 +382,7 @@ func getUserName(gothUser *goth.User) string {
 
 // HandleSignOut resets the session and sets the cookies
 func HandleSignOut(ctx *context.Context) {
+	auth.DeleteKitspaceAccessToken(ctx.Doer.ID)
 	_ = ctx.Session.Flush()
 	_ = ctx.Session.Destroy(ctx.Resp, ctx.Req)
 	ctx.DeleteCookie(setting.CookieUserName)

routers/web/auth/kitspace_auth.go

@@ -3,6 +3,7 @@ package auth
 import (
 	"net/http"
 
+	auth_model "code.gitea.io/gitea/models/auth"
 	"code.gitea.io/gitea/models/db"
 	user_model "code.gitea.io/gitea/models/user"
 	"code.gitea.io/gitea/modules/context"
@@ -110,7 +111,6 @@ func KitspaceSignUp(ctx *context.Context) {
 
 	// Return the success response with user details
 	response["user"] = convert.ToUser(u, u)
-
 	ctx.JSON(http.StatusCreated, response)
 }
 
@@ -206,9 +206,24 @@ func GetKitspaceSession(ctx *context.Context) {
 		user = convert.ToUser(ctx.Doer, ctx.Doer)
 	}
 
+	if user != nil && ctx.Session.Get(auth_model.KitspaceApiToken) == nil {
+		token, err := auth_model.GetKitspaceAccessToken(user.ID)
+		if err != nil {
+			ctx.ServerError("Kitspace session error", err)
+			return
+		}
+
+		err = ctx.Session.Set(auth_model.KitspaceApiToken, token.Token)
+		if err != nil {
+			ctx.ServerError("Kitspace session error", err)
+			return
+		}
+	}
+
 	response := make(map[string]interface{})
 	response["user"] = user
 	response["csrf"] = ctx.Data["CsrfToken"]
+	response["ApiToken"] = ctx.Session.Get(auth_model.KitspaceApiToken)
 
 	ctx.JSON(http.StatusOK, response)
 }