const static = require("node-static"); const app = new static.Server("./app"); const port = Number(process.argv[2]) || process.env.PORT || 8080; let tickets = []; const cmd = "youtube-dl"; function escape(arg) { return `'${arg.replace(/'/g, `'\\''`)}'`; } function searchYoutube(q, limit, response) { response.setHeader("Content-Type", "text/plain"); // necessary for firefox to read by chunks console.log("YouTube searching", q, limit); q = escape(`ytsearch${limit}:${q}`); const command = `set -o pipefail; ${cmd} -j ${q} | jq "{id,title}" | jq -s .`; require("child_process").exec(command, {shell:"/bin/bash"}, (error, stdout, stderr) => { if (error) { console.log("error", error); response.writeHead(500); response.end(error.message); } else { response.end(stdout); } }); } function downloadYoutube(id, response) { response.setHeader("Content-Type", "text/plain"); // necessary for firefox to read by chunks console.log("YouTube downloading", id); let args = [ "-f", "bestaudio", "-o", `${__dirname}/_youtube/%(title)s-%(id)s.%(ext)s`, id ] let child = require("child_process").spawn(cmd, args); child.stdout.setEncoding("utf8").on("data", chunk => response.write(chunk)); child.stderr.setEncoding("utf8").on("data", chunk => response.write(chunk)); child.on("error", error => { console.log(error); response.writeHead(500); response.end(error.message); }); child.on("close", code => { if (code != 0) { // fixme } response.end(); }); } function handleYoutubeSearch(url, response) { let q = url.searchParams.get("q"); let limit = url.searchParams.get("limit") || ""; if (q) { searchYoutube(q, limit, response); } else { response.writeHead(404); response.end(); } } function handleYoutubeDownload(request, response) { let str = ""; request.setEncoding("utf8"); request.on("data", chunk => str += chunk); request.on("end", () => { let id = require("querystring").parse(str)["id"]; if (id) { downloadYoutube(id, response); } else { response.writeHead(404); response.end(); } }); } function handleTicket(request, response) { request.resume().on("end", () => { let ticket = require("crypto").randomBytes(16).toString("hex"); tickets.push(ticket); if (tickets.length > 10) { tickets.shift(); } let data = {ticket}; response.setHeader("Content-Type", "application/json"); response.end(JSON.stringify(data)); }); } function onRequest(request, response) { const url = new URL(request.url, "http://localhost"); switch (true) { case request.method == "GET" && url.pathname == "/youtube": return handleYoutubeSearch(url, response); case request.method == "POST" && url.pathname == "/youtube": return handleYoutubeDownload(request, response); case request.method == "POST" && url.pathname == "/ticket": return handleTicket(request, response); default: return request.on("end", () => app.serve(request, response)).resume(); } } function requestValidator(request) { let ticket = request.resourceURL.query["ticket"]; let index = tickets.indexOf(ticket); if (index > -1) { tickets.splice(index, 1); return true; } else { return false; } } let httpServer = require("http").createServer(onRequest).listen(port); let passwords = {}; try { passwords = require("./passwords.json"); console.log("loaded passwords.json file"); } catch (e) { console.log("no passwords.json found"); } require("ws2mpd").ws2mpd(httpServer, requestValidator, passwords); require("ws2mpd").logging(false);