#  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/.

from __future__ import print_function, unicode_literals, absolute_import

import sys
import os
import json

import buildconfig

sourcestamp_tmpl = """{buildid}
{comm_repo}/rev/{comm_rev}
{gecko_repo}/rev/{gecko_rev}
"""


def mk_hg_url(repo, revision):
    """
    Return a URL to a specific revision in the given repo.
    """
    return "{}/rev/{}".format(repo, revision)


def gen_treeherder_build_links(output):
    """
    Create a JSON file that is used by Treeherder to display "Built from" links.
    """
    gecko_repo = buildconfig.substs.get('MOZ_GECKO_SOURCE_REPO')
    gecko_rev = buildconfig.substs.get('MOZ_GECKO_SOURCE_CHANGESET')
    comm_repo = buildconfig.substs.get('MOZ_COMM_SOURCE_REPO')
    comm_rev = buildconfig.substs.get('MOZ_COMM_SOURCE_CHANGESET')

    def mk_built_from_line(repo, revision):
        repo_name = repo.split('/')[-1]  # Last component of base URL
        title = "Built from {} revision {}".format(repo_name, revision)
        url = mk_hg_url(repo, revision)
        return dict(title=title,
                    value=revision,
                    url=url)

    built_from = [mk_built_from_line(gecko_repo, gecko_rev),
                  mk_built_from_line(comm_repo, comm_rev)]
    json.dump(built_from, output)


def gen_platformini(output, platform_ini):
    gecko_repo = buildconfig.substs.get('MOZ_GECKO_SOURCE_REPO', '')
    gecko_rev = buildconfig.substs.get('MOZ_GECKO_SOURCE_CHANGESET', '')

    with open(platform_ini, 'r') as fp:
        data = fp.readlines()

    for i in range(len(data)):
        if data[i].startswith('SourceRepository='):
            data[i] = 'SourceRepository=%s\n' % gecko_repo
        elif data[i].startswith('SourceStamp='):
            data[i] = 'SourceStamp=%s\n' % gecko_rev

    with open(platform_ini, 'w') as fp:
        fp.writelines(data)

    output.write('platform.ini updated.\n')


def gen_sourcestamp(output):
    data = dict(buildid=os.environ.get('MOZ_BUILD_DATE', 'unknown'),
        gecko_repo=buildconfig.substs.get('MOZ_GECKO_SOURCE_REPO', None),
        gecko_rev=buildconfig.substs.get('MOZ_GECKO_SOURCE_CHANGESET', None),
        comm_repo=buildconfig.substs.get('MOZ_COMM_SOURCE_REPO', None),
        comm_rev=buildconfig.substs.get('MOZ_COMM_SOURCE_CHANGESET', None))

    output.write(sourcestamp_tmpl.format(**data))


def source_repo_header(output):
    """
    Appends the Gecko source repository information to source-repo.h
    This information should be set in buildconfig.substs by moz.configure
    """
    gecko_repo = buildconfig.substs.get('MOZ_GECKO_SOURCE_REPO', None)
    gecko_rev = buildconfig.substs.get('MOZ_GECKO_SOURCE_CHANGESET', None)
    comm_repo = buildconfig.substs.get('MOZ_COMM_SOURCE_REPO', None)
    comm_rev = buildconfig.substs.get('MOZ_COMM_SOURCE_CHANGESET', None)

    if None in [gecko_repo, gecko_rev, comm_repo, comm_rev]:
        Exception("Source information not found in buildconfig."
                  "Try setting GECKO_HEAD_REPOSITORY and GECKO_HEAD_REV"
                  "as well as MOZ_SOURCE_REPO and MOZ_SOURCE_CHANGESET"
                  "environment variables and running mach configure again.")

    output.write('#define MOZ_GECKO_SOURCE_STAMP {}\n'.format(gecko_rev))
    output.write('#define MOZ_COMM_SOURCE_STAMP {}\n'.format(comm_rev))
    output.write('#define MOZ_SOURCE_STAMP {}\n'.format(comm_rev))

    if buildconfig.substs.get('MOZ_INCLUDE_SOURCE_INFO'):
        gecko_source_url = mk_hg_url(gecko_repo, gecko_rev)
        comm_source_url = mk_hg_url(comm_repo, comm_rev)
        output.write('#define MOZ_GECKO_SOURCE_REPO {}\n'.format(gecko_repo))
        output.write('#define MOZ_GECKO_SOURCE_URL {}\n'.format(gecko_source_url))
        output.write('#define MOZ_COMM_SOURCE_REPO {}\n'.format(comm_repo))
        output.write('#define MOZ_COMM_SOURCE_URL {}\n'.format(comm_source_url))
        output.write('#define MOZ_SOURCE_REPO {}\n'.format(comm_repo))
        output.write('#define MOZ_SOURCE_URL {}\n'.format(comm_source_url))


def main(args):
    if args:
        func = globals().get(args[0])
        if func:
            return func(sys.stdout, *args[1:])

    return 1


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