import { NotFoundException } from '@nestjs/common'
import { ObjectLiteral, Repository } from 'typeorm'

import { PagerDto } from '~/common/dto/pager.dto'

import { paginate } from '../paginate'
import { Pagination } from '../paginate/pagination'

export class BaseService<E extends ObjectLiteral, R extends Repository<E> = Repository<E>> {
  constructor(private repository: R) {
  }

  async list({
    page,
    pageSize,
  }: PagerDto): Promise<Pagination<E>> {
    return paginate(this.repository, { page, pageSize })
  }

  async findOne(id: number): Promise<E> {
    const item = await this.repository.createQueryBuilder().where({ id }).getOne()
    if (!item)
      throw new NotFoundException('未找到该记录')

    return item
  }

  async create(dto: any): Promise<E> {
    return await this.repository.save(dto)
  }

  async update(id: number, dto: any): Promise<void> {
    await this.repository.update(id, dto)
  }

  async delete(id: number): Promise<void> {
    const item = await this.findOne(id)
    await this.repository.remove(item)
  }
}