Example that replaces youtube video pages with a custom page that is …

…capable of streaming the video through the proxy. Relates to #103 and #162
nfriedly · Mar 3, 2021 · ba09b57 · ba09b57
1 parent f456dc4
commit ba09b57
Show file tree

Hide file tree

Showing 5 changed files with 210 additions and 1 deletion.
diff --git a/examples/youtube/package-lock.json b/examples/youtube/package-lock.json
diff --git a/examples/youtube/package.json b/examples/youtube/package.json
@@ -0,0 +1,12 @@
+{
+  "name": "unblocker-youtube-example",
+  "private": true,
+  "version": "0.0.0",
+  "dependencies": {
+    "unblocker": "*",
+    "ytdl-core": "^4.5.0"
+  },
+  "scripts": {
+    "start": "node server.js"
+  }
+}
diff --git a/examples/youtube/server.js b/examples/youtube/server.js
@@ -0,0 +1,34 @@
+"use strict";
+
+var http = require("http");
+var Unblocker = require("unblocker");
+var youtube = require('./youtube.js');
+
+var unblocker = Unblocker({
+  requestMiddleware: [youtube.processRequest],
+});
+
+http
+  .createServer(function(req, res) {
+    // first let unblocker try to handle the requests
+    unblocker(req, res, function(err) {
+      // this callback will be fired for any request that unblocker does not serve
+      var headers = { "content-type": "text/html" };
+      if (err) {
+        res.writeHead(500, headers);
+        return res.end(err.stack || err);
+      }
+      if (req.url == "/") {
+        res.writeHead(200, headers);
+        return res.end(
+          'Visit a link such as <a href="/proxy/https://www.youtube.com/watch?v=dQw4w9WgXcQ"><script>document.write(window.location)</script>proxy/https://www.youtube.com/watch?v=dQw4w9WgXcQ</a> to see the magic.'
+        );
+      } else {
+        res.writeHead(404, headers);
+        return res.end("Error 404: file not found.");
+      }
+    });
+  })
+  .listen(8080);
+
+console.log("proxy server live at http://localhost:8080/");
diff --git a/examples/youtube/youtube.js b/examples/youtube/youtube.js
@@ -0,0 +1,47 @@
+"use strict";
+
+const ytdl = require('ytdl-core');
+
+// handles youtube video pages and replaces them with a custom page that just streams the video
+function processRequest(data) {
+    const { hostname, pathname } = new URL(data.url);
+    if (hostname === 'www.youtube.com' && pathname === "/watch") {
+        const res = data.clientResponse;
+        // if we write headers, unblocker will detect that and stop trying to process this request
+        res.writeHead(200, {"content-type": "text/html; charset=utf-8"});
+
+
+        // todo: use config.prefix instead of hard-coding '/proxy/' into the url
+        ytdl.getInfo(data.url).then(info => {
+
+            // only use formats with combined audio and video (note these tend to be lower resolution)
+            const formats = ytdl.filterFormats(info.formats, 'audioandvideo');
+
+            const thumb = info.videoDetails.thumbnails.pop();
+
+            res.end(
+`<!DOCTYPE html>
+<head>
+<title>${info.videoDetails.title}</title>
+<meta name="ROBOTS" content="NOINDEX, NOFOLLOW"/>
+</head>
+<body>
+<h1>${info.videoDetails.title}</h1>
+<video controls poster="${thumb.url}" style="width: 100%">
+${formats.map(format => `  <source type="${format.mimeType.split(';').shift()}" src="/proxy/${format.url.replace(/&/g, '&amp;')}">`).join('\n')}
+</video>
+<p>${info.videoDetails.description.replace(/[\n]/g, '\n<br>')}</p>
+</body>
+</html>
+`);
+        }).catch(err => {
+            console.error(`Error getting info for ${data.url}`, err);
+            res.end('Error retrieving video info');
+        })
+
+    }
+}
+
+module.exports = {
+    processRequest,
+}
diff --git a/readme.md b/readme.md
@@ -30,7 +30,9 @@ near term.
 
 Additionally, websockets are not currently supported. However, some websocket libraries, such as socket.io and engine.io will start with or fall back to long-poling automatically, which _is_ supported.
 
-More advanced websites, such as Roblox, Discord, Youtube, Instagram, etc. do not currently work. At the moment, there is no timeframe for when these might be supported.
+More advanced websites, such as Roblox, Discord, YouTube*, Instagram, etc. do not currently work. At the moment, there is no timeframe for when these might be supported.
+
+* There is an example that detects YouTube video pages and [replaces them with a custom page that just streams the video](examples/youtube/).
 
 Patches are welcome, including both general-purpose improvements to go into the main library, and site-specific
 fixes to go in the examples folder.