update

Author: devarogundade

Diff for: backend/.dockerignore

@@ -0,0 +1,5 @@
+Dockerfile
+.dockerignore
+node_modules
+npm-debug.log
+dist

Diff for: backend/Dockerfile

@@ -0,0 +1,21 @@
+# Use the official Node.js image as a base
+FROM node:18
+
+# Create and change to the app directory
+WORKDIR /app
+
+# Install app dependencies
+COPY package*.json ./
+RUN npm install
+
+# Copy app source code
+COPY . .
+
+# Build the app
+RUN npm run build
+
+# Expose the port the app runs on
+EXPOSE 80
+
+# Start the app
+CMD ["npm", "run", "start:prod"]

Diff for: backend/docker-compose.yml

@@ -0,0 +1,30 @@
+services:
+  thube-backend:
+    depends_on:
+      - redis
+    image: thube.azurecr.io/thube-backend:latest
+    container_name: thube-backend
+    restart: always
+    ports:
+      - "80:80"
+    env_file: .env
+    networks:
+      - thube-net
+  
+  redis:
+    image: "redis:alpine"
+    container_name: redis
+    restart: always
+    ports:
+      - "6379:6379"
+    volumes:
+      - "redis-data:/data"
+    networks:
+      - thube-net
+
+volumes:
+  redis-data:
+
+networks:
+  thube-net:
+    driver: bridge

Diff for: backend/src/app.controller.ts

@@ -5,6 +5,7 @@ import { AppService } from './app.service';
 import { Account } from './database/schemas/account';
 import { Stream } from './database/schemas/stream';
 import { Paged } from './types';
+import { Video } from './database/schemas/video';
 
 @Controller()
 export class AppController {
@@ -72,6 +73,32 @@ export class AppController {
     );
   }
 
+  @Post('/upload-video')
+  uploadVideo(
+    @Body() dto: Video
+  ): Promise<Video | null> {
+    return this.appService.uploadVideo(
+      dto.videoId,
+      dto.creator as string,
+      dto.name,
+      dto.thumbnail,
+      dto.collection,
+      dto.playback_uri,
+      dto.tips
+    );
+  }
+
+  @Post('/watch-video/:address')
+  watchVideo(
+    @Param('address') address: string,
+    @Body() videoId: string
+  ): Promise<boolean> {
+    return this.appService.watchVideo(
+      address,
+      videoId,
+    );
+  }
+
   @Get('/streams')
   getStreams(
     @Query('page') page: number,
@@ -83,7 +110,7 @@ export class AppController {
     );
   }
 
-  @Get('/stream/:id')
+  @Get('/streams/:id')
   getStream(
     @Param('id') streamId: string,
   ): Promise<Stream | null> {
@@ -92,6 +119,26 @@ export class AppController {
     );
   }
 
