1 files changed, 61 insertions, 0 deletions
diff --git a/util/src/entities/BaseClass.ts b/util/src/entities/BaseClass.ts
new file mode 100644
index 00000000..38f6a71b
--- /dev/null
+++ b/util/src/entities/BaseClass.ts
@@ -0,0 +1,61 @@
+import "reflect-metadata";
+import { BaseEntity, BeforeInsert, BeforeUpdate, PrimaryColumn } from "typeorm";
+import { Snowflake } from "../util/Snowflake";
+import Ajv, { ValidateFunction } from "ajv";
+import schema from "./schema.json";
+
+const ajv = new Ajv({
+ removeAdditional: "all",
+ useDefaults: true,
+ coerceTypes: true,
+ validateFormats: false,
+ allowUnionTypes: true,
+});
+// const validator = ajv.compile<BaseClass>(schema);
+
+export class BaseClass extends BaseEntity {
+ @PrimaryColumn()
+ id: string;
+
+ // @ts-ignore
+ constructor(props?: any, public opts: { id?: string } = {}) {
+ super();
+ this.assign(props);
+
+ if (!this.construct.schema) this.construct.schema = { ...schema, $ref: `#/definitions/${this.construct.name}` };
+ }
+
+ get construct(): any {
+ return this.constructor;
+ }
+
+ assign(props: any) {
+ if (!props || typeof props !== "object") return;
+
+ delete props.opts;
+
+ for (const key in props) {
+ if (this.hasOwnProperty(key)) continue;
+
+ Object.defineProperty(this, key, { value: props[key] });
+ }
+
+ this.id = this.opts.id || Snowflake.generate();
+ }
+
+ @BeforeUpdate()
+ @BeforeInsert()
+ validate() {
+ const valid = ajv.validate(this.construct.schema, this.toJSON());
+ if (!valid) throw ajv.errors;
+ }
+
+ get metadata() {
+ return this.construct.getRepository().metadata;
+ }
+
+ toJSON(): any {
+ // @ts-ignore
+ return Object.fromEntries(this.metadata.columns.map((x) => [x.propertyName, this[x.propertyName]]));
+ }
+}
|
