'''
docker & docker-compose frustrated me, podctl unfrustrates me.
'''

import asyncio
import cli2
import inspect
import os
import sys

from .container import Container
from .pod import Pod
from .exceptions import Mistake, WrongResult
from .service import Service


class ConsoleScript(cli2.ConsoleScript):
    class Parser(cli2.Parser):
        def parse(self):
            super().parse()
            if str(self.command) == 'help':
                return

            self.forward_args = []

            found_dash = False
            for arg in self.argv:
                if arg == '--':
                    found_dash = True
                if not found_dash:
                    continue
                self.forward_args.append(arg)

            self.funckwargs['cmd'] = self.forward_args

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.options = dict()

    def __call__(self, *args, **kwargs):
        import inspect
        from podctl.podfile import Podfile
        self.podfile = Podfile.factory(os.getenv('PODFILE', 'pod.py'))
        for name, script in self.podfile.pod.scripts.items():
            cb = self.podfile.pod.script(name)
            cb.__doc__ = inspect.getdoc(script) or script.doc
            self[name] = cli2.Callable(
                name,
                cb,
                options={o.name: o for o in script.options},
                color=getattr(script, 'color', cli2.YELLOW),
            )
        return super().__call__(*args, **kwargs)

    def call(self, command):
        self.options = self.parser.options

        try:
            return super().call(command)
        except Mistake as e:
            print(e)
            sys.exit(1)
        except WrongResult as e:
            print(e)
            sys.exit(e.proc.rc)


console_script = ConsoleScript(__doc__).add_module('podctl.console_script')