Feat(Wallet): This allow the creation of wallet during account creation

migrations/20221119164750_wallets.ts

@@ -0,0 +1,15 @@
+import { Knex } from 'knex';
+
+export async function up(knex: Knex): Promise<void> {
+  return knex.schema.createTable('wallets', (table) => {
+    table.increments('id');
+    table.string('address', 255).unique().notNullable();
+    table.double('balance').notNullable().defaultTo(0.0).unsigned();
+    table.timestamps(true, true, true);
+    table.integer('owner').index().references('id').inTable('users').unsigned();
+  });
+}
+
+export async function down(knex: Knex): Promise<void> {
+  return knex.schema.dropTable('wallets');
+}

migrations/20221119164830_transactions.ts

@@ -0,0 +1,17 @@
+import { Knex } from 'knex';
+
+export async function up(knex: Knex): Promise<void> {
+  return knex.schema.createTable('transactions', (table) => {
+    table.increments('id');
+    table.enum('type', ['credit', 'debit']);
+    table.string('purpose', 255).nullable();
+    table.double('amount').notNullable().unsigned();
+    table.timestamps(true, true, true);
+    table.string('from').references('address').inTable('wallets');
+    table.string('to').references('address').inTable('wallets');
+  });
+}
+
+export async function down(knex: Knex): Promise<void> {
+  return knex.schema.dropTable('transactions');
+}

src/app.module.ts

@@ -6,6 +6,7 @@ import { AppService } from './app.service';
 import { UsersModule } from './users/users.module';
 import appConfig from './config/app.config';
 import { AuthModule } from './auth/auth.module';
+import { WalletsModule } from './wallets/wallets.module';
 import jwtConfig from './config/jwt.config';
 import databaseConfig from './config/database.config';
 
@@ -25,7 +26,8 @@ import databaseConfig from './config/database.config';
     }),
 
     UsersModule,
-    AuthModule
+    AuthModule,
+    WalletsModule
   ],
   controllers: [AppController],
   providers: []

src/auth/auth.controller.ts

@@ -1,32 +1,24 @@
 import { Auth } from '../common/decorators/http.decorator';
 import { ErrorResponseDTO } from '../common/dtos/response.dto';
-import {
-  BadRequestException,
-  Body,
-  Controller,
-  HttpCode,
-  HttpStatus,
-  Post,
-  Put
-} from '@nestjs/common';
+import { Body, Controller, HttpCode, HttpStatus, Post } from '@nestjs/common';
 import {
   ApiBadRequestResponse,
   ApiOkResponse,
   ApiOperation,
   ApiTags
 } from '@nestjs/swagger';
-import {
-  CreateUserDto,
-  CreateUserResponseDTO,
-  LoginDTO
-  // LoginResponseDTO
-} from './auth.dto';
+import { CreateUserDto, CreateUserResponseDTO, LoginDTO } from './auth.dto';
 import { AuthService } from './auth.service';
+import { WalletsService } from 'src/wallets/wallets.service';
+import { CreateWalletDto } from 'src/wallets/wallet.dto';
 
 @Controller('auth')
 @ApiTags('auth')
 export class AuthController {
-  constructor(private authService: AuthService) {}
+  constructor(
+    private authService: AuthService,
+    private walletService: WalletsService
+  ) {}
 
   @Post('/login')
   @HttpCode(HttpStatus.OK)
@@ -55,6 +47,17 @@ export class AuthController {
     type: ErrorResponseDTO
   })
   async auth(@Body() body: CreateUserDto) {
-    return await this.authService.auth(body);
+    const { data, access_token } = await this.authService.auth(body);
+    const createWalletDto: CreateWalletDto = {
+      owner: data.id
+    };
+
+    const { balance, address } = await this.walletService.create(
+      createWalletDto
+    );
+
+    return {
+      data: { ...data, wallet: { address, balance }, access_token }
+    };
   }
 }

src/auth/auth.dto.ts

