1#!/usr/bin/env python
2#
3# Copyright 2016 Google Inc.
4#
5# Use of this source code is governed by a BSD-style license that can be
6# found in the LICENSE file.
7
8from __future__ import print_function
9
10import collections
11import json
12import os
13import subprocess
14import sys
15
16# Finds all public sources in include directories then write them to skia.h.
17
18# Also write skia.h.deps, which Ninja uses to track dependencies. It's the
19# very same mechanism Ninja uses to know which .h files affect which .cpp files.
20
21gn              = sys.argv[1]
22absolute_source = sys.argv[2]
23skia_h          = sys.argv[3]
24include_dirs    = sys.argv[4:]
25
26absolute_source = os.path.normpath(absolute_source)
27
28include_dirs = [os.path.join(os.path.normpath(include_dir), '')
29                for include_dir in include_dirs]
30include_dirs.sort(key=len, reverse=True)
31
32gn_desc_cmd = [gn, 'desc', '.', '--root=%s' % absolute_source, '--format=json',
33               '*']
34
35desc_json_txt = ''
36try:
37  desc_json_txt = subprocess.check_output(gn_desc_cmd)
38except subprocess.CalledProcessError as e:
39  print(e.output)
40  raise
41
42desc_json = {}
43try:
44  desc_json = json.loads(desc_json_txt)
45except ValueError:
46  print(desc_json_txt)
47  raise
48
49sources = set()
50
51for target in desc_json.values():
52  # We'll use `public` headers if they're listed, or pull them from `sources`
53  # if not.  GN sneaks in a default "public": "*" into the JSON if you don't
54  # set one explicitly.
55  search_list = target.get('public')
56  if search_list == '*':
57    search_list = target.get('sources', [])
58
59  for name in search_list:
60    sources.add(os.path.join(absolute_source, os.path.normpath(name[2:])))
61
62Header = collections.namedtuple('Header', ['absolute', 'include'])
63headers = {}
64for source in sources:
65  source_as_include = [os.path.relpath(source, absolute_source)
66                       for include_dir in include_dirs
67                       if source.startswith(include_dir)]
68  if not source_as_include:
69    continue
70  statinfo = os.stat(source)
71  key = str(statinfo.st_ino) + ':' + str(statinfo.st_dev)
72  # On Windows os.stat st_ino is 0 until 3.3.4 and st_dev is 0 until 3.4.0.
73  if key == '0:0':
74    key = source
75  include_path = source_as_include[0]
76  if key not in headers or len(include_path) < len(headers[key].include):
77    headers[key] = Header(source, include_path)
78
79headers = sorted(headers.values(), key=lambda x: x.include)
80
81with open(skia_h, 'w') as f:
82  f.write('// skia.h generated by GN.\n')
83  f.write('#ifndef skia_h_DEFINED\n')
84  f.write('#define skia_h_DEFINED\n')
85  for header in headers:
86    f.write('#include "' + header.include + '"\n')
87  f.write('#endif//skia_h_DEFINED\n')
88
89with open(skia_h + '.deps', 'w') as f:
90  f.write(skia_h + ':')
91  for header in headers:
92    f.write(' ' + header.absolute)
93  f.write(' build.ninja.d')
94  f.write('\n')
95
96# Temporary: during development this file wrote skia.h.d, not skia.h.deps,
97# and I think we have some bad versions of those files laying around.
98if os.path.exists(skia_h + '.d'):
99  os.remove(skia_h + '.d')
100