supertokenEndpoint.go

package super

import (
	"github.com/gofiber/fiber/v2"
	log "github.com/sirupsen/logrus"
	"github.com/zachmann/mytoken/internal/config"
	"github.com/zachmann/mytoken/internal/endpoints/token/super/polling"
	"github.com/zachmann/mytoken/internal/model"
	"github.com/zachmann/mytoken/internal/oidc/authcode"
	"github.com/zachmann/mytoken/internal/supertoken"
	"github.com/zachmann/mytoken/internal/utils/ctxUtils"
)

func HandleSuperTokenEndpoint(ctx *fiber.Ctx) error {
	grantType, err := ctxUtils.GetGrantType(ctx)
	if err != nil {
		return model.ErrorToBadRequestErrorResponse(err).Send(ctx)
	}
	log.WithField("grant_type", grantType).Trace("Received super token request")
	switch grantType {
	case model.GrantTypeSuperToken:
		return supertoken.HandleSuperTokenFromSuperToken(ctx).Send(ctx)
	case model.GrantTypeOIDCFlow:
		return handleOIDCFlow(ctx)
	case model.GrantTypePollingCode:
		if config.Get().Features.Polling.Enabled {
			return polling.HandlePollingCode(ctx)
		}
	case model.GrantTypeAccessToken:
		if config.Get().Features.AccessTokenGrant.Enabled {
			return model.ResponseNYI.Send(ctx)
		}
	case model.GrantTypePrivateKeyJWT:
		if config.Get().Features.SignedJWTGrant.Enabled {
			return model.ResponseNYI.Send(ctx)
		}
	case model.GrantTypeTransferCode:
		if config.Get().Features.TransferCodes.Enabled {
			return model.ResponseNYI.Send(ctx)
		}
	}
	res := model.Response{
		Status:   fiber.StatusBadRequest,
		Response: model.APIErrorUnsupportedGrantType,
	}
	return res.Send(ctx)
}

func handleOIDCFlow(ctx *fiber.Ctx) error {
	flow := ctxUtils.GetOIDCFlow(ctx)
	switch flow {
	case model.OIDCFlowAuthorizationCode:
		return authcode.InitAuthCodeFlow(ctx.Body()).Send(ctx)
	case model.OIDCFlowDevice:
		return model.ResponseNYI.Send(ctx)
	default:
		res := model.Response{
			Status:   fiber.StatusBadRequest,
			Response: model.APIErrorUnsupportedOIDCFlow,
		}
		return res.Send(ctx)
	}
}