summaryrefslogtreecommitdiffstats
path: root/application/basilisk/components/translation/TranslationContentHandler.jsm
diff options
context:
space:
mode:
authorMatt A. Tobin <email@mattatobin.com>2018-02-02 03:32:58 -0500
committerMatt A. Tobin <email@mattatobin.com>2018-02-02 03:32:58 -0500
commite72ef92b5bdc43cd2584198e2e54e951b70299e8 (patch)
tree01ceb4a897c33eca9e7ccf2bc3aefbe530169fe5 /application/basilisk/components/translation/TranslationContentHandler.jsm
parent0d19b77d3eaa5b8d837bf52c19759e68e42a1c4c (diff)
downloadUXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.gz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.lz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.xz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.zip
Add Basilisk
Diffstat (limited to 'application/basilisk/components/translation/TranslationContentHandler.jsm')
-rw-r--r--application/basilisk/components/translation/TranslationContentHandler.jsm181
1 files changed, 181 insertions, 0 deletions
diff --git a/application/basilisk/components/translation/TranslationContentHandler.jsm b/application/basilisk/components/translation/TranslationContentHandler.jsm
new file mode 100644
index 000000000..62376d4aa
--- /dev/null
+++ b/application/basilisk/components/translation/TranslationContentHandler.jsm
@@ -0,0 +1,181 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+"use strict";
+
+this.EXPORTED_SYMBOLS = [ "TranslationContentHandler" ];
+
+const {classes: Cc, interfaces: Ci, utils: Cu} = Components;
+
+Cu.import("resource://gre/modules/Services.jsm");
+Cu.import("resource://gre/modules/XPCOMUtils.jsm");
+XPCOMUtils.defineLazyModuleGetter(this, "LanguageDetector",
+ "resource:///modules/translation/LanguageDetector.jsm");
+
+const STATE_OFFER = 0;
+const STATE_TRANSLATED = 2;
+const STATE_ERROR = 3;
+
+this.TranslationContentHandler = function(global, docShell) {
+ let webProgress = docShell.QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIWebProgress);
+ webProgress.addProgressListener(this, Ci.nsIWebProgress.NOTIFY_STATE_DOCUMENT);
+
+ global.addEventListener("pageshow", this);
+
+ global.addMessageListener("Translation:TranslateDocument", this);
+ global.addMessageListener("Translation:ShowTranslation", this);
+ global.addMessageListener("Translation:ShowOriginal", this);
+ this.global = global;
+}
+
+TranslationContentHandler.prototype = {
+ handleEvent(aEvent) {
+ // We are only listening to pageshow events.
+ let target = aEvent.target;
+
+ // Only handle top-level frames.
+ let win = target.defaultView;
+ if (win.parent !== win)
+ return;
+
+ let content = this.global.content;
+ if (!content.detectedLanguage)
+ return;
+
+ let data = {};
+ let trDoc = content.translationDocument;
+ if (trDoc) {
+ data.state = trDoc.translationError ? STATE_ERROR : STATE_TRANSLATED;
+ data.translatedFrom = trDoc.translatedFrom;
+ data.translatedTo = trDoc.translatedTo;
+ data.originalShown = trDoc.originalShown;
+ } else {
+ data.state = STATE_OFFER;
+ data.originalShown = true;
+ }
+ data.detectedLanguage = content.detectedLanguage;
+
+ this.global.sendAsyncMessage("Translation:DocumentState", data);
+ },
+
+ /* nsIWebProgressListener implementation */
+ onStateChange(aWebProgress, aRequest, aStateFlags, aStatus) {
+ if (!aWebProgress.isTopLevel ||
+ !(aStateFlags & Ci.nsIWebProgressListener.STATE_STOP) ||
+ !this.global.content)
+ return;
+
+ let url = aRequest.name;
+ if (!url.startsWith("http://") && !url.startsWith("https://"))
+ return;
+
+ let content = this.global.content;
+ if (content.detectedLanguage)
+ return;
+
+ // Grab a 60k sample of text from the page.
+ let encoder = Cc["@mozilla.org/layout/documentEncoder;1?type=text/plain"]
+ .createInstance(Ci.nsIDocumentEncoder);
+ encoder.init(content.document, "text/plain", encoder.SkipInvisibleContent);
+ let string = encoder.encodeToStringWithMaxLength(60 * 1024);
+
+ // Language detection isn't reliable on very short strings.
+ if (string.length < 100)
+ return;
+
+ LanguageDetector.detectLanguage(string).then(result => {
+ // Bail if we're not confident.
+ if (!result.confident) {
+ return;
+ }
+
+ // The window might be gone by now.
+ if (Cu.isDeadWrapper(content)) {
+ return;
+ }
+
+ content.detectedLanguage = result.language;
+
+ let data = {
+ state: STATE_OFFER,
+ originalShown: true,
+ detectedLanguage: result.language
+ };
+ this.global.sendAsyncMessage("Translation:DocumentState", data);
+ });
+ },
+
+ // Unused methods.
+ onProgressChange() {},
+ onLocationChange() {},
+ onStatusChange() {},
+ onSecurityChange() {},
+
+ QueryInterface: XPCOMUtils.generateQI([Ci.nsIWebProgressListener,
+ Ci.nsISupportsWeakReference]),
+
+ receiveMessage(msg) {
+ switch (msg.name) {
+ case "Translation:TranslateDocument":
+ {
+ Cu.import("resource:///modules/translation/TranslationDocument.jsm");
+
+ // If a TranslationDocument already exists for this document, it should
+ // be used instead of creating a new one so that we can use the original
+ // content of the page for the new translation instead of the newly
+ // translated text.
+ let translationDocument = this.global.content.translationDocument ||
+ new TranslationDocument(this.global.content.document);
+
+ let preferredEngine = Services.prefs.getCharPref("browser.translation.engine");
+ let translator = null;
+ if (preferredEngine == "yandex") {
+ Cu.import("resource:///modules/translation/YandexTranslator.jsm");
+ translator = new YandexTranslator(translationDocument,
+ msg.data.from,
+ msg.data.to);
+ } else {
+ Cu.import("resource:///modules/translation/BingTranslator.jsm");
+ translator = new BingTranslator(translationDocument,
+ msg.data.from,
+ msg.data.to);
+ }
+
+ this.global.content.translationDocument = translationDocument;
+ translationDocument.translatedFrom = msg.data.from;
+ translationDocument.translatedTo = msg.data.to;
+ translationDocument.translationError = false;
+
+ translator.translate().then(
+ result => {
+ this.global.sendAsyncMessage("Translation:Finished", {
+ characterCount: result.characterCount,
+ from: msg.data.from,
+ to: msg.data.to,
+ success: true
+ });
+ translationDocument.showTranslation();
+ },
+ error => {
+ translationDocument.translationError = true;
+ let data = {success: false};
+ if (error == "unavailable")
+ data.unavailable = true;
+ this.global.sendAsyncMessage("Translation:Finished", data);
+ }
+ );
+ break;
+ }
+
+ case "Translation:ShowOriginal":
+ this.global.content.translationDocument.showOriginal();
+ break;
+
+ case "Translation:ShowTranslation":
+ this.global.content.translationDocument.showTranslation();
+ break;
+ }
+ }
+};