You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
4.6 KiB
136 lines
4.6 KiB
# Copyright 2018 - The Android Open Source Project
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
r"""Setup entry point.
|
|
|
|
Setup will handle all of the necessary steps to enable acloud to create a local
|
|
or remote instance of an Android Virtual Device.
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from acloud.internal import constants
|
|
from acloud.internal.lib import utils
|
|
from acloud.public import config
|
|
from acloud.setup import host_setup_runner
|
|
from acloud.setup import gcp_setup_runner
|
|
|
|
|
|
def Run(args):
|
|
"""Run setup.
|
|
|
|
Setup options:
|
|
-host: Setup host settings.
|
|
-gcp_init: Setup gcp settings.
|
|
-None, default behavior will setup host and gcp settings.
|
|
|
|
Args:
|
|
args: Namespace object from argparse.parse_args.
|
|
"""
|
|
if args.update_config:
|
|
_UpdateConfig(args.config_file, args.update_config[0], args.update_config[1])
|
|
return
|
|
|
|
_RunPreSetup()
|
|
|
|
# Setup process will be in the following manner:
|
|
# 1.Print welcome message.
|
|
_PrintWelcomeMessage()
|
|
|
|
# 2.Init all subtasks in queue and traverse them.
|
|
host_base_runner = host_setup_runner.HostBasePkgInstaller()
|
|
host_avd_runner = host_setup_runner.AvdPkgInstaller()
|
|
host_cf_common_runner = host_setup_runner.CuttlefishCommonPkgInstaller()
|
|
host_env_runner = host_setup_runner.CuttlefishHostSetup()
|
|
gcp_runner = gcp_setup_runner.GcpTaskRunner(args.config_file)
|
|
task_queue = []
|
|
# User must explicitly specify --host to install the avd host packages.
|
|
if args.host:
|
|
task_queue.append(host_base_runner)
|
|
task_queue.append(host_avd_runner)
|
|
task_queue.append(host_cf_common_runner)
|
|
task_queue.append(host_env_runner)
|
|
# We should do these setup tasks if specified or if no args were used.
|
|
if args.host_base or (not args.host and not args.gcp_init):
|
|
task_queue.append(host_base_runner)
|
|
if args.gcp_init or (not args.host and not args.host_base):
|
|
task_queue.append(gcp_runner)
|
|
|
|
for subtask in task_queue:
|
|
subtask.Run(force_setup=args.force)
|
|
|
|
# 3.Print the usage hints.
|
|
_PrintUsage()
|
|
|
|
|
|
def _PrintWelcomeMessage():
|
|
"""Print welcome message when acloud setup been called."""
|
|
|
|
# pylint: disable=anomalous-backslash-in-string
|
|
asc_art = " \n" \
|
|
" ___ _______ ____ __ _____ \n" \
|
|
" / _ |/ ___/ / / __ \/ / / / _ \\ \n" \
|
|
" / __ / /__/ /__/ /_/ / /_/ / // / \n" \
|
|
"/_/ |_\\___/____/\\____/\\____/____/ \n" \
|
|
" \n"
|
|
|
|
print("\nWelcome to")
|
|
print(asc_art)
|
|
|
|
|
|
def _PrintUsage():
|
|
"""Print cmd usage hints when acloud setup been finished."""
|
|
utils.PrintColorString("")
|
|
utils.PrintColorString("Setup process finished")
|
|
|
|
|
|
def _RunPreSetup():
|
|
"""This will run any pre-setup scripts.
|
|
|
|
If we can find any pre-setup scripts, run it and don't care about the
|
|
results. Pre-setup scripts will do any special setup before actual
|
|
setup occurs (e.g. copying configs).
|
|
"""
|
|
if constants.ENV_ANDROID_BUILD_TOP not in os.environ:
|
|
print("Can't find $%s." % constants.ENV_ANDROID_BUILD_TOP)
|
|
print("Please run '#source build/envsetup.sh && lunch <target>' first.")
|
|
sys.exit(constants.EXIT_BY_USER)
|
|
|
|
pre_setup_sh = os.path.join(os.environ.get(constants.ENV_ANDROID_BUILD_TOP),
|
|
"tools",
|
|
"acloud",
|
|
"setup",
|
|
"pre_setup_sh",
|
|
"acloud_pre_setup.sh")
|
|
|
|
if os.path.exists(pre_setup_sh):
|
|
subprocess.call([pre_setup_sh])
|
|
|
|
def _UpdateConfig(config_file, field, value):
|
|
"""Update the user config.
|
|
|
|
Args:
|
|
config_file: String of config file path.
|
|
field: String, field name in user config.
|
|
value: String, the value of field.
|
|
"""
|
|
config_mgr = config.AcloudConfigManager(config_file)
|
|
config_mgr.Load()
|
|
user_config = config_mgr.user_config_path
|
|
print("Your config (%s) is updated." % user_config)
|
|
gcp_setup_runner.UpdateConfigFile(user_config, field, value)
|
|
_PrintUsage()
|