fix asynchronous code

This commit is contained in:
raiper34
2024-06-26 23:00:01 +02:00
parent 903697646d
commit 3455db8d71
3 changed files with 39 additions and 32 deletions
+3 -15
View File
@@ -1,16 +1,13 @@
import {Body, Controller, Get, Post} from '@nestjs/common'; import {Body, Controller, Get, Post} from '@nestjs/common';
const fetch = require('isomorphic-unfetch');
const { getData, getPreview, getTracks, getDetails } = require('spotify-url-info')(fetch);
import {PlaylistService} from "./playlist.service"; import {PlaylistService} from "./playlist.service";
import {PlaylistModel} from "./playlist.model"; import {PlaylistModel} from "./playlist.model";
import {TrackService} from "../track/track.service"; import {PlaylistEntity} from "./playlist.entity";
@Controller('playlist') @Controller('playlist')
export class PlaylistController { export class PlaylistController {
constructor( constructor(
private readonly service: PlaylistService, private readonly service: PlaylistService,
private readonly trackService: TrackService,
) { ) {
} }
@@ -20,16 +17,7 @@ export class PlaylistController {
} }
@Post() @Post()
async create(@Body() playlist: PlaylistModel): Promise<void> { async create(@Body() playlist: PlaylistModel): Promise<PlaylistEntity> {
await getDetails(playlist.spotifyUrl).then(async details => { return await this.service.create(playlist);
const savedPlaylist = await this.service.create({...playlist, name: details.preview.title});
details.tracks.forEach(track => {
this.trackService.create({
artist: track.artist,
song: track.name,
spotifyUrl: track.previewUrl,
}, savedPlaylist);
});
});
} }
} }
+15 -2
View File
@@ -2,12 +2,16 @@ import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm'; import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm'; import { Repository } from 'typeorm';
import {PlaylistEntity} from "./playlist.entity"; import {PlaylistEntity} from "./playlist.entity";
import {TrackService} from "../track/track.service";
const fetch = require('isomorphic-unfetch');
const { getData, getPreview, getTracks, getDetails } = require('spotify-url-info')(fetch);
@Injectable() @Injectable()
export class PlaylistService { export class PlaylistService {
constructor( constructor(
@InjectRepository(PlaylistEntity) @InjectRepository(PlaylistEntity)
private repository: Repository<PlaylistEntity>, private repository: Repository<PlaylistEntity>,
private readonly trackService: TrackService,
) {} ) {}
findAll(): Promise<PlaylistEntity[]> { findAll(): Promise<PlaylistEntity[]> {
@@ -22,8 +26,17 @@ export class PlaylistService {
await this.repository.delete(id); await this.repository.delete(id);
} }
async create(track: PlaylistEntity): Promise<PlaylistEntity> { async create(playlist: PlaylistEntity): Promise<PlaylistEntity> {
return this.repository.save(track); const details = await getDetails(playlist.spotifyUrl);
const savedPlaylist = await this.repository.save({...playlist, name: details.preview.title});
for(let track of details.tracks) {
await this.trackService.create({
artist: track.artist,
song: track.name,
spotifyUrl: track.previewUrl,
}, savedPlaylist);
}
return savedPlaylist;
} }
async update(id: number, track: PlaylistEntity): Promise<void> { async update(id: number, track: PlaylistEntity): Promise<void> {
+20 -14
View File
@@ -39,28 +39,34 @@ export class TrackService {
await this.repository.update(id, track); await this.repository.update(id, track);
} }
@Interval(3000) @Interval(10000)
async findOnYoutube() { async findOnYoutube() {
const newTracks = await this.findAll({status: TrackStatusEnum.New}); const newTracks = await this.findAll({status: TrackStatusEnum.New});
newTracks.forEach(async track => { for(let track of newTracks) {
const youtubeResult = await yts(`${track.artist} - ${track.song}`); const youtubeResult = await yts(`${track.artist} - ${track.song}`);
await this.update(track.id, {...track, youtubeUrl: youtubeResult.videos[0].url, status: TrackStatusEnum.Queued}) await this.update(track.id, {
...track,
youtubeUrl: youtubeResult.videos[0].url,
status: TrackStatusEnum.Queued
}); });
} }
}
@Interval(3000) @Interval(15000)
async download() { async download() {
const queuedTracks = await this.findAll({status: TrackStatusEnum.Queued}); const queuedTracks = await this.findAll({status: TrackStatusEnum.Queued});
queuedTracks.forEach(async track => { for(let track of queuedTracks) {
await ytdl(track.youtubeUrl, {quality: "highestaudio", filter: "audioonly"}).pipe( await this.youtubeDownload(track);
fs.createWriteStream(resolve(
__dirname,
'..',
this.configService.get<string>('DOWNLOADS'),
`${track.artist} - ${track.song.replace('/', '')}.mp3`,
))
);
await this.update(track.id, {...track, status: TrackStatusEnum.Completed}) await this.update(track.id, {...track, status: TrackStatusEnum.Completed})
}); }
}
private youtubeDownload(track: TrackEntity): Promise<void> {
return new Promise((res, _reject) =>
ytdl(track.youtubeUrl, {quality: "highestaudio", filter: "audioonly"}).pipe(
fs.createWriteStream(resolve(__dirname, '..', this.configService.get<string>('DOWNLOADS'), `${track.artist} - ${track.song.replace('/', '')}.mp3`))
.on('finish', () => res())
)
);
} }
} }