base.entity.ts

import { Exclude } from 'class-transformer';
import { DateTime, Duration } from 'luxon';
import {
  AfterLoad,
  BaseEntity,
  BeforeInsert,
  Column,
  CreateDateColumn,
  PrimaryColumn,
  PrimaryGeneratedColumn,
  UpdateDateColumn,
} from 'typeorm';
import { MetaInfo } from '../../common/decorators';
import { SimpleIdGenerator } from '../../ids';

export type EntityConstructorObject<Entity> = Omit<Entity, keyof typeof BaseEntity | 'reload'>;
export type Constructor<T = {}> = new (...args: any[]) => T;

// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
export const Publishable = <TBase extends Constructor>(Base: TBase) => {
  class ExtendableEntity extends Base {
    @MetaInfo({ name: '是否发布？' })
    @Column({ nullable: true, name: 'is_published' })
    isPublished: boolean;
  }

  return ExtendableEntity;
};

// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
export const NameDescAttachable = <TBase extends Constructor>(Base: TBase) => {
  class ExtendableEntity extends Base {
    @MetaInfo({ name: '名称' })
    @Column({ nullable: false, length: 100, unique: true, name: 'name' })
    name: string;

    @MetaInfo({ name: '描述' })
    @Column('text', { nullable: true, name: 'description' })
    description: string;
  }
  return ExtendableEntity;
};

export class AbstractBaseEntity extends BaseEntity {
  @PrimaryGeneratedColumn() id?: number;

  @CreateDateColumn({ name: 'created_at' })
  createdAt?: Date;

  @UpdateDateColumn({ name: 'updated_at' })
  updatedAt?: Date;

  @MetaInfo({ accessible: 'hidden' })
  @Column({ nullable: true, length: 100, name: 'updated_by' })
  updatedBy?: string;

  @AfterLoad()
  afterLoad(): void {
    this.createdAt = DateTime.fromJSDate(this.createdAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
    this.updatedAt = DateTime.fromJSDate(this.updatedAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
  }
}

/**
 * 生成基于时间的 id，prefix 可以作为一个特殊的前缀用于识别不同的类型
 */
export class AbstractTimeBasedBaseEntity extends BaseEntity {
  @Exclude()
  readonly idPrefix: string;
  @Exclude()
  readonly generator: SimpleIdGenerator;

  @PrimaryColumn({ length: 36 }) id?: string;

  @CreateDateColumn({ name: 'created_at' })
  createdAt?: Date;

  @UpdateDateColumn({ name: 'updated_at' })
  updatedAt?: Date;

  @MetaInfo({ accessible: 'hidden' })
  @Column({ nullable: true, length: 100, name: 'updated_by' })
  updatedBy?: string;

  constructor(idPrefix: string = '') {
    super();
    this.idPrefix = idPrefix;
    this.generator = new SimpleIdGenerator(idPrefix);
  }

  @BeforeInsert()
  beforeInsert(): void {
    this.id = this.generator.nextId();
  }

  @AfterLoad()
  afterLoad(): void {
    this.createdAt = DateTime.fromJSDate(this.createdAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
    this.updatedAt = DateTime.fromJSDate(this.updatedAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
  }
}

export class AbstractTimeBasedNameEntity extends AbstractTimeBasedBaseEntity {
  @MetaInfo({ name: '名称' })
  @Column({ nullable: false, length: 100, unique: true, name: 'name' })
  name: string;

  @MetaInfo({ name: '描述' })
  @Column('text', { nullable: true, name: 'description' })
  description?: string;
}

export class AbstractNameEntity extends AbstractBaseEntity {
  @MetaInfo({ name: '名称' })
  @Column({ nullable: false, length: 100, unique: true, name: 'name' })
  name: string;

  @MetaInfo({ name: '描述' })
  @Column('text', { nullable: true, name: 'description' })
  description?: string;
}

export class AbstractUUIDBaseEntity extends BaseEntity {
  @PrimaryGeneratedColumn('uuid') uuid!: string;

  @CreateDateColumn({ name: 'created_at' })
  createdAt: Date;

  @UpdateDateColumn({ name: 'updated_at' })
  updatedAt: Date;

  @MetaInfo({ accessible: 'hidden' })
  @Column({ nullable: true, length: 100, name: 'updated_by' })
  updatedBy: string;

  @AfterLoad()
  afterLoad(): void {
    this.createdAt = DateTime.fromJSDate(this.createdAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
    this.updatedAt = DateTime.fromJSDate(this.updatedAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
  }
}

export class AbstractUUIDNameEntity extends AbstractUUIDBaseEntity {
  @MetaInfo({ name: '名称' })
  @Column({ nullable: false, length: 100, unique: true, name: 'name' })
  name: string;

  @MetaInfo({ name: '描述' })
  @Column('text', { nullable: true, name: 'description' })
  description: string;
}

export class AbstractUUID2BaseEntity extends BaseEntity {
  @PrimaryGeneratedColumn('uuid') id!: string;

  @CreateDateColumn({ name: 'created_at' })
  createdAt: Date;

  @UpdateDateColumn({ name: 'updated_at' })
  updatedAt: Date;

  @MetaInfo({ accessible: 'hidden' })
  @Column({ nullable: true, length: 100, name: 'updated_by' })
  updatedBy: string;

  @AfterLoad()
  afterLoad(): void {
    this.createdAt = DateTime.fromJSDate(this.createdAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
    this.updatedAt = DateTime.fromJSDate(this.updatedAt)
      .plus(Duration.fromObject({ hours: 8 }))
      .toJSDate();
  }
}

export class AbstractUUID2NameEntity extends AbstractUUID2BaseEntity {
  @MetaInfo({ name: '名称' })
  @Column({ nullable: false, length: 100, unique: true, name: 'name' })
  name: string;

  @MetaInfo({ name: '描述' })
  @Column('text', { nullable: true, name: 'description' })
  description: string;
}

export class AbstractCategoryEntity extends Publishable(AbstractBaseEntity) {
  @MetaInfo({ name: '名称' })
  @Column({ nullable: false, length: 100, unique: true, name: 'name' })
  name: string;

  @MetaInfo({ name: '描述' })
  @Column('text', { nullable: true, name: 'description' })
  description: string;

  @MetaInfo({
    name: '是否系统数据？',
    type: 'Deletable',
    help: '系统数据无法删除',
  })
  @Column({ nullable: true, name: 'is_system' })
  isSystem: boolean;
}