Infrastructure/safetwitch
0
Fork 0
mirror of https://codeberg.org/SafeTwitch/safetwitch.git synced 2024-12-22 13:22:58 -05:00
Code Issues Activity

Add new methods and general fixes

Browse source
This commit is contained in:
dragongoose 2023-03-17 21:19:48 -04:00
parent 42e4503383
commit 07668cc092
1 changed files with 21 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
server/routes/proxyRoute.ts
View file

@ -1,9 +1,9 @@
import { Streamlink } from '@dragongoose/streamlink';
import { Socket } from 'dgram';
import { Router, Response, Request, NextFunction } from 'express'
import { TwitchAPI } from '../util/scraping/extractor';
import ws from 'ws';
const proxyRouter = Router();
const twitch = new TwitchAPI()
proxyRouter.get('/img', async (req: Request, res: Response, next: NextFunction) => {
const imageUrl = req.query.imageUrl?.toString()
@ -27,22 +27,11 @@ proxyRouter.get('/img', async (req: Request, res: Response, next: NextFunction)
.catch((err) => next(err))
})
proxyRouter.get('/stream/:username/hls.m3u8', (req: Request, res: Response, next: NextFunction) => {
proxyRouter.get('/stream/:username/hls.m3u8', async (req: Request, res: Response, next: NextFunction) => {
console.log(req.params.username)
const streamlink = new Streamlink(`https://twitch.tv/${req.params.username}`, {
otherArgs: ['--stream-url']
})
streamlink.begin()
streamlink.on('log', async (data) => {
// m3u8 url
let twitchM3u8url = data.toString()
let m3u8Data = await twitch.getStream(req.params.username)
const urlRegex =/(\b(https?|ftp|file):\/\/[-A-Z0-9+&@#\/%?=~_|!:,.;]*[-A-Z0-9+&@#\/%=~_|])/ig;
const twitchRes = await fetch(twitchM3u8url)
let m3u8Data = await twitchRes.text()
const matches = m3u8Data.match(urlRegex)
if (!matches) return next(new Error('Error proxying HLS'));
@ -53,34 +42,22 @@ proxyRouter.get('/stream/:username/hls.m3u8', (req: Request, res: Response, next
res.setHeader('Content-type','application/vnd.apple.mpegurl')
res.send(m3u8Data)
})
})
proxyRouter.get('/hls/:encodedUrl' , (req: Request, res: Response, next: NextFunction) => {
proxyRouter.get('/hls/:encodedUrl' , async (req: Request, res: Response, next: NextFunction) => {
console.log('hi')
const unencodedUrl = Buffer.from(req.params.encodedUrl, 'base64url').toString()
fetch(unencodedUrl).then((response) => {
response.body!.pipeTo(
new WritableStream({
start() {
response.headers.forEach((v, n) => res.setHeader(n, v));
},
write(chunk) {
res.write(chunk);
},
close() {
res.end();
},
})
);
})
.catch((err) => next(err))
const m3u8Fetch = await fetch(unencodedUrl)
var m3u8Data = await m3u8Fetch.text()
res.send(m3u8Data)
})
// IRC PROXY
export const wsServer = new ws.Server({ noServer: true });
wsServer.on('connection', (socket: Socket) => {
wsServer.on('connection', (socket: ws.WebSocket) => {
socket.send('Welcome! Send a comma seperated list to get the IRC')
socket.on('message', message => console.log(message.toString()));
});