package model

import (
	"time"

	"github.com/caos/zitadel/internal/model"
)

type UserGrantView struct {
	ID            string
	ResourceOwner string
	UserID        string
	ProjectID     string
	GrantID       string
	UserName      string
	FirstName     string
	LastName      string
	DisplayName   string
	Email         string
	ProjectName   string
	OrgName       string
	OrgDomain     string
	RoleKeys      []string

	CreationDate time.Time
	ChangeDate   time.Time
	State        UserGrantState

	Sequence uint64
}

type UserGrantSearchRequest struct {
	Offset        uint64
	Limit         uint64
	SortingColumn UserGrantSearchKey
	Asc           bool
	Queries       []*UserGrantSearchQuery
}

type UserGrantSearchKey int32

const (
	UserGrantSearchKeyUnspecified UserGrantSearchKey = iota
	UserGrantSearchKeyUserID
	UserGrantSearchKeyProjectID
	UserGrantSearchKeyResourceOwner
	UserGrantSearchKeyState
	UserGrantSearchKeyGrantID
	UserGrantSearchKeyOrgName
	UserGrantSearchKeyRoleKey
	UserGrantSearchKeyID
)

type UserGrantSearchQuery struct {
	Key    UserGrantSearchKey
	Method model.SearchMethod
	Value  interface{}
}

type UserGrantSearchResponse struct {
	Offset      uint64
	Limit       uint64
	TotalResult uint64
	Result      []*UserGrantView
	Sequence    uint64
	Timestamp   time.Time
}

func (r *UserGrantSearchRequest) EnsureLimit(limit uint64) {
	if r.Limit == 0 || r.Limit > limit {
		r.Limit = limit
	}
}

func (r *UserGrantSearchRequest) GetSearchQuery(key UserGrantSearchKey) (int, *UserGrantSearchQuery) {
	for i, q := range r.Queries {
		if q.Key == key {
			return i, q
		}
	}
	return -1, nil
}

func (r *UserGrantSearchRequest) AppendMyOrgQuery(orgID string) {
	r.Queries = append(r.Queries, &UserGrantSearchQuery{Key: UserGrantSearchKeyResourceOwner, Method: model.SearchMethodEquals, Value: orgID})
}

func (r *UserGrantSearchRequest) AppendProjectIDQuery(projectID string) {
	r.Queries = append(r.Queries, &UserGrantSearchQuery{Key: UserGrantSearchKeyProjectID, Method: model.SearchMethodEquals, Value: projectID})
}