kopia lustrzana https://github.com/snarfed/bridgy-fed
38 wiersze
1.2 KiB
Python
38 wiersze
1.2 KiB
Python
"""HTTP proxy that injects our webmention endpoint.
|
|
"""
|
|
import datetime
|
|
import urllib.parse
|
|
|
|
import flask
|
|
from flask import request
|
|
from oauth_dropins.webutil import flask_util
|
|
from oauth_dropins.webutil.flask_util import error
|
|
import requests
|
|
|
|
from app import app, cache
|
|
import common
|
|
|
|
LINK_HEADER = '<%s>; rel="webmention"'
|
|
CACHE_TIME = datetime.timedelta(seconds=15)
|
|
|
|
|
|
@app.get(r'/wm/<path:url>')
|
|
@flask_util.cached(cache, CACHE_TIME)
|
|
def add_wm(url=None):
|
|
"""Proxies HTTP requests and adds Link header to our webmention endpoint."""
|
|
url = urllib.parse.unquote(url)
|
|
if not url.startswith('http://') and not url.startswith('https://'):
|
|
error('URL must start with http:// or https://')
|
|
|
|
try:
|
|
got = common.requests_get(url)
|
|
except requests.exceptions.Timeout as e:
|
|
error(str(e), status=504, exc_info=True)
|
|
except requests.exceptions.RequestException as e:
|
|
error(str(e), status=502, exc_info=True)
|
|
|
|
resp = flask.make_response(got.content, got.status_code, dict(got.headers))
|
|
resp.headers.add('Link', LINK_HEADER % (request.args.get('endpoint') or
|
|
request.host_url + 'webmention'))
|
|
return resp
|