#! /usr/bin/python

# Copyright 2017, 2018 Max H. Parke KA1RBI
# 
# This file is part of OP25
# 
# OP25 is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
# 
# OP25 is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
# License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with OP25; see the file COPYING. If not, write to the Free
# Software Foundation, Inc., 51 Franklin Street, Boston, MA
# 02110-1301, USA.

import sys
import os
import time
import re
import json
import socket
import traceback
import threading
import glob
import subprocess
import zmq

from gnuradio import gr
from waitress.server import create_server
from optparse import OptionParser
from multi_rx import byteify
from tsvfile import load_tsv, make_config

my_input_q = None
my_output_q = None
my_recv_q = None
my_port = None
my_backend = None
CFG_DIR = '../www/config/'
TSV_DIR = './'

"""
fake http and ajax server module
TODO: make less fake
"""

def static_file(environ, start_response):
    content_types = { 'png': 'image/png', 'jpeg': 'image/jpeg', 'jpg': 'image/jpeg', 'gif': 'image/gif', 'css': 'text/css', 'js': 'application/javascript', 'html': 'text/html'}
    img_types = 'png jpg jpeg gif'.split()
    if environ['PATH_INFO'] == '/':
        filename = 'index.html'
    else:
        filename = re.sub(r'[^a-zA-Z0-9_.\-]', '', environ['PATH_INFO'])
    suf = filename.split('.')[-1]
    pathname = '../www/www-static'
    if suf in img_types:
        pathname = '../www/images'
    pathname = '%s/%s' % (pathname, filename)
    if suf not in content_types.keys() or '..' in filename or not os.access(pathname, os.R_OK):
        sys.stderr.write('404 %s\n' % pathname)
        status = '404 NOT FOUND'
        content_type = 'text/plain'
        output = status
    else:
        output = open(pathname).read()
        content_type = content_types[suf]
        status = '200 OK'
    return status, content_type, output

def valid_tsv(filename):
    if not os.access(filename, os.R_OK):
        return False
    line = open(filename).readline()
    for word in 'Sysname Offset NAC Modulation TGID Whitelist Blacklist'.split():
        if word not in line:
            return False
    return True

def tsv_config(filename):
    DEFAULT_CFG = '../www/config/default.json'
    filename = '%s%s' % (TSV_DIR, filename)
    filename = filename.replace('[TSV]', '.tsv')
    if not valid_tsv(filename):
        return None
    cfg = make_config(load_tsv(filename))
    default_cfg = json.loads(open(DEFAULT_CFG).read())

    result = default_cfg
    channels = [ {'active': True,
                  'blacklist': cfg[nac]['blacklist'],
                  'whitelist': cfg[nac]['whitelist'],
                  'cclist': cfg[nac]['cclist'],
                  'demod_type': 'cqpsk',
                  'destination': 'udp://127.0.0.1:23456',
                  'filter_type': 'rc',
                  'frequency': 500000000,
                  'if_rate': 24000,
                  'nac': nac,
                  'name': cfg[nac]['sysname'],
                  'phase2_tdma': False,
                  'plot': "",
                  'tgids': cfg[nac]['tgid_map'],
                  'trunked': True
                 }
                for nac in cfg.keys() ]
    result['channels'] = channels
    return {'json_type':'config_data', 'data': result}

def do_request(d):
    global my_backend
    if d['command'].startswith('rx-'):
        msg = gr.message().make_from_string(json.dumps(d), -2, 0, 0)
        if not my_backend.input_q.full_p():
            my_backend.input_q.insert_tail(msg)
        return None
    elif d['command'] == 'config-load':
        if '[TSV]' in d['data']:
            return tsv_config(d['data'])
        filename = '%s%s.json' % (CFG_DIR, d['data'])
        if not os.access(filename, os.R_OK):
            return None
        js_msg = json.loads(open(filename).read())
        return {'json_type':'config_data', 'data': js_msg}
    elif d['command'] == 'config-list':
        files = glob.glob('%s*.json' % CFG_DIR)
        files = [x.replace('.json', '') for x in files]
        files = [x.replace(CFG_DIR, '') for x in files]
        if d['data'] == 'tsv':
            tsvfiles = glob.glob('%s*.tsv' % TSV_DIR)
            tsvfiles = [x for x in tsvfiles if valid_tsv(x)]
            tsvfiles = [x.replace('.tsv', '[TSV]') for x in tsvfiles]
            tsvfiles = [x.replace(TSV_DIR, '') for x in tsvfiles]
            files += tsvfiles
        return {'json_type':'config_list', 'data': files}
    elif d['command'] == 'config-save':
        name = d['data']['name']
        if '..' in name or '.json' in name or '/' in name:
            return None
        filename = '%s%s.json' % (CFG_DIR, d['data']['name'])
        open(filename, 'w').write(json.dumps(d['data']['value'], indent=4, separators=[',',':'], sort_keys=True))
        return None

