feat(server): business owned by user data model

server/package.json

@@ -16,7 +16,8 @@
     "teardown": "docker compose down",
     "keygen": "ts-node src/scripts/keygen.ts",
     "hash": "ts-node src/scripts/hash.ts",
-    "verify": "ts-node src/scripts/verify.ts",
+    "verify-hash": "ts-node src/scripts/verify-hash.ts",
+    "create-user": "ts-node src/scripts/create-user.ts",
     "dev": "npm run setup && npm run node; npm run teardown"
   },
   "dependencies": {

server/src/messages/User.ts

@@ -1,4 +1,4 @@
-import User from "../model/user";
+import Account from "../model/user";
 
 export class UserRequest{
     constructor(
@@ -6,6 +6,6 @@ export class UserRequest{
 }
 export class UserResponse{
     constructor(
-        public user: User
+        public user: Account
     ){}
 }

server/src/model/transaction.ts

@@ -1,5 +1,5 @@
 import { Table, Column, Model, CreatedAt, ForeignKey, BelongsTo} from 'sequelize-typescript';
-import User from './user';
+import Account from './user';
 
 @Table
 export default class Transaction extends Model{
@@ -10,18 +10,18 @@ export default class Transaction extends Model{
     declare reference: string;
 
     @Column
-    @ForeignKey(()=> User)
+    @ForeignKey(()=> Account)
     declare senderID: string;
 
-    @BelongsTo(() => User, 'senderID')
-    declare sender: User;
+    @BelongsTo(() => Account, 'senderID')
+    declare sender: Account;
 
     @Column
-    @ForeignKey(()=> User)
+    @ForeignKey(()=> Account)
     declare receiverID: string;
 
-    @BelongsTo(() => User, 'receiverID')
-    declare receiver: User;
+    @BelongsTo(() => Account, 'receiverID')
+    declare receiver: Account;
 
     @CreatedAt
     declare date: Date;

server/src/model/user.ts

@@ -1,4 +1,4 @@
-import { Table, Column, Model, CreatedAt, DataType, Scopes, DefaultScope} from 'sequelize-typescript';
+import { Table, Column, Model, DataType, Scopes, DefaultScope, DeletedAt, BelongsToMany, ForeignKey} from 'sequelize-typescript';
 
 @DefaultScope(() => ({
   attributes:{ exclude: ['password']}
@@ -9,10 +9,13 @@ import { Table, Column, Model, CreatedAt, DataType, Scopes, DefaultScope} from '
   }
 }))
 @Table
-export default class User extends Model{
+export default class Account extends Model{
 
     @Column({primaryKey: true, unique: true, allowNull: false})
-    declare userID: string;
+    declare id: string;
+
+    @Column
+    declare isBusiness: boolean;
 
     @Column
     declare displayName: string;
@@ -22,8 +25,21 @@ export default class User extends Model{
 
     @Column
     declare password: string;
+    
+    @DeletedAt
+    declare deletedAt: Date | null;
+    
+}
+@Table
+export class BusinessOwnership extends Model {
+  @ForeignKey(() => Account)
+  @Column
+  ownerAccountId!: number;
 
-    @CreatedAt
-    declare creationDate: Date;
+  @ForeignKey(() => Account)
+  @Column
+  ownedAccountId!: number;
 
+  @DeletedAt
+  declare deletedAt: Date | null;
 }

server/src/routes/auth.ts

@@ -1,7 +1,7 @@
 import { compare } from 'bcrypt';
 import express from 'express';
 import { logger } from '../util/logging';
-import User from '../model/user';
+import Account from '../model/user';
 import { Scope } from '../util/db';
 import { getJWT, requireAuth } from '../util/auth';
 import { LoginRequest } from '../messages/Login';
@@ -12,7 +12,7 @@ const router = express.Router();
 router.post('/login', async (req, res) => {
   try {
     const data : LoginRequest = req.body;
-    const user = await User.scope(Scope.withPassword).findOne({where: { userID: data.username}});
+    const user = await Account.scope(Scope.withPassword).findOne({where: { id: data.username}});
     if (!user) return res.status(401).json(new Msg('Invalid credentials'));
     const isMatch = await compare(data.password, user.password);
     if (!isMatch) return res.status(401).json(new Msg('Invalid credentials'));

server/src/routes/send.ts

@@ -1,7 +1,7 @@
 import express  from 'express';
 import { logger } from '../util/logging';
 import { requireAuth } from '../util/auth';
-import User from '../model/user';
+import Account from '../model/user';
 import { db } from '../util/db';
 import Transaction from '../model/transaction';
 import { SendRequest, SendResponse} from '../messages/Send';
@@ -12,9 +12,9 @@ const router = express.Router();
 
 router.post('/', requireAuth, async (req, res) => {
   try {
-    const sender = res.locals.user as User;
+    const sender = res.locals.user as Account;
     const data : SendRequest = req.body;
-    const recipient = await User.findOne({where: {userID: data.recipientID}})
+    const recipient = await Account.findOne({where: {id: data.recipientID}})
     if ( Number(data.amount) <= 0) {
       return res.status(400).json(new Err('Invalid transfer amount'));
     }
@@ -31,8 +31,8 @@ router.post('/', requireAuth, async (req, res) => {
       await recipient.increment({balance: data.amount});
       await Transaction.create({
         amount: data.amount,
-        senderID: sender.userID,
-        receiverID: recipient.userID,
+        senderID: sender.id,
+        receiverID: recipient.id,
         reference: data.reference
       });
     })

server/src/scripts/create-user.ts

@@ -0,0 +1,12 @@
+import { db, testConnection } from "../util/db";
+import { hash } from "bcrypt";
+import Account  from '../model/user';
+(async () =>{
+    await testConnection();
+    await Account.create({
+        id: process.argv[2],
+        displayName: process.argv[3],
+        balance: process.argv[4],
+        password: await hash(process.argv[5], 10),
+    })
+})();

server/src/util/auth.ts

@@ -1,5 +1,5 @@
 import { NextFunction, Request, Response } from "express";
-import User from "../model/user"
+import Account from "../model/user"
 import { importJWK, SignJWT, jwtVerify } from "jose";
 
 
@@ -9,9 +9,9 @@ async function setKeyFromEnv() {
     key = await importJWK(JSON.parse(process.env.FM_PRIVATE_KEY));
 }
 
-async function getJWT(user: User){
+async function getJWT(user: Account){
     let jwt = await new SignJWT()
-        .setSubject(user.userID)
+        .setSubject(user.id)
         .setProtectedHeader({ alg: 'HS256' })
         .setIssuedAt()
         .sign(key);
@@ -26,7 +26,7 @@ async function requireAuth(req: Request, res: Response, next: NextFunction) {
       return res.status(401).json({ error: 'Unauthorized: No token provided' });
     }
     const jwt= await jwtVerify(token, key);
-    const user = await User.findOne({where: { userID: jwt.payload.sub}});
+    const user = await Account.findOne({where: { id: jwt.payload.sub}});
     if (!user) {
       return res.status(401).json({ error: 'Unauthorized: User not found' });
     }

server/src/util/db.ts

@@ -1,6 +1,6 @@
 import { Sequelize } from 'sequelize-typescript';
 import { logger } from './logging';
-import User from '../model/user';
+import Account, { BusinessOwnership } from '../model/user';
 import Transaction from '../model/transaction';
 
 enum Scope{
@@ -19,7 +19,7 @@ const db = new Sequelize({
   logging: logger.debug.bind(logger),
 });
 
-db.addModels([User, Transaction ])
+db.addModels([Account, BusinessOwnership, Transaction ])
 
 // Test the connection
 async function testConnection() {