+  @Get('/videos')
+  getVideos(
+    @Query('page') page: number,
+    @Query('creator') creator: string | null
+  ): Promise<Paged<Video[]> | null> {
+    return this.appService.getVideos(
+      page,
+      creator,
+    );
+  }
+
+  @Get('/videos/:id')
+  getVideo(
+    @Param('id') videoId: string,
+  ): Promise<Video | null> {
+    return this.appService.getVideo(
+      videoId
+    );
+  }
+
   @Get('/accounts/:id')
   getAccount(
     @Param('address') address: string,

Diff for: backend/src/app.module.ts

@@ -7,11 +7,13 @@ import { BullModule } from '@nestjs/bullmq';
 import { MongooseModule } from '@nestjs/mongoose';
 import { ConfigModule } from '@nestjs/config';
 import { Stream } from 'stream';
+import { VideoSchema } from './database/schemas/video';
 import { StreamSchema } from './database/schemas/stream';
 import { Account, AccountSchema } from './database/schemas/account';
 import { MailWorker } from './worker/mail';
 // import { ScheduleModule } from '@nestjs/schedule';
 import { MainGateway } from './socket/main.gateway';
+import { Video } from './database/schemas/video';
 
 @Module({
   imports: [
@@ -28,6 +30,7 @@ import { MainGateway } from './socket/main.gateway';
     MongooseModule.forFeature([
       { name: Stream.name, schema: StreamSchema },
       { name: Account.name, schema: AccountSchema },
+      { name: Video.name, schema: VideoSchema },
     ])
   ],
   controllers: [AppController],

Diff for: backend/src/app.service.ts

@@ -6,6 +6,7 @@ import { Injectable } from '@nestjs/common';
 import { Account } from './database/schemas/account';
 import { Stream } from './database/schemas/stream';
 import { Paged } from './types';
+import { Video } from './database/schemas/video';
 
 const TAKE_SIZE: number = 25;
 
@@ -14,6 +15,7 @@ export class AppService {
   constructor(
     @InjectModel(Stream.name) private streamModel: Model<Stream>,
     @InjectModel(Account.name) private accountModel: Model<Account>,
+    @InjectModel(Video.name) private videoModel: Model<Video>,
   ) {
   }
 
@@ -140,6 +142,59 @@ export class AppService {
     }
   }
 
+  async uploadVideo(
+    videoId: string,
+    address: string,
+    name: string,
+    thumbnail: string,
+    collection: string,
+    playback_uri: string | null,
+    tips: boolean,
+  ): Promise<Video | null> {
+    try {
+      const video: Video = {
+        videoId,
+        name,
+        thumbnail,
+        creator: address,
+        playback_uri,
+        tips,
+        collection,
+        created_at: new Date(),
+        viewers: [],
+        views: 0
+      };
+
+      return await this.videoModel.create(video);
+    } catch (error) {
+      console.error(error);
+      return null;
+    }
+  }
+
+  async watchVideo(
+    videoId: string,
+    streamer: string
+  ): Promise<boolean> {
+    try {
+      await this.videoModel.updateOne({
+        videoId
+      }, {
+        $addToSet: {
+          viewers: streamer
+        },
+        $inc: {
+          views: 1
+        }
+      });
+
+      return true;
+    } catch (error) {
+      console.error(error);
+      return false;
+    }
+  }
+
   async getStreams(
     page: number,
     creator?: string
@@ -178,6 +233,44 @@ export class AppService {
     }
   }
 
+  async getVideos(
+    page: number,
+    creator?: string
+  ): Promise<Paged<Video[]> | null> {
+    try {
+      const filter = (creator != 'undefined') ? { creator } : {};
+
+      const total = await this.videoModel.countDocuments(filter);
+
+      const data = await this.videoModel.find(filter)
+        .limit(TAKE_SIZE * 1)
+        .skip((page - 1) * TAKE_SIZE)
+        .sort({ start_at: 'desc' })
+        .populate(['creator'])
+        .exec();
+
+      const lastPage = Math.ceil(total / TAKE_SIZE);
+
+      return { total, lastPage, data };
+    } catch (error) {
+      console.error(error);
+      return null;
+    }
+  }
+
+  async getVideo(
+    videoId: string
+  ): Promise<Video | null> {
+    try {
+      return this.videoModel.findOne({ videoId })
+        .populate(['creator'])
+        .exec();
+    } catch (error) {
+      console.error(error);
+      return null;
+    }
+  }
+
   async getAccount(
     address: string
   ): Promise<Account | null> {

Diff for: backend/src/database/schemas/video.ts

@@ -0,0 +1,42 @@
+/* eslint-disable prettier/prettier */
+
+import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
+import { HydratedDocument, Types } from 'mongoose';
+import { Account } from './account';
+
+export type VideoDocument = HydratedDocument<Video>;
+
+@Schema()
+export class Video {
+    @Prop({ required: true, _id: true, unique: true })
+    videoId: string;
+
+    @Prop({ required: true })
+    name: string;
+
+    @Prop({ required: true })
+    thumbnail: string;
+
+    @Prop({ required: true, type: Types.ObjectId, ref: Account.name })
+    creator: Account | string;
+
+    @Prop({ default: null })
+    playback_uri: string | null;
+
+    @Prop({ required: true })
+    tips: boolean;
+
+    @Prop({ required: true, type: [Types.ObjectId], ref: Account.name })
+    viewers: Account[];
+
+    @Prop({ default: 0 })
+    views: number;
+
+    @Prop({ default: null })
+    collection: string | null;
+
+    @Prop({ required: true })
+    created_at: Date;
+}
+
+export const VideoSchema = SchemaFactory.createForClass(Video);

Diff for: backend/src/main.ts

@@ -5,6 +5,8 @@ import { AppModule } from './app.module';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);
-  await app.listen(3000);
+  app.enableCors();
+  await app.listen(80);
 }
+
 bootstrap();