1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/* 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/. */
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
TranslationsParent: "resource://gre/actors/TranslationsParent.sys.mjs",
EngineProcess: "chrome://global/content/ml/EngineProcess.sys.mjs",
});
/**
* The translations engine is in its own content process. This actor handles the
* marshalling of the data such as the engine payload and port passing.
*/
export class TranslationsEngineParent extends JSWindowActorParent {
/**
* Keep track of the live actors by InnerWindowID.
*
* @type {Map<InnerWindowID, TranslationsParent | AboutTranslationsParent>}
*/
#translationsParents = new Map();
async receiveMessage({ name, data }) {
switch (name) {
case "TranslationsEngine:Ready":
if (!lazy.EngineProcess.resolveTranslationsEngineParent) {
throw new Error(
"Unable to find the resolve function for when the translations engine is ready."
);
}
lazy.EngineProcess.resolveTranslationsEngineParent(this);
return undefined;
case "TranslationsEngine:RequestEnginePayload": {
const { fromLanguage, toLanguage } = data;
const payloadPromise =
lazy.TranslationsParent.getTranslationsEnginePayload(
fromLanguage,
toLanguage
);
payloadPromise.catch(error => {
lazy.TranslationsParent.telemetry().onError(String(error));
});
return payloadPromise;
}
case "TranslationsEngine:ReportEngineStatus": {
const { innerWindowId, status } = data;
const translationsParent = this.#translationsParents.get(innerWindowId);
// about:translations will not have a TranslationsParent associated with
// this call.
if (translationsParent) {
switch (status) {
case "ready":
translationsParent.languageState.isEngineReady = true;
break;
case "error":
translationsParent.languageState.error = "engine-load-failure";
break;
default:
throw new Error("Unknown engine status: " + status);
}
}
return undefined;
}
case "TranslationsEngine:DestroyEngineProcess":
lazy.EngineProcess.destroyTranslationsEngine().catch(error =>
console.error(error)
);
return undefined;
default:
return undefined;
}
}
/**
* @param {string} fromLanguage
* @param {string} toLanguage
* @param {MessagePort} port
* @param {number} innerWindowId
* @param {TranslationsParent} [translationsParent]
*/
startTranslation(
fromLanguage,
toLanguage,
port,
innerWindowId,
translationsParent
) {
if (translationsParent) {
this.#translationsParents.set(
translationsParent.innerWindowId,
translationsParent
);
}
if (this.#isDestroyed) {
throw new Error("The translation engine process was already destroyed.");
}
const transferables = [port];
this.sendAsyncMessage(
"TranslationsEngine:StartTranslation",
{
fromLanguage,
toLanguage,
innerWindowId,
port,
},
transferables
);
}
/**
* Remove all the translations that are currently queued, and remove
* the communication port.
*
* @param {number} innerWindowId
*/
discardTranslations(innerWindowId) {
this.#translationsParents.delete(innerWindowId);
this.sendAsyncMessage("TranslationsEngine:DiscardTranslations", {
innerWindowId,
});
}
/**
* Manually shut down the engines, typically for testing purposes.
*/
forceShutdown() {
return this.sendQuery("TranslationsEngine:ForceShutdown");
}
#isDestroyed = false;
didDestroy() {
this.#isDestroyed = true;
}
}
|
