import type { TPaginationQuery } from './pagination'

import { z } from 'zod'
import { isEqual } from '../utils'
import { Pagination, paginationQuerySchema } from './pagination'
import { bFieldsSchema, cFieldsSchema, fieldSchema } from '../forms'

export const querySchema = cFieldsSchema
	.pick({ q: true })
	.extend(paginationQuerySchema.shape)
	.extend({
		sort: fieldSchema(bFieldsSchema.string.min(1).max(64), {
			label: 'Сортировка'
		})
	})
	.partial()
	.describe('Параметры базового запроса')

export type TQuery = z.infer<typeof querySchema>

/**
 * Класс для работы с запросами (для удобства).
 */
export type TFindFilter<T extends TQuery> = {
	obj: Omit<T, 'page' | 'limit' | 'sort'>
	pagination: TPaginationQuery
	sort?: string
}

/**
 * Класс для работы с запросами (для удобства).
 */
export class FindFilter<T extends TQuery> implements TFindFilter<T> {
	obj: Omit<T, 'page' | 'limit' | 'sort'>
	pagination: TPaginationQuery
	sort?: string

	constructor(query?: T) {
		// Copy fiends.
		let queryCopy: T = Object.assign({}, query)

		// Pagination.
		this.pagination = new Pagination(queryCopy).getQuery()
		// Delete pagination fields.
		if (this.pagination as TPaginationQuery) {
			const paginationKeys = Object.keys(this.pagination) as [keyof T]
			for (const key of paginationKeys) {
				if (queryCopy[key]) delete queryCopy[key]
			}
		}

		// Sort.
		if (queryCopy.sort) {
			this.sort = queryCopy.sort
			delete queryCopy.sort
		}

		// Obj.
		this.obj = queryCopy
	}

	setPagination(pagination?: TPaginationQuery) {
		this.pagination = new Pagination(pagination).getQuery()
	}

	toObject(): TFindFilter<T> {
		return {
			obj: this.obj,
			pagination: this.pagination,
			sort: this.sort
		}
	}

	isEqual(filters: TFindFilter<T>[]) {
		return isEqual([this, ...filters])
	}
}