def post_req(environ, start_response, postdata):
    global my_input_q, my_output_q, my_recv_q, my_port
    resp_msg = []
    try:
        data = json.loads(postdata)
    except:
        sys.stderr.write('post_req: error processing input: %s:\n' % (postdata))
        traceback.print_exc(limit=None, file=sys.stderr)
        sys.stderr.write('*** end traceback ***\n')
    for d in data:
        if d['command'].startswith('config-') or d['command'].startswith('rx-'):
            resp = do_request(d)
            if resp:
                resp_msg.append(resp)
            continue
        if d['command'].startswith('settings-'):
            msg = gr.message().make_from_string(json.dumps(d), -4, 0, 0)
        else:
            msg = gr.message().make_from_string(str(d['command']), -2, d['data'], 0)
        if my_output_q.full_p():
            my_output_q.delete_head_nowait()   # ignores result
        if not my_output_q.full_p():
            my_output_q.insert_tail(msg)
    time.sleep(0.2)

    while not my_recv_q.empty_p():
        msg = my_recv_q.delete_head()
        if msg.type() == -4:
            resp_msg.append(json.loads(msg.to_string()))
    status = '200 OK'
    content_type = 'application/json'
    output = json.dumps(resp_msg)
    return status, content_type, output

def http_request(environ, start_response):
    if environ['REQUEST_METHOD'] == 'GET':
        status, content_type, output = static_file(environ, start_response)
    elif environ['REQUEST_METHOD'] == 'POST':
        postdata = environ['wsgi.input'].read()
        status, content_type, output = post_req(environ, start_response, postdata)
    else:
        status = '200 OK'
        content_type = 'text/plain'
        output = status
        sys.stderr.write('http_request: unexpected input %s\n' % environ['PATH_INFO'])
    
    response_headers = [('Content-type', content_type),
                        ('Content-Length', str(len(output)))]
    start_response(status, response_headers)

    return [output]

def application(environ, start_response):
    failed = False
    try:
        result = http_request(environ, start_response)
    except:
        failed = True
        sys.stderr.write('application: request failed:\n%s\n' % traceback.format_exc())
        sys.exit(1)
    return result

def process_qmsg(msg):
    if my_recv_q.full_p():
        my_recv_q.delete_head_nowait()   # ignores result
    if my_recv_q.full_p():
        return
    my_recv_q.insert_tail(msg)

class http_server(object):
    def __init__(self, input_q, output_q, endpoint, **kwds):
        global my_input_q, my_output_q, my_recv_q, my_port
        host, port = endpoint.split(':')
        if my_port is not None:
            raise AssertionError('this server is already active on port %s' % my_port)
        my_input_q = input_q
        my_output_q = output_q
        my_port = int(port)

        my_recv_q = gr.msg_queue(10)
        self.q_watcher = queue_watcher(my_input_q, process_qmsg)

        self.server = create_server(application, host=host, port=my_port)

    def run(self):
        self.server.run()

class queue_watcher(threading.Thread):
    def __init__(self, msgq,  callback, **kwds):
        threading.Thread.__init__ (self, **kwds)
        self.setDaemon(1)
        self.msgq = msgq
        self.callback = callback
        self.keep_running = True
        self.start()

    def run(self):
        while(self.keep_running):
            msg = self.msgq.delete_head()
            self.callback(msg)

