summaryrefslogtreecommitdiff
path: root/tools/inspector_protocol/ConcatenateProtocols.py
blob: a7cbc992c76e4010d3053f25c63c51f3c28133b0 (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
32
33
34
35
36
37
38
39
#!/usr/bin/env python
# Copyright 2016 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os.path
import sys

try:
    import json
except ImportError:
    import simplejson as json


def main(argv):
    if len(argv) < 1:
        sys.stderr.write("Usage: %s <protocol-1> [<protocol-2> [, <protocol-3>...]] <output-file>\n" % sys.argv[0])
        return 1

    domains = []
    version = None
    for protocol in argv[:-1]:
        file_name = os.path.normpath(protocol)
        if not os.path.isfile(file_name):
            sys.stderr.write("Cannot find %s\n" % file_name)
            return 1
        input_file = open(file_name, "r")
        json_string = input_file.read()
        parsed_json = json.loads(json_string)
        domains += parsed_json["domains"]
        version = parsed_json["version"]

    output_file = open(argv[-1], "w")
    json.dump({"version": version, "domains": domains}, output_file, indent=4, sort_keys=False, separators=(',', ': '))
    output_file.close()


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))