model.ts

import { faker } from '@faker-js/faker'
import { Builder } from './builder/builder'
import { RelationType } from './contracts'
import type {
  DefineFactoryCallback,
  DefineStateCallback,
  RelationshipMeta,
  RelationshipMetaOptions,
} from './contracts'

export class FactoryModel<Model extends Record<string, any>, States extends string | null = null> {
  /**
   * Store the factory callback
   */
  public callback: DefineFactoryCallback<Model>

  /**
   * Store each state callbacks
   */
  public states: Record<string, DefineStateCallback<Model>> = {}

  /**
   * Store relations metadata
   */
  public relations: Record<string, RelationshipMeta> = {}

  constructor(callback: DefineFactoryCallback<Model>) {
    this.callback = callback
  }

  private addRelation(name: string, type: RelationType, meta: RelationshipMetaOptions) {
    const { tableName } = this.callback({ faker })

    this.relations[name] = {
      foreignKey: `${tableName}_id`,
      localKey: 'id',
      ...meta,
      type,
    }

    return this
  }

  /**
   * Allows you to define a new state for the factory.
   */
  public state<S extends string>(
    name: S,
    stateCb: DefineStateCallback<Model>
  ): FactoryModel<Model, Extract<States | S, string>> {
    this.states[name] = stateCb
    return this
  }

  /**
   * Add hasOne relationship
   */
  public hasOne(name: string, meta: RelationshipMetaOptions) {
    return this.addRelation(name, RelationType.HasOne, meta)
  }

  /**
   * Add hasMany relationship
   */
  public hasMany(name: string, meta: RelationshipMetaOptions) {
    return this.addRelation(name, RelationType.HasMany, meta)
  }

  /**
   * Add belongsTo relationship
   */
  public belongsTo(name: string, meta: RelationshipMetaOptions) {
    return this.addRelation(name, RelationType.BelongsTo, meta)
  }

  /**
   * Returns the Builder
   */
  public build(): Builder<typeof this> {
    return new Builder(this)
  }
}