Add scripts to semi-automate publishing of ChromeVox webstore extension using webstore API.

chrome/browser/resources/chromeos/chromevox/tools/chromevox_webstore_util.py

+#!/usr/bin/env python
+
+# Copyright 2014 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.
+
+'''A set of utilities to interface with the Chrome Webstore API.'''
+
+import os
+import SimpleHTTPServer
+import SocketServer
+import httplib
+import json
+import re
+import thread
+import urllib
+import webbrowser
+
+PROJECT_ARGS = {
+  'client_id':
+    '937534751394-gbj5334v9144c57qjqghl7d283plj5r4.apps.googleusercontent.com',

[dmazzoni, 2014/07/28 05:29:07]

nit: 4-space indent for line continuation?

[David Tseng, 2014/07/29 23:52:18]

Done.

+  'grant_type': 'authorization_code',
+  'redirect_uri': 'http://localhost:8000'
+}
+
+PORT = 8000
+
+APP_ID = 'kgejglhpjiefppelpmljglcjbhoiplfn'
+OAUTH_DOMAIN = 'accounts.google.com'
+OAUTH_AUTH_COMMAND = '/o/oauth2/auth'
+OAUTH_TOKEN_COMMAND = '/o/oauth2/token'
+WEBSTORE_API_SCOPE = 'https://www.googleapis.com/auth/chromewebstore'
+
+API_ENDPOINT_DOMAIN = 'www.googleapis.com'
+COMMAND_GET_UPLOAD_STATUS = (
+    '/chromewebstore/v1.1/items/%s?projection=draft' % APP_ID)
+COMMAND_POST_PUBLISH = '/chromewebstore/v1.1/items/%s/publish' % APP_ID
+COMMAND_POST_UPLOAD = '/upload/chromewebstore/v1.1/items/%s' % APP_ID
+
+class CodeRequestHandler(SocketServer.StreamRequestHandler):
+  def handle(self):
+    content = self.rfile.readline()
+    self.server.code = re.search('code=(.*) ', content).groups()[0]
+    self.rfile.close()
+
+def GetAuthCode():
+  Handler = CodeRequestHandler
+  httpd = SocketServer.TCPServer(("", PORT), Handler)
+  auth_url = ''.join([' https://%s%s?' % (OAUTH_DOMAIN, OAUTH_AUTH_COMMAND),
+         'response_type=code&',

[dmazzoni, 2014/07/28 05:29:07]

How about something like this to aid readability:

query = '&'.join([
    'response_type=code',
    'scope=%s' % WEBSTORE_API_SCOPE,
   ...]
auth_url = 'https://%s%s?%s' % (OAUTH_DOMAIN, OAUTH_AUTH_COMMAND, query)

[David Tseng, 2014/07/29 23:52:18]

Done.

+         'scope=%s&' % WEBSTORE_API_SCOPE,
+         ('client_id=%(client_id)s&'
+          'redirect_uri=%(redirect_uri)s' % PROJECT_ARGS)])
+  print 'Navigating to %s' % auth_url
+  webbrowser.open(auth_url)
+  httpd.handle_request()
+  return httpd.code
+
+def GetOauthToken(code, client_secret):
+  PROJECT_ARGS['code'] = code
+  PROJECT_ARGS['client_secret'] = client_secret
+  body = urllib.urlencode(PROJECT_ARGS)
+  conn = httplib.HTTPSConnection(OAUTH_DOMAIN)
+  conn.putrequest('POST', OAUTH_TOKEN_COMMAND)
+  conn.putheader('content-type', 'application/x-www-form-urlencoded')
+  conn.putheader('content-length', len(body))
+  conn.endheaders()
+  conn.send(body)
+  content = conn.getresponse().read()
+  return json.loads(content)
+
+def GetPopulatedHeader(client_secret):
+  code = GetAuthCode()
+  access_token = GetOauthToken(code, client_secret)
+  url = 'www.googleapis.com'
+
+  return {'Authorization': 'Bearer %(access_token)s' % access_token,
+             'x-goog-api-version': 2,
+             'Content-Length': 0
+            }
+
+def SendGetCommand(command, client_secret):
+  headers = GetPopulatedHeader(client_secret)
+  conn = httplib.HTTPSConnection(API_ENDPOINT_DOMAIN)
+  conn.request('GET', command, '', headers)
+  return conn.getresponse()
+
+def SendPostCommand(command, body, client_secret):
+  headers = GetPopulatedHeader(client_secret)
+  conn = httplib.HTTPSConnection(API_ENDPOINT_DOMAIN)
+  conn.request('PUT', command, body, headers)
+  return conn.getresponse()
+
+def GetUploadStatus(client_secret):
+  '''Gets the status of a previous upload.
+  Args:
+    client_secret ChromeVox's client secret creds.
+  '''
+  return SendGetCommand(COMMAND_GET_UPLOAD_STATUS, client_secret)
+
+# httplib fails to persist the connection during upload; use curl instead.
+def PostUpload(file, client_secret):
+  '''Posts an uploaded version of ChromeVox.
+  Args:
+    file A string path to the ChromeVox extension zip.
+    client_secret ChromeVox's client secret creds.
+  '''
+  header = GetPopulatedHeader(client_secret)
+  curl_command = ''.join(['curl ',

[dmazzoni, 2014/07/28 05:29:07]

I think ' '.join with no spaces in the arguments would be more clear.

[David Tseng, 2014/07/29 23:52:18]

Done.

+                  '-H "Authorization: %(Authorization)s" ' % header,
+                '-H "x-goog-api-version: 2" ',

[dmazzoni, 2014/07/28 05:29:07]

nit: line up with previous line, and below

[David Tseng, 2014/07/29 23:52:18]

Done.

+                '-X PUT ',
+                '-T %s ' % file,
+                '-v ',
+                          'https://%s%s' % (API_ENDPOINT_DOMAIN,
+                                            COMMAND_POST_UPLOAD)])
+  os.system(curl_command)

[dmazzoni, 2014/07/28 05:29:07]

Return the result of this system call or throw if it's nonzero?

[David Tseng, 2014/07/29 23:52:18]

Done.

+
+def PostPublish(client_secret):
+  '''Publishes an previously uploaded ChromeVox extension.
+  Args:
+    client_secret ChromeVox's client secret creds.
+  '''
+  return SendPostCommand(COMMAND_POST_PUBLISH, client_secret)