#!/usr/bin/env python3
#
# (c) 2016 Fetal-Neonatal Neuroimaging & Developmental Science Center
#                   Boston Children's Hospital
#
#              http://childrenshospital.org/FNNDSC/
#                        dev@babyMRI.org
#

import sys
import os
# sys.path.insert(1, os.path.join(os.path.dirname(__file__), '..'))
from    distutils.sysconfig import get_python_lib

from    argparse            import RawTextHelpFormatter
from    argparse            import ArgumentParser
import  socket
# import  pman
import  threading

sys.path.insert(1, os.path.join(os.path.dirname(__file__), '../pman'))
try:
    import pman
except:
    sys.path.insert(1, os.path.join(get_python_lib(True, True), 'site-packages'))
    sys.path.insert(1, os.path.join(get_python_lib(True, True), 'site-packages/pman'))
    import pman

try:
    from    _colors         import Colors
except:
    from    pman._colors    import Colors
    

str_defIP = [l for l in ([ip for ip in socket.gethostbyname_ex(socket.gethostname())[2] if not ip.startswith("127.")][:1], [[(s.connect(('8.8.8.8', 53)), s.getsockname()[0], s.close()) for s in [socket.socket(socket.AF_INET, socket.SOCK_DGRAM)]][0][1]]) if l][0][0]

str_version = "1.5.4"
str_name    = 'pman'
str_desc    = Colors.CYAN + """

 _ __  _ __ ___   __ _ _ __
| '_ \| '_ ` _ \ / _` | '_ \\
| |_) | | | | | | (_| | | | |
| .__/|_| |_| |_|\__,_|_| |_|
| |
|_|


font generated by:
http://patorjk.com/software/taag/#p=display&f=Doom&t=pman


                            Process-MANager

           A simple compute process tracking and management framework.

                              -- version """ + \
           Colors.YELLOW + str_version + Colors.CYAN + """ --
           
    'pman' is a server that spawns and tracks processes executed on an underlying
    shell/system.
    
    'pman' listens for REST comms and provides facilities for running, stopping,
    querying, and managing command-line processes.
    
""" + \
           Colors.BLINK_RED +  """

              +---------------------------------------------------+
              | NOTE THAT 'pman' DOES NOT CURRENTLY AUTHENTICATE  |
              | AND WILL HONOR *ALL* PROCESS MANAGEMENT REQUESTS! |
              +---------------------------------------------------+
              
""" + Colors.NO_COLOUR

parser  = ArgumentParser(description = str_desc, formatter_class = RawTextHelpFormatter)

parser.add_argument(
    '--ip',
    action  = 'store',
    dest    = 'ip',
    default = str_defIP,
    help    = 'IP to connect.'
)
parser.add_argument(
    '--port',
    action  = 'store',
    dest    = 'port',
    default = '5010',
    help    = 'Port to use.'
)
parser.add_argument(
    '--protocol',
    action  = 'store',
    dest    = 'protocol',
    default = 'tcp',
    help    = 'Protocol to use.'
)
parser.add_argument(
    '--rawmode',
    action  = 'store',
    dest    = 'raw',
    default = '1',
    help    = 'Router raw mode.'
)
parser.add_argument(
    '--listeners',
    action  = 'store',
    dest    = 'listeners',
    default = '1',
    help    = 'Number of listeners.'
)
parser.add_argument(
    '--http',
    action  = 'store_true',
    dest    = 'http',
    default = False,
    help    = 'Send HTTP formatted replies with content-type html; otherwise content-type json.'
)
parser.add_argument(
    '--debugToFile',
    action  = 'store_true',
    dest    = 'debugToFile',
    default = False,
    help    = 'If true, stream debugging info to file.'
)
parser.add_argument(
    '--debugFile',
    action  = 'store',
    dest    = 'debugFile',
    default = '%s/tmp/debug-pman.log' % os.environ['HOME'],
    help    = 'In conjunction with --debugToFile, stream debugging info to specified file.'
)
parser.add_argument(
    '--listenerSleep',
    action  = 'store',
    dest    = 'listenerSleep',
    default = '0.1',
    help    = 'A delay in each listener loop to prevent busy-wait.'
)
parser.add_argument(
    '--DBsavePeriod',
    action  = 'store',
    dest    = 'DBsavePeriod',
    default = '60',
    help    = 'The periodicity, in seconds, for the internal DB to be flushed to disk.'
)
parser.add_argument(
    '--version',
    help    = 'if specified, print version number',
    dest    = 'b_version',
    action  = 'store_true',
    default = False
)

args = parser.parse_args()

# server          = ThreadedHTTPServer((args.ip, int(args.port)), StoreHandler)
# server.setup(args = vars(args))

if args.b_version:
    print("Version: %s" % str_version)
    sys.exit(1)

comm = pman.pman(
    IP              = args.ip,
    port            = args.port,
    protocol        = args.protocol,
    raw             = args.raw,
    listeners       = args.listeners,
    http            = args.http,
    listenerSleep   = args.listenerSleep,
    DBsavePeriod    = args.DBsavePeriod,
    debugToFile     = args.debugToFile,
    debugFile       = args.debugFile,
    desc            = str_desc,
    name            = str_name,
    version         = str_version
    )

# This starts the main thread and "detaches" execution from the main
# program.
t1 = threading.Thread(target = comm.thread_serve)
t1.start()

# Execution will block until pman receives an asynchronous http quit
# request, i.e.
# pfurl   --content-type application/vnd.collection+json      \
#         --verb POST                                         \
#         --rawmode --http 192.168.1.189:5010/api/v1/cmd      \
#         --jsonwrapper 'payload'                             \
#         --msg '{    "action": "quit",
#                     "meta": {
#                         "when":         "now",
#                         "saveDB":       true
#                     }
#                 }'

