import cyclone.web, cyclone.websocket, cyclone.httpclient

import greplin.scales.twistedweb, greplin.scales.formats
from greplin import scales


# Like scales.twistedweb.StatsResource, but modified for cyclone. May
# be missing features.
class StatsForCyclone(cyclone.web.RequestHandler):

    def get(self):
        parts = []
        statDict = greplin.scales.twistedweb.util.lookup(
            scales.getStats(), parts)

        if statDict is None:
            self.set_status(404)
            self.write("Path not found.")
            return

        query = self.get_argument('query', default=None)

        if self.get_argument('format', default=None) == 'json':
            self.set_header('content-type', 'text/javascript; charset=UTF-8')
            greplin.scales.formats.jsonFormat(self, statDict, query)
        elif self.get_argument('format', default=None) == 'prettyjson':
            self.set_header('content-type', 'text/javascript; charset=UTF-8')
            greplin.scales.formats.jsonFormat(self,
                                              statDict,
                                              query,
                                              pretty=True)
        else:
            greplin.scales.formats.htmlHeader(self, '/' + '/'.join(parts),
                                              'svr', query)
            greplin.scales.formats.htmlFormat(self, tuple(parts), statDict,
                                              query)