main.py

import StringIO
import csv
import json
import logging
import urllib

import webapp2
from webapp2 import uri_for

from common import common_request
from model import CrashReport, GlobalPreferences, Link
from search_model import Search
from util import CrashReports


class RequestHandlerUtils(object):
    @classmethod
    def add_brand(cls, handler):
        brand = Link('Tessel Error Reporter', uri_for('home'))
        handler.add_parameter('brand', brand)

    @classmethod
    def add_nav_links(cls, handler):
        nav_links = list()
        nav_links.append(Link('Who we are', 'https://tessel.io/'))
        nav_links.append(Link('About', 'https://tessel.io/about'))
        handler.add_parameter('nav_links', nav_links)


class RootHandler(webapp2.RequestHandler):
    @common_request
    def get(self):
        self.add_parameter('title', 'Tessel Error Reporter')
        self.add_breadcrumb('Tessel Error Reporter', uri_for('home'))
        RequestHandlerUtils.add_brand(self)
        RequestHandlerUtils.add_nav_links(self)
        directory_links = list()
        directory_links.append(Link('Trending Crashes', uri_for('trending_crashes')))
        directory_links.append(Link('Submit Crash', uri_for('submit_crash')))
        directory_links.append(Link('View Crash', uri_for('view_crash')))
        directory_links.append(Link('Update Crash Report', uri_for('update_crash_state')))
        directory_links.append(Link('Search', uri_for('search')))
        directory_links.append(Link('Update Global Preferences', uri_for('update_global_preferences')))
        self.add_parameter('directory_links', directory_links)
        self.render('index.html')


class SubmitCrashHandler(webapp2.RequestHandler):
    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Submit Crash Report')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('Submit Crash', uri_for('submit_crash'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    @common_request
    def get(self):
        SubmitCrashHandler.common(self)
        self.render('submit-crash.html')

    @classmethod
    def csv_to_list(cls, input_as_csv):
        if not input_as_csv:
            return None
        else:
            input_as_io = StringIO.StringIO(input_as_csv)
            reader = csv.reader(input_as_io, delimiter=',')
            tokens = list()
            for token in reader:
                tokens.extend(token)
            return tokens

    @common_request
    def post(self):
        SubmitCrashHandler.common(self)
        if self.empty_query_string('crash', 'labels'):
            self.request_handler.redirect(uri_for('submit_crash'))
        else:
            crash = self.get_parameter('crash')
            argv = SubmitCrashHandler.csv_to_list(self.get_parameter('argv'))
            labels = SubmitCrashHandler.csv_to_list(self.get_parameter('labels'))
            # strip spaces around the crash report
            crash_report = CrashReports.add_crash_report(crash.strip(), argv=argv, labels=labels)
            message = 'Added Crash Report with fingerprint, count) => ({0}, {1})'.format(
                crash_report.fingerprint, CrashReport.get_count(crash_report.name))

            self.add_message(message)
            self.add_to_json('crash_report', CrashReport.to_json(crash_report))
            self.render('submit-crash.html')


class ViewCrashHandler(webapp2.RequestHandler):
    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Show Crash')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('View Crash', uri_for('view_crash'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    @common_request
    def get(self):
        ViewCrashHandler.common(self)
        if not self.empty_query_string('fingerprint'):
            fingerprint = self.get_parameter('fingerprint')
            crash_report = CrashReport.get_crash(fingerprint)
            if crash_report:
                crash_report_item = CrashReport.to_json(crash_report)
                self.add_parameter('crash_report', crash_report_item)
                self.add_to_json('crash_report', crash_report_item)
        self.render('show-crash.html')


class UpdateCrashStateHandler(webapp2.RequestHandler):
    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Update Crash State')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('Update Crash State', uri_for('update_crash_state'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    @common_request
    def get(self):
        UpdateCrashStateHandler.common(self)
        self.render('update-crash-state.html')

    @common_request
    def post(self):
        UpdateCrashStateHandler.common(self)
        if not self.empty_query_string('fingerprint', 'state'):
            fingerprint = self.get_parameter('fingerprint')
            state = self.get_parameter('state', default_value='unresolved',
                                       valid_iter=['unresolved', 'pending', 'submitted', 'resolved'])
            crash_report = CrashReports.update_report_state(fingerprint, state)
            if crash_report:
                crash_report_item = CrashReport.to_json(crash_report)
                self.add_parameter('crash_report', crash_report_item)
                self.add_to_json('crash_report', crash_report_item)
        self.render('update-crash-state.html')


