build.py

#!/usr/bin/env python2

# tz.js - Library for working with timezones in JavaScript

# Written in 2011 by L. David Baron <dbaron@dbaron.org>

# To the extent possible under law, the author(s) have dedicated all
# copyright and related and neighboring rights to this software to the
# public domain worldwide.  This software is distributed without any
# warranty.
#
# You should have received a copy of the CC0 Public Domain Dedication
# along with this software.  If not, see
# <http://creativecommons.org/publicdomain/zero/1.0/>.

import os
import os.path
import subprocess
import gzip
import imp
import re
import shutil
import tempfile
import json

INPUT_DIR = os.path.dirname(os.path.abspath(__file__))
OUTPUT_DIR = os.path.join(INPUT_DIR, "output")

if not os.path.exists(OUTPUT_DIR):
    os.mkdir(OUTPUT_DIR)

version_proc = subprocess.Popen(["git", "log", "--format=oneline", "HEAD"],
                                 cwd=INPUT_DIR,
                                 stdout=subprocess.PIPE)
version = version_proc.stdout.read(40)
version_proc.stdout.close()

version_proc = subprocess.Popen(["git", "describe", "--tags", "--exact-match"],
                                 cwd=INPUT_DIR,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE)
(describe_stdout, describe_stderr) = version_proc.communicate()
if version_proc.returncode == 0:
    # Use the tag as primary description and the hash in parentheses.
    version = "{0} ({1})".format(describe_stdout.rstrip("\n"), version)

# If this looks like a github repository, compute the HTTP URL from the
# git URL and include that as part of the version.
version_proc = subprocess.Popen(["git", "remote", "show", "-n", "origin"],
                                 cwd=INPUT_DIR,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE)
(describe_stdout, describe_stderr) = version_proc.communicate()
if version_proc.returncode == 0:
    fetch_re = re.compile("^  Fetch URL: git@github.com:(.*)\.git$",
                          re.MULTILINE)
    m = fetch_re.search(describe_stdout)
    if m is not None:
        url = "https://github.com/" + m.groups()[0]
        version = "{0} {1}".format(url, version)

get_tz_module = imp.load_source("get_tz",
                                os.path.join(INPUT_DIR,
                                             "get-latest-tz-release.py"))
tzversions = get_tz_module.get_latest_tz_release()

tztempdir = tempfile.mkdtemp()

build_tz_module = imp.load_source("build_tz",
                                  os.path.join(INPUT_DIR,
                                               "build-tz.py"))
build_tz_module.build_tz(tztempdir,
                         "tzcode" + tzversions["tzcode"] + ".tar.gz",
                         "tzdata" + tzversions["tzdata"] + ".tar.gz")

zoneinfo_dir = os.path.join(tztempdir, "output", "usr", "share", "zoneinfo")
zdump_command = os.path.join(tztempdir, "output", "usr", "bin", "zdump")

tz_json_module = imp.load_source("compiled_to_json",
                                 os.path.join(INPUT_DIR,
                                              "compiled-to-json.py"))
tz_json = tz_json_module.json_zones(zoneinfo_dir)

source_dir = os.path.join(tztempdir, "source")
link_json_module = imp.load_source("links_to_json",
                                   os.path.join(INPUT_DIR,
                                                "links-to-json.py"))
link_json = link_json_module.json_zones(source_dir)

tz_js_in = open(os.path.join(INPUT_DIR, "tz.js.in"), "rb")
tz_js_source = tz_js_in.read()
tz_js_in.close()

# Most Links are included in zone.tab (though some are not).  So let's
# delete them from the JSON.  (It saves about 10% uncompressed size of
# tz.js, although only 5% compressed.)
tzs = json.loads(tz_json)
for zone in json.loads(link_json):
    if zone in tzs:
        del tzs[zone]
tz_json = json.dumps(tzs, sort_keys=True)

tz_js_source = tz_js_source.replace("@@VERSION@@", version)
tz_js_source = tz_js_source.replace("@@TZDATA_VERSION@@", tzversions["tzdata"])
tz_js_source = tz_js_source.replace("@@LINKS@@", link_json)
tz_js_source = tz_js_source.replace("@@ZONES@@", tz_json)

tz_js_gz = gzip.open(os.path.join(OUTPUT_DIR, "tz.js.gz"), "wb")
tz_js_gz.write(tz_js_source)
tz_js_gz.close()

tz_js = open(os.path.join(OUTPUT_DIR, "tz.js"), "wb")
tz_js.write(tz_js_source)
tz_js.close()

test_input_file = os.path.join(INPUT_DIR, "build-tests.py")
test_output_file = os.path.join(OUTPUT_DIR, "test-tz.js")
if (not os.path.exists(test_output_file)) or \
   os.stat(test_input_file).st_mtime > os.stat(test_output_file).st_mtime:
    tz_tests_module = imp.load_source("tz_test_generator", test_input_file)
    tests_io = open(test_output_file, "wb")
    tz_tests_module.output_tests(zoneinfo_dir, zdump_command, tests_io)
    tests_io.close()

shutil.copy(os.path.join(INPUT_DIR, "test-tz-browser.html"), OUTPUT_DIR)
shutil.copy(os.path.join(INPUT_DIR, "test-tz-node.js"), OUTPUT_DIR)

shutil.rmtree(tztempdir)