#!/usr/bin/env python3

# Libervia CLI
# Copyright (C) 2009-2026 Jérôme Poisson (goffi@goffi.org)

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from . import base
from libervia.backend.core.i18n import _
from libervia.backend.plugins.plugin_misc_app_manager.models import AppUpdateResult
from libervia.backend.tools.common import data_format
from libervia.cli.constants import Const as C

__commands__ = ["Application"]


class List(base.CommandBase):
    """List available applications"""

    def __init__(self, host):
        super(List, self).__init__(
            host,
            "list",
            use_profile=False,
            use_output=C.OUTPUT_LIST,
            help=_("list available applications"),
        )

    def add_parser_options(self):
        # FIXME: "extend" would be better here, but it's only available from Python 3.8+
        #   so we use "append" until minimum version of Python is raised.
        self.parser.add_argument(
            "-f",
            "--filter",
            dest="filters",
            action="append",
            choices=["available", "running"],
            help=_("show applications with this status"),
        )

    async def start(self):

        # FIXME: this is only needed because we can't use "extend" in
        #   add_parser_options, see note there
        if self.args.filters:
            self.args.filters = list(set(self.args.filters))
        else:
            self.args.filters = ["available"]

        try:
            found_apps = await self.host.bridge.applications_list(self.args.filters)
        except Exception as e:
            self.disp(f"can't get applications list: {e}", error=True)
            self.host.quit(C.EXIT_BRIDGE_ERRBACK)
        else:
            await self.output(found_apps)
            self.host.quit()


class Start(base.CommandBase):
    """Start an application"""

    def __init__(self, host):
        super(Start, self).__init__(
            host, "start", use_profile=False, help=_("start an application")
        )

    def add_parser_options(self):
        self.parser.add_argument(
            "name",
            help=_("name of the application to start"),
        )

    async def start(self):
        try:
            await self.host.bridge.application_start(
                self.args.name,
                "",
            )
        except Exception as e:
            self.disp(f"can't start {self.args.name}: {e}", error=True)
            self.host.quit(C.EXIT_BRIDGE_ERRBACK)
        else:
            self.host.quit()


class Stop(base.CommandBase):

    def __init__(self, host):
        super(Stop, self).__init__(
            host, "stop", use_profile=False, help=_("stop a running application")
        )

    def add_parser_options(self):
        id_group = self.parser.add_mutually_exclusive_group(required=True)
        id_group.add_argument(
            "name",
            nargs="?",
            help=_("name of the application to stop"),
        )
        id_group.add_argument(
            "-i",
            "--id",
            help=_("identifier of the instance to stop"),
        )

    async def start(self):
        try:
            if self.args.name is not None:
                args = [self.args.name, "name"]
            else:
                args = [self.args.id, "instance"]
            await self.host.bridge.application_stop(
                *args,
                "",
            )
        except Exception as e:
            if self.args.name is not None:
                self.disp(f"can't stop application {self.args.name!r}: {e}", error=True)
            else:
                self.disp(
                    f"can't stop application instance with id {self.args.id!r}: {e}",
                    error=True,
                )
            self.host.quit(C.EXIT_BRIDGE_ERRBACK)
        else:
            self.host.quit()


class Exposed(base.CommandBase):

    def __init__(self, host):
        super(Exposed, self).__init__(
            host,
            "exposed",
            use_profile=False,
            use_output=C.OUTPUT_DICT,
            help=_("show data exposed by a running application"),
        )

    def add_parser_options(self):
        id_group = self.parser.add_mutually_exclusive_group(required=True)
        id_group.add_argument(
            "name",
            nargs="?",
            help=_("name of the application to check"),
        )
        id_group.add_argument(
            "-i",
            "--id",
            help=_("identifier of the instance to check"),
        )

    async def start(self):
        try:
            if self.args.name is not None:
                args = [self.args.name, "name"]
            else:
                args = [self.args.id, "instance"]
            exposed_data_raw = await self.host.bridge.application_exposed_get(
                *args,
                "",
            )
        except Exception as e:
            if self.args.name is not None:
                self.disp(
                    f"can't get values exposed from application {self.args.name!r}: {e}",
                    error=True,
                )
            else:
                self.disp(
                    f"can't values exposed from  application instance with id {self.args.id!r}: {e}",
                    error=True,
                )
            self.host.quit(C.EXIT_BRIDGE_ERRBACK)
        else:
            exposed_data = data_format.deserialise(exposed_data_raw)
            await self.output(exposed_data)
            self.host.quit()


class Update(base.CommandBase):
    """Update an application"""

    def __init__(self, host):
        extra_outputs = {"default": self.default_output}
        super(Update, self).__init__(
            host,
            "update",
            use_profile=False,
            use_output=C.OUTPUT_MODEL,
            extra_outputs=extra_outputs,
            help=_("update an application"),
        )

    def add_parser_options(self):
        self.parser.add_argument(
            "name",
            nargs="?",
            help=_("name of the application to update"),
        )
        self.parser.add_argument(
            "-a",
            "--all",
            action="store_true",
            help=_("update all applications"),
        )

    def default_output(self, result: AppUpdateResult) -> None:
        for info in result.updated:
            self.console.print(
                f"  [bold]{info.service}[/bold]: {info.old_version} → {info.new_version}"
            )

    async def _update_one(self, name: str, index: int, total: int) -> None:
        self.console.print(f"[dim]updating {name}[/dim] [bold]{index}/{total}[/bold]")
        try:
            result_s = await self.host.bridge.application_update(name)
        except Exception as e:
            self.disp(f"can't update {name}: {e}", error=True)
            return

        if result_s == "":
            self.disp(f"{name} is already up to date")
            return

        result = AppUpdateResult.model_validate_json(result_s)
        await self.output(result)

    async def start(self):
        if self.args.all:
            if self.args.name:
                self.disp("--all and name are mutually exclusive", error=True)
                self.host.quit(C.EXIT_BRIDGE_ERRBACK)
                return
            try:
                apps = await self.host.bridge.applications_list(["available"])
            except Exception as e:
                self.disp(f"can't get applications list: {e}", error=True)
                self.host.quit(C.EXIT_BRIDGE_ERRBACK)
                return
            app_names = [app["name"] for app in apps]
            for i, name in enumerate(app_names, 1):
                await self._update_one(name, i, len(app_names))
        else:
            if not self.args.name:
                self.disp("name is required (or use --all)", error=True)
                self.host.quit(C.EXIT_BRIDGE_ERRBACK)
                return
            await self._update_one(self.args.name, 1, 1)
        self.host.quit()


class Application(base.CommandBase):
    subcommands = (List, Start, Stop, Exposed, Update)

    def __init__(self, host):
        super(Application, self).__init__(
            host,
            "application",
            use_profile=False,
            help=_("manage applications"),
            aliases=["app"],
        )