package service_auth

import (
	"context"
	"nixcn-cms/internal/authtoken"
	"nixcn-cms/internal/exception"
	"nixcn-cms/service/shared"

	"github.com/spf13/viper"
)

type RefreshData struct {
	RefreshToken string `json:"refresh_token"`
}

type RefreshPayload struct {
	Context context.Context
	Data    *RefreshData
}

type RefreshResult struct {
	Common shared.CommonResult
	Data   *TokenResponse
}

func (self *AuthServiceImpl) Refresh(payload *RefreshPayload) (result *RefreshResult) {
	JwtTool := authtoken.Token{
		Application: viper.GetString("server.application"),
	}

	// 1. Refresh Access Token
	accessToken, err := JwtTool.RefreshAccessToken(payload.Context, payload.Data.RefreshToken)
	if err != nil {
		exception := new(exception.Builder).
			SetStatus(exception.StatusUser).
			SetService(exception.ServiceAuth).
			SetEndpoint(exception.EndpointAuthServiceRefresh).
			SetType(exception.TypeSpecific).
			SetOriginal(exception.AuthRefreshInvalidToken).
			SetError(err).
			Throw(payload.Context)

		result = &RefreshResult{
			Common: shared.CommonResult{
				HttpCode:  401,
				Exception: exception,
			},
			Data: nil,
		}

		return
	}

	// 2. Renew Refresh Token (Rotation)
	refreshToken, err := JwtTool.RenewRefreshToken(payload.Context, payload.Data.RefreshToken)
	if err != nil {
		exception := new(exception.Builder).
			SetStatus(exception.StatusServer).
			SetService(exception.ServiceAuth).
			SetEndpoint(exception.EndpointAuthServiceRefresh).
			SetType(exception.TypeSpecific).
			SetOriginal(exception.AuthRefreshRenewFailed).
			SetError(err).
			Throw(payload.Context)

		result = &RefreshResult{
			Common: shared.CommonResult{
				HttpCode:  500,
				Exception: exception,
			},
			Data: nil,
		}

		return
	}

	// 3. Success Assignment
	exception := new(exception.Builder).
		SetStatus(exception.StatusSuccess).
		SetService(exception.ServiceAuth).
		SetEndpoint(exception.EndpointAuthServiceRefresh).
		SetType(exception.TypeCommon).
		SetOriginal(exception.CommonSuccess).
		SetError(nil).
		Throw(payload.Context)

	result = &RefreshResult{
		Common: shared.CommonResult{
			HttpCode:  200,
			Exception: exception,
		},
		Data: &TokenResponse{
			AccessToken:  accessToken,
			RefreshToken: refreshToken,
		},
	}

	return
}