import { Logger } from "@nestjs/common";
import { DataSource } from "typeorm";

import { Role } from "../../src/modules/users/entities/role.entity";
import { RoleEnum } from "../../src/modules/users/enums/role.enum";

const roles = [{ name: RoleEnum.SUPER_ADMIN }, { name: RoleEnum.ADMIN }, { name: RoleEnum.VISITOR }, { name: RoleEnum.USER }];

export const seedRole = async (dataSource: DataSource, logger: Logger) => {
  try {
    const roleRepo = dataSource.getRepository(Role);
    for (const role of roles) {
      const newRole = roleRepo.create(role);
      await roleRepo.save(newRole);
    }
    logger.log("Role seeded successfully");
  } catch (error) {
    logger.error("Error in seeding role", error);
    process.exit(1);
  }
};