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