added referers to .torrent grabbing calls #314

Author: bogenpirat

miscapis/background.js

@@ -73,7 +73,7 @@ chrome.extension.onRequest.addListener(function(request, sender, sendResponse) {
 		if(request.server) {
 			server = request.server;
 		}
-		RTA.getTorrent(server, request.url, request.label, request.dir);
+		RTA.getTorrent(server, request.url, request.label, request.dir, request.referer);
 		sendResponse({});
 	} else if(request.action == "getStorageData") {
 		sendResponse(localStorage);

miscapis/content_rta.js

@@ -72,7 +72,9 @@ function registerLinks(response) {
 							showLabelDirChooser(response, url);
 						}
 						else {
-							chrome.extension.sendRequest({"action": "addTorrent", "url": url, "label": undefined, "dir": undefined});
+							var ref = new URL(window.location);
+							ref.hash = '';
+							chrome.extension.sendRequest({"action": "addTorrent", "url": url, "label": undefined, "dir": undefined, "referer": ref.toString()});
 						}
 					}
 				});
@@ -164,7 +166,9 @@ function showLabelDirChooser(settings, url, theServer) {
 		var targetLabel = (inputLabel=="")? ((selectedLabel==null)? "" : selectedLabel) : inputLabel;
 		var targetDir = (inputDir=="")? ((selectedDir==null)? "" : selectedDir) : inputDir;
 
-		chrome.extension.sendRequest({"action": "addTorrent", "url": url, "label": targetLabel, "dir": targetDir, "server": server});
+		var ref = new URL(window.location);
+		ref.hash = '';
+		chrome.extension.sendRequest({"action": "addTorrent", "url": url, "label": targetLabel, "dir": targetDir, "server": server, "referer": ref.toString()});
 
 		setNewSettings(settings, dirlist, labellist, targetDir, targetLabel, serverIndex);
 

miscapis/functions.js

@@ -60,13 +60,17 @@ RTA.dispatchTorrent = function(server, data, name, label, dir) {
 }
 
 
-RTA.getTorrent = function(server, url, label, dir) {
+RTA.getTorrent = function(server, url, label, dir, referer) {
 	if(url.substring(0,7) == "magnet:" || server.rutorrentalwaysurl) {
 		RTA.dispatchTorrent(server, url, "", label, dir);
 	} else {
 		RTA.getTorrentLink = url;
 
-		fetch(url)
+		fetch(url, { 
+			headers: {
+				'Referer': referer
+			}
+		})
 		.then(RTA.handleFetchError)
 		.then(async function(response) {
 			var name = "file.torrent";
@@ -182,7 +186,7 @@ RTA.genericOnClick = function(info, tab) {
 
 	if(serverId === -1) { // send to all servers
 		for(var i in servers) {
-			RTA.getTorrent(servers[i], info.linkUrl);
+			RTA.getTorrent(servers[i], info.linkUrl, null, null, tab.url);
 		}
 	} else { // only one server specified
 		var server = JSON.parse(localStorage.getItem("servers"))[serverId];
@@ -197,7 +201,7 @@ RTA.genericOnClick = function(info, tab) {
 			chrome.tabs.sendRequest(tab.id, {"action": "showLabelDirChooser", "url": info.linkUrl, "settings": localStorage, "server": server});
 		} 
 		else {
-			RTA.getTorrent(server, info.linkUrl);
+			RTA.getTorrent(server, info.linkUrl, null, null, tab.url);
 		}
 	}
 }