83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
|
# Copyright 2012 Google Inc. All Rights Reserved.
|
||
|
|
|
||
|
|
"""Common command utilities.
|
||
|
|
"""
|
||
|
|
|
||
|
|
__author__ = 'benvanik@google.com (Ben Vanik)'
|
||
|
|
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
from anvil.context import BuildEnvironment, BuildContext
|
||
|
|
from anvil.project import FileModuleResolver, Project
|
||
|
|
from anvil.task import InProcessTaskExecutor, MultiProcessTaskExecutor
|
||
|
|
|
||
|
|
|
||
|
|
def clean_output(cwd):
|
||
|
|
"""Cleans all build-related output and caches.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
cwd: Current working directory.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
True if the clean succeeded.
|
||
|
|
"""
|
||
|
|
nuke_paths = [
|
||
|
|
'.build-cache',
|
||
|
|
'build-out',
|
||
|
|
'build-gen',
|
||
|
|
'build-bin',
|
||
|
|
]
|
||
|
|
any_failed = False
|
||
|
|
for path in nuke_paths:
|
||
|
|
full_path = os.path.join(cwd, path)
|
||
|
|
if os.path.isdir(full_path):
|
||
|
|
try:
|
||
|
|
shutil.rmtree(full_path)
|
||
|
|
except Exception as e:
|
||
|
|
print 'Unable to remove %s: %s' % (full_path, e)
|
||
|
|
any_failed = True
|
||
|
|
return not any_failed
|
||
|
|
|
||
|
|
|
||
|
|
def run_build(cwd, parsed_args):
|
||
|
|
"""Runs a build with the given arguments.
|
||
|
|
Assumes that add_common_args and add_common_build_args was called on the
|
||
|
|
ArgumentParser.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
cwd: Current working directory.
|
||
|
|
parsed_args: Argument namespace from an ArgumentParser.
|
||
|
|
"""
|
||
|
|
build_env = BuildEnvironment(root_path=cwd)
|
||
|
|
|
||
|
|
module_resolver = FileModuleResolver(cwd)
|
||
|
|
project = Project(module_resolver=module_resolver)
|
||
|
|
|
||
|
|
# -j/--jobs switch to change execution mode
|
||
|
|
# TODO(benvanik): force -j 1 on Cygwin?
|
||
|
|
if parsed_args.jobs == 1:
|
||
|
|
task_executor = InProcessTaskExecutor()
|
||
|
|
else:
|
||
|
|
task_executor = MultiProcessTaskExecutor(worker_count=parsed_args.jobs)
|
||
|
|
|
||
|
|
# TODO(benvanik): good logging/info - resolve rules in project and print
|
||
|
|
# info?
|
||
|
|
print 'building %s' % (parsed_args.targets)
|
||
|
|
|
||
|
|
# TODO(benvanik): take additional args from command line
|
||
|
|
all_target_outputs = set([])
|
||
|
|
with BuildContext(build_env, project,
|
||
|
|
task_executor=task_executor,
|
||
|
|
force=parsed_args.force,
|
||
|
|
stop_on_error=parsed_args.stop_on_error,
|
||
|
|
raise_on_error=False) as build_ctx:
|
||
|
|
result = build_ctx.execute_sync(parsed_args.targets)
|
||
|
|
if result:
|
||
|
|
for target in parsed_args.targets:
|
||
|
|
(state, target_outputs) = build_ctx.get_rule_results(target)
|
||
|
|
all_target_outputs.update(target_outputs)
|
||
|
|
|
||
|
|
return (result == True, all_target_outputs)
|