from flask import current_app, g, make_response
from flask.ext.login import login_required
from piecrust.app import PieCrustFactory
from piecrust.serving.server import Server
from ..blueprint import foodtruck_bp


@foodtruck_bp.route('/site/<sitename>/')
@login_required
def preview_site_root(sitename):
    return preview_site(sitename, '/')


@foodtruck_bp.route('/site/<sitename>/<path:url>')
@login_required
def preview_site(sitename, url):
    root_dir = g.sites.get_root_dir(sitename)
    appfactory = PieCrustFactory(
        root_dir,
        cache_key='foodtruck',
        debug=current_app.debug)
    server = Server(appfactory,
                    root_url='/site/%s/' % sitename)
    return make_response(server._run_request)