@@ -58,6 +58,9 @@ class _CreateUserResponseDTO extends CreateUserDto {
   @ApiProperty()
   id: string;
 
+  @ApiProperty()
+  wallet: string;
+
   @ApiProperty()
   createdAt: Date;
 

src/auth/auth.module.ts

@@ -7,9 +7,11 @@ import { JwtModule } from '@nestjs/jwt';
 import { SignOptions } from 'jsonwebtoken';
 import { JwtStrategy } from './auth.strategy';
 import { UsersService } from 'src/users/users.service';
+import { WalletsModule } from 'src/wallets/wallets.module';
 
 @Module({
   imports: [
+    WalletsModule,
     UsersModule,
     JwtModule.registerAsync({
       inject: [ConfigService],

src/users/users.service.ts

@@ -46,8 +46,6 @@ export class UsersService {
 
     const user = await this.findOne({ email: createUserDto.email });
 
-    console.log(user);
-
     return { user };
   }
 }

src/wallets/wallet.dto.ts

@@ -0,0 +1,44 @@
+import {
+  IsNotEmpty,
+  IsNumber,
+  IsOptional,
+  IsPositive,
+  IsString
+} from 'class-validator';
+
+export class CreateWalletDto {
+  @IsString()
+  @IsNotEmpty()
+  address?: string;
+
+  @IsNumber()
+  @IsPositive()
+  @IsNotEmpty()
+  balance? = 0;
+
+  @IsNumber()
+  @IsPositive()
+  @IsNotEmpty()
+  owner: number;
+}
+
+export class UpdateWalletDto {
+  @IsNumber()
+  @IsPositive()
+  @IsOptional()
+  balance?: number;
+}
+
+export class FindWalletDto {
+  @IsString()
+  @IsOptional()
+  address?: string;
+
+  @IsNumber()
+  @IsOptional()
+  id?: string;
+
+  @IsNumber()
+  @IsOptional()
+  owner?: number;
+}

src/wallets/wallets.module.ts

@@ -0,0 +1,8 @@
+import { Module } from '@nestjs/common';
+import { WalletsService } from './wallets.service';
+
+@Module({
+  providers: [WalletsService],
+  exports: [WalletsService]
+})
+export class WalletsModule {}

src/wallets/wallets.service.ts

@@ -0,0 +1,41 @@
+import { Injectable } from '@nestjs/common';
+import { Knex } from 'knex';
+import { InjectConnection } from 'nest-knexjs';
+import { CreateWalletDto, FindWalletDto, UpdateWalletDto } from './wallet.dto';
+import { randomUUID } from 'crypto';
+
+@Injectable()
+export class WalletsService {
+  constructor(@InjectConnection() private readonly knex: Knex) {}
+
+  async create(createWalletDto: CreateWalletDto) {
+    const newAddress = randomUUID().replaceAll('-', '');
+    createWalletDto.address = newAddress;
+
+    const result = await this.knex.table('wallets').insert(createWalletDto);
+
+    const wallet = await this.findOne({ address: newAddress });
+
+    return wallet;
+  }
+
+  async update(filter: FindWalletDto, update: UpdateWalletDto) {
+    const count = await this.knex('wallets')
+      .where({ FindWalletDto })
+      .update(update);
+
+    const updates = await this.findOne(filter);
+
+    return updates[0];
+  }
+
+  async findOne(filter: FindWalletDto) {
+    const wallet = await this.knex.table('wallets').where(filter);
+
+    if (!wallet.length) {
+      return undefined;
+    }
+
+    return wallet[0];
+  }
+}

tsconfig.json

@@ -6,7 +6,7 @@
     "emitDecoratorMetadata": true,
     "experimentalDecorators": true,
     "allowSyntheticDefaultImports": true,
-    "target": "es2017",
+    "target": "ES2021",
     "sourceMap": true,
     "outDir": "./dist",
     "baseUrl": "./",
@@ -16,6 +16,9 @@
     "noImplicitAny": false,
     "strictBindCallApply": false,
     "forceConsistentCasingInFileNames": false,
-    "noFallthroughCasesInSwitch": false
+    "noFallthroughCasesInSwitch": false,
+    "lib": [
+      "ES2021.String"
+    ]
   }
 }