class TrendingCrashesHandler(webapp2.RequestHandler):
    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Show Crash')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('Trending Crashes', uri_for('trending_crashes'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    @common_request
    def get(self):
        TrendingCrashesHandler.common(self)
        start = self.get_parameter('start')
        trending_result = CrashReports.trending(start=start)
        self.add_parameter('trending', trending_result.get('trending', list()))
        self.add_parameter('has_more', trending_result.get('has_more', False))
        self.add_to_json('trending', trending_result)
        self.render('trending.html')


class SearchCrashesHandler(webapp2.RequestHandler):
    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Search crashes')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('Search', uri_for('search'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    def get(self):
        self.post()

    @common_request
    def post(self):
        SearchCrashesHandler.common(self)
        if not self.empty_query_string('query'):
            query = self.get_parameter('query')
            cursor = self.get_parameter('cursor')
            try:
                search_results = Search.search(query, cursor=cursor)
                results = search_results.get('results', list())
                if results and len(results) > 0:
                    self.add_parameter('results', results)
                    self.add_to_json('results', results)

                cursor = search_results.get('cursor', None)
                if cursor:
                    query_fragment = {
                        'query': query,
                        'cursor': cursor
                    }
                    encoded_fragment = urllib.urlencode(query_fragment)
                    self.add_parameter('query_fragment', encoded_fragment)
                    self.add_to_json('query_fragment', encoded_fragment)
            except Exception, e:
                logging.exception('Exception : %s' % unicode(e))
                self.add_error('Exception : %s' % unicode(e))
                self.add_to_json('error', unicode(e))

        self.render('search.html')


class UpdatePreferencesHandler(webapp2.RequestHandler):

    # a list of all global prefences
    __PREFERENCES__ = [
        GlobalPreferences.__INTEGRATE_WITH_GITHUB__,
    ]

    @classmethod
    def common(cls, handler):
        handler.add_parameter('title', 'Update Global Preferences')
        handler.add_breadcrumb('Home', uri_for('home'))
        handler.add_breadcrumb('Update Global Preferences', uri_for('update_global_preferences'))
        RequestHandlerUtils.add_brand(handler)
        RequestHandlerUtils.add_nav_links(handler)

    def get(self):
        self.post()

    @common_request
    def post(self):
        UpdatePreferencesHandler.common(self)
        for preference in UpdatePreferencesHandler.__PREFERENCES__:
            if not self.empty_query_string(preference):
                preference_value = self.get_parameter(preference)
                GlobalPreferences.update(preference, preference_value)

                self.add_message('Updated %s to %s' % (preference, preference_value))
                self.add_to_json('success', True)

        self.render('update-global-preferences.html')


class GitHubWebHooksHandler(webapp2.RequestHandler):
    @common_request
    def post(self):
        """
        Handle GitHub webhooks
        """
        headers = self.request_handler.request.headers
        event_type = headers.get('X-GitHub-Event')
        request_body = json.loads(self.request_handler.request.body)

        if event_type is not None and request_body is not None:
            logging.info('X-GitHub-Event {0}'.format(event_type))
            if event_type == 'issues':
                '''
                We are looking for issue closed event types.
                https://developer.github.com/v3/activity/events/types/#issuesevent
                '''
                action = request_body.get('action')
                if action == 'closed':
                    '''
                    Handle closed event
                    '''
                    # the full issue body
                    issue = request_body.get('issue')
                    # issue number (treated as a string in the datastore)
                    number = str(issue.get('number'))
                    CrashReports.close_github_issue(number)
                    logging.info('Marked GitHub issue {0} as resolved'.format(number))
                else:
                    logging.info('Other action {0}. Ignoring.'.format(action))


application = webapp2.WSGIApplication(
    [
        webapp2.Route('/', handler='main.RootHandler', name='home'),
        webapp2.Route('/crashes/state/update', handler='main.UpdateCrashStateHandler', name='update_crash_state'),
        webapp2.Route('/crashes/submit', handler='main.SubmitCrashHandler', name='submit_crash'),
        webapp2.Route('/crashes', handler='main.ViewCrashHandler', name='view_crash'),
        webapp2.Route('/trending', handler='main.TrendingCrashesHandler', name='trending_crashes'),
        webapp2.Route('/search', handler='main.SearchCrashesHandler', name='search'),
        webapp2.Route('/preferences/update', handler='main.UpdatePreferencesHandler', name='update_global_preferences'),
        webapp2.Route('/webhooks/github', handler='main.GitHubWebHooksHandler', name='github_webhooks'),
    ]
    , debug=True
)