package webhooks

import (
	"context"
	"encoding/hex"
	"fmt"
	"time"

	"git.sr.ht/~sircmpwn/core-go/auth"
	"git.sr.ht/~sircmpwn/core-go/config"
	sq "github.com/Masterminds/squirrel"
)

// The following invariants apply to AuthConfig:
//  1. AuthMethod will be either OAUTH2 or INTERNAL
//  2. If OAUTH2, TokenHash, Grants, and Expires will be non-nil, and ClientID
//     may be non-nil, and NodeID will be nil.
//  3. If INTERNAL, TokenHash, Grants, Expires, and ClientID will be nil, and
//     NodeID will be non-nil.
type AuthConfig struct {
	AuthMethod string
	TokenHash  *string
	Grants     *string
	ClientID   *string
	Expires    *time.Time
	NodeID     *string
}

// Pulls auth details out of the config context and returns a structure of all
// of the information necessary to build a webhook context with the same
// authentication parameters.
func NewAuthConfig(ctx context.Context) (AuthConfig, error) {
	user := auth.ForContext(ctx)
	switch user.AuthMethod {
	case auth.AUTH_OAUTH_LEGACY:
		return AuthConfig{}, fmt.Errorf("Native webhooks are not supported with legacy OAuth")
	case auth.AUTH_OAUTH2:
		tokenHash := hex.EncodeToString(user.TokenHash[:])
		grants := user.BearerToken.Grants
		expires := user.BearerToken.Expires.Time()
		var clientID *string
		if user.BearerToken.ClientID != "" {
			_clientID := user.BearerToken.ClientID
			clientID = &_clientID
		}
		return AuthConfig{
			AuthMethod: user.AuthMethod,
			TokenHash:  &tokenHash,
			Grants:     &grants,
			Expires:    &expires,
			ClientID:   clientID,
		}, nil
	case auth.AUTH_COOKIE:
		// TODO: Should this work?
		return AuthConfig{}, fmt.Errorf("Native webhooks are not supported with web authentication")
	case auth.AUTH_INTERNAL:
		nodeID := config.ServiceName(ctx)
		return AuthConfig{
			AuthMethod: user.AuthMethod,
			NodeID:     &nodeID,
		}, nil
	case auth.AUTH_WEBHOOK:
		panic(fmt.Errorf("Recursive webhook auth is not supported"))
	}
	panic(fmt.Errorf("Unreachable"))
}

// Returns an SQL expression to filter webhooks for the authenticated user.
func FilterWebhooks(ctx context.Context) (sq.Sqlizer, error) {
	user := auth.ForContext(ctx)
	ac, err := NewAuthConfig(ctx)
	if err != nil {
		return nil, err
	}
	if ac.AuthMethod == auth.AUTH_INTERNAL {
		return sq.And{
			sq.Expr(`user_id = ?`, user.UserID),
			sq.Expr(`auth_method = 'INTERNAL'`),
		}, nil
	} else if ac.ClientID != nil {
		// XXX: Should we maybe return all webhooks configured by client ID?
		return sq.And{
			sq.Expr(`NOW() at time zone 'utc' < expires`),
			sq.Expr(`token_hash = ?`, ac.TokenHash),
			sq.Expr(`client_id = ?`, *ac.ClientID),
			sq.Expr(`user_id = ?`, user.UserID),
		}, nil
	} else {
		return sq.And{
			sq.Expr(`NOW() at time zone 'utc' < expires`),
			sq.Expr(`user_id = ?`, user.UserID),
		}, nil
	}
}