88 lines
2.4 KiB
TypeScript
88 lines
2.4 KiB
TypeScript
import { Stream } from 'stremio-addon-sdk';
|
|
import { flag } from 'country-emoji';
|
|
import winston from 'winston';
|
|
import bytes from 'bytes';
|
|
import { Context, UrlResult } from '../types';
|
|
import { Handler } from '../handler';
|
|
|
|
export class StreamResolver {
|
|
private readonly logger: winston.Logger;
|
|
|
|
constructor(logger: winston.Logger) {
|
|
this.logger = logger;
|
|
}
|
|
|
|
readonly resolve = async (ctx: Context, handlers: Handler[], type: string, id: string): Promise<Stream[]> => {
|
|
if (handlers.length === 0) {
|
|
return [{
|
|
name: 'WebStreamr',
|
|
title: '⚠️ No handlers found. Please re-configure the plugin.',
|
|
ytId: 'E4WlUXrJgy4',
|
|
}];
|
|
}
|
|
|
|
const streams: Stream[] = [];
|
|
|
|
const urlResults: UrlResult[] = [];
|
|
const handlerPromises = handlers.map(async (handler) => {
|
|
if (!handler.contentTypes.includes(type)) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const handlerUrlResults = await handler.handle(ctx, type, id);
|
|
this.logger.info(`${handler.id} returned ${handlerUrlResults.length} urls`);
|
|
|
|
urlResults.push(...(handlerUrlResults.filter(handlerUrlResult => handlerUrlResult !== undefined)));
|
|
} catch (err) {
|
|
streams.push({
|
|
name: 'WebStreamr',
|
|
title: `❌ Error with handler "${handler.id}". Please check the logs or create an issue if this persists.`,
|
|
ytId: 'E4WlUXrJgy4',
|
|
});
|
|
|
|
this.logger.error(`${handler.id} error: ` + err);
|
|
}
|
|
});
|
|
await Promise.all(handlerPromises);
|
|
|
|
urlResults.sort((a, b) => {
|
|
const heightComparison = b.height - a.height;
|
|
if (heightComparison !== 0) {
|
|
return heightComparison;
|
|
}
|
|
|
|
return b.bytes - a.bytes;
|
|
});
|
|
|
|
this.logger.info(`Return ${urlResults.length} streams`);
|
|
|
|
streams.push(
|
|
...urlResults.map((urlResult) => {
|
|
let name = 'WebStreamr';
|
|
if (urlResult.height) {
|
|
name += ` ${urlResult.height}p`;
|
|
}
|
|
|
|
let title = urlResult.label;
|
|
if (urlResult.bytes) {
|
|
title += ` | 💾 ${bytes.format(urlResult.bytes, { unitSeparator: ' ' })}`;
|
|
}
|
|
if (urlResult.countryCode) {
|
|
title += ` | ${flag(urlResult.countryCode)}`;
|
|
}
|
|
|
|
return {
|
|
url: urlResult.url.toString(),
|
|
name,
|
|
title,
|
|
behaviourHints: {
|
|
group: `webstreamr-${urlResult.sourceId}`,
|
|
},
|
|
};
|
|
}),
|
|
);
|
|
|
|
return streams;
|
|
};
|
|
}
|