1#!/usr/bin/python
2
3# Copyright (C) 2013 Apple Inc. All rights reserved.
4#
5# Redistribution and use in source and binary forms, with or without
6# modification, are permitted provided that the following conditions
7# are met:
8# 1. Redistributions of source code must retain the above copyright
9#    notice, this list of conditions and the following disclaimer.
10# 2. Redistributions in binary form must reproduce the above copyright
11#    notice, this list of conditions and the following disclaimer in the
12#    documentation and/or other materials provided with the distribution.
13#
14# THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
15# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
16# THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
18# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
24# THE POSSIBILITY OF SUCH DAMAGE.
25
26import glob
27import json
28import os
29import sys
30
31if len(sys.argv) < 2:
32    print("usage: %s [json files or directory of json files ...]" % os.path.basename(sys.argv[0]))
33    sys.exit(1)
34
35files = []
36for arg in sys.argv[1:]:
37    if not os.access(arg, os.F_OK):
38        raise Exception("File \"%s\" not found" % arg)
39    elif os.path.isdir(arg):
40        files.extend(glob.glob(os.path.join(arg, "*.json")))
41    else:
42        files.append(arg)
43files.sort()
44
45# To keep as close to the original JSON formatting as possible, just
46# dump each JSON input file unmodified into an array of "domains".
47# Validate each file is valid JSON and that there is a "domain" key.
48
49first = True
50print("{\"domains\":[")
51for file in files:
52    if first:
53        first = False
54    else:
55        print(",")
56
57    string = open(file).read()
58
59    try:
60        dictionary = json.loads(string)
61        if not "domain" in dictionary:
62            raise Exception("File \"%s\" does not contains a \"domain\" key." % file)
63    except ValueError:
64        sys.stderr.write("File \"%s\" does not contain valid JSON:\n" % file)
65        raise
66
67    print(string.rstrip())
68print("]}")
69
70