✨ Entity mapping

Author: AnandChowdhary

src/entities/backup-code.ts

@@ -3,16 +3,20 @@ import {
   PrimaryGeneratedColumn,
   CreateDateColumn,
   UpdateDateColumn,
-  Column
+  Column,
+  ManyToOne,
+  JoinColumn
 } from "typeorm";
+import { User } from "./user";
 
 @Entity()
 export class BackupCode {
   @PrimaryGeneratedColumn("uuid")
   code!: number;
 
-  @Column()
-  userId!: number;
+  @ManyToOne(type => User)
+  @JoinColumn()
+  user!: User;
 
   @Column()
   used!: boolean;

src/entities/email.ts

@@ -3,16 +3,20 @@ import {
   PrimaryGeneratedColumn,
   CreateDateColumn,
   UpdateDateColumn,
-  Column
+  Column,
+  ManyToOne,
+  JoinColumn
 } from "typeorm";
+import { User } from "./user";
 
 @Entity()
 export class Email {
   @PrimaryGeneratedColumn()
   id!: number;
 
-  @Column()
-  userId!: number;
+  @ManyToOne(type => User)
+  @JoinColumn()
+  user!: User;
 
   @Column()
   email!: string;

src/entities/event.ts

@@ -2,17 +2,21 @@ import {
   Entity,
   PrimaryGeneratedColumn,
   CreateDateColumn,
-  Column
+  Column,
+  ManyToOne,
+  JoinColumn
 } from "typeorm";
 import { EventType } from "../interfaces/enum";
+import { User } from "./user";
 
 @Entity()
 export class Event {
   @PrimaryGeneratedColumn()
   id!: number;
 
-  @Column()
-  userId!: number;
+  @ManyToOne(type => User)
+  @JoinColumn()
+  user: User;
 
   @Column({
     type: "enum",

src/entities/user.ts

@@ -3,50 +3,54 @@ import {
   PrimaryGeneratedColumn,
   CreateDateColumn,
   UpdateDateColumn,
-  Column
+  Column,
+  OneToOne,
+  JoinColumn
 } from "typeorm";
+import { Email } from "./email";
 
 @Entity()
 export class User {
   @PrimaryGeneratedColumn()
-  id!: number;
+  id?: number;
 
   @Column()
-  name!: string;
+  name?: string;
 
   @Column()
-  nickname!: string;
+  nickname?: string;
 
-  @Column()
-  primaryEmailId!: number;
+  @OneToOne(type => Email)
+  @JoinColumn()
+  primaryEmail?: Email;
 
   @Column()
-  password!: string;
+  password?: string;
 
   @Column()
-  twoFactorEnabled!: boolean;
+  twoFactorEnabled?: boolean;
 
   @Column()
-  twoFactorSecret!: string;
+  twoFactorSecret?: string;
 
   @Column()
-  country!: string;
+  country?: string;
 
   @Column()
-  timezone!: string;
+  timezone?: string;
 
   @Column()
-  notificationEmails!: 1 | 2 | 3 | 4;
+  notificationEmails?: 1 | 2 | 3 | 4;
 
   @Column()
-  preferredLanguage!: string;
+  preferredLanguage?: string;
 
   @Column()
-  prefersReducedMotion!: boolean;
+  prefersReducedMotion?: boolean;
 
   @CreateDateColumn()
-  createdAt!: Date;
+  createdAt?: Date;
 
   @UpdateDateColumn()
-  updatedAt!: Date;
+  updatedAt?: Date;
 }

src/helpers/mapStringToEntity.ts

@@ -0,0 +1,17 @@
+import { User } from "../entities/user";
+import { Email } from "../entities/email";
+import { Organization } from "../entities/organization";
+import { Membership } from "../entities/membership";
+import { BackupCode } from "../entities/backup-code";
+import { HTTPError } from "../interfaces/error";
+
+export const mapStringToEntity = (string: string) => {
+  if (string === "user") return User;
+  if (string === "email") return Email;
+  if (string === "organization") return Organization;
+  if (string === "membership") return Membership;
+  if (string === "backupCode") return BackupCode;
+  throw new Error(
+    JSON.stringify({ status: 500, code: "invalid_entity" } as HTTPError)
+  );
+};

src/interfaces/error.ts

@@ -0,0 +1,5 @@
+export interface HTTPError {
+  status: number;
+  code: string;
+  message?: string;
+}

src/queries/crud.ts

@@ -0,0 +1,63 @@
+import { connect } from "../helpers/database";
+import { User } from "../entities/user";
+import { Email } from "../entities/email";
+import { Organization } from "../entities/organization";
+import { Membership } from "../entities/membership";
+import { BackupCode } from "../entities/backup-code";
+import { mapStringToEntity } from "../helpers/mapStringToEntity";
+
+export const create = async (
+  entity: string,
+  data: User | Email | Organization | Membership | BackupCode
+) => {
+  const connection = await connect();
+  const into = mapStringToEntity(entity);
+  await connection
+    .createQueryBuilder()
+    .insert()
+    .into(into)
+    .values(data)
+    .execute();
+  connection.close();
+};
+
+export const read = async (entity: string, id: number) => {
+  const connection = await connect();
+  const into = mapStringToEntity(entity);
+  const record = await connection
+    .getRepository(into)
+    .createQueryBuilder()
+    .where(`${entity}.id = :id`, { id })
+    .getOne();
+  connection.close();
+  return record;
+};
+
+export const update = async (
+  entity: string,
+  id: number,
+  data: User | Email | Organization | Membership | BackupCode
+) => {
+  const connection = await connect();
+  const into = mapStringToEntity(entity);
+  const updatedRecord = await connection
+    .createQueryBuilder()
+    .update(into)
+    .set(data)
+    .where("id = :id", { id })
+    .execute();
+  connection.close();
+  return updatedRecord;
+};
+
+export const remove = async (entity: string, id: number) => {
+  const connection = await connect();
+  const into = mapStringToEntity(entity);
+  await connection
+    .createQueryBuilder()
+    .delete()
+    .from(into)
+    .where("id = :id", { id })
+    .execute();
+  connection.close();
+};

src/queries/user.ts

@@ -0,0 +1,8 @@
+import { create } from "../queries/crud";
+
+export const register = async () => {
+  await create("user", {
+    name: "Anand Chowdhary",
+    nickname: "Anand"
+  });
+};