class Backend(threading.Thread):
    def __init__(self, options, input_q, output_q, init_config=None, **kwds):
        threading.Thread.__init__ (self, **kwds)
        self.setDaemon(1)
        self.keep_running = True
        self.rx_options = None
        self.input_q = input_q
        self.output_q = output_q
        self.verbosity = options.verbosity

        self.zmq_context = zmq.Context()
        self.zmq_port = options.zmq_port

        self.zmq_sub = self.zmq_context.socket(zmq.SUB)
        self.zmq_sub.connect('tcp://localhost:%d' % self.zmq_port)
        self.zmq_sub.setsockopt(zmq.SUBSCRIBE, '')

        self.zmq_pub = self.zmq_context.socket(zmq.PUB)
        self.zmq_pub.sndhwm = 5
        self.zmq_pub.bind('tcp://*:%d' % (self.zmq_port+1))

        self.start()
        self.subproc = None
        self.msg = None

        self.q_watcher = queue_watcher(self.input_q, self.process_msg)

        if init_config:
            d = {'command': 'rx-start', 'data': init_config}
            msg = gr.message().make_from_string(json.dumps(d), -4, 0, 0)
            self.input_q.insert_tail(msg)

    def publish(self, msg):
        t = msg.type()
        s = msg.to_string()
        a = msg.arg1()
        self.zmq_pub.send(json.dumps({'command': s, 'data': a, 'msgtype': t}))

    def check_subproc(self):	# return True if subprocess is active
        if not self.subproc:
            return False
        rc = self.subproc.poll()
        if rc is None:
            return True
        else:
            self.subproc.wait()
            self.subproc = None
            return False

    def process_msg(self, msg):
        def make_command(options, config_file):
            trunked_ct = [True for x in options._js_config['channels'] if x['trunked']]
            total_ct = [True for x in options._js_config['channels']]
            if trunked_ct and len(trunked_ct) != len(total_ct):
                self.msg = 'no suitable backend found for this configuration'
                return None
            if not trunked_ct:
                self.backend = '%s/%s' % (os.getcwd(), 'multi_rx.py')
                opts = [self.backend]
                filename = '%s%s.json' % (CFG_DIR, config_file)
                opts.append('--config-file')
                opts.append(filename)
                return opts

            types = {'costas-alpha': 'float', 'trunk-conf-file': 'str', 'demod-type': 'str', 'logfile-workers': 'int', 'decim-amt': 'int', 'wireshark-host': 'str', 'gain-mu': 'float', 'phase2-tdma': 'bool', 'seek': 'int', 'ifile': 'str', 'pause': 'bool', 'antenna': 'str', 'calibration': 'float', 'fine-tune': 'float', 'raw-symbols': 'str', 'audio-output': 'str', 'vocoder': 'bool', 'input': 'str', 'wireshark': 'bool', 'gains': 'str', 'args': 'str', 'sample-rate': 'int', 'terminal-type': 'str', 'gain': 'float', 'excess-bw': 'float', 'offset': 'float', 'audio-input': 'str', 'audio': 'bool', 'plot-mode': 'str', 'audio-if': 'bool', 'tone-detect': 'bool', 'frequency': 'int', 'freq-corr': 'float', 'hamlib-model': 'int', 'udp-player': 'bool', 'verbosity': 'int'}
            self.backend = '%s/%s' % (os.getcwd(), 'rx.py')
            opts = [self.backend]
            for k in [ x for x in dir(options) if not x.startswith('_') ]:
                kw = k.replace('_', '-')
                val = getattr(options, k)
                if kw not in types.keys():
                    self.msg = 'make_command: unknown option: %s %s type %s' % (k, val, type(val))
                    return None
                elif types[kw] == 'str':
                    if val:
                        opts.append('--%s' % kw)
                        opts.append('%s' % (val))
                elif types[kw] == 'float':
                    opts.append('--%s' % kw)
                    if val:
                        opts.append('%f' % (val))
                    else:
                        opts.append('%f' % (0))
                elif types[kw] == 'int':
                    opts.append('--%s' % kw)
                    if val:
                        opts.append('%d' % (val))
                    else:
                        opts.append('%d' % (0))
                elif types[kw] == 'bool':
                    if val:
                        opts.append('--%s' % kw)
                else:
                    self.msg = 'make_command: unknown2 option: %s %s type %s' % (k, val, type(val))
                    return None
            return opts

        msg = json.loads(msg.to_string())
        if msg['command'] == 'rx-start':
            if self.check_subproc():
                self.msg = 'start command failed: subprocess pid %d already active' % self.subproc.pid
                return
            options = rx_options(msg['data'])
            if getattr(options, '_js_config', None) is None:
                self.msg = 'start command failed: rx_options: unable to initialize config=%s' % (msg['data'])
                return
            options.verbosity = self.verbosity
            options.terminal_type = 'zmq:tcp:%d' % (self.zmq_port)
            cmd = make_command(options, msg['data'])
            if cmd:
                self.subproc = subprocess.Popen(cmd)
        elif msg['command'] == 'rx-stop':
            if not self.check_subproc():
                self.msg = 'stop command failed: subprocess not active'
                return
            if msg['data'] == 'kill':
                self.subproc.kill()
            else:
                self.subproc.terminate()
        elif msg['command'] == 'rx-state':
            d = {}
            if self.check_subproc():
                d['rx-state'] = 'subprocess pid %d active' % self.subproc.pid
            else:
                d['rx-state'] = 'subprocess not active, last msg: %s' % self.msg
            msg = gr.message().make_from_string(json.dumps(d), -4, 0, 0)
            if not self.output_q.full_p():
                self.output_q.insert_tail(msg)

    def run(self):
        while self.keep_running:
            js = self.zmq_sub.recv()
            if not self.keep_running:
                break
            msg = gr.message().make_from_string(js, -4, 0, 0)
            if not self.output_q.full_p():
                self.output_q.insert_tail(msg)

