this repo has no description
edavis.dev
1#!/usr/bin/env python3
2
3from flask import Flask, request, jsonify
4from prometheus_client import Counter, make_wsgi_app
5from werkzeug.middleware.dispatcher import DispatcherMiddleware
6from werkzeug.datastructures import LanguageAccept
7
8from feed_manager import feed_manager
9from feeds.rapidfire import RapidFireFeed
10from feeds.popular import PopularFeed
11
12feed_requests = Counter('feed_requests', 'requests by feed URI', ['feed'])
13
14app = Flask(__name__)
15
16@app.route('/xrpc/app.bsky.feed.getFeedSkeleton')
17def get_feed_skeleton():
18 try:
19 limit = int(request.args.get('limit', 50))
20 except ValueError:
21 limit = 50
22
23 try:
24 offset = int(request.args.get('cursor', 0))
25 except ValueError:
26 offset = 0
27
28 feed_uri = request.args['feed']
29 if feed_uri.endswith('-dev'):
30 feed_uri = feed_uri.replace('-dev', '')
31 else:
32 (prefix, sep, rkey) = feed_uri.rpartition('/')
33 feed_requests.labels(rkey).inc()
34
35 if request.args.getlist('langs'):
36 req_langs = request.args.getlist('langs')
37 langs = LanguageAccept([(l, 1) for l in req_langs])
38 else:
39 langs = request.accept_languages
40
41 if request.args.get('debug', '0') == '1':
42 headers = {'Content-Type': 'text/plain; charset=utf-8'}
43 debug = feed_manager.serve_feed(feed_uri, limit, offset, langs, debug=True)
44 return debug, headers
45
46 posts = feed_manager.serve_feed(feed_uri, limit, offset, langs, debug=False)
47 offset += len(posts)
48
49 return dict(cursor=str(offset), feed=[dict(post=uri) for uri in posts])
50
51app.wsgi_app = DispatcherMiddleware(app.wsgi_app, {
52 '/metrics': make_wsgi_app()
53})
54
55if __name__ == '__main__':
56 from feedweb_utils import did_doc
57 app.add_url_rule('/.well-known/did.json', view_func=did_doc)
58
59 app.run(debug=True)