pretty codes and LiveStream issues fixed 100%

This commit is contained in:
killer069 2021-09-28 21:05:43 +05:30
parent 9ab641d1f0
commit 20984ce9e9
5 changed files with 120 additions and 96 deletions

View File

@ -1,6 +1,6 @@
import { PassThrough } from 'stream';
import { IncomingMessage } from 'http';
import { StreamType } from '../stream';
import { parseAudioFormats, StreamType } from '../stream';
import { request, request_stream } from '../utils/request';
import { video_info } from '..';
@ -132,6 +132,7 @@ export class Stream {
private cookie: string;
private data_ended: boolean;
private playing_count: number;
private quality: number;
private request: IncomingMessage | null;
constructor(
url: string,
@ -139,10 +140,12 @@ export class Stream {
duration: number,
contentLength: number,
video_url: string,
cookie: string
cookie: string,
quality: number
) {
this.stream = new PassThrough({ highWaterMark: 10 * 1000 * 1000 });
this.url = url;
this.quality = quality;
this.type = type;
this.bytes_count = 0;
this.video_url = video_url;
@ -175,7 +178,8 @@ export class Stream {
private async retry() {
const info = await video_info(this.video_url, { cookie: this.cookie });
this.url = info.format[info.format.length - 1].url;
const audioFormat = parseAudioFormats(info.format);
this.url = audioFormat[this.quality].url;
}
private cleanup() {
@ -221,7 +225,7 @@ export class Stream {
stream.pipe(this.stream, { end: false });
stream.once('error', async (err) => {
this.cleanup()
this.cleanup();
await this.retry();
this.loop();
if (!this.timer) {

View File

@ -1,5 +1,6 @@
import { video_info } from '.';
import { LiveStreaming, Stream } from './classes/LiveStream';
import { Proxy } from './utils/request';
export enum StreamType {
Arbitrary = 'arbitrary',
@ -12,6 +13,7 @@ export enum StreamType {
export interface StreamOptions {
quality?: number;
cookie?: string;
proxy?: Proxy[];
}
export interface InfoData {
@ -25,7 +27,7 @@ export interface InfoData {
video_details: any;
}
function parseAudioFormats(formats: any[]) {
export function parseAudioFormats(formats: any[]) {
const result: any[] = [];
formats.forEach((format) => {
const type = format.mimeType as string;
@ -41,7 +43,7 @@ function parseAudioFormats(formats: any[]) {
export type YouTubeStream = Stream | LiveStreaming;
export async function stream(url: string, options: StreamOptions = {}): Promise<YouTubeStream> {
const info = await video_info(url, { cookie : options.cookie });
const info = await video_info(url, { cookie: options.cookie, proxy: options.proxy });
const final: any[] = [];
if (
info.LiveStreamData.isLive === true &&
@ -70,7 +72,8 @@ export async function stream(url: string, options: StreamOptions = {}): Promise<
info.video_details.durationInSec,
Number(final[0].contentLength),
info.video_details.url,
options.cookie as string
options.cookie as string,
options.quality
);
}
@ -103,6 +106,7 @@ export async function stream_from_info(info: InfoData, options: StreamOptions =
info.video_details.durationInSec,
Number(final[0].contentLength),
info.video_details.url,
options.cookie as string
options.cookie as string,
options.quality
);
}

View File

@ -5,12 +5,12 @@ import { YouTubePlayList } from '../classes/Playlist';
interface InfoOptions {
cookie?: string;
proxy? : Proxy[]
proxy?: Proxy[];
}
interface PlaylistOptions {
incomplete?: boolean;
proxy? : Proxy[]
proxy?: Proxy[];
}
const DEFAULT_API_KEY = 'AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8';

View File

@ -3,15 +3,15 @@ import tls from 'tls';
import http, { ClientRequest, IncomingMessage } from 'http';
import { URL } from 'url';
export type Proxy = ProxyOpts | string
export type Proxy = ProxyOpts | string;
interface ProxyOpts {
host : string,
port : number,
host: string;
port: number;
authentication?: {
username: string;
password: string;
}
};
}
interface ProxyOutput {
@ -23,7 +23,7 @@ interface ProxyOutput {
interface RequestOpts extends RequestOptions {
body?: string;
method?: 'GET' | 'POST';
proxies? : Proxy[]
proxies?: Proxy[];
}
function https_getter(req_url: string, options: RequestOpts = {}): Promise<IncomingMessage> {
@ -47,18 +47,18 @@ function https_getter(req_url: string, options: RequestOpts = {}): Promise<Incom
}
function randomIntFromInterval(min: number, max: number): number {
let x = Math.floor(Math.random() * (max - min + 1) + min)
if(x === 0) return 0
else return x - 1
let x = Math.floor(Math.random() * (max - min + 1) + min);
if (x === 0) return 0;
else return x - 1;
}
async function proxy_getter(req_url: string, req_proxy: Proxy[]): Promise<ProxyOutput> {
return new Promise((resolve, reject) => {
const proxy : string | ProxyOpts = req_proxy[randomIntFromInterval(0, req_proxy.length)]
const parsed_url = new URL(req_url)
let opts : ProxyOpts
const proxy: string | ProxyOpts = req_proxy[randomIntFromInterval(0, req_proxy.length)];
const parsed_url = new URL(req_url);
let opts: ProxyOpts;
if (typeof proxy === 'string') {
const parsed = new URL(proxy)
const parsed = new URL(proxy);
opts = {
host: parsed.hostname,
port: Number(parsed.port),
@ -66,61 +66,66 @@ async function proxy_getter(req_url : string, req_proxy : Proxy[]): Promise<Prox
username: parsed.username,
password: parsed.password
}
}
}
else opts = proxy
let req : ClientRequest
};
} else opts = proxy;
let req: ClientRequest;
if (opts.authentication?.username.length === 0) {
req = http.request({
host: opts.host,
port: opts.port,
method: 'CONNECT',
path: `${parsed_url.host}:443`,
path: `${parsed_url.host}:443`
});
}
else {
} else {
req = http.request({
host: opts.host,
port: opts.port,
method: 'CONNECT',
path: `${parsed_url.host}:443`,
headers: {
"Proxy-Authorization" : `Basic ${Buffer.from(`${opts.authentication?.username}:${opts.authentication?.password}`).toString('base64')}`
'Proxy-Authorization': `Basic ${Buffer.from(
`${opts.authentication?.username}:${opts.authentication?.password}`
).toString('base64')}`
}
});
}
req.on('connect', function (res, socket, head) {
console.log('Connected')
const tlsConnection = tls.connect({
console.log('Connected');
const tlsConnection = tls.connect(
{
host: parsed_url.hostname,
port: 443,
socket: socket,
rejectUnauthorized: false
}, function() {
tlsConnection.write(`GET ${parsed_url.pathname}${parsed_url.search} HTTP/1.1\r\n`
+ `Host : ${parsed_url.hostname}\r\n`
+ 'Connection: close\r\n'
+ '\r\n')
})
},
function () {
tlsConnection.write(
`GET ${parsed_url.pathname}${parsed_url.search} HTTP/1.1\r\n` +
`Host : ${parsed_url.hostname}\r\n` +
'Connection: close\r\n' +
'\r\n'
);
}
);
tlsConnection.setEncoding('utf-8')
let data = ''
tlsConnection.once('error', (e) => reject(e))
tlsConnection.on('data', (c) => data+=c)
tlsConnection.setEncoding('utf-8');
let data = '';
tlsConnection.once('error', (e) => reject(e));
tlsConnection.on('data', (c) => (data += c));
tlsConnection.on('end', () => {
const y = data.split('\r\n\r\n')
const head = y.shift() as string
const y = data.split('\r\n\r\n');
const head = y.shift() as string;
resolve({
statusCode: Number(head.split('\n')[0].split(' ')[1]),
head: head,
body: y.join('\n')
})
})
})
req.on('error', (e : Error) => reject(e))
req.end()
})
});
});
});
req.on('error', (e: Error) => reject(e));
req.end();
});
}
export async function request(url: string, options?: RequestOpts): Promise<string> {
@ -140,19 +145,18 @@ export async function request(url: string, options?: RequestOpts): Promise<strin
res.setEncoding('utf-8');
res.on('data', (c) => (data += c));
res.on('end', () => resolve(data));
}
else {
let res = await proxy_getter(url, options.proxies).catch((e : Error) => e)
} else {
let res = await proxy_getter(url, options.proxies).catch((e: Error) => e);
if (res instanceof Error) {
reject(res);
return;
}
if (res.statusCode >= 300 && res.statusCode < 400) {
res = await proxy_getter(res.head.split('Location: ')[1].split('\n')[0], options.proxies)
res = await proxy_getter(res.head.split('Location: ')[1].split('\n')[0], options.proxies);
} else if (res.statusCode > 400) {
reject(new Error(`GOT ${res.statusCode} from proxy request`))
reject(new Error(`GOT ${res.statusCode} from proxy request`));
}
resolve(res.body)
resolve(res.body);
}
});
}

View File

@ -13,7 +13,16 @@ interface SearchOptions {
import readline from 'readline';
import fs from 'fs';
import { sp_validate, yt_validate, so_validate, YouTubeStream, SoundCloudStream, YouTube, SoundCloud, Spotify } from '.';
import {
sp_validate,
yt_validate,
so_validate,
YouTubeStream,
SoundCloudStream,
YouTube,
SoundCloud,
Spotify
} from '.';
import { SpotifyAuthorize, sp_search } from './Spotify';
import { check_id, so_search, stream as so_stream, stream_from_info as so_stream_info } from './SoundCloud';
import { InfoData, stream as yt_stream, StreamOptions, stream_from_info as yt_stream_info } from './YouTube/stream';
@ -39,13 +48,16 @@ export async function stream(url: string, options: StreamOptions = {}): Promise<
* @param options contains limit and source to choose.
* @returns
*/
export async function search(query: string, options: SearchOptions = {}): Promise<YouTube[] | Spotify[] | SoundCloud[]> {
export async function search(
query: string,
options: SearchOptions = {}
): Promise<YouTube[] | Spotify[] | SoundCloud[]> {
if (!options.source) options.source = { youtube: 'video' };
if (options.source.youtube) return await yt_search(query, { limit: options.limit, type: options.source.youtube });
else if (options.source.spotify) return await sp_search(query, options.source.spotify, options.limit);
else if (options.source.soundcloud) return await so_search(query, options.source.soundcloud, options.limit);
else throw new Error('Not possible to reach Here LOL. Easter Egg of play-dl if someone get this.')
else throw new Error('Not possible to reach Here LOL. Easter Egg of play-dl if someone get this.');
}
/**