Spaces:
Build error
Build error
File size: 2,403 Bytes
0bfe2e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
import { ParsedStream, UserData } from '../db/schemas';
import { createLogger } from '../utils';
import { createProxy } from '../proxy';
const logger = createLogger('proxifier');
class Proxifier {
private userData: UserData;
constructor(userData: UserData) {
this.userData = userData;
}
private shouldProxyStream(stream: ParsedStream): boolean {
const streamService = stream.service ? stream.service.id : 'none';
const proxy = this.userData.proxy;
if (!stream.url || !proxy?.enabled) {
return false;
}
const proxyAddon =
!proxy.proxiedAddons?.length ||
proxy.proxiedAddons.includes(stream.addon.presetInstanceId);
const proxyService =
!proxy.proxiedServices?.length ||
proxy.proxiedServices.includes(streamService);
if (proxy.enabled && proxyAddon && proxyService) {
return true;
}
return false;
}
public async proxify(streams: ParsedStream[]): Promise<ParsedStream[]> {
if (!this.userData.proxy?.enabled) {
return streams;
}
const streamsToProxy = streams
.map((stream, index) => ({ stream, index }))
.filter(({ stream }) => stream.url && this.shouldProxyStream(stream));
if (streamsToProxy.length === 0) {
return streams;
}
logger.info(`Proxying ${streamsToProxy.length} streams`);
const proxy = createProxy(this.userData.proxy);
const proxiedUrls = streamsToProxy.length
? await proxy.generateUrls(
streamsToProxy.map(({ stream }) => ({
url: stream.url!,
filename: stream.filename,
headers: {
request: stream.requestHeaders,
response: stream.responseHeaders,
},
}))
)
: [];
logger.info(`Generated ${(proxiedUrls || []).length} proxied URLs`);
const removeIndexes = new Set<number>();
streamsToProxy.forEach(({ stream, index }, i) => {
const proxiedUrl = proxiedUrls?.[i];
if (proxiedUrl) {
stream.url = proxiedUrl;
stream.proxied = true;
} else {
removeIndexes.add(index);
}
});
if (removeIndexes.size > 0) {
logger.warn(
`Failed to proxy ${removeIndexes.size} streams. Removing them from the list.`
);
streams = streams.filter((_, index) => !removeIndexes.has(index));
}
return streams;
}
}
export default Proxifier;
|