123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185
import { Reflect } from "./utils/reflect.ts";import { metadata } from "./constants.ts";import { getDataType } from "./utils/models.ts";
// --------------------------------------------------------------------------------// MODELS// --------------------------------------------------------------------------------
/** * Define a class as a database model. *  * @param tableName a custom table name for this model */export function Model(tableName?: string) {  return (target: Function) => {    Reflect.defineMetadata(      metadata.tableName,      tableName || target.name.toLowerCase(),      target,    );  };}
// --------------------------------------------------------------------------------// COLUMNS// --------------------------------------------------------------------------------
/** * Transform database value to JavaScript types */export enum DataType {  String = "string",  Date = "date",  Number = "number",  Boolean = "boolean",}
/** * Information about table the table column */export interface ColumnOptions {  /** JavaScript type which will be converted from the database */  type: DataType;
  /** The column name on the database */  name: string;
  /** The default value */  default?: any;}
export type ColumnDescription = ColumnOptions & {  propertyKey: string;  isPrimaryKey: boolean;};
/** * Model field *  * @param options field options */export function Column(options?: Partial<ColumnOptions>) {  return (target: Object, propertyKey: string) => {    let columns: ColumnDescription[] = [];    if (Reflect.hasMetadata(metadata.columns, target)) {      columns = Reflect.getMetadata(metadata.columns, target);    }
    const typeMetadata = Reflect.getMetadata(      "design:type",      target,      propertyKey,    );    const type = getDataType(typeMetadata);    if (!type && !options?.type) {      throw new Error(`Column '${propertyKey}' must have a type!`);    }
    const description: ColumnDescription = Object.assign({}, {      propertyKey: propertyKey,      name: propertyKey,      type: type,      isPrimaryKey: false,    }, options);
    columns.push(description);
    Reflect.defineMetadata(metadata.columns, columns, target);  };}
/** Options for model's primary field */export interface PrimaryFieldOptions {  name: string;}
/** * Model primary key field * * @param options primary field options */export function Primary(options?: PrimaryFieldOptions) {  return (target: Object, propertyKey: string) => {    let columns: ColumnDescription[] = [];    if (Reflect.hasMetadata(metadata.columns, target)) {      columns = Reflect.getMetadata(metadata.columns, target);    }
    columns.push({      propertyKey,      name: options?.name || "id",      type: DataType.Number,      isPrimaryKey: true,    });
    Reflect.defineMetadata(metadata.columns, columns, target);  };}
// --------------------------------------------------------------------------------// RELATIONSHIPS// --------------------------------------------------------------------------------
export enum RelationType {  HasMany = 1,  BelongsTo = 2,}
export interface RelationDescription {  propertyKey: string;  type: RelationType;  getModel: () => { new (): any };  targetColumn: string;}
/** * Define a "belongs to" relational property. *  * @param getModel a callback function that returns the model associated with this relation. * @param column the column name that being targetted in this relation. */export function BelongsTo(  getModel: () => { new (): any },  column: string,) {  return (target: Object, propertyKey: string) => {    addRelation(target, RelationType.BelongsTo, propertyKey, getModel, column);  };}
/** * Define a "has many" relational property. *  * @param getModel a callback function that returns the model associated with this relation. * @param column the column name that being targetted in this relation. */export function HasMany(  getModel: () => { new (): any },  column: string,) {  return (target: Object, propertyKey: string) => {    addRelation(target, RelationType.HasMany, propertyKey, getModel, column);  };}
/** * Add relation metadata to a targetted model. */function addRelation(  target: Object,  type: RelationType,  propertyKey: string,  getModel: () => any,  targetColumn: string,) {  let relations: RelationDescription[] = [];  if (Reflect.hasMetadata(metadata.relations, target)) {    relations = Reflect.getMetadata(metadata.relations, target);  }
  relations.push({ propertyKey, type, getModel, targetColumn });
  Reflect.defineMetadata(metadata.relations, relations, target);}