import {
  Entity,
  PrimaryGeneratedColumn,
  Column,
  ManyToOne,
  JoinColumn,
  BeforeInsert,
} from "typeorm"
import { BaseEntity } from "../../common/entities/base.entity"
import { Company } from "../../company/entities/company.entity"
import { Employee } from "../../employees/entities/employee.entity"
import { LeaveType } from "../../leave-types/entities/leave-type.entity"

@Entity("employee_leave_balances")
export class EmployeeLeaveBalance extends BaseEntity {
  @PrimaryGeneratedColumn()
  id: number

  @Column({ type: "int", nullable: false })
  company_id: number

  @Column({ type: "int", nullable: false })
  employee_id: number

  @Column({ type: "int", nullable: false })
  leave_type_id: number

  @Column({ type: "int", nullable: false })
  total_allocated: number

  @Column({
    type: "decimal",
    precision: 5,
    scale: 2,
    nullable: false,
    default: 0,
  })
  used_days: number

  @Column({
    type: "decimal",
    precision: 5,
    scale: 2,
    nullable: false,
    default: 0,
  })
  carried_forward_days: number

  @Column({ type: "decimal", precision: 5, scale: 2, nullable: false })
  available_days: number

  @Column({ type: "int", nullable: false })
  year: number

  @ManyToOne(() => Company)
  @JoinColumn({ name: "company_id" })
  company: Company

  @ManyToOne(() => Employee)
  @JoinColumn({ name: "employee_id" })
  employee: Employee

  @ManyToOne(() => LeaveType)
  @JoinColumn({ name: "leave_type_id" })
  leaveType: LeaveType

  @BeforeInsert()
  calculateAvailableDaysOnInsert() {
    this.available_days =
      this.total_allocated + this.carried_forward_days - this.used_days
  }
}