1 # Copyright (C) 2015 Ipsilon Contributors see COPYING for license
4 from ipsilon.util.log import Log
5 from ipsilon.util.user import UserSession
6 from urllib import unquote
8 from urlparse import urlparse
10 # pylint: disable=no-name-in-module, import-error
11 from urllib.parse import urlparse
15 def __init__(self, site):
17 self.basepath = cherrypy.config.get('base.mount', "")
19 self.default_headers = {
20 'Cache-Control': 'no-cache, no-store, must-revalidate, private',
23 self.auth_protect = False
26 return cherrypy.url(relative=False)
28 def instance_base_url(self):
30 s = urlparse(unquote(url))
31 return '%s://%s%s' % (s.scheme, s.netloc, self.basepath)
33 def _check_referer(self, referer, url):
34 r = urlparse(unquote(referer))
35 u = urlparse(unquote(url))
36 if r.scheme != u.scheme:
38 if r.netloc != u.netloc:
40 if r.path.startswith(self.basepath):
44 def __call__(self, *args, **kwargs):
45 # pylint: disable=star-args
46 cherrypy.response.headers.update(self.default_headers)
48 self.user = UserSession().get_user()
50 if self.auth_protect and self.user.is_anonymous:
51 raise cherrypy.HTTPError(401)
53 self._debug("method: %s" % cherrypy.request.method)
54 op = getattr(self, cherrypy.request.method, None)
56 # Basic CSRF protection
57 if cherrypy.request.method != 'GET':
59 if 'referer' not in cherrypy.request.headers:
60 self._debug("Missing referer in %s request to %s"
61 % (cherrypy.request.method, url))
62 raise cherrypy.HTTPError(403)
63 referer = cherrypy.request.headers['referer']
64 if not self._check_referer(referer, url):
65 self._debug("Wrong referer %s in request to %s"
67 raise cherrypy.HTTPError(403)
68 return op(*args, **kwargs)
70 op = getattr(self, 'root', None)
72 return op(*args, **kwargs)
74 return self.default(*args, **kwargs)
76 def default(self, *args, **kwargs):
77 raise cherrypy.NotFound()
79 def add_subtree(self, name, page):
80 self.__dict__[name] = page
82 def del_subtree(self, name):
83 del self.__dict__[name]