code cleanup in connector and prohibit GET requests

This commit is contained in:
Simon Kornblith 2010-10-03 19:11:48 +00:00
parent 072fc721ca
commit 1640d07776

View File

@ -84,7 +84,7 @@ Zotero.Connector = new function() {
/**
* Decodes application/x-www-form-urlencoded data
*
* @param {String} postData application/x-www-form-urlencoded data, as sent in a POST request
* @param {String} postData application/x-www-form-urlencoded data, as sent in a g request
* @return {Object} data in object form
*/
this.decodeURLEncodedData = function(postData) {
@ -253,7 +253,7 @@ Zotero.Connector.DataListener.prototype._headerFinished = function() {
if(method[1] == "HEAD" || method[1] == "OPTIONS") {
this._requestFinished(Zotero.Connector.generateResponse(200));
} else if(method[1] == "GET") {
this._requestFinished(this._processEndpoint(method[3]));
this._requestFinished(this._processEndpoint("GET", method[3]));
} else if(method[1] == "POST") {
const contentLengthRe = /[\r\n]Content-Length: *([0-9]+)/i;
@ -273,8 +273,7 @@ Zotero.Connector.DataListener.prototype._headerFinished = function() {
}
/*
* checks to see if Content-Length bytes of body have been read and, if they
* have, processes the body
* checks to see if Content-Length bytes of body have been read and, if so, processes the body
*/
Zotero.Connector.DataListener.prototype._bodyData = function() {
if(this.body.length >= this.bodyLength) {
@ -294,21 +293,21 @@ Zotero.Connector.DataListener.prototype._bodyData = function() {
}
// handle envelope
this._processEndpoint(this.body);
this._processEndpoint("POST", this.body);
}
}
/**
* Generates a response based on calling the function associated with the endpoint
*/
Zotero.Connector.DataListener.prototype._processEndpoint = function(postData) {
Zotero.Connector.DataListener.prototype._processEndpoint = function(method, postData) {
try {
var endpoint = new this.endpoint;
var me = this;
endpoint.return = function(code, contentType, arg) {
var sendResponseCallback = function(code, contentType, arg) {
me._requestFinished(Zotero.Connector.generateResponse(code, contentType, arg));
}
endpoint.init(postData ? postData : undefined);
endpoint.init(method, postData ? postData : undefined, sendResponseCallback);
} catch(e) {
Zotero.debug(e);
this._requestFinished(Zotero.Connector.generateResponse(500));
@ -343,40 +342,26 @@ Zotero.Connector.Data = {};
Zotero.Connector.Translate = function() {};
Zotero.Connector.Translate._waitingForSelection = {};
Zotero.Connector.Translate.constructTranslateInstance = function(postData, browser, translate) {
Zotero.Connector.Data[postData["uri"]] = "<html>"+postData["html"]+"</html>";
var ioService = Components.classes["@mozilla.org/network/io-service;1"]
.getService(Components.interfaces.nsIIOService);
var uri = ioService.newURI(postData["uri"], "UTF-8", null);
var pageShowCalled = false;
browser.addEventListener("DOMContentLoaded", function() {
try {
if(browser.contentDocument.location.href == "about:blank") return;
if(pageShowCalled) return;
pageShowCalled = true;
delete Zotero.Connector.Data[postData["uri"]];
browser.contentDocument.cookie = postData["cookie"];
// get translators
translate.setDocument(browser.contentDocument);
translate.getTranslators();
} catch(e) {
Zotero.debug(e);
throw e;
}
}, false);
browser.loadURI("zotero://connector/"+encodeURIComponent(postData["uri"]));
}
/**
* Lists all available translators, including code for translators that should be run on every page
*/
Zotero.Connector.Translate.List = function() {};
Zotero.Connector.Translate.List.prototype = {
"init":function(postData) {
/**
* Gets available translator list
* @param {String} method "GET" or "POST"
* @param {String} data POST data or GET query string
* @param {Function} sendResponseCallback function to send HTTP response
*/
"init":function(method, data, sendResponseCallback) {
if(method != "POST") {
sendResponseCallback(400);
return;
}
var translators = Zotero.Translators.getAllForType("web");
var jsons = [];
for each(var translator in translators) {
@ -384,7 +369,7 @@ Zotero.Connector.Translate.List.prototype = {
for each(var key in ["translatorID", "label", "creator", "target", "priority", "detectXPath"]) {
json[key] = translator[key];
}
json["localExecution"] = translator.browserSupport.indexOf(postData["browser"]) !== -1;
json["localExecution"] = translator.browserSupport.indexOf(data["browser"]) !== -1;
// Do not pass targetless translators that do not support this browser (since that
// would mean passing each page back to Zotero)
@ -393,27 +378,69 @@ Zotero.Connector.Translate.List.prototype = {
}
}
this.return(200, "application/json", JSON.stringify(jsons));
sendResponseCallback(200, "application/json", JSON.stringify(jsons));
}
}
/**
* Checks whether there is a translator available to handle the current page
* Detects whether there is an available translator to handle a given page
*/
Zotero.Connector.Translate.Detect = function() {};
Zotero.Connector.Translate.Detect.prototype = {
"init":function(postData) {
var postData = JSON.parse(postData);
// get data into a browser
var translate = new Zotero.Translate("web");
var me = this;
translate.setHandler("translators", function(obj, item) { me._translatorsAvailable(obj, item) });
this._browser = Zotero.Browser.createHiddenBrowser();
Zotero.Connector.Translate.constructTranslateInstance(postData, this._browser, translate);
Zotero.Connector.Translate.Detect.prototype = {
/**
* Loads HTML into a hidden browser and initiates translator detection
* @param {String} method "GET" or "POST"
* @param {String} data POST data or GET query string
* @param {Function} sendResponseCallback function to send HTTP response
*/
"init":function(method, data, sendResponseCallback) {
if(method != "POST") {
sendResponseCallback(400);
return;
}
this.sendResponse = sendResponseCallback;
this._parsedPostData = JSON.parse(data);
this._translate = new Zotero.Translate("web");
this._translate.setHandler("translators", function(obj, item) { me._translatorsAvailable(obj, item) });
Zotero.Connector.Data[this._parsedPostData["uri"]] = "<html>"+this._parsedPostData["html"]+"</html>";
this._browser = Zotero.Browser.createHiddenBrowser();
var ioService = Components.classes["@mozilla.org/network/io-service;1"]
.getService(Components.interfaces.nsIIOService);
var uri = ioService.newURI(this._parsedPostData["uri"], "UTF-8", null);
var pageShowCalled = false;
var me = this;
this._browser.addEventListener("DOMContentLoaded", function() {
try {
if(me._browser.contentDocument.location.href == "about:blank") return;
if(pageShowCalled) return;
pageShowCalled = true;
delete Zotero.Connector.Data[me._parsedPostData["uri"]];
me._browser.contentDocument.cookie = me._parsedPostData["cookie"];
// get translators
me._translate.setDocument(me._browser.contentDocument);
me._translate.getTranslators();
} catch(e) {
Zotero.debug(e);
throw e;
}
}, false);
me._browser.loadURI("zotero://connector/"+encodeURIComponent(this._parsedPostData["uri"]));
},
/**
* Callback to be executed when list of translators becomes available. Sends response with
* item types, translator IDs, labels, and icons for available translators.
* @param {Zotero.Translate} translate
* @param {Zotero.Translator[]} translators
*/
"_translatorsAvailable":function(obj, translators) {
var jsons = [];
for each(var translator in translators) {
@ -427,62 +454,34 @@ Zotero.Connector.Translate.Detect.prototype = {
"label":translator.label, "icon":icon}
jsons.push(json);
}
this.return(200, "application/json", JSON.stringify(jsons));
this.sendResponse(200, "application/json", JSON.stringify(jsons));
Zotero.Browser.deleteHiddenBrowser(this._browser);
}
}
/**
* Perform translation
* Performs translation of a given page
*/
Zotero.Connector.Translate.Save = function() {};
Zotero.Connector.Translate.Save.prototype = {
"init":function(postData) {
var postData = JSON.parse(postData);
// get data into a browser
this._uri = postData.url;
this._browser = Zotero.Browser.createHiddenBrowser();
var translate = new Zotero.Translate("web");
var me = this;
var win = Components.classes["@mozilla.org/appshell/window-mediator;1"]
.getService(Components.interfaces.nsIWindowMediator)
.getMostRecentWindow("navigator:browser");
var progressWindow = win.Zotero_Browser.progress;
if(Zotero.locked) {
progressWindow.changeHeadline(Zotero.getString("ingester.scrapeError"));
var desc = Zotero.localeJoin([
Zotero.getString('general.operationInProgress'), Zotero.getString('general.operationInProgress.waitUntilFinishedAndTryAgain')
]);
progressWindow.addDescription(desc);
progressWindow.show();
progressWindow.startCloseTimer(8000);
return;
}
progressWindow.show();
this._libraryID = null;
var collection = null;
try {
this._libraryID = win.ZoteroPane.getSelectedLibraryID();
collection = win.ZoteroPane.getSelectedCollection();
} catch(e) {}
translate.setHandler("select", function(obj, item) { return me._selectItems(obj, item, progressWindow) });
translate.setHandler("itemDone", function(obj, item) { win.Zotero_Browser.itemDone(obj, item, collection) });
translate.setHandler("done", function(obj, item) { win.Zotero_Browser.finishScraping(obj, item, collection); me.return(201); })
translate.setHandler("translators", function(obj, item) { me._translatorsAvailable(obj, item, postData.translatorID) });
Zotero.Connector.Translate.constructTranslateInstance(postData, this._browser, translate);
},
"_selectItems":function(translate, itemList, progressWindow) {
/**
* Init method inherited from Zotero.Connector.Translate.Detect
* @borrows Zotero.Connector.Translate.Detect as this.init
*/
"init":Zotero.Connector.Translate.Detect.prototype.init,
/**
* Callback to be executed when items must be selected
* @param {Zotero.Translate} translate
* @param {Object} itemList ID=>text pairs representing available items
*/
"_selectItems":function(translate, itemList) {
var instanceID = Zotero.randomString();
Zotero.Connector.Translate._waitingForSelection[instanceID] = this;
// Send "Multiple Choices" HTTP response
this.return(300, "application/json", JSON.stringify({"items":itemList, "instanceID":instanceID, "uri":this._uri}));
this.sendResponse(300, "application/json", JSON.stringify({"items":itemList, "instanceID":instanceID, "uri":this._parsedPostData.uri}));
// We need this to make sure that we won't stop Firefox from quitting, even if the user
// didn't close the selectItems window
@ -500,17 +499,58 @@ Zotero.Connector.Translate.Save.prototype = {
}
observerService.removeObserver(quitObserver, "quit-application");
if(!this.selectedItems) progressWindow.close();
if(!this.selectedItems) this._progressWindow.close();
return this.selectedItems;
},
"_translatorsAvailable":function(translate, translators, translatorID) {
if(translators.length) {
translate.setTranslator(translatorID);
translate.translate(this._libraryID);
} else {
/**
* Callback to be executed when list of translators becomes available. Opens progress window,
* selects specified translator, and initiates translation.
* @param {Zotero.Translate} translate
* @param {Zotero.Translator[]} translators
*/
"_translatorsAvailable":function(translate, translators) {
// make sure translatorsAvailable succeded
if(!translators.length) {
Zotero.Browser.deleteHiddenBrowser(this._browser);
this.return(500);
this.sendResponse(500);
return;
}
// set up progress window
var win = Components.classes["@mozilla.org/appshell/window-mediator;1"]
.getService(Components.interfaces.nsIWindowMediator)
.getMostRecentWindow("navigator:browser");
this._progressWindow = win.Zotero_Browser.progress;
if(Zotero.locked) {
this._progressWindow.changeHeadline(Zotero.getString("ingester.scrapeError"));
var desc = Zotero.localeJoin([
Zotero.getString('general.operationInProgress'), Zotero.getString('general.operationInProgress.waitUntilFinishedAndTryAgain')
]);
pthis._rogressWindow.addDescription(desc);
this._progressWindow.show();
this._progressWindow.startCloseTimer(8000);
return;
}
this._progressWindow.show();
// set save callbacks
this._libraryID = null;
var collection = null;
try {
this._libraryID = win.ZoteroPane.getSelectedLibraryID();
collection = win.ZoteroPane.getSelectedCollection();
} catch(e) {}
var me = this;
translate.setHandler("select", function(obj, item) { return me._selectItems(obj, item) });
translate.setHandler("itemDone", function(obj, item) { win.Zotero_Browser.itemDone(obj, item, collection) });
translate.setHandler("done", function(obj, item) { win.Zotero_Browser.finishScraping(obj, item, collection); me.sendResponse(201); })
// set translator and translate
translate.setTranslator(this._parsedPostData.translatorID);
translate.translate(this._libraryID);
}
}
@ -519,11 +559,21 @@ Zotero.Connector.Translate.Save.prototype = {
*/
Zotero.Connector.Translate.Select = function() {};
Zotero.Connector.Translate.Select.prototype = {
"init":function(postData) {
Zotero.debug(postData);
/**
* Finishes up translation when item selection is complete
* @param {String} method "GET" or "POST"
* @param {String} data POST data or GET query string
* @param {Function} sendResponseCallback function to send HTTP response
*/
"init":function(method, postData, sendResponseCallback) {
if(method != "POST") {
sendResponseCallback(400);
return;
}
var postData = JSON.parse(postData);
var saveInstance = Zotero.Connector.Translate._waitingForSelection[postData.instanceID];
saveInstance.return = this.return;
saveInstance.sendResponse = sendResponseCallback;
saveInstance.selectedItems = false;
for(var i in postData.items) {
@ -535,6 +585,14 @@ Zotero.Connector.Translate.Select.prototype = {
/**
* Endpoints for the Connector HTTP server
*
* Each endpoint should take the form of an object. The init() method of this object will be passed:
* method - the method of the request ("GET" or "POST")
* data - the query string (for a "GET" request) or POST data (for a "POST" request)
* sendResponseCallback - a function to send a response to the HTTP request. This can be passed
* a response code alone (e.g., sendResponseCallback(404)) or a response
* code, MIME type, and response body
* (e.g., sendResponseCallback(200, "text/plain", "Hello World!"))
*/
Zotero.Connector.Endpoints = {
"/translate/list":Zotero.Connector.Translate.List,