#!/usr/bin/env python3
# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Generates compile DBs that are more amenable for clang tooling."""

import argparse
import json
import os
import sys

script_dir = os.path.dirname(os.path.realpath(__file__))
tool_dir = os.path.abspath(os.path.join(script_dir, '../pylib'))
sys.path.insert(0, tool_dir)

from clang import compile_db


def main(argv):
  parser = argparse.ArgumentParser()
  parser.add_argument(
      '--filter_arg',
      nargs='*',
      help='Additional argument(s) to filter out from compilation database.')
  parser.add_argument(
      '-o',
      help='File to write the compilation database to. Defaults to stdout')
  parser.add_argument('-p', required=True, help='Path to build directory')
  parser.add_argument(
      '--target_os',
      choices=[
          'android',
          'chromeos',
          'fuchsia',
          'ios',
          'linux',
          'mac',
          'nacl',
          'win',
      ],
      help='Target OS - see `gn help target_os`. Set to "win" when ' +
      'cross-compiling Windows from Linux or another host')
  parser.add_argument('targets',
                      nargs='*',
                      help='Additional targets to pass to ninja')

  args = parser.parse_args()

  compdb_text = json.dumps(compile_db.ProcessCompileDatabase(
      compile_db.GenerateWithNinja(args.p, args.targets), args.filter_arg,
      args.target_os),
                           indent=2)
  if args.o is None:
    print(compdb_text)
  else:
    with open(args.o, 'w') as f:
      f.write(compdb_text)


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