108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
from flask import g, request, safe_join, url_for
|
|
import os.path
|
|
|
|
from i2p2www import CANONICAL_DOMAIN, CURRENT_I2P_VERSION, STATIC_DIR, THIS_DOMAIN, app
|
|
|
|
INPROXY = '.us'
|
|
|
|
I2P_TO_CLEAR = {
|
|
'www.i2p2.i2p': THIS_DOMAIN,
|
|
'forum.i2p': 'forum.i2p', # Don't convert forum.i2p, it is not accessible outside I2P
|
|
'trac.i2p2.i2p': 'trac.i2p2.de',
|
|
'mail.i2p': 'i2pmail.org',
|
|
}
|
|
|
|
|
|
####################
|
|
# Template functions
|
|
|
|
@app.context_processor
|
|
def utility_processor():
|
|
# Shorthand for getting a site url
|
|
def get_site_url(path=None):
|
|
lang = 'en'
|
|
if hasattr(g, 'lang') and g.lang:
|
|
lang = g.lang
|
|
if path:
|
|
return url_for('site_show', lang=lang, page=path)
|
|
else:
|
|
return url_for('site_show', lang=lang)
|
|
|
|
# Shorthand for getting a language-specific url
|
|
def get_url_with_lang(endpoint, **args):
|
|
lang = 'en'
|
|
if hasattr(g, 'lang') and g.lang:
|
|
lang = g.lang
|
|
return url_for(endpoint, lang=lang, **args)
|
|
|
|
# Get a specific language flag, or the flag for the current language
|
|
def get_flag(lang=None):
|
|
if not lang:
|
|
if hasattr(g, 'lang') and g.lang:
|
|
lang = g.lang
|
|
else:
|
|
lang = 'en'
|
|
return url_for('static', filename='images/flags/'+lang+'.png')
|
|
|
|
# Provide the canonical link to the current page
|
|
def get_canonical_link():
|
|
protocol = request.url.split('//')[0]
|
|
return protocol + '//' + CANONICAL_DOMAIN + request.path
|
|
|
|
# Convert an I2P url to an equivalent clearnet one
|
|
def convert_url_to_clearnet(value):
|
|
if not value.endswith('.i2p'):
|
|
# The url being passed in isn't an I2P url, so just return it
|
|
return value
|
|
if request.headers.get('X-I2P-Desthash') and not request.headers.get('X-Forwarded-Server'):
|
|
# The request is from within I2P, so use I2P url
|
|
return value
|
|
# The request is either directly from clearnet or through an inproxy
|
|
try:
|
|
# Return the known clearnet url corresponding to the I2P url
|
|
return I2P_TO_CLEAR[value]
|
|
except KeyError:
|
|
# The I2P site has no known clearnet address, so use an inproxy
|
|
return value + INPROXY
|
|
|
|
# Convert a paginated URL to that of another page
|
|
def url_for_other_page(page):
|
|
args = request.view_args.copy()
|
|
args['page'] = page
|
|
return url_for(request.endpoint, **args)
|
|
|
|
# Change the theme of the current page
|
|
def change_theme(theme):
|
|
args = {}
|
|
if request.view_args:
|
|
args = request.view_args.copy()
|
|
args['theme'] = theme
|
|
if request.endpoint:
|
|
return url_for(request.endpoint, **args)
|
|
# Probably a 404 error page
|
|
return url_for('main_index', **args)
|
|
|
|
# Shorthand for getting the logo for the current theme
|
|
def get_logo_for_theme():
|
|
logo = 'styles/' + g.theme + '/images/i2plogo.png'
|
|
if not os.path.isfile(safe_join(STATIC_DIR, logo)):
|
|
logo = 'images/i2plogo.png'
|
|
return logo
|
|
|
|
def get_current_version(string=None, ver=None):
|
|
if string:
|
|
if ver:
|
|
return string % ver
|
|
return string % CURRENT_I2P_VERSION
|
|
return CURRENT_I2P_VERSION
|
|
|
|
return dict(i2pconv=convert_url_to_clearnet,
|
|
url_for_other_page=url_for_other_page,
|
|
change_theme=change_theme,
|
|
logo_url=get_logo_for_theme,
|
|
site_url=get_site_url,
|
|
get_url=get_url_with_lang,
|
|
get_flag=get_flag,
|
|
ver=get_current_version,
|
|
canonical=get_canonical_link)
|