Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 1 | #!/usr/bin/env python |
Jamie Gennis | d1270ce | 2012-05-06 13:27:13 -0700 | [diff] [blame] | 2 | |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 3 | import optparse |
| 4 | import os |
| 5 | import shutil |
| 6 | import subprocess |
| 7 | import sys |
Jamie Gennis | d1270ce | 2012-05-06 13:27:13 -0700 | [diff] [blame] | 8 | |
Dan Sinclair | cfc706a | 2015-08-17 14:04:28 -0400 | [diff] [blame] | 9 | upstream_git = 'https://github.com/catapult-project/catapult.git' |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 10 | PACKAGE_DIRS = [ |
| 11 | 'common', |
| 12 | 'dependency_manager', |
| 13 | 'devil', |
| 14 | 'systrace', |
| 15 | 'third_party/pyserial', |
| 16 | 'third_party/zipfile', |
| 17 | 'tracing/tracing/trace_data', |
| 18 | ] |
| 19 | PACKAGE_FILES = [ |
| 20 | 'tracing/tracing/__init__.py', |
| 21 | 'tracing/tracing_project.py', |
| 22 | ] |
Chris Craik | 5f019af | 2017-02-21 17:11:58 -0800 | [diff] [blame] | 23 | IGNORE_PATTERNS = ['OWNERS'] # doesn't make sense to sync owners files |
Jamie Gennis | 2da489c | 2012-09-19 18:06:29 -0700 | [diff] [blame] | 24 | |
| 25 | script_dir = os.path.dirname(os.path.abspath(sys.argv[0])) |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 26 | catapult_src_dir = os.path.join(script_dir, 'catapult-upstream') |
| 27 | catapult_dst_dir = os.path.join(script_dir, 'catapult') |
Jamie Gennis | 2da489c | 2012-09-19 18:06:29 -0700 | [diff] [blame] | 28 | |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 29 | parser = optparse.OptionParser() |
| 30 | parser.add_option('--local', dest='local_dir', metavar='DIR', |
Zhen Wang | 46b43bf | 2015-08-28 09:54:29 -0700 | [diff] [blame] | 31 | help='use a local catapult') |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 32 | parser.add_option('--no-min', dest='no_min', default=False, action='store_true', |
| 33 | help='skip minification') |
| 34 | options, args = parser.parse_args() |
Jamie Gennis | 2da489c | 2012-09-19 18:06:29 -0700 | [diff] [blame] | 35 | |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 36 | ## Update the source if needed. |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 37 | if options.local_dir is None: |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 38 | # Remove the old source tree. |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 39 | shutil.rmtree(catapult_src_dir, True) |
Jamie Gennis | 2da489c | 2012-09-19 18:06:29 -0700 | [diff] [blame] | 40 | |
Chris Craik | beca7ae | 2015-04-07 13:29:55 -0700 | [diff] [blame] | 41 | # Pull the latest source from the upstream git. |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 42 | git_args = ['git', 'clone', upstream_git, catapult_src_dir] |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 43 | p = subprocess.Popen(git_args, stdout=subprocess.PIPE, cwd=script_dir) |
| 44 | p.communicate() |
| 45 | if p.wait() != 0: |
| 46 | print 'Failed to checkout source from upstream git.' |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 47 | sys.exit(1) |
| 48 | |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 49 | catapult_git_dir = os.path.join(catapult_src_dir, '.git') |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 50 | # Update the UPSTREAM_REVISION file |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 51 | git_args = ['git', 'rev-parse', 'HEAD'] |
Chris Craik | 51b7635 | 2015-06-11 15:03:53 -0700 | [diff] [blame] | 52 | p = subprocess.Popen(git_args, |
| 53 | stdout=subprocess.PIPE, |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 54 | cwd=catapult_src_dir, |
Zhen Wang | 46b43bf | 2015-08-28 09:54:29 -0700 | [diff] [blame] | 55 | env={"GIT_DIR":catapult_git_dir}) |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 56 | out, err = p.communicate() |
| 57 | if p.wait() != 0: |
| 58 | print 'Failed to get revision.' |
| 59 | sys.exit(1) |
| 60 | |
Zhen Wang | 46b43bf | 2015-08-28 09:54:29 -0700 | [diff] [blame] | 61 | shutil.rmtree(catapult_git_dir, True) |
Chris Craik | 51b7635 | 2015-06-11 15:03:53 -0700 | [diff] [blame] | 62 | |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 63 | rev = out.strip() |
| 64 | with open('UPSTREAM_REVISION', 'wt') as f: |
| 65 | f.write(rev + '\n') |
Jamie Gennis | 6907377 | 2012-09-20 12:14:58 -0700 | [diff] [blame] | 66 | else: |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 67 | catapult_src_dir = options.local_dir |
Jamie Gennis | 2da489c | 2012-09-19 18:06:29 -0700 | [diff] [blame] | 68 | |
Chris Craik | b122baf | 2015-03-05 13:58:42 -0800 | [diff] [blame] | 69 | |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 70 | ## Update systrace_trace_viewer.html |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 71 | systrace_dir = os.path.join(catapult_src_dir, 'systrace', 'systrace') |
Zhen Wang | 46b43bf | 2015-08-28 09:54:29 -0700 | [diff] [blame] | 72 | sys.path.append(systrace_dir) |
| 73 | import update_systrace_trace_viewer |
| 74 | update_systrace_trace_viewer.update(no_auto_update=True, no_min=options.no_min) |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 75 | |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 76 | ## Package the result |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 77 | shutil.rmtree(catapult_dst_dir) |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 78 | |
John Reck | ee838d1 | 2016-10-19 08:18:32 -0700 | [diff] [blame] | 79 | for d in PACKAGE_DIRS: |
Chris Craik | 576fd2a | 2017-06-02 13:39:51 -0700 | [diff] [blame] | 80 | src = os.path.join(catapult_src_dir, d) |
| 81 | dst = os.path.join(catapult_dst_dir, d) |
| 82 | |
| 83 | # make parent dir by creating dst + ancestors, and deleting dst |
| 84 | if not os.path.isdir(dst): |
| 85 | os.makedirs(dst) |
| 86 | shutil.rmtree(dst) |
| 87 | |
| 88 | # copy tree |
| 89 | shutil.copytree(src, dst, ignore=shutil.ignore_patterns(*IGNORE_PATTERNS)) |
| 90 | |
| 91 | for f in PACKAGE_FILES: |
| 92 | src = os.path.join(catapult_src_dir, f) |
| 93 | dst = os.path.join(catapult_dst_dir, f) |
| 94 | shutil.copy(src, dst) |