""" Defines helper functions for creating kernel entry points and process
launchers.
"""

# Standard library imports.
import logging
import atexit
import os
import socket
from subprocess import Popen, PIPE
import sys

# System library imports.
import zmq
from zmq.log import handlers
# Local imports.
from IPython.core.ultratb import FormattedTB
from IPython.external.argparse import ArgumentParser
from IPython.zmq.log import logger

def split_ports(s, n):
    """Parser helper for multiport strings"""
    if not s:
        return tuple([0]*n)
    ports = map(int, s.split(','))
    if len(ports) != n:
        raise ValueError
    return ports

def select_random_ports(n):
    """Selects and return n random ports that are open."""
    ports = []
    for i in xrange(n):
        sock = socket.socket()
        sock.bind(('', 0))
        ports.append(sock)
    for i, sock in enumerate(ports):
        port = sock.getsockname()[1]
        sock.close()
        ports[i] = port
    return ports

def parse_url(args):
    if args.url:
        iface = args.url.split('://',1)
        if len(args) == 2:
            args.transport,iface = iface
        iface = iface.split(':')
        args.ip = iface[0]
        if iface[1]:
            args.regport = iface[1]
    args.url = "%s://%s:%i"%(args.transport, args.ip,args.regport)
    
        

def make_base_argument_parser():
    """ Creates an ArgumentParser for the generic arguments supported by all 
    ipcluster entry points.
    """
    parser = ArgumentParser()
    parser.add_argument('--ip', type=str, default='127.0.0.1',
                        help='set the controller\'s IP address [default: local]')
    parser.add_argument('--transport', type=str, default='tcp',
                        help='set the transport to use [default: tcp]')
    parser.add_argument('--regport', type=int, metavar='PORT', default=10101,
                        help='set the XREP port for registration [default: 10101]')
    parser.add_argument('--logport', type=int, metavar='PORT', default=20202,
                        help='set the PUB port for logging [default: 10201]')
    parser.add_argument('--loglevel', type=int, metavar='LEVEL', default=logging.DEBUG,
                        help='set the log level [default: DEBUG]')
    parser.add_argument('--ident', type=str,
                        help='set the ZMQ identity [default: random]')
    parser.add_argument('--packer', type=str, default='json',
                        choices=['json','pickle'],
                        help='set the message format method [default: json]')
    parser.add_argument('--url', type=str,
                        help='set transport,ip,regport in one arg, e.g. tcp://127.0.0.1:10101')

    return parser


def connect_logger(context, iface, root="ip", loglevel=logging.DEBUG):
    lsock = context.socket(zmq.PUB)
    lsock.connect(iface)
    handler = handlers.PUBHandler(lsock)
    handler.setLevel(loglevel)
    handler.root_topic = root
    logger.addHandler(handler)