package model

import (
	"github.com/zitadel/zitadel/internal/domain"
	usr_model "github.com/zitadel/zitadel/internal/user/model"
	"github.com/zitadel/zitadel/internal/view/repository"
)

type UserSessionSearchRequest usr_model.UserSessionSearchRequest
type UserSessionSearchQuery usr_model.UserSessionSearchQuery
type UserSessionSearchKey usr_model.UserSessionSearchKey

func (req UserSessionSearchRequest) GetLimit() uint64 {
	return req.Limit
}

func (req UserSessionSearchRequest) GetOffset() uint64 {
	return req.Offset
}

func (req UserSessionSearchRequest) GetSortingColumn() repository.ColumnKey {
	if req.SortingColumn == usr_model.UserSessionSearchKeyUnspecified {
		return nil
	}
	return UserSessionSearchKey(req.SortingColumn)
}

func (req UserSessionSearchRequest) GetAsc() bool {
	return req.Asc
}

func (req UserSessionSearchRequest) GetQueries() []repository.SearchQuery {
	result := make([]repository.SearchQuery, len(req.Queries))
	for i, q := range req.Queries {
		result[i] = UserSessionSearchQuery{Key: q.Key, Value: q.Value, Method: q.Method}
	}
	return result
}

func (req UserSessionSearchQuery) GetKey() repository.ColumnKey {
	return UserSessionSearchKey(req.Key)
}

func (req UserSessionSearchQuery) GetMethod() domain.SearchMethod {
	return req.Method
}

func (req UserSessionSearchQuery) GetValue() interface{} {
	return req.Value
}

func (key UserSessionSearchKey) ToColumnName() string {
	switch usr_model.UserSessionSearchKey(key) {
	case usr_model.UserSessionSearchKeyUserAgentID:
		return UserSessionKeyUserAgentID
	case usr_model.UserSessionSearchKeyUserID:
		return UserSessionKeyUserID
	case usr_model.UserSessionSearchKeyState:
		return UserSessionKeyState
	case usr_model.UserSessionSearchKeyResourceOwner:
		return UserSessionKeyResourceOwner
	case usr_model.UserSessionSearchKeyInstanceID:
		return UserSessionKeyInstanceID
	case usr_model.UserSessionSearchKeyOwnerRemoved:
		return UserSessionKeyOwnerRemoved
	default:
		return ""
	}
}