Got the basics working with angular

2026-03-05 13:20:27 -05:00 · 2021-07-31 08:56:49 -04:00
parent e5ef4d8847
commit edbfd8bb9e
18 changed files with 3131 additions and 27 deletions
--- a/apps/api/src/web/web.controller.ts
+++ b/apps/api/src/web/web.controller.ts
@@ -0,0 +1,123 @@
+import { InjectQueue } from '@nestjs/bull';
+import {
+  Body,
+  Controller,
+  Get,
+  HttpCode,
+  Logger,
+  Post,
+  Render,
+  Req,
+  UsePipes,
+  ValidationPipe,
+} from '@nestjs/common';
+import { EventEmitter2 } from '@nestjs/event-emitter';
+import { Queue } from 'bull';
+import { pick, trimEnd } from 'lodash';
+
+import { QueueDto, UploadDto } from '@vidgrab2/api-interfaces';
+import { YtdlService } from '../ytdl/ytdl.service';
+
+@Controller()
+export class WebController {
+  private readonly logger = new Logger(WebController.name);
+
+  constructor(
+    @InjectQueue('vidgrab') private readonly vidgrabQueue: Queue,
+    private readonly ytdlService: YtdlService,
+    private readonly eventEmitter: EventEmitter2,
+  ) {}
+
+  @Get()
+  @Render('Index')
+  async root(@Req() request: any) {
+    const fullUrl =
+      request.protocol + '://' + request.get('host') + request.originalUrl;
+    const bookmarklet = `javascript:(function(){var xhr=new XMLHttpRequest();xhr.open('POST',encodeURI('${trimEnd(
+      fullUrl,
+      '/',
+    )}/queue'));xhr.setRequestHeader('Content-Type','application/x-www-form-urlencoded');xhr.send('url='+document.location.href.replace(/ /g,'+'));}());`;
+
+    const jobs = await this.vidgrabQueue.getJobs([
+      'completed',
+      'waiting',
+      'active',
+      'delayed',
+      'failed',
+      'paused',
+    ]);
+
+    const jobList = await Promise.all(
+      jobs.map(async (j) => {
+        const state = await j.getState();
+        const progress = j.progress();
+
+        return {
+          ...pick(j, ['id', 'name', 'data']),
+          progress: progress ? `${progress}%` : 'n/a',
+          state: state,
+        };
+      }),
+    );
+
+    return {
+      bookmarklet,
+      jobs: JSON.parse(
+        JSON.stringify(
+          jobList
+            .sort(
+              (a: {id: string}, b: { id: string }) => parseInt(b.id.toString()) - parseInt(a.id.toString()),
+            )
+            .slice(0, 10),
+        ),
+      ),
+    };
+  }
+
+  @Get('/extractors')
+  async listExtractors() {
+    const extractors = await this.ytdlService.listExtractors();
+
+    return {
+      extractors,
+    };
+  }
+
+  @Post('/getinfo')
+  @HttpCode(200)
+  @UsePipes(new ValidationPipe({ transform: true }))
+  async getInfo(@Body() body: UploadDto) {
+    const videoInfo = await this.ytdlService.getVideoInfo(body.url);
+
+    return {
+      title: videoInfo.title,
+      extractor: videoInfo.extractor,
+      description: videoInfo.description,
+      videoUrl: videoInfo.webpage_url,
+      thumbnails: videoInfo.thumbnails,
+      upload_date: videoInfo.upload_date,
+      duration: videoInfo.duration,
+      formats: videoInfo.formats.map((f) =>
+        pick(f, ['format_id', 'format', 'filesize', 'format_note', 'ext']),
+      ),
+    };
+  }
+
+  @Post('/queue')
+  @UsePipes(new ValidationPipe({ transform: true }))
+  async queue(@Body() body: QueueDto) {
+    if (!body.extractor || !body.title) {
+      const videoInfo = await this.ytdlService.getVideoInfo(body.url);
+      body.extractor = videoInfo.extractor;
+      body.title = videoInfo.title;
+    }
+
+    const job = await this.vidgrabQueue.add('download', body);
+
+    this.eventEmitter.emit('job.added', { job });
+
+    return {
+      jobId: job.id,
+    };
+  }
+}