Index: infra/bots/recipe_modules/skia/ios_flavor.py |
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 0000000000000000000000000000000000000000..d83cf213de9013482a2b28d9346591eafa1639a6 |
--- /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') |