/*
Copyright 2023 CAcert Inc.
SPDX-License-Identifier: Apache-2.0

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package services

import (
	"errors"
	"fmt"
	"time"

	"github.com/gorilla/sessions"
	"github.com/lestrrat-go/jwx/jwk"
	"github.com/lestrrat-go/jwx/jwt"
	"github.com/lestrrat-go/jwx/jwt/openid"
	log "github.com/sirupsen/logrus"
)

type OIDCTokenInfo struct {
	AccessToken  string
	IDToken      string
	RefreshToken string
	Expires      time.Time
	Name         string
	Email        string
}

type TokenInfoService struct {
	logger *log.Logger
	keySet jwk.Set
}

func (s *TokenInfoService) GetTokenInfo(session *sessions.Session) (*OIDCTokenInfo, error) {
	tokenInfo := &OIDCTokenInfo{}

	var ok bool

	if tokenInfo.AccessToken, ok = session.Values[SessionAccessToken].(string); ok {
		s.logger.WithField("access_token", tokenInfo.AccessToken).Debug("found access token in session")
	}

	if tokenInfo.RefreshToken, ok = session.Values[SessionRefreshToken].(string); ok {
		s.logger.WithField("refresh_token", tokenInfo.RefreshToken).Debug("found refresh token in session")
	}

	if tokenInfo.IDToken, ok = session.Values[SessionIDToken].(string); ok {
		s.logger.WithField("id_token", tokenInfo.IDToken).Debug("found ID token in session")
	}

	if tokenInfo.IDToken == "" {
		return tokenInfo, nil
	}

	oidcToken, err := ParseIDToken(tokenInfo.IDToken, s.keySet)
	if err != nil {
		return nil, fmt.Errorf("could not parse ID token: %w", err)
	}

	tokenInfo.Expires = oidcToken.Expiration()

	tokenInfo.Name = oidcToken.Name()

	tokenInfo.Email = oidcToken.Email()

	return tokenInfo, nil
}

func InitTokenInfoService(logger *log.Logger, oidcInfo *OIDCInformation) (*TokenInfoService, error) {
	return &TokenInfoService{logger: logger, keySet: oidcInfo.KeySet}, nil
}

func ParseIDToken(token string, keySet jwk.Set) (openid.Token, error) {
	var (
		parsedIDToken jwt.Token
		err           error
	)

	if parsedIDToken, err = jwt.ParseString(token, jwt.WithKeySet(keySet), jwt.WithToken(openid.New())); err != nil {
		return nil, fmt.Errorf("could not parse ID token: %w", err)
	}

	if v, ok := parsedIDToken.(openid.Token); ok {
		return v, nil
	}

	return nil, errors.New("ID token is no OpenID Connect Identity Token")
}