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
|
/* 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";
const { CFRMessageProvider } = ChromeUtils.importESModule(
"resource://activity-stream/lib/CFRMessageProvider.sys.mjs"
);
const { OnboardingMessageProvider } = ChromeUtils.import(
"resource://activity-stream/lib/OnboardingMessageProvider.jsm"
);
const { PanelTestProvider } = ChromeUtils.importESModule(
"resource://activity-stream/lib/PanelTestProvider.sys.mjs"
);
const CWD = Services.dirsvc.get("CurWorkD", Ci.nsIFile).path;
const CORPUS_DIR = PathUtils.join(CWD, "corpus");
const CORPUS = [
{
name: "CFRMessageProvider.messages.json",
provider: CFRMessageProvider,
},
{
name: "OnboardingMessageProvider.messages.json",
provider: OnboardingMessageProvider,
},
{
name: "PanelTestProvider.messages.json",
provider: PanelTestProvider,
},
{
name: "PanelTestProvider_toast_notification.messages.json",
provider: PanelTestProvider,
filter: message => message.template === "toast_notification",
},
];
let exit = false;
async function main() {
try {
await IOUtils.makeDirectory(CORPUS_DIR);
for (const entry of CORPUS) {
const { name, provider } = entry;
const filter = entry.filter ?? (() => true);
const messages = await provider.getMessages();
const json = `${JSON.stringify(messages.filter(filter), undefined, 2)}\n`;
const path = PathUtils.join(CORPUS_DIR, name);
await IOUtils.writeUTF8(path, json);
}
} finally {
exit = true;
}
}
main();
// We need to spin the event loop here, otherwise everything goes out of scope.
Services.tm.spinEventLoopUntil(
"extract-test-corpus.js: waiting for completion",
() => exit
);
|
