redirect.py

from apiclient.discovery import build
import logging
import webapp2
from google.appengine.api import urlfetch
from google.appengine.api import app_identity
from oauth2client.contrib.appengine import AppAssertionCredentials
from httplib2 import Http
import json

class MainPage(webapp2.RequestHandler):
  Monorail_cans = {
    '': 'open',
    '1': 'all',
    '2': 'open',
    'closed': 'all',
  }
  for x in ['all', 'new', 'open', 'owned', 'reported', 'starred', 'to_verify']:
    Monorail_cans[x] = x

  def get(self):
      scope = 'https://www.googleapis.com/auth/userinfo.email'
      credentials = AppAssertionCredentials(scope)
      http = credentials.authorize(Http())

      MONORAIL_DISCOVERY_URL = (
        'https://monorail-prod.appspot.com/_ah/api/discovery/v1/apis/'
        '{api}/{apiVersion}/rest'
      )

      monorail = build(
        'monorail', 'v1',
        discoveryServiceUrl=MONORAIL_DISCOVERY_URL,
        http=http
      )

      scope = "https://www.googleapis.com/auth/gerritcodereview"
      auth_token, _ = app_identity.get_access_token(scope)

      if self.request.get('site') == 'issues':
          urlfetch.set_default_fetch_deadline(10)
          startIndex = self.request.get('start')
          try:
            startIndex = int(startIndex)
          except:
            startIndex = 0
          self.response.headers.add_header("Access-Control-Allow-Origin", "*")
          result = monorail.issues().list(projectId='chromium', q=self.request.get('q'),
                                          can=MainPage.Monorail_cans.get(self.request.get('can',''),'open'),
                                          startIndex=startIndex).execute()
          self.response.write(json.dumps(result))
      elif self.request.get('site') == 'issue':
          urlfetch.set_default_fetch_deadline(10)
          self.response.headers.add_header("Access-Control-Allow-Origin", "*")
          result = monorail.issues().get(projectId='chromium', issueId=self.request.get('issueId')).execute()
          self.response.write(json.dumps(result))
      elif self.request.get('site') == 'comments':
          urlfetch.set_default_fetch_deadline(10)
          startIndex = self.request.get('start')
          try:
            startIndex = int(startIndex)
          except:
            startIndex = 0
          maxResults = self.request.get('max')
          try:
            maxResults = int(maxResults)
            if (maxResults > 200):
              maxResults = 200
              logging.error('You can only fetch 200 comments at a time. If you need to fetch more comments than that please paginate using the startIndex.')
          except:
            maxResults = 0
          self.response.headers.add_header("Access-Control-Allow-Origin", "*")
          result = monorail.issues().comments().list(projectId='chromium', issueId=self.request.get('issueId'), maxResults=maxResults, startIndex=startIndex).execute()
          self.response.write(json.dumps(result))
      elif self.request.get('site') == 'gerrit':
          urlfetch.set_default_fetch_deadline(10)
          self.response.headers.add_header("Access-Control-Allow-Origin", "*")
          result = urlfetch.fetch(url="https://chromium-review.googlesource.com/a/changes/?q=" 
            + self.request.get('q') + "&o=" + self.request.get('o') + "&o=DETAILED_ACCOUNTS&o=REVIEWER_UPDATES",
            headers={"Authorization": "Bearer " + auth_token});
          self.response.write(json.dumps(result.content[5:]));
app = webapp2.WSGIApplication([
    ('/.*', MainPage),
], debug=True)