import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { ContentManagement } from './entities/content_management.entity';

export class ContentManagementRepository {
  constructor(
    @InjectRepository(ContentManagement)
    private readonly contentRepository: Repository<ContentManagement>,
  ) {}

  async findOne(type: string) {
    const data = await this.contentRepository.findOne({
      where: {
        type,
      },
    });

    if (data) {
      return data;
    }
  }

  async findOneById(id: string) {
    const data = await this.contentRepository.findOne({
      where: {
        id,
      },
    });

    if (data) {
      return data;
    }
  }
}