Update index.js

index.js

@@ -45,6 +45,90 @@ app.post('/imagetopdf', async (req, res) => {
 	}
 })
 
+app.all('/yt', async (req, res) => {
+	if (!['GET', 'POST'].includes(req.method)) return res.status(405).json({ success: false, message: 'Method Not Allowed' })
+	
+	try {
+		const url = req.method !== 'GET' ? req.body.url : req.query.url
+		if (!url) return res.json({ success: false, message: 'Input parameter url' })
+		if (!/^https?:\/\//.test(url)) return res.json({ success: false, message: 'Invalid url' })
+		
+		const response = await fetch(`https://srvcdn3.2convert.me/api/json?url=${url}`, { headers: { referer: 'https://y2mate.is/' }})
+		const json = await response.json()
+		if (json.error) return res.json({ success: false, message: json.message || json.errorMessage })
+		
+		const { duration } = json.formats
+		const result = {
+			...json.formats,
+			lengthSeconds: +duration,
+			duration: new Date(duration * 1000).toUTCString().split(' ')[4],
+			video: {},
+			audio: {}
+		}
+		
+		for (const type of ['video', 'audio']) {
+			json.formats[type].forEach((data) => {
+				const { quality: q, fileType, needConvert, fileSize, url: dlUrl } = data
+				const quality = type === 'video' ? q : `${q}kbps`
+				
+				result[type][quality] = {
+					quality,
+					fileType,
+					fileSizeH: formatSize(fileSize),
+					fileSize,
+					needConvert,
+					url: !needConvert ? dlUrl : `https://${process.env.SPACE_HOST}${req.path}/convert?hash=${dlUrl}`
+				}
+			})
+		}
+		
+		res.json({ success: true, result })
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.status(500).json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
+	}
+})
+
+app.get('/yt/convert', async (req, res) => {
+	try {
+		const { hash } = req.query
+		if (!hash) return res.json({ success: false, message: 'Invalid hash' })
+		
+		const createPayload = (payload) => ({
+			method: 'POST',
+			body: new URLSearchParams(Object.entries(payload)),
+			headers: { referer: 'https://y2mate.is/' }
+		})
+		
+		const reqTaskId = await fetch('https://srvcdn3.2convert.me/api/json', createPayload({ hash }))
+		if (!reqTaskId.ok) return res.json({ success: false, message: 'Hash not found' })
+		
+		const reqTaskJson = await reqTaskId.json()
+		if (reqTaskJson.error) return res.json({ success: false, message: reqTaskJson.message })
+		
+		let downloadUrl
+		while (!downloadUrl) {
+			const { taskId } = reqTaskJson
+			const fetchTaskId = await fetch('https://srvcdn3.2convert.me/api/json/task', createPayload({ taskId }))
+			const fetchTaskJson = await fetchTaskId.json()
+			
+			if (fetchTaskJson.status === 'finished') {
+				downloadUrl = fetchTaskJson.download
+				break
+			}
+			
+			await new Promise(resolve => setTimeout(resolve, 1000))
+		}
+		
+		res.redirect(downloadUrl)
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.status(500).json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
+	}
+})
+
 const PORT = process.env.PORT || 7860
 app.listen(PORT, () => console.log('App running on port', PORT))