1#!/usr/bin/env python
2# Copyright (c) 2012 The Chromium Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6"""
7This script invokes the go build tool.
8Must be called as follows:
9python go.py <go-binary> <build directory> <output file> <src directory>
10<CGO_CFLAGS> <CGO_LDFLAGS> <go-binary options>
11eg.
12python go.py /usr/lib/google-golang/bin/go out/build out/a.out .. "-I."
13"-L. -ltest" test -c test/test.go
14"""
15
16import argparse
17import os
18import shutil
19import sys
20
21def main():
22  parser = argparse.ArgumentParser()
23  parser.add_argument('go_binary')
24  parser.add_argument('build_directory')
25  parser.add_argument('output_file')
26  parser.add_argument('src_root')
27  parser.add_argument('cgo_cflags')
28  parser.add_argument('cgo_ldflags')
29  parser.add_argument('go_option', nargs='*')
30  args = parser.parse_args()
31  go_binary = args.go_binary
32  build_dir = args.build_directory
33  out_file = os.path.abspath(args.output_file)
34  # The src directory specified is relative. We need this as an absolute path.
35  src_root = os.path.abspath(args.src_root)
36  # GOPATH must be absolute, and point to one directory up from |src_Root|
37  go_path = os.path.abspath(os.path.join(src_root, ".."))
38  go_options = args.go_option
39  try:
40    shutil.rmtree(build_dir, True)
41    os.mkdir(build_dir)
42  except:
43    pass
44  old_directory = os.getcwd()
45  os.chdir(build_dir)
46  os.environ["GOPATH"] = go_path
47  os.environ["CGO_CFLAGS"] = args.cgo_cflags
48  os.environ["CGO_LDFLAGS"] = args.cgo_ldflags
49  os.system("%s %s" % (go_binary, " ".join(go_options)))
50  out_files = [ f for f in os.listdir(".") if os.path.isfile(f)]
51  if (len(out_files) > 0):
52    shutil.move(out_files[0], out_file)
53  os.chdir(old_directory)
54  try:
55    shutil.rmtree(build_dir, True)
56  except:
57    pass
58
59if __name__ == '__main__':
60  sys.exit(main())
61