package template_repo

import (
	"github.com/XingfenD/yoresee_doc/internal/model"
	"gorm.io/gorm"
)

type ListTemplateOperation struct {
	repo            *TemplateRepository
	model           *model.Template
	userID          *int64
	scope           *string
	knowledgeBaseID *int64
	nameKeyword     *string
	documentType    *model.DocumentType
	sortField       string
	sortDesc        bool
	page            int
	pageSize        int
	tx              *gorm.DB
}

func (r *TemplateRepository) List(m *model.Template) (op *ListTemplateOperation) {
	return &ListTemplateOperation{
		repo:  r,
		model: m,
	}
}

func (op *ListTemplateOperation) WithTx(tx *gorm.DB) *ListTemplateOperation {
	op.tx = tx
	return op
}

func (op *ListTemplateOperation) WithUserID(userID *int64) *ListTemplateOperation {
	op.userID = userID
	return op
}

func (op *ListTemplateOperation) WithScope(scope *string) *ListTemplateOperation {
	op.scope = scope
	return op
}

func (op *ListTemplateOperation) WithKnowledgeBaseID(knowledgeBaseID *int64) *ListTemplateOperation {
	op.knowledgeBaseID = knowledgeBaseID
	return op
}

func (op *ListTemplateOperation) WithNameKeyword(nameKeyword *string) *ListTemplateOperation {
	op.nameKeyword = nameKeyword
	return op
}

func (op *ListTemplateOperation) WithDocumentType(documentType *model.DocumentType) *ListTemplateOperation {
	op.documentType = documentType
	return op
}

func (op *ListTemplateOperation) WithSort(field string, desc bool) *ListTemplateOperation {
	op.sortField = field
	op.sortDesc = desc
	return op
}

func (op *ListTemplateOperation) WithPagination(page, pageSize int) *ListTemplateOperation {
	op.page = page
	op.pageSize = pageSize
	return op
}

func (op *ListTemplateOperation) ExecWithTotal() (templates []*model.Template, total int64, err error) {
	if op.tx == nil {
		op.tx = op.repo.db
	}

	dbQuery := op.tx.Model(op.model)
	if op.model != nil {
		dbQuery = dbQuery.Where(op.model)
	}

	if op.userID != nil {
		dbQuery = dbQuery.Where("user_id = ?", *op.userID)
	}
	if op.scope != nil {
		dbQuery = dbQuery.Where("scope = ?", *op.scope)
	}
	if op.knowledgeBaseID != nil {
		dbQuery = dbQuery.Where("knowledge_base_id = ?", *op.knowledgeBaseID)
	}
	if op.nameKeyword != nil && *op.nameKeyword != "" {
		dbQuery = dbQuery.Where("name ILIKE ?", "%"+*op.nameKeyword+"%")
	}
	if op.documentType != nil && *op.documentType != "" {
		dbQuery = dbQuery.Where("document_type = ?", *op.documentType)
	}

	if err = dbQuery.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	orderStr := "created_at DESC"
	if op.sortField != "" {
		if op.sortDesc {
			orderStr = op.sortField + " DESC"
		} else {
			orderStr = op.sortField + " ASC"
		}
	}
	dbQuery = dbQuery.Order(orderStr)

	if op.page > 0 && op.pageSize > 0 {
		offset := (op.page - 1) * op.pageSize
		dbQuery = dbQuery.Offset(offset).Limit(op.pageSize)
	}

	err = dbQuery.Find(&templates).Error
	return
}