column.ts

import { ModelAttributeColumnOptions, DataType } from 'sequelize';

import { addAttribute } from './attribute-service';
import { isDataType } from '../../sequelize/data-type/data-type-service';
import { getSequelizeTypeByDesignType } from '../shared/model-service';

export function Column(dataType: DataType): Function;
export function Column(options: Partial<ModelAttributeColumnOptions>): Function;
export function Column(
  target: any,
  propertyName: string,
  propertyDescriptor?: PropertyDescriptor
): void;
export function Column(...args: any[]): Function | void {
  // In case of no specified options, we infer the
  // sequelize data type by the type of the property
  if (args.length >= 2) {
    const target = args[0];
    const propertyName = args[1];
    const propertyDescriptor = args[2];

    annotate(target, propertyName, propertyDescriptor);
    return;
  }

  return (target: any, propertyName: string, propertyDescriptor?: PropertyDescriptor) => {
    annotate(
      target,
      propertyName,
      propertyDescriptor ?? Object.getOwnPropertyDescriptor(target, propertyName),
      args[0]
    );
  };
}

function annotate(
  target: any,
  propertyName: string,
  propertyDescriptor?: PropertyDescriptor,
  optionsOrDataType: Partial<ModelAttributeColumnOptions> | DataType = {}
): void {
  let options: Partial<ModelAttributeColumnOptions>;

  if (isDataType(optionsOrDataType)) {
    options = {
      type: optionsOrDataType,
    };
  } else {
    options = { ...(optionsOrDataType as ModelAttributeColumnOptions) };

    if (!options.type) {
      options.type = getSequelizeTypeByDesignType(target, propertyName);
    }
  }

  if (propertyDescriptor) {
    if (propertyDescriptor.get) {
      options.get = propertyDescriptor.get;
    }
    if (propertyDescriptor.set) {
      options.set = propertyDescriptor.set;
    }
  }

  addAttribute(target, propertyName, options);
}