import base

import tempfile
import os
import shutil
from sat.core.i18n import _


parser = base.subparser.add_parser('pipe',
                                           help = "File managment")

parser.add_argument("-p", "--profile", action="store", type=str, default='@DEFAULT@',
            help=_("Use PROFILE profile key (default: %(default)s)"))

subparser = parser.add_subparsers()
pipein = subparser.add_parser('out',
                                      help = "Send a pipe to a contact")
pipein.add_argument("jid", type=str,
                         help=_("The destination jid"))

pipein.set_defaults(func=lambda args : PipeOut(args.profile, args.jid).go())

pipeout = subparser.add_parser('in',
                                      help = "Send a pipe to a contact")
pipeout.add_argument("jids", type=str, nargs="*",
                     help=_("The destination jid"))

pipeout.set_defaults(func=lambda args : PipeIn(args.profile, args.jids).go())

class PipeOut(base.JPWithProfile):
    def __init__(self, profile, dest_jid):
        base.JPWithProfile.__init__(self,profile)
        self.dest_jid = dest_jid

    def pipe_out(self, dest_jid):
        """Create named pipe, and send stdin to it"""
        tmp_dir = tempfile.mkdtemp()
        fifopath = os.path.join(tmp_dir,"pipe_out")
        os.mkfifo(fifopath)
        self.bridge.pipeOut(self._getFullJid(dest_jid), fifopath, {}, self.profile)
        with open(fifopath, 'w') as f:
            shutil.copyfileobj(sys.stdin, f)
        shutil.rmtree(tmp_dir)

    def run(self):
        jids = self.check_jids([self.dest_jid])
        jid = jids[0]
        self.pipe_out(jid)

class PipeIn(base.JPAsk):
    def __init__(self, profile, dest_jids):
        base.JPAsk.__init__(self,profile, start_mainloop = True)
        self._dest_jids = dest_jids

    def dest_jids(self):
        return self._dest_jids

    def confirm_type(self):
        return "PIPE_TRANSFER"

    def ask(self, data):
        answer_data = {}
        tmp_dir = tempfile.mkdtemp()
        fifopath = os.path.join(tmp_dir,"pipe_in")
        answer_data["dest_path"] = fifopath
        os.mkfifo(fifopath)
        self.answer(True, answer_data)
        with open(fifopath, 'r') as f:
            shutil.copyfileobj(f, sys.stdout)
        shutil.rmtree(tmp_dir)
        self.loop.quit()