diff options
Diffstat (limited to 'infra/bots/recipe_modules/skia/ios_flavor.py')
-rw-r--r-- | infra/bots/recipe_modules/skia/ios_flavor.py | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/infra/bots/recipe_modules/skia/ios_flavor.py b/infra/bots/recipe_modules/skia/ios_flavor.py new file mode 100644 index 0000000000..d83cf213de --- /dev/null +++ b/infra/bots/recipe_modules/skia/ios_flavor.py @@ -0,0 +1,176 @@ +# Copyright 2015 The Chromium Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. + + +import copy +import default_flavor + + +"""iOS flavor utils, used for building for and running tests on iOS.""" + + +class iOSFlavorUtils(default_flavor.DefaultFlavorUtils): + def __init__(self, skia_api): + super(iOSFlavorUtils, self).__init__(skia_api) + self.default_env = {} + self.default_env['XCODEBUILD'] = ( + self._skia_api.slave_dir.join('xcodebuild')) + self.ios_bin = self._skia_api.skia_dir.join( + 'platform_tools', 'ios', 'bin') + + def step(self, name, cmd, **kwargs): + args = [self.ios_bin.join('ios_run_skia')] + env = {} + env.update(kwargs.pop('env', {})) + env.update(self.default_env) + # Convert 'dm' and 'nanobench' from positional arguments + # to flags, which is what iOSShell expects to select which + # one is being run. + cmd = ["--" + c if c in ['dm', 'nanobench'] else c + for c in cmd] + return self._skia_api.run(self._skia_api.m.step, name=name, cmd=args + cmd, + env=env, + **kwargs) + + def compile(self, target): + """Build the given target.""" + cmd = [self.ios_bin.join('ios_ninja')] + self._skia_api.run(self._skia_api.m.step, 'build iOSShell', cmd=cmd, + cwd=self._skia_api.m.path['checkout']) + + def device_path_join(self, *args): + """Like os.path.join(), but for paths on a connected iOS device.""" + return '/'.join(args) + + def device_path_exists(self, path): + """Like os.path.exists(), but for paths on a connected device.""" + return self._skia_api.run( + self._skia_api.m.step, + 'exists %s' % path, + cmd=[self.ios_bin.join('ios_path_exists'), path], + env=self.default_env, + infra_step=True, + ) # pragma: no cover + + def _remove_device_dir(self, path): + """Remove the directory on the device.""" + return self._skia_api.run( + self._skia_api.m.step, + 'rmdir %s' % path, + cmd=[self.ios_bin.join('ios_rm'), path], + env=self.default_env, + infra_step=True, + ) + + def _create_device_dir(self, path): + """Create the directory on the device.""" + return self._skia_api.run( + self._skia_api.m.step, + 'mkdir %s' % path, + cmd=[self.ios_bin.join('ios_mkdir'), path], + env=self.default_env, + infra_step=True, + ) + + def copy_directory_contents_to_device(self, host_dir, device_dir): + """Like shutil.copytree(), but for copying to a connected device.""" + return self._skia_api.run( + self._skia_api.m.step, + name='push %s to %s' % (self._skia_api.m.path.basename(host_dir), + self._skia_api.m.path.basename(device_dir)), + cmd=[self.ios_bin.join('ios_push_if_needed'), + host_dir, device_dir], + env=self.default_env, + infra_step=True, + ) + + def copy_directory_contents_to_host(self, device_dir, host_dir): + """Like shutil.copytree(), but for copying from a connected device.""" + self._skia_api.run( + self._skia_api.m.step, + name='pull %s' % self._skia_api.m.path.basename(device_dir), + cmd=[self.ios_bin.join('ios_pull_if_needed'), + device_dir, host_dir], + env=self.default_env, + infra_step=True, + ) + + def copy_file_to_device(self, host_path, device_path): + """Like shutil.copyfile, but for copying to a connected device.""" + self._skia_api.run( + self._skia_api.m.step, + name='push %s' % host_path, + cmd=[self.ios_bin.join('ios_push_file'), host_path, device_path], + env=self.default_env, + infra_step=True, + ) # pragma: no cover + + def copy_extra_build_products(self, swarming_out_dir): + xcode_dir = self._skia_api.m.path.join( + 'xcodebuild', '%s-iphoneos' % self._skia_api.configuration) + self._skia_api.copy_build_products( + self._skia_api.skia_dir.join(xcode_dir), + swarming_out_dir.join(xcode_dir)) + + def create_clean_device_dir(self, path): + """Like shutil.rmtree() + os.makedirs(), but on a connected device.""" + self._remove_device_dir(path) + self._create_device_dir(path) + + def install(self): + """Run device-specific installation steps.""" + self._skia_api.run( + self._skia_api.m.step, + name='install iOSShell', + cmd=[self.ios_bin.join('ios_install')], + env=self.default_env, + infra_step=True) + + def cleanup_steps(self): + """Run any device-specific cleanup steps.""" + if self._skia_api.do_test_steps or self._skia_api.do_perf_steps: + self._skia_api.run( + self._skia_api.m.step, + name='reboot', + cmd=[self.ios_bin.join('ios_restart')], + env=self.default_env, + infra_step=True) + self._skia_api.run( + self._skia_api.m.step, + name='wait for reboot', + cmd=['sleep', '20'], + env=self.default_env, + infra_step=True) + + def read_file_on_device(self, path): + """Read the given file.""" + ret = self._skia_api.run( + self._skia_api.m.step, + name='read %s' % self._skia_api.m.path.basename(path), + cmd=[self.ios_bin.join('ios_cat_file'), path], + env=self.default_env, + stdout=self._skia_api.m.raw_io.output(), + infra_step=True) + return ret.stdout.rstrip() if ret.stdout else ret.stdout + + def remove_file_on_device(self, path): + """Remove the file on the device.""" + return self._skia_api.run( + self._skia_api.m.step, + 'rm %s' % path, + cmd=[self.ios_bin.join('ios_rm'), path], + env=self.default_env, + infra_step=True, + ) + + def get_device_dirs(self): + """ Set the directories which will be used by the build steps.""" + prefix = self.device_path_join('skiabot', 'skia_') + return default_flavor.DeviceDirs( + dm_dir=prefix + 'dm', + perf_data_dir=prefix + 'perf', + resource_dir=prefix + 'resources', + images_dir=prefix + 'images', + skp_dir=prefix + 'skp/skps', + tmp_dir=prefix + 'tmp_dir') |