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';
const fetch = require('isomorphic-unfetch');
const { getData, getPreview, getTracks, getDetails } = require('spotify-url-info')(fetch);
import {PlaylistService} from "./playlist.service";
import {PlaylistModel} from "./playlist.model";
import {TrackService} from "../track/track.service";
import {PlaylistEntity} from "./playlist.entity";
@Controller('playlist')
export class PlaylistController {
constructor(
private readonly service: PlaylistService,
private readonly trackService: TrackService,
) {
}
@@ -20,16 +17,7 @@ export class PlaylistController {
}
@Post()
async create(@Body() playlist: PlaylistModel): Promise<void> {
await getDetails(playlist.spotifyUrl).then(async details => {
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);
});
});
async create(@Body() playlist: PlaylistModel): Promise<PlaylistEntity> {
return await this.service.create(playlist);
}
}
+15 -2
View File
@@ -2,12 +2,16 @@ import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
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()
export class PlaylistService {
constructor(
@InjectRepository(PlaylistEntity)
private repository: Repository<PlaylistEntity>,
private readonly trackService: TrackService,
) {}
findAll(): Promise<PlaylistEntity[]> {
@@ -22,8 +26,17 @@ export class PlaylistService {
await this.repository.delete(id);
}
async create(track: PlaylistEntity): Promise<PlaylistEntity> {
return this.repository.save(track);
async create(playlist: PlaylistEntity): Promise<PlaylistEntity> {
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> {
+21 -15
View File
@@ -39,28 +39,34 @@ export class TrackService {
await this.repository.update(id, track);
}
@Interval(3000)
@Interval(10000)
async findOnYoutube() {
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}`);
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() {
const queuedTracks = await this.findAll({status: TrackStatusEnum.Queued});
queuedTracks.forEach(async track => {
await ytdl(track.youtubeUrl, {quality: "highestaudio", filter: "audioonly"}).pipe(
fs.createWriteStream(resolve(
__dirname,
'..',
this.configService.get<string>('DOWNLOADS'),
`${track.artist} - ${track.song.replace('/', '')}.mp3`,
))
);
for(let track of queuedTracks) {
await this.youtubeDownload(track);
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())
)
);
}
}