#!/usr/bin/env python3

"""
Stand-alone script to manage the GNU Taler
survey frontend.
"""

import argparse
import sys
import os
import site


os.environ.setdefault("TALER_PREFIX", "@prefix@")
site.addsitedir("%s/lib/python%d.%d/site-packages" % (
    "@prefix@", 
    sys.version_info.major,
    sys.version_info.minor))

from talersurvey.talerconfig import TalerConfig
import logging
logger = logging.getLogger(__name__)

# No perfect match to our logging format, but good enough ...
uwsgi_logfmt = "%(ltime) %(proto) %(method) %(uri) %(proto) => %(status)"

def handle_serve_http(args):
    tc = TalerConfig.from_file(os.environ.get("TALER_CONFIG_FILE"))
    port = args.port
    if port is None:
        port = tc["survey"]["http_port"].value_int(required=True)
    spec = ":%d" % (port,)
    os.execlp("uwsgi", "uwsgi",
            "--master",
            "--die-on-term",
            "--log-format", uwsgi_logfmt,
            "--http", spec,
            "--wsgi-file", "@prefix@/share/taler/frontend-survey.wsgi")

def handle_serve_uwsgi(args):
    tc = TalerConfig.from_file(os.environ.get("TALER_CONFIG_FILE"))
    serve_uwsgi = tc["survey"]["uwsgi_serve"].value_string(required=True).lower()
    params = ["uwsgi", "uwsgi",
        "--master",
        "--die-on-term",
        "--log-format", uwsgi_logfmt,
        "--wsgi-file", "@prefix@/share/taler/frontend-survey.wsgi"]
    if serve_uwsgi == "tcp":
        port = tc["survey"]["uwsgi_port"].value_int(required=True)
        spec = ":%d" % (port,)
        params.extend(["--socket", spec])
    elif serve_uwsgi == "unix":
        spec = tc["survey"]["uwsgi_unixpath"].value_filename(required=True)
        mode = tc["survey"]["uwsgi_unixpath_mode"].value_filename(required=True)
        params.extend(["--socket", spec])
        params.extend(["--chmod-socket="+mode])
        os.makedirs(os.path.dirname(spec), exist_ok=True)
    logging.info("launching uwsgi with argv %s", params[1:])
    os.execlp(*params)



parser = argparse.ArgumentParser()
parser.set_defaults(func=None)
parser.add_argument('--config', '-c', help="configuration file to use", metavar="CONFIG", type=str, dest="config", default=None)
sub = parser.add_subparsers()

p = sub.add_parser('serve-http', help="Serve over HTTP")
p.add_argument("--port", "-p", dest="port", type=int, default=None, metavar="PORT")
p.set_defaults(func=handle_serve_http)

p = sub.add_parser('serve-uwsgi', help="Serve over UWSGI")
p.set_defaults(func=handle_serve_uwsgi)

args = parser.parse_args()
if getattr(args, 'func', None) is None:
    parser.print_help()
    sys.exit(1)

if args.config is not None:
    os.environ["TALER_CONFIG_FILE"] = args.config

args.func(args)