class rx_options(object):
    def __init__(self, name):
        def map_name(k):
            return k.replace('-', '_')

        filename = '%s%s.json' % (CFG_DIR, name)
        if not os.access(filename, os.R_OK):
            sys.stderr.write('unable to access config file %s\n' % (filename))
            return
        config = byteify(json.loads(open(filename).read()))
        dev = [x for x in config['devices'] if x['active']][0]
	if not dev:
            return
        chan = [x for x in config['channels'] if x['active']][0]
	if not chan:
            return
        options = object()
        for k in config['backend-rx'].keys():
            setattr(self, map_name(k), config['backend-rx'][k])
        for k in 'args frequency gains offset'.split():
            setattr(self, k, dev[k])
        self.demod_type = chan['demod_type']
        self.freq_corr = dev['ppm']
        self.sample_rate = dev['rate']
        self.plot_mode = chan['plot']
        self.phase2_tdma = chan['phase2_tdma']
        self.trunk_conf_file = filename
        self._js_config = config

def http_main():
    global my_backend
    # command line argument parsing
    parser = OptionParser()
    parser.add_option("-c", "--config", type="string", default=None, help="config json name, without prefix/suffix")
    parser.add_option("-e", "--endpoint", type="string", default="127.0.0.1:8080", help="address:port to listen on (use addr 0.0.0.0 to enable external clients)")
    parser.add_option("-v", "--verbosity", type="int", default=0, help="message debug level")
    parser.add_option("-p", "--pause", action="store_true", default=False, help="block on startup")
    parser.add_option("-z", "--zmq-port", type="int", default=25000, help="backend sub port")
    (options, args) = parser.parse_args()

    # wait for gdb
    if options.pause:
        print 'Ready for GDB to attach (pid = %d)' % (os.getpid(),)
        raw_input("Press 'Enter' to continue...")

    input_q = gr.msg_queue(20)
    output_q = gr.msg_queue(20)
    backend_input_q = gr.msg_queue(20)
    backend_output_q = gr.msg_queue(20)

    my_backend = Backend(options, backend_input_q, backend_output_q, init_config=options.config)
    server = http_server(input_q, output_q, options.endpoint)
    q_watcher = queue_watcher(output_q, lambda msg : my_backend.publish(msg))
    backend_q_watcher = queue_watcher(backend_output_q, lambda msg : process_qmsg(msg))

    server.run()

if __name__ == '__main__':
    http_main()