58 lines
2.8 KiB
TypeScript
58 lines
2.8 KiB
TypeScript
import { QueryType } from "../types/types";
|
|
import play from "play-dl";
|
|
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
|
|
|
|
// scary things below *sigh*
|
|
const spotifySongRegex = /https?:\/\/(?:embed\.|open\.)(?:spotify\.com\/)(?:track\/|\?uri=spotify:track:)((\w|-){22})/;
|
|
const spotifyPlaylistRegex = /https?:\/\/(?:embed\.|open\.)(?:spotify\.com\/)(?:playlist\/|\?uri=spotify:playlist:)((\w|-){22})/;
|
|
const spotifyAlbumRegex = /https?:\/\/(?:embed\.|open\.)(?:spotify\.com\/)(?:album\/|\?uri=spotify:album:)((\w|-){22})/;
|
|
const vimeoRegex = /(http|https)?:\/\/(www\.|player\.)?vimeo\.com\/(?:channels\/(?:\w+\/)?|groups\/([^/]*)\/videos\/|video\/|)(\d+)(?:|\/\?)/;
|
|
const facebookRegex = /(https?:\/\/)(www\.|m\.)?(facebook|fb).com\/.*\/videos\/.*/;
|
|
const reverbnationRegex = /https:\/\/(www.)?reverbnation.com\/(.+)\/song\/(.+)/;
|
|
const attachmentRegex =
|
|
/^(?:(?:https?|ftp):\/\/)?(?:(?!(?:10|127)(?:\.\d{1,3}){3})(?!(?:169\.254|192\.168)(?:\.\d{1,3}){2})(?!172\.(?:1[6-9]|2\d|3[0-1])(?:\.\d{1,3}){2})(?:[1-9]\d?|1\d\d|2[01]\d|22[0-3])(?:\.(?:1?\d{1,2}|2[0-4]\d|25[0-5])){2}(?:\.(?:[1-9]\d?|1\d\d|2[0-4]\d|25[0-4]))|(?:(?:[a-z\u00a1-\uffff0-9]-*)*[a-z\u00a1-\uffff0-9]+)(?:\.(?:[a-z\u00a1-\uffff0-9]-*)*[a-z\u00a1-\uffff0-9]+)*(?:\.(?:[a-z\u00a1-\uffff]{2,})))(?::\d{2,5})?(?:\/\S*)?$/;
|
|
// scary things above *sigh*
|
|
|
|
class QueryResolver {
|
|
/**
|
|
* Query resolver
|
|
*/
|
|
private constructor() {} // eslint-disable-line @typescript-eslint/no-empty-function
|
|
|
|
/**
|
|
* Resolves the given search query
|
|
* @param {string} query The query
|
|
* @returns {QueryType}
|
|
*/
|
|
static async resolve(query: string): Promise<QueryType> {
|
|
if ((await play.so_validate(query)) === "track") return QueryType.SOUNDCLOUD_TRACK;
|
|
if ((await play.so_validate(query)) === "playlist" || query.includes("/sets/")) return QueryType.SOUNDCLOUD_PLAYLIST;
|
|
if (play.yt_validate(query) === "playlist") return QueryType.YOUTUBE_PLAYLIST;
|
|
if (play.yt_validate(query) === "video") return QueryType.YOUTUBE_VIDEO;
|
|
if (spotifySongRegex.test(query)) return QueryType.SPOTIFY_SONG;
|
|
if (spotifyPlaylistRegex.test(query)) return QueryType.SPOTIFY_PLAYLIST;
|
|
if (spotifyAlbumRegex.test(query)) return QueryType.SPOTIFY_ALBUM;
|
|
if (vimeoRegex.test(query)) return QueryType.VIMEO;
|
|
if (facebookRegex.test(query)) return QueryType.FACEBOOK;
|
|
if (reverbnationRegex.test(query)) return QueryType.REVERBNATION;
|
|
if (attachmentRegex.test(query)) return QueryType.ARBITRARY;
|
|
|
|
return QueryType.YOUTUBE_SEARCH;
|
|
}
|
|
|
|
/**
|
|
* Parses vimeo id from url
|
|
* @param {string} query The query
|
|
* @returns {string}
|
|
*/
|
|
static async getVimeoID(query: string): Promise<string> {
|
|
return (await QueryResolver.resolve(query)) === QueryType.VIMEO
|
|
? query
|
|
.split("/")
|
|
.filter((x) => !!x)
|
|
.pop()
|
|
: null;
|
|
}
|
|
}
|
|
|
|
export { QueryResolver };
|