58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"otaupdate/services"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/redis"
|
|
"github.com/fiskerinc/cloud-services/pkg/superset"
|
|
"github.com/fiskerinc/cloud-services/pkg/utils"
|
|
"github.com/fiskerinc/cloud-services/pkg/loggerdataresp"
|
|
)
|
|
|
|
type GuestTokenResp struct {
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
var (
|
|
getAccessTokenFunc = superset.GetAccessToken
|
|
getGuestTokenFunc = superset.GetGuestToken
|
|
)
|
|
|
|
// HandleDashboardToken godoc
|
|
// @Summary Get token for accessing dashboard
|
|
// @Description Returns token
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param Authorization header string false "Bearer <ID token>"
|
|
// @Param Api-Key header string false "<API token>"
|
|
// @Success 200 {object} GuestTokenResp
|
|
// @Failure 401 {object} common.JSONError "Unauthorized"
|
|
// @Failure 503 {object} common.JSONError "Service unavailable"
|
|
// @Router /dashboard/guest-token [get]
|
|
func HandleDashboardToken(w http.ResponseWriter, r *http.Request) {
|
|
redisClient := services.RedisClientPool().GetFromPool()
|
|
defer redisClient.Close()
|
|
|
|
accToken, err := getAccessTokenFunc(redisClient)
|
|
if loggerdataresp.BadDataErrorResp(w, err, http.StatusServiceUnavailable) {
|
|
return
|
|
}
|
|
|
|
guestToken, err := getGuestTokenFunc(redisClient, accToken)
|
|
if loggerdataresp.BadDataErrorResp(w, err, http.StatusServiceUnavailable) {
|
|
return
|
|
}
|
|
|
|
utils.RespJSON(w, http.StatusOK, GuestTokenResp{Token: guestToken})
|
|
}
|
|
|
|
func SetGetAccessTokenFunc(f func(r redis.Client) (string, error)) {
|
|
getAccessTokenFunc = f
|
|
}
|
|
|
|
func SetGetGuestTokenFunc(f func(r redis.Client, accToken string) (string, error)) {
|
|
getGuestTokenFunc = f
|
|
}
|