summaryrefslogtreecommitdiffstats
path: root/toolkit/content/neterror/gen_aboutneterror_codes.py
blob: 806756422f74d432590c2ae95213a59d471971b1 (plain)
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
#!/usr/bin/env python3
# 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/.

import sys

from fluent.syntax import parse
from fluent.syntax.ast import Message


def find_error_ids(filename, known_strings):
    with open(filename, "r", encoding="utf-8") as f:
        known_strings += [
            m.id.name for m in parse(f.read()).body if isinstance(m, Message)
        ]


def main(output, *filenames):
    known_strings = []
    for filename in filenames:
        find_error_ids(filename, known_strings)

    output.write("const KNOWN_ERROR_MESSAGE_IDS = new Set([\n")
    for known_string in known_strings:
        output.write('  "{}",\n'.format(known_string))
    output.write("]);\n")


if __name__ == "__main__":
    sys.exit(main(sys.stdout, *sys.argv[1:]))