import os

import click

from openpype.modules import OpenPypeModule, IHostAddon

WEBPUBLISHER_ROOT_DIR = os.path.dirname(os.path.abspath(__file__))


class WebpublisherAddon(OpenPypeModule, IHostAddon):
    name = "webpublisher"
    host_name = "webpublisher"

    def initialize(self, module_settings):
        self.enabled = True

    def headless_publish(self, log, close_plugin_name=None, is_test=False):
        """Runs publish in a opened host with a context.

        Close Python process at the end.
        """

        from openpype.pipeline.publish.lib import remote_publish
        from .lib import get_webpublish_conn, publish_and_log

        if is_test:
            remote_publish(log, close_plugin_name)
            return

        dbcon = get_webpublish_conn()
        _id = os.environ.get("BATCH_LOG_ID")
        if not _id:
            log.warning("Unable to store log records, "
                        "batch will be unfinished!")
            return

        publish_and_log(
            dbcon, _id, log, close_plugin_name=close_plugin_name
        )

    def cli(self, click_group):
        click_group.add_command(cli_main)


@click.group(
    WebpublisherAddon.name,
    help="Webpublisher related commands.")
def cli_main():
    pass


@cli_main.command()
@click.argument("path")
@click.option("-u", "--user", help="User email address")
@click.option("-p", "--project", help="Project")
@click.option("-t", "--targets", help="Targets", default=None,
              multiple=True)
def publish(project, path, user=None, targets=None):
    """Start publishing (Inner command).

    Publish collects json from paths provided as an argument.
    More than one path is allowed.
    """

    from .publish_functions import cli_publish

    cli_publish(project, path, user, targets)


@cli_main.command()
@click.argument("path")
@click.option("-p", "--project", help="Project")
@click.option("-h", "--host", help="Host")
@click.option("-u", "--user", help="User email address")
@click.option("-t", "--targets", help="Targets", default=None,
              multiple=True)
def publishfromapp(project, path, host, user=None, targets=None):
    """Start publishing through application (Inner command).

    Publish collects json from paths provided as an argument.
    More than one path is allowed.
    """

    from .publish_functions import cli_publish_from_app

    cli_publish_from_app(project, path, host, user, targets)


@cli_main.command()
@click.option("-e", "--executable", help="Executable")
@click.option("-u", "--upload_dir", help="Upload dir")
@click.option("-h", "--host", help="Host", default=None)
@click.option("-p", "--port", help="Port", default=None)
def webserver(executable, upload_dir, host=None, port=None):
    """Start service for communication with Webpublish Front end.

        OP must be congigured on a machine, eg. OPENPYPE_MONGO filled AND
        FTRACK_BOT_API_KEY provided with api key from Ftrack.

        Expect "pype.club" user created on Ftrack.
    """

    from .webserver_service import run_webserver

    run_webserver(executable, upload_dir, host, port)