Add initial userscript

gdrive-userscript/cytube-google-drive.user.js

@@ -0,0 +1,155 @@
+// ==UserScript==
+// @name Google Drive Video Player for {SITENAME}
+// @namespace gdcytube
+// @description Play Google Drive videos on {SITENAME}
+// {INCLUDE_BLOCK}
+// @grant unsafeWindow
+// @grant GM_xmlhttpRequest
+// @connect docs.google.com
+// @run-at document-end
+// @version 1.0.0
+// ==/UserScript==
+
+(function () {
+    if (!unsafeWindow.enableCyTubeGoogleDriveUserscript) {
+        return;
+    }
+
+    function debug(message) {
+        if (!unsafeWindow.enableCyTubeGoogleDriveUserscriptDebug) {
+            return;
+        }
+
+        unsafeWindow.console.log.apply(unsafeWindow.console, arguments);
+    }
+
+    var ITAG_QMAP = {
+        37: 1080,
+        46: 1080,
+        22: 720,
+        45: 720,
+        59: 480,
+        44: 480,
+        35: 480,
+        18: 360,
+        43: 360,
+        34: 360
+    };
+
+    var ITAG_CMAP = {
+        43: 'video/webm',
+        44: 'video/webm',
+        45: 'video/webm',
+        46: 'video/webm',
+        18: 'video/mp4',
+        22: 'video/mp4',
+        37: 'video/mp4',
+        59: 'video/mp4',
+        35: 'video/flv',
+        34: 'video/flv'
+    };
+
+    function getVideoInfo(id, cb) {
+        var url = 'https://docs.google.com/file/d/' + id + '/get_video_info';
+        debug('Fetching ' + url);
+
+        GM_xmlhttpRequest({
+            method: 'GET',
+            url: url,
+            onload: function (res) {
+                var data = {};
+                res.responseText.split('&').forEach(function (kv) {
+                    var pair = kv.split('=');
+                    data[decodeURIComponent(pair[0])] = decodeURIComponent(pair[1]);
+                });
+
+                if (data.status === 'fail') {
+                    var error = new Error('Google Docs request failed: ' +
+                            'metadata indicated status=fail');
+                    error.response = res.responseText;
+                    error.reason = 'RESPONSE_STATUS_FAIL';
+                    return cb(error);
+                }
+
+                if (!data.fmt_stream_map) {
+                    var error = new Error('Google Docs request failed: ' +
+                            'metadata lookup returned no valid links');
+                    error.response = res.responseText;
+                    error.reason = 'MISSING_LINKS';
+                    return cb(error);
+                }
+
+                data.links = {};
+                data.fmt_stream_map.split(',').forEach(function (item) {
+                    var pair = item.split('|');
+                    data.links[pair[0]] = pair[1];
+                });
+
+                cb(null, data);
+            },
+
+            onerror: function () {
+                var error = new Error('Google Docs request failed: ' +
+                        'metadata lookup HTTP request failed');
+                error.reason = 'HTTP_ONERROR';
+                return cb(error);
+            }
+        });
+    }
+
+    function mapLinks(links) {
+        var videos = {
+            1080: [],
+            720: [],
+            480: [],
+            360: []
+        };
+
+        Object.keys(links).forEach(function (itag) {
+            itag = parseInt(itag, 10);
+            if (!ITAG_QMAP.hasOwnProperty(itag)) {
+                return;
+            }
+
+            videos[ITAG_QMAP[itag]].push({
+                itag: itag,
+                contentType: ITAG_CMAP[itag],
+                link: links[itag]
+            });
+        });
+
+        return videos;
+    }
+
+    function GoogleDrivePlayer(data) {
+        if (!(this instanceof GoogleDrivePlayer)) {
+            return new GoogleDrivePlayer(data);
+        }
+
+        this.setMediaProperties(data);
+        this.load(data);
+    }
+
+    GoogleDrivePlayer.prototype = Object.create(unsafeWindow.VideoJSPlayer.prototype);
+
+    GoogleDrivePlayer.prototype.load = function (data) {
+        var self = this;
+        getVideoInfo(data.id, function (err, videoData) {
+            if (err) {
+                debug(err);
+                var alertBox = unsafeWindow.document.createElement('div');
+                alertBox.className = 'alert alert-danger';
+                alertBox.textContent = err.message;
+                document.getElementById('ytapiplayer').appendChild(alertBox);
+                return;
+            }
+
+            debug('Retrieved links: ' + JSON.stringify(videoData.links));
+            data.meta.direct = mapLinks(videoData.links);
+            unsafeWindow.VideoJSPlayer.prototype.loadPlayer.call(self, data);
+        });
+    };
+
+    unsafeWindow.GoogleDrivePlayer = GoogleDrivePlayer;
+    unsafeWindow.console.log('Initialized userscript Google Drive player');
+})();

gdrive-userscript/generate-userscript.js

@@ -0,0 +1,19 @@
+var fs = require('fs');
+var path = require('path');
+
+var sitename = process.argv[2];
+var includes = process.argv.slice(3).map(function (include) {
+    return '// @include ' + include;
+}).join('\n');
+
+var lines = String(fs.readFileSync(
+        path.resolve(__dirname, 'cytube-google-drive.user.js'))).split('\n');
+lines.forEach(function (line) {
+    if (line.match(/\{INCLUDE_BLOCK\}/)) {
+        console.log(includes);
+    } else if (line.match(/\{SITENAME\}/)) {
+        console.log(line.replace(/\{SITENAME\}/, sitename));
+    } else {
+        console.log(line);
+    }
+});