mirror of
https://github.com/BluemediaDev/muse.git
synced 2025-06-27 17:22:42 +02:00
commit
250c7d316b
9 changed files with 201 additions and 166 deletions
|
@ -37,7 +37,7 @@
|
||||||
"@types/debug": "^4.1.5",
|
"@types/debug": "^4.1.5",
|
||||||
"@types/fluent-ffmpeg": "^2.1.17",
|
"@types/fluent-ffmpeg": "^2.1.17",
|
||||||
"@types/fs-capacitor": "^2.0.0",
|
"@types/fs-capacitor": "^2.0.0",
|
||||||
"@types/ms": "0.7.31",
|
"@types/ms": "0.7.34",
|
||||||
"@types/node": "^17.0.0",
|
"@types/node": "^17.0.0",
|
||||||
"@types/node-emoji": "^1.8.1",
|
"@types/node-emoji": "^1.8.1",
|
||||||
"@types/spotify-web-api-node": "^5.0.2",
|
"@types/spotify-web-api-node": "^5.0.2",
|
||||||
|
@ -106,14 +106,14 @@
|
||||||
"got": "^12.0.2",
|
"got": "^12.0.2",
|
||||||
"hasha": "^5.2.2",
|
"hasha": "^5.2.2",
|
||||||
"inversify": "^6.0.1",
|
"inversify": "^6.0.1",
|
||||||
"iso8601-duration": "^1.3.0",
|
"iso8601-duration": "^2.1.2",
|
||||||
"libsodium-wrappers": "^0.7.9",
|
"libsodium-wrappers": "^0.7.9",
|
||||||
"make-dir": "^3.1.0",
|
"make-dir": "^3.1.0",
|
||||||
"node-emoji": "^1.10.0",
|
"node-emoji": "^1.10.0",
|
||||||
"nodesplash": "^0.1.1",
|
"nodesplash": "^0.1.1",
|
||||||
"ora": "^8.1.0",
|
"ora": "^8.1.0",
|
||||||
"p-event": "^5.0.1",
|
"p-event": "^5.0.1",
|
||||||
"p-limit": "^4.0.0",
|
"p-limit": "^6.1.0",
|
||||||
"p-queue": "^7.2.0",
|
"p-queue": "^7.2.0",
|
||||||
"p-retry": "6.2.0",
|
"p-retry": "6.2.0",
|
||||||
"pagination.djs": "^4.0.10",
|
"pagination.djs": "^4.0.10",
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
import {AutocompleteInteraction, ChatInputCommandInteraction} from 'discord.js';
|
import {AutocompleteInteraction, ChatInputCommandInteraction} from 'discord.js';
|
||||||
import {URL} from 'url';
|
import {URL} from 'url';
|
||||||
import {SlashCommandBuilder} from '@discordjs/builders';
|
import {SlashCommandBuilder, SlashCommandSubcommandsOnlyBuilder} from '@discordjs/builders';
|
||||||
import {inject, injectable} from 'inversify';
|
import {inject, injectable, optional} from 'inversify';
|
||||||
import Spotify from 'spotify-web-api-node';
|
import Spotify from 'spotify-web-api-node';
|
||||||
import Command from './index.js';
|
import Command from './index.js';
|
||||||
import {TYPES} from '../types.js';
|
import {TYPES} from '../types.js';
|
||||||
|
@ -13,12 +13,29 @@ import AddQueryToQueue from '../services/add-query-to-queue.js';
|
||||||
|
|
||||||
@injectable()
|
@injectable()
|
||||||
export default class implements Command {
|
export default class implements Command {
|
||||||
public readonly slashCommand = new SlashCommandBuilder()
|
public readonly slashCommand: Partial<SlashCommandBuilder | SlashCommandSubcommandsOnlyBuilder> & Pick<SlashCommandBuilder, 'toJSON'>;
|
||||||
|
|
||||||
|
public requiresVC = true;
|
||||||
|
|
||||||
|
private readonly spotify?: Spotify;
|
||||||
|
private readonly cache: KeyValueCacheProvider;
|
||||||
|
private readonly addQueryToQueue: AddQueryToQueue;
|
||||||
|
|
||||||
|
constructor(@inject(TYPES.ThirdParty) @optional() thirdParty: ThirdParty, @inject(TYPES.KeyValueCache) cache: KeyValueCacheProvider, @inject(TYPES.Services.AddQueryToQueue) addQueryToQueue: AddQueryToQueue) {
|
||||||
|
this.spotify = thirdParty?.spotify;
|
||||||
|
this.cache = cache;
|
||||||
|
this.addQueryToQueue = addQueryToQueue;
|
||||||
|
|
||||||
|
const queryDescription = thirdParty === undefined
|
||||||
|
? 'YouTube URL or search query'
|
||||||
|
: 'YouTube URL, Spotify URL, or search query';
|
||||||
|
|
||||||
|
this.slashCommand = new SlashCommandBuilder()
|
||||||
.setName('play')
|
.setName('play')
|
||||||
.setDescription('play a song')
|
.setDescription('play a song')
|
||||||
.addStringOption(option => option
|
.addStringOption(option => option
|
||||||
.setName('query')
|
.setName('query')
|
||||||
.setDescription('YouTube URL, Spotify URL, or search query')
|
.setDescription(queryDescription)
|
||||||
.setAutocomplete(true)
|
.setAutocomplete(true)
|
||||||
.setRequired(true))
|
.setRequired(true))
|
||||||
.addBooleanOption(option => option
|
.addBooleanOption(option => option
|
||||||
|
@ -33,17 +50,6 @@ export default class implements Command {
|
||||||
.addBooleanOption(option => option
|
.addBooleanOption(option => option
|
||||||
.setName('skip')
|
.setName('skip')
|
||||||
.setDescription('skip the currently playing track'));
|
.setDescription('skip the currently playing track'));
|
||||||
|
|
||||||
public requiresVC = true;
|
|
||||||
|
|
||||||
private readonly spotify: Spotify;
|
|
||||||
private readonly cache: KeyValueCacheProvider;
|
|
||||||
private readonly addQueryToQueue: AddQueryToQueue;
|
|
||||||
|
|
||||||
constructor(@inject(TYPES.ThirdParty) thirdParty: ThirdParty, @inject(TYPES.KeyValueCache) cache: KeyValueCacheProvider, @inject(TYPES.Services.AddQueryToQueue) addQueryToQueue: AddQueryToQueue) {
|
|
||||||
this.spotify = thirdParty.spotify;
|
|
||||||
this.cache = cache;
|
|
||||||
this.addQueryToQueue = addQueryToQueue;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public async execute(interaction: ChatInputCommandInteraction): Promise<void> {
|
public async execute(interaction: ChatInputCommandInteraction): Promise<void> {
|
||||||
|
|
|
@ -57,11 +57,20 @@ container.bind<Client>(TYPES.Client).toConstantValue(new Client({intents}));
|
||||||
// Managers
|
// Managers
|
||||||
container.bind<PlayerManager>(TYPES.Managers.Player).to(PlayerManager).inSingletonScope();
|
container.bind<PlayerManager>(TYPES.Managers.Player).to(PlayerManager).inSingletonScope();
|
||||||
|
|
||||||
|
// Config values
|
||||||
|
container.bind(TYPES.Config).toConstantValue(new ConfigProvider());
|
||||||
|
|
||||||
// Services
|
// Services
|
||||||
container.bind<GetSongs>(TYPES.Services.GetSongs).to(GetSongs).inSingletonScope();
|
container.bind<GetSongs>(TYPES.Services.GetSongs).to(GetSongs).inSingletonScope();
|
||||||
container.bind<AddQueryToQueue>(TYPES.Services.AddQueryToQueue).to(AddQueryToQueue).inSingletonScope();
|
container.bind<AddQueryToQueue>(TYPES.Services.AddQueryToQueue).to(AddQueryToQueue).inSingletonScope();
|
||||||
container.bind<YoutubeAPI>(TYPES.Services.YoutubeAPI).to(YoutubeAPI).inSingletonScope();
|
container.bind<YoutubeAPI>(TYPES.Services.YoutubeAPI).to(YoutubeAPI).inSingletonScope();
|
||||||
container.bind<SpotifyAPI>(TYPES.Services.SpotifyAPI).to(SpotifyAPI).inSingletonScope();
|
|
||||||
|
// Only instanciate spotify dependencies if the Spotify client ID and secret are set
|
||||||
|
const config = container.get<ConfigProvider>(TYPES.Config);
|
||||||
|
if (config.SPOTIFY_CLIENT_ID !== '' && config.SPOTIFY_CLIENT_SECRET !== '') {
|
||||||
|
container.bind<SpotifyAPI>(TYPES.Services.SpotifyAPI).to(SpotifyAPI).inSingletonScope();
|
||||||
|
container.bind(TYPES.ThirdParty).to(ThirdParty);
|
||||||
|
}
|
||||||
|
|
||||||
// Commands
|
// Commands
|
||||||
[
|
[
|
||||||
|
@ -91,12 +100,7 @@ container.bind<SpotifyAPI>(TYPES.Services.SpotifyAPI).to(SpotifyAPI).inSingleton
|
||||||
container.bind<Command>(TYPES.Command).to(command).inSingletonScope();
|
container.bind<Command>(TYPES.Command).to(command).inSingletonScope();
|
||||||
});
|
});
|
||||||
|
|
||||||
// Config values
|
|
||||||
container.bind(TYPES.Config).toConstantValue(new ConfigProvider());
|
|
||||||
|
|
||||||
// Static libraries
|
// Static libraries
|
||||||
container.bind(TYPES.ThirdParty).to(ThirdParty);
|
|
||||||
|
|
||||||
container.bind(TYPES.FileCache).to(FileCacheProvider);
|
container.bind(TYPES.FileCache).to(FileCacheProvider);
|
||||||
container.bind(TYPES.KeyValueCache).to(KeyValueCacheProvider);
|
container.bind(TYPES.KeyValueCache).to(KeyValueCacheProvider);
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
/* eslint-disable complexity */
|
/* eslint-disable complexity */
|
||||||
import {ChatInputCommandInteraction, GuildMember} from 'discord.js';
|
import {ChatInputCommandInteraction, GuildMember} from 'discord.js';
|
||||||
import {URL} from 'node:url';
|
|
||||||
import {inject, injectable} from 'inversify';
|
import {inject, injectable} from 'inversify';
|
||||||
import shuffle from 'array-shuffle';
|
import shuffle from 'array-shuffle';
|
||||||
import {TYPES} from '../types.js';
|
import {TYPES} from '../types.js';
|
||||||
|
@ -60,74 +59,7 @@ export default class AddQueryToQueue {
|
||||||
|
|
||||||
await interaction.deferReply({ephemeral: queueAddResponseEphemeral});
|
await interaction.deferReply({ephemeral: queueAddResponseEphemeral});
|
||||||
|
|
||||||
let newSongs: SongMetadata[] = [];
|
let [newSongs, extraMsg] = await this.getSongs.getSongs(query, playlistLimit, shouldSplitChapters);
|
||||||
let extraMsg = '';
|
|
||||||
|
|
||||||
// Test if it's a complete URL
|
|
||||||
try {
|
|
||||||
const url = new URL(query);
|
|
||||||
|
|
||||||
const YOUTUBE_HOSTS = [
|
|
||||||
'www.youtube.com',
|
|
||||||
'youtu.be',
|
|
||||||
'youtube.com',
|
|
||||||
'music.youtube.com',
|
|
||||||
'www.music.youtube.com',
|
|
||||||
];
|
|
||||||
|
|
||||||
if (YOUTUBE_HOSTS.includes(url.host)) {
|
|
||||||
// YouTube source
|
|
||||||
if (url.searchParams.get('list')) {
|
|
||||||
// YouTube playlist
|
|
||||||
newSongs.push(...await this.getSongs.youtubePlaylist(url.searchParams.get('list')!, shouldSplitChapters));
|
|
||||||
} else {
|
|
||||||
const songs = await this.getSongs.youtubeVideo(url.href, shouldSplitChapters);
|
|
||||||
|
|
||||||
if (songs) {
|
|
||||||
newSongs.push(...songs);
|
|
||||||
} else {
|
|
||||||
throw new Error('that doesn\'t exist');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (url.protocol === 'spotify:' || url.host === 'open.spotify.com') {
|
|
||||||
const [convertedSongs, nSongsNotFound, totalSongs] = await this.getSongs.spotifySource(query, playlistLimit, shouldSplitChapters);
|
|
||||||
|
|
||||||
if (totalSongs > playlistLimit) {
|
|
||||||
extraMsg = `a random sample of ${playlistLimit} songs was taken`;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (totalSongs > playlistLimit && nSongsNotFound !== 0) {
|
|
||||||
extraMsg += ' and ';
|
|
||||||
}
|
|
||||||
|
|
||||||
if (nSongsNotFound !== 0) {
|
|
||||||
if (nSongsNotFound === 1) {
|
|
||||||
extraMsg += '1 song was not found';
|
|
||||||
} else {
|
|
||||||
extraMsg += `${nSongsNotFound.toString()} songs were not found`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
newSongs.push(...convertedSongs);
|
|
||||||
} else {
|
|
||||||
const song = await this.getSongs.httpLiveStream(query);
|
|
||||||
|
|
||||||
if (song) {
|
|
||||||
newSongs.push(song);
|
|
||||||
} else {
|
|
||||||
throw new Error('that doesn\'t exist');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (_: unknown) {
|
|
||||||
// Not a URL, must search YouTube
|
|
||||||
const songs = await this.getSongs.youtubeVideoSearch(query, shouldSplitChapters);
|
|
||||||
|
|
||||||
if (songs) {
|
|
||||||
newSongs.push(...songs);
|
|
||||||
} else {
|
|
||||||
throw new Error('that doesn\'t exist');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (newSongs.length === 0) {
|
if (newSongs.length === 0) {
|
||||||
throw new Error('no songs found');
|
throw new Error('no songs found');
|
||||||
|
|
|
@ -12,8 +12,8 @@ export const DATA_DIR = path.resolve(process.env.DATA_DIR ? process.env.DATA_DIR
|
||||||
const CONFIG_MAP = {
|
const CONFIG_MAP = {
|
||||||
DISCORD_TOKEN: process.env.DISCORD_TOKEN,
|
DISCORD_TOKEN: process.env.DISCORD_TOKEN,
|
||||||
YOUTUBE_API_KEY: process.env.YOUTUBE_API_KEY,
|
YOUTUBE_API_KEY: process.env.YOUTUBE_API_KEY,
|
||||||
SPOTIFY_CLIENT_ID: process.env.SPOTIFY_CLIENT_ID,
|
SPOTIFY_CLIENT_ID: process.env.SPOTIFY_CLIENT_ID ?? '',
|
||||||
SPOTIFY_CLIENT_SECRET: process.env.SPOTIFY_CLIENT_SECRET,
|
SPOTIFY_CLIENT_SECRET: process.env.SPOTIFY_CLIENT_SECRET ?? '',
|
||||||
REGISTER_COMMANDS_ON_BOT: process.env.REGISTER_COMMANDS_ON_BOT === 'true',
|
REGISTER_COMMANDS_ON_BOT: process.env.REGISTER_COMMANDS_ON_BOT === 'true',
|
||||||
DATA_DIR,
|
DATA_DIR,
|
||||||
CACHE_DIR: path.join(DATA_DIR, 'cache'),
|
CACHE_DIR: path.join(DATA_DIR, 'cache'),
|
||||||
|
|
|
@ -1,34 +1,120 @@
|
||||||
import {inject, injectable} from 'inversify';
|
import {inject, injectable, optional} from 'inversify';
|
||||||
import * as spotifyURI from 'spotify-uri';
|
import * as spotifyURI from 'spotify-uri';
|
||||||
import {SongMetadata, QueuedPlaylist, MediaSource} from './player.js';
|
import {SongMetadata, QueuedPlaylist, MediaSource} from './player.js';
|
||||||
import {TYPES} from '../types.js';
|
import {TYPES} from '../types.js';
|
||||||
import ffmpeg from 'fluent-ffmpeg';
|
import ffmpeg from 'fluent-ffmpeg';
|
||||||
import YoutubeAPI from './youtube-api.js';
|
import YoutubeAPI from './youtube-api.js';
|
||||||
import SpotifyAPI, {SpotifyTrack} from './spotify-api.js';
|
import SpotifyAPI, {SpotifyTrack} from './spotify-api.js';
|
||||||
|
import {URL} from 'node:url';
|
||||||
|
|
||||||
@injectable()
|
@injectable()
|
||||||
export default class {
|
export default class {
|
||||||
private readonly youtubeAPI: YoutubeAPI;
|
private readonly youtubeAPI: YoutubeAPI;
|
||||||
private readonly spotifyAPI: SpotifyAPI;
|
private readonly spotifyAPI?: SpotifyAPI;
|
||||||
|
|
||||||
constructor(@inject(TYPES.Services.YoutubeAPI) youtubeAPI: YoutubeAPI, @inject(TYPES.Services.SpotifyAPI) spotifyAPI: SpotifyAPI) {
|
constructor(@inject(TYPES.Services.YoutubeAPI) youtubeAPI: YoutubeAPI, @inject(TYPES.Services.SpotifyAPI) @optional() spotifyAPI?: SpotifyAPI) {
|
||||||
this.youtubeAPI = youtubeAPI;
|
this.youtubeAPI = youtubeAPI;
|
||||||
this.spotifyAPI = spotifyAPI;
|
this.spotifyAPI = spotifyAPI;
|
||||||
}
|
}
|
||||||
|
|
||||||
async youtubeVideoSearch(query: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
async getSongs(query: string, playlistLimit: number, shouldSplitChapters: boolean): Promise<[SongMetadata[], string]> {
|
||||||
|
const newSongs: SongMetadata[] = [];
|
||||||
|
let extraMsg = '';
|
||||||
|
|
||||||
|
// Test if it's a complete URL
|
||||||
|
try {
|
||||||
|
const url = new URL(query);
|
||||||
|
|
||||||
|
const YOUTUBE_HOSTS = [
|
||||||
|
'www.youtube.com',
|
||||||
|
'youtu.be',
|
||||||
|
'youtube.com',
|
||||||
|
'music.youtube.com',
|
||||||
|
'www.music.youtube.com',
|
||||||
|
];
|
||||||
|
|
||||||
|
if (YOUTUBE_HOSTS.includes(url.host)) {
|
||||||
|
// YouTube source
|
||||||
|
if (url.searchParams.get('list')) {
|
||||||
|
// YouTube playlist
|
||||||
|
newSongs.push(...await this.youtubePlaylist(url.searchParams.get('list')!, shouldSplitChapters));
|
||||||
|
} else {
|
||||||
|
const songs = await this.youtubeVideo(url.href, shouldSplitChapters);
|
||||||
|
|
||||||
|
if (songs) {
|
||||||
|
newSongs.push(...songs);
|
||||||
|
} else {
|
||||||
|
throw new Error('that doesn\'t exist');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if (url.protocol === 'spotify:' || url.host === 'open.spotify.com') {
|
||||||
|
if (this.spotifyAPI === undefined) {
|
||||||
|
throw new Error('Spotify is not enabled!');
|
||||||
|
}
|
||||||
|
|
||||||
|
const [convertedSongs, nSongsNotFound, totalSongs] = await this.spotifySource(query, playlistLimit, shouldSplitChapters);
|
||||||
|
|
||||||
|
if (totalSongs > playlistLimit) {
|
||||||
|
extraMsg = `a random sample of ${playlistLimit} songs was taken`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (totalSongs > playlistLimit && nSongsNotFound !== 0) {
|
||||||
|
extraMsg += ' and ';
|
||||||
|
}
|
||||||
|
|
||||||
|
if (nSongsNotFound !== 0) {
|
||||||
|
if (nSongsNotFound === 1) {
|
||||||
|
extraMsg += '1 song was not found';
|
||||||
|
} else {
|
||||||
|
extraMsg += `${nSongsNotFound.toString()} songs were not found`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
newSongs.push(...convertedSongs);
|
||||||
|
} else {
|
||||||
|
const song = await this.httpLiveStream(query);
|
||||||
|
|
||||||
|
if (song) {
|
||||||
|
newSongs.push(song);
|
||||||
|
} else {
|
||||||
|
throw new Error('that doesn\'t exist');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (err: any) {
|
||||||
|
if (err instanceof Error && err.message === 'Spotify is not enabled!') {
|
||||||
|
throw err;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Not a URL, must search YouTube
|
||||||
|
const songs = await this.youtubeVideoSearch(query, shouldSplitChapters);
|
||||||
|
|
||||||
|
if (songs) {
|
||||||
|
newSongs.push(...songs);
|
||||||
|
} else {
|
||||||
|
throw new Error('that doesn\'t exist');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return [newSongs, extraMsg];
|
||||||
|
}
|
||||||
|
|
||||||
|
private async youtubeVideoSearch(query: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
||||||
return this.youtubeAPI.search(query, shouldSplitChapters);
|
return this.youtubeAPI.search(query, shouldSplitChapters);
|
||||||
}
|
}
|
||||||
|
|
||||||
async youtubeVideo(url: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
private async youtubeVideo(url: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
||||||
return this.youtubeAPI.getVideo(url, shouldSplitChapters);
|
return this.youtubeAPI.getVideo(url, shouldSplitChapters);
|
||||||
}
|
}
|
||||||
|
|
||||||
async youtubePlaylist(listId: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
private async youtubePlaylist(listId: string, shouldSplitChapters: boolean): Promise<SongMetadata[]> {
|
||||||
return this.youtubeAPI.getPlaylist(listId, shouldSplitChapters);
|
return this.youtubeAPI.getPlaylist(listId, shouldSplitChapters);
|
||||||
}
|
}
|
||||||
|
|
||||||
async spotifySource(url: string, playlistLimit: number, shouldSplitChapters: boolean): Promise<[SongMetadata[], number, number]> {
|
private async spotifySource(url: string, playlistLimit: number, shouldSplitChapters: boolean): Promise<[SongMetadata[], number, number]> {
|
||||||
|
if (this.spotifyAPI === undefined) {
|
||||||
|
return [[], 0, 0];
|
||||||
|
}
|
||||||
|
|
||||||
const parsed = spotifyURI.parse(url);
|
const parsed = spotifyURI.parse(url);
|
||||||
|
|
||||||
switch (parsed.type) {
|
switch (parsed.type) {
|
||||||
|
@ -58,7 +144,7 @@ export default class {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async httpLiveStream(url: string): Promise<SongMetadata> {
|
private async httpLiveStream(url: string): Promise<SongMetadata> {
|
||||||
return new Promise((resolve, reject) => {
|
return new Promise((resolve, reject) => {
|
||||||
ffmpeg(url).ffprobe((err, _) => {
|
ffmpeg(url).ffprobe((err, _) => {
|
||||||
if (err) {
|
if (err) {
|
||||||
|
|
|
@ -95,7 +95,7 @@ export default class {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!firstVideo) {
|
if (!firstVideo) {
|
||||||
throw new Error('No video found.');
|
return [];
|
||||||
}
|
}
|
||||||
|
|
||||||
return this.getVideo(firstVideo.url, shouldSplitChapters);
|
return this.getVideo(firstVideo.url, shouldSplitChapters);
|
||||||
|
|
|
@ -14,28 +14,18 @@ const filterDuplicates = <T extends {name: string}>(items: T[]) => {
|
||||||
return results;
|
return results;
|
||||||
};
|
};
|
||||||
|
|
||||||
const getYouTubeAndSpotifySuggestionsFor = async (query: string, spotify: SpotifyWebApi, limit = 10): Promise<APIApplicationCommandOptionChoice[]> => {
|
const getYouTubeAndSpotifySuggestionsFor = async (query: string, spotify?: SpotifyWebApi, limit = 10): Promise<APIApplicationCommandOptionChoice[]> => {
|
||||||
const [youtubeSuggestions, spotifyResults] = await Promise.all([
|
// Only search Spotify if enabled
|
||||||
getYouTubeSuggestionsFor(query),
|
const spotifySuggestionPromise = spotify === undefined
|
||||||
spotify.search(query, ['track', 'album'], {limit: 5}),
|
? undefined
|
||||||
]);
|
: spotify.search(query, ['album', 'track'], {limit});
|
||||||
|
|
||||||
|
const youtubeSuggestions = await getYouTubeSuggestionsFor(query);
|
||||||
|
|
||||||
const totalYouTubeResults = youtubeSuggestions.length;
|
const totalYouTubeResults = youtubeSuggestions.length;
|
||||||
|
const numOfYouTubeSuggestions = Math.min(limit, totalYouTubeResults);
|
||||||
|
|
||||||
const spotifyAlbums = filterDuplicates(spotifyResults.body.albums?.items ?? []);
|
let suggestions: APIApplicationCommandOptionChoice[] = [];
|
||||||
const spotifyTracks = filterDuplicates(spotifyResults.body.tracks?.items ?? []);
|
|
||||||
|
|
||||||
const totalSpotifyResults = spotifyAlbums.length + spotifyTracks.length;
|
|
||||||
|
|
||||||
// Number of results for each source should be roughly the same.
|
|
||||||
// If we don't have enough Spotify suggestions, prioritize YouTube results.
|
|
||||||
const maxSpotifySuggestions = Math.floor(limit / 2);
|
|
||||||
const numOfSpotifySuggestions = Math.min(maxSpotifySuggestions, totalSpotifyResults);
|
|
||||||
|
|
||||||
const maxYouTubeSuggestions = limit - numOfSpotifySuggestions;
|
|
||||||
const numOfYouTubeSuggestions = Math.min(maxYouTubeSuggestions, totalYouTubeResults);
|
|
||||||
|
|
||||||
const suggestions: APIApplicationCommandOptionChoice[] = [];
|
|
||||||
|
|
||||||
suggestions.push(
|
suggestions.push(
|
||||||
...youtubeSuggestions
|
...youtubeSuggestions
|
||||||
|
@ -46,10 +36,26 @@ const getYouTubeAndSpotifySuggestionsFor = async (query: string, spotify: Spotif
|
||||||
}),
|
}),
|
||||||
));
|
));
|
||||||
|
|
||||||
|
if (spotify !== undefined && spotifySuggestionPromise !== undefined) {
|
||||||
|
const spotifyResponse = (await spotifySuggestionPromise).body;
|
||||||
|
const spotifyAlbums = filterDuplicates(spotifyResponse.albums?.items ?? []);
|
||||||
|
const spotifyTracks = filterDuplicates(spotifyResponse.tracks?.items ?? []);
|
||||||
|
|
||||||
|
const totalSpotifyResults = spotifyAlbums.length + spotifyTracks.length;
|
||||||
|
|
||||||
|
// Number of results for each source should be roughly the same.
|
||||||
|
// If we don't have enough Spotify suggestions, prioritize YouTube results.
|
||||||
|
const maxSpotifySuggestions = Math.floor(limit / 2);
|
||||||
|
const numOfSpotifySuggestions = Math.min(maxSpotifySuggestions, totalSpotifyResults);
|
||||||
|
|
||||||
const maxSpotifyAlbums = Math.floor(numOfSpotifySuggestions / 2);
|
const maxSpotifyAlbums = Math.floor(numOfSpotifySuggestions / 2);
|
||||||
const numOfSpotifyAlbums = Math.min(maxSpotifyAlbums, spotifyResults.body.albums?.items.length ?? 0);
|
const numOfSpotifyAlbums = Math.min(maxSpotifyAlbums, spotifyResponse.albums?.items.length ?? 0);
|
||||||
const maxSpotifyTracks = numOfSpotifySuggestions - numOfSpotifyAlbums;
|
const maxSpotifyTracks = numOfSpotifySuggestions - numOfSpotifyAlbums;
|
||||||
|
|
||||||
|
// Make room for spotify results
|
||||||
|
const maxYouTubeSuggestions = limit - numOfSpotifySuggestions;
|
||||||
|
suggestions = suggestions.slice(0, maxYouTubeSuggestions);
|
||||||
|
|
||||||
suggestions.push(
|
suggestions.push(
|
||||||
...spotifyAlbums.slice(0, maxSpotifyAlbums).map(album => ({
|
...spotifyAlbums.slice(0, maxSpotifyAlbums).map(album => ({
|
||||||
name: `Spotify: 💿 ${album.name}${album.artists.length > 0 ? ` - ${album.artists[0].name}` : ''}`,
|
name: `Spotify: 💿 ${album.name}${album.artists.length > 0 ? ` - ${album.artists[0].name}` : ''}`,
|
||||||
|
@ -63,6 +69,7 @@ const getYouTubeAndSpotifySuggestionsFor = async (query: string, spotify: Spotif
|
||||||
value: `spotify:track:${track.id}`,
|
value: `spotify:track:${track.id}`,
|
||||||
})),
|
})),
|
||||||
);
|
);
|
||||||
|
}
|
||||||
|
|
||||||
return suggestions;
|
return suggestions;
|
||||||
};
|
};
|
||||||
|
|
34
yarn.lock
34
yarn.lock
|
@ -550,10 +550,10 @@
|
||||||
resolved "https://registry.npmjs.org/@types/libsodium-wrappers/-/libsodium-wrappers-0.7.9.tgz"
|
resolved "https://registry.npmjs.org/@types/libsodium-wrappers/-/libsodium-wrappers-0.7.9.tgz"
|
||||||
integrity sha512-LisgKLlYQk19baQwjkBZZXdJL0KbeTpdEnrAfz5hQACbklCY0gVFnsKUyjfNWF1UQsCSjw93Sj5jSbiO8RPfdw==
|
integrity sha512-LisgKLlYQk19baQwjkBZZXdJL0KbeTpdEnrAfz5hQACbklCY0gVFnsKUyjfNWF1UQsCSjw93Sj5jSbiO8RPfdw==
|
||||||
|
|
||||||
"@types/ms@*", "@types/ms@0.7.31":
|
"@types/ms@*", "@types/ms@0.7.34":
|
||||||
version "0.7.31"
|
version "0.7.34"
|
||||||
resolved "https://registry.npmjs.org/@types/ms/-/ms-0.7.31.tgz"
|
resolved "https://registry.yarnpkg.com/@types/ms/-/ms-0.7.34.tgz#10964ba0dee6ac4cd462e2795b6bebd407303433"
|
||||||
integrity sha512-iiUgKzV9AuaEkZqkOLDIvlQiL6ltuZd9tGcW3gwpnX8JbuiuhFlEGmmFXEXkN50Cvq7Os88IY2v0dkDqXYWVgA==
|
integrity sha512-nG96G3Wp6acyAgJqGasjODb+acrI7KltPiRxzHPXnP3NgI28bpQDRv53olbqGXbfcgF5aiiHmO3xpwEpS5Ld9g==
|
||||||
|
|
||||||
"@types/node-emoji@^1.8.1":
|
"@types/node-emoji@^1.8.1":
|
||||||
version "1.8.1"
|
version "1.8.1"
|
||||||
|
@ -3149,10 +3149,10 @@ isexe@^2.0.0:
|
||||||
resolved "https://registry.npmjs.org/isexe/-/isexe-2.0.0.tgz"
|
resolved "https://registry.npmjs.org/isexe/-/isexe-2.0.0.tgz"
|
||||||
integrity sha1-6PvzdNxVb/iUehDcsFctYz8s+hA=
|
integrity sha1-6PvzdNxVb/iUehDcsFctYz8s+hA=
|
||||||
|
|
||||||
iso8601-duration@^1.3.0:
|
iso8601-duration@^2.1.2:
|
||||||
version "1.3.0"
|
version "2.1.2"
|
||||||
resolved "https://registry.npmjs.org/iso8601-duration/-/iso8601-duration-1.3.0.tgz"
|
resolved "https://registry.yarnpkg.com/iso8601-duration/-/iso8601-duration-2.1.2.tgz#b13f14068fe5890c91b91e1f74e474a49f355028"
|
||||||
integrity sha512-K4CiUBzo3YeWk76FuET/dQPH03WE04R94feo5TSKQCXpoXQt9E4yx2CnY737QZnSAI3PI4WlKo/zfqizGx52QQ==
|
integrity sha512-yXteYUiKv6x8seaDzyBwnZtPpmx766KfvQuaVNyPifYOjmPdOo3ajd4phDNa7Y5mTQGnXsNEcXFtVun1FjYXxQ==
|
||||||
|
|
||||||
iterate-iterator@^1.0.1:
|
iterate-iterator@^1.0.1:
|
||||||
version "1.0.2"
|
version "1.0.2"
|
||||||
|
@ -3823,12 +3823,12 @@ p-limit@^3.0.2:
|
||||||
dependencies:
|
dependencies:
|
||||||
yocto-queue "^0.1.0"
|
yocto-queue "^0.1.0"
|
||||||
|
|
||||||
p-limit@^4.0.0:
|
p-limit@^6.1.0:
|
||||||
version "4.0.0"
|
version "6.1.0"
|
||||||
resolved "https://registry.npmjs.org/p-limit/-/p-limit-4.0.0.tgz"
|
resolved "https://registry.yarnpkg.com/p-limit/-/p-limit-6.1.0.tgz#d91f9364d3fdff89b0a45c70d04ad4e0df30a0e8"
|
||||||
integrity sha512-5b0R4txpzjPWVw/cXXUResoD4hb6U/x9BH08L7nw+GN1sezDzPdxeRvpc9c433fZhBan/wusjbCsqwqm4EIBIQ==
|
integrity sha512-H0jc0q1vOzlEk0TqAKXKZxdl7kX3OFUzCnNVUnq5Pc3DGo0kpeaMuPqxQn235HibwBEb0/pm9dgKTjXy66fBkg==
|
||||||
dependencies:
|
dependencies:
|
||||||
yocto-queue "^1.0.0"
|
yocto-queue "^1.1.1"
|
||||||
|
|
||||||
p-locate@^5.0.0:
|
p-locate@^5.0.0:
|
||||||
version "5.0.0"
|
version "5.0.0"
|
||||||
|
@ -5394,10 +5394,10 @@ yocto-queue@^0.1.0:
|
||||||
resolved "https://registry.npmjs.org/yocto-queue/-/yocto-queue-0.1.0.tgz"
|
resolved "https://registry.npmjs.org/yocto-queue/-/yocto-queue-0.1.0.tgz"
|
||||||
integrity sha512-rVksvsnNCdJ/ohGc6xgPwyN8eheCxsiLM8mxuE/t/mOVqJewPuO1miLpTHQiRgTKCLexL4MeAFVagts7HmNZ2Q==
|
integrity sha512-rVksvsnNCdJ/ohGc6xgPwyN8eheCxsiLM8mxuE/t/mOVqJewPuO1miLpTHQiRgTKCLexL4MeAFVagts7HmNZ2Q==
|
||||||
|
|
||||||
yocto-queue@^1.0.0:
|
yocto-queue@^1.1.1:
|
||||||
version "1.0.0"
|
version "1.1.1"
|
||||||
resolved "https://registry.npmjs.org/yocto-queue/-/yocto-queue-1.0.0.tgz"
|
resolved "https://registry.yarnpkg.com/yocto-queue/-/yocto-queue-1.1.1.tgz#fef65ce3ac9f8a32ceac5a634f74e17e5b232110"
|
||||||
integrity sha512-9bnSc/HEW2uRy67wc+T8UwauLuPJVn28jb+GtJY16iiKWyvmYJRXVT4UamsAEGQfPohgr2q4Tq0sQbQlxTfi1g==
|
integrity sha512-b4JR1PFR10y1mKjhHY9LaGo6tmrgjit7hxVIeAmyMw3jegXR4dhYqLaQF5zMXZxY7tLpMyJeLjr1C4rLmkVe8g==
|
||||||
|
|
||||||
ytsr@^3.8.4:
|
ytsr@^3.8.4:
|
||||||
version "3.8.4"
|
version "3.8.4"
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue