ferris.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright 2019 Tobi Schäfer.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import entrecote
import json

from datetime import datetime, timedelta, timezone
from gzip import compress
from wsgiref.handlers import format_date_time

def ferris(request, make_response):
    response_headers = {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': '*'
    }

    group = request.args.get('group')

    if not group:
        return make_response('', 400)

    key = request.args.get('url')
    callback = request.args.get('callback')

    if key:
        metadata = request.args.get('metadata')
        entry = entrecote.add(group, key, metadata)
        return make_response(str(entry['count']), 201)

    else:
        referrers = sorted(
            map(
                lambda entry: {
                    'url': entry[0],
                    'hits': entry[1]['count'],
                    'metadata': entry[1]['metadata'] if 'metadata' in entry[1] else {}
                },
                entrecote.get(group)
            ),
            key=lambda entry: entry['hits'],
            reverse=True
        )

        # Let the browser cache the referrer list for 10 minutes
        response_headers['Expires'] = format_date_time(datetime.now(timezone.utc).timestamp() + 600)

        data = json.dumps(list(referrers))

        if callback:
            data = '%s(%s)' % (callback, data)
            response_headers['Content-Type'] = 'application/javascript'

        if request.headers.get('Accept-Encoding', '').find('gzip') > -1:
            response_headers['Content-Encoding'] = 'gzip'
            data = compress(data.encode('utf-8'))

        response = make_response(data)
        response.headers = response_headers

        return response


def cleanup(request, make_response):
    if request.remote_addr != '127.0.0.1':
        return make_response('', 401)

    return str(entrecote.truncate(request.args.get('group')))