Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(301)

Unified Diff: sky/sdk/packages/sky/bin/sky

Issue 1021933004: Unbreak the build by adding the sky tool (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: sky/sdk/packages/sky/bin/sky
diff --git a/sky/sdk/packages/sky/bin/sky b/sky/sdk/packages/sky/bin/sky
new file mode 100755
index 0000000000000000000000000000000000000000..7224bcb4f32564700ef612c259232801db88a4db
--- /dev/null
+++ b/sky/sdk/packages/sky/bin/sky
@@ -0,0 +1,221 @@
+#!/usr/bin/env python
+# 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 argparse
+import json
+import logging
+import os
+import signal
+import socket
+import subprocess
+import sys
+import urlparse
+
+SDK_TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
+SDK_ROOT = os.path.dirname(SDK_TOOLS_DIR)
+
+SKY_SERVER_PORT = 9888
+DEFAULT_URL = os.path.join(SDK_ROOT, "examples/index.sky")
+APK_NAME = 'SkyDemo.apk'
+ANDROID_PACKAGE = "org.domokit.sky.demo"
+# FIXME: This assumes adb is in $PATH, we could look for ANDROID_HOME, etc?
+ADB_PATH = 'adb'
+
+PID_FILE_PATH = "/tmp/skydemo.pids"
+PID_FILE_KEYS = frozenset([
+ 'remote_sky_server_port',
+ 'sky_server_pid',
+ 'sky_server_port',
+ 'sky_server_root',
+ 'build_dir',
+])
+
+
+def _port_in_use(port):
+ sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+ return sock.connect_ex(('localhost', port)) == 0
+
+
+# We need something to serve sky files, python's httpserver is sufficient.
+def _start_http_server(port, root):
+ server_command = [
+ 'python', '-m', 'SimpleHTTPServer', str(port),
+ ]
+ return subprocess.Popen(server_command, cwd=root).pid
+
+
+# This 'strict dictionary' approach is useful for catching typos.
+class Pids(object):
+ def __init__(self, known_keys, contents=None):
+ self._known_keys = known_keys
+ self._dict = contents if contents is not None else {}
+
+ def __len__(self):
+ return len(self._dict)
+
+ def get(self, key, default=None):
+ assert key in self._known_keys, '%s not in known_keys' % key
+ return self._dict.get(key, default)
+
+ def __getitem__(self, key):
+ assert key in self._known_keys, '%s not in known_keys' % key
+ return self._dict[key]
+
+ def __setitem__(self, key, value):
+ assert key in self._known_keys, '%s not in known_keys' % key
+ self._dict[key] = value
+
+ def __delitem__(self, key):
+ assert key in self._known_keys, '%s not in known_keys' % key
+ del self._dict[key]
+
+ def __iter__(self):
+ return iter(self._dict)
+
+ def __contains__(self, key):
+ assert key in self._known_keys, '%s not in allowed_keys' % key
+ return key in self._dict
+
+ def clear(self):
+ self._dict = {}
+
+ def pop(self, key, default=None):
+ assert key in self._known_keys, '%s not in known_keys' % key
+ return self._dict.pop(key, default)
+
+ @classmethod
+ def read_from(cls, path, known_keys):
+ contents = {}
+ try:
+ with open(path, 'r') as pid_file:
+ contents = json.load(pid_file)
+ except:
+ if os.path.exists(path):
+ logging.warn('Failed to read pid file: %s' % path)
+ return cls(known_keys, contents)
+
+ def write_to(self, path):
+ try:
+ with open(path, 'w') as pid_file:
+ json.dump(self._dict, pid_file, indent=2, sort_keys=True)
+ except:
+ logging.warn('Failed to write pid file: %s' % path)
+
+
+def _url_for_path(port, root, path):
+ relative_path = os.path.relpath(path, root)
+ return 'sky://localhost:%s/%s' % (port, relative_path)
+
+
+class StartSky(object):
+ def add_subparser(self, subparsers):
+ start_parser = subparsers.add_parser('start',
+ help='launch SKyShell.apk on the device')
+ start_parser.add_argument('--install', action='store_true')
+ start_parser.add_argument('project_or_path', nargs='?', type=str,
+ default='main.sky')
+ start_parser.set_defaults(func=self.run)
+
+ def run(self, args, pids):
+ StopSky().run(args, pids)
+ if args.install:
+ apk_path = os.path.join(SDK_ROOT, 'apks', APK_NAME)
+ if not os.path.exists(apk_path):
+ print "'%s' does not exist?" % apk_path
+ return 2
+
+ subprocess.check_call([ADB_PATH, 'install', '-r', apk_path])
+
+ project_or_path = os.path.abspath(args.project_or_path)
+
+ if os.path.isdir(project_or_path):
+ sky_server_root = project_or_path
+ main_sky = os.path.join(project_or_path, 'main.sky')
+ missing_msg = "Missing main.sky in project: %s" % sky_server_root
+ else:
+ sky_server_root = os.path.dirname(project_or_path)
+ main_sky = project_or_path
+ missing_msg = "%s does not exist." % main_sky
+
+ if not os.path.isfile(main_sky):
+ print missing_msg
+ return 2
+
+ sky_server_port = SKY_SERVER_PORT
+ pids['sky_server_port'] = sky_server_port
+ if _port_in_use(sky_server_port):
+ logging.warn(('Port %s already in use. '
+ ' Not starting server for %s') % (sky_server_port, sky_server_root))
+ else:
+ sky_server_pid = _start_http_server(sky_server_port, sky_server_root)
+ pids['sky_server_pid'] = sky_server_pid
+ pids['sky_server_root'] = sky_server_root
+
+ port_string = 'tcp:%s' % sky_server_port
+ subprocess.check_call([
+ ADB_PATH, 'reverse', port_string, port_string
+ ])
+ pids['remote_sky_server_port'] = sky_server_port
+
+ # The load happens on the remote device, use the remote port.
+ sky_url = _url_for_path(pids['remote_sky_server_port'], sky_server_root,
+ main_sky)
+
+ subprocess.check_call([ADB_PATH, 'shell',
+ 'am', 'start',
+ '-a', 'android.intent.action.VIEW',
+ '-d', sky_url])
+
+
+class StopSky(object):
+ def add_subparser(self, subparsers):
+ stop_parser = subparsers.add_parser('stop',
+ help=('kill all running SkyShell.apk processes'))
+ stop_parser.set_defaults(func=self.run)
+
+ def _kill_if_exists(self, pids, key, name):
+ pid = pids.pop(key, None)
+ if not pid:
+ logging.info('No pid for %s, nothing to do.' % name)
+ return
+ logging.info('Killing %s (%d).' % (name, pid))
+ try:
+ os.kill(pid, signal.SIGTERM)
+ except OSError:
+ logging.info('%s (%d) already gone.' % (name, pid))
+
+ def run(self, args, pids):
+ self._kill_if_exists(pids, 'sky_server_pid', 'sky_server')
+
+ if 'remote_sky_server_port' in pids:
+ port_string = 'tcp:%s' % pids['remote_sky_server_port']
+ subprocess.call([ADB_PATH, 'reverse', '--remove', port_string])
+
+ subprocess.call([
+ ADB_PATH, 'shell', 'am', 'force-stop', ANDROID_PACKAGE])
+
+ pids.clear()
+
+
+class SkyShellRunner(object):
+ def main(self):
+ logging.basicConfig(level=logging.WARNING)
+
+ parser = argparse.ArgumentParser(description='Sky Demo Runner')
+ subparsers = parser.add_subparsers(help='sub-command help')
+
+ for command in [StartSky(), StopSky()]:
+ command.add_subparser(subparsers)
+
+ args = parser.parse_args()
+ pids = Pids.read_from(PID_FILE_PATH, PID_FILE_KEYS)
+ exit_code = args.func(args, pids)
+ # We could do this with an at-exit handler instead?
+ pids.write_to(PID_FILE_PATH)
+ sys.exit(exit_code)
+
+
+if __name__ == '__main__':
+ SkyShellRunner().main()
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698