#!/usr/bin/env python
# Copyright (C) 2017 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse
import os
import subprocess
import sys

MAC_BUILD_CONFIGS = {
  'mac_debug': ('is_clang=true', 'is_debug=true'),
  'mac_release': ('is_clang=true', 'is_debug=false'),
  'mac_asan': ('is_clang=true', 'is_debug=false', 'is_asan=true'),
  'mac_tsan': ('is_clang=true', 'is_debug=false', 'is_tsan=true'),
  'mac_ubsan': ('is_clang=true', 'is_debug=false', 'is_ubsan=true'),
}

ANDROID_BUILD_CONFIGS = {
  'android_debug': ('target_os="android"', 'is_clang=true', 'is_debug=true'),
  'android_release': ('target_os="android"', 'is_clang=true', 'is_debug=false'),
  'android_release_incl_heapprofd': ('target_os="android"', 'is_clang=true',
                                     'is_debug=false', 'android_api_level=26'),
  'android_asan': ('target_os="android"', 'is_clang=true', 'is_debug=false',
                   'is_asan=true'),
  'android_lsan': ('target_os="android"', 'is_clang=true', 'is_debug=false',
                   'is_lsan=true'),
}

ANDROID_ARCHS = ('arm', 'arm64')

LINUX_BUILD_CONFIGS = {
  'linux_gcc_debug': ('is_clang=false', 'is_debug=true'),
  'linux_gcc_release': ('is_clang=false', 'is_debug=false'),
  'linux_clang_debug': ('is_clang=true', 'is_debug=true'),
  'linux_clang_release': ('is_clang=true', 'is_debug=false'),
  'linux_asan': ('is_clang=true', 'is_debug=false', 'is_asan=true'),
  'linux_lsan': ('is_clang=true', 'is_debug=false', 'is_lsan=true'),
  'linux_msan': ('is_clang=true', 'is_debug=false', 'is_msan=true'),
  'linux_tsan': ('is_clang=true', 'is_debug=false', 'is_tsan=true'),
  'linux_ubsan': ('is_clang=true', 'is_debug=false', 'is_ubsan=true'),
  'linux_fuzzer': ('is_clang=true', 'is_debug=false', 'is_fuzzer=true',
                   'is_asan=true'),
}

LINUX_ARCHS = ('arm64',)

ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('--host-only', action='store_true', default=False)
  parser.add_argument('--build', default=None)
  args = parser.parse_args()

  configs = {}
  if not args.host_only:
    for config_name, gn_args in ANDROID_BUILD_CONFIGS.iteritems():
      for arch in ANDROID_ARCHS:
        full_config_name = '%s_%s' % (config_name, arch)
        configs[full_config_name] = gn_args + ('target_cpu="%s"' % arch,)
    for config_name, gn_args in LINUX_BUILD_CONFIGS.iteritems():
      if dict(a.split('=') for a in gn_args).get('is_clang', None) == 'true':
        continue
      for arch in LINUX_ARCHS:
        full_config_name = '%s_%s' % (config_name, arch)
        configs[full_config_name] = gn_args + ('target_cpu="%s"' % arch,
                                               'target_os="linux"')

  if sys.platform == 'linux2':
    configs.update(LINUX_BUILD_CONFIGS)
  elif sys.platform == 'darwin':
    configs.update(MAC_BUILD_CONFIGS)
  else:
    assert (False)

  out_base_dir = os.path.join(ROOT_DIR, 'out')
  if not os.path.isdir(out_base_dir):
    os.mkdir(out_base_dir)

  gn = os.path.join(ROOT_DIR, 'tools', 'gn')

  for config_name, gn_args in configs.iteritems():
    print '\n\033[32mBuilding %-20s[%s]\033[0m' % (config_name,
                                                   ','.join(gn_args))
    out_dir = os.path.join(ROOT_DIR, 'out', config_name)
    if not os.path.isdir(out_dir):
      os.mkdir(out_dir)
    gn_cmd = (gn, 'gen', out_dir, '--args=%s' % (' '.join(gn_args)), '--check')
    print ' '.join(gn_cmd)
    subprocess.check_call(gn_cmd, cwd=ROOT_DIR)
    if args.build:
      ninja = os.path.join(ROOT_DIR, 'tools', 'ninja')
      ninja_cmd = (ninja, '-C', '.', args.build)
      subprocess.check_call(ninja_cmd, cwd=out_dir)


if __name__ == '__main__':
  sys.exit(main())