import logging
from pathlib import Path

import docker
from docker.models.containers import Container
from injectable import injectable

from app.model.containers import Tree, Compose, SimpleContainer

logger = logging.getLogger(__name__)


@injectable(singleton=True)
class DockerService:
  def __init__(self):
    self._client = docker.from_env()
    # logger.info(f"Docker client initialized. Plugins: {self._client.plugins()}")
    self._tree = self._init_tree()

  def _init_tree(self) -> Tree:
    tree = Tree()
    container: Container
    for container in self._client.containers.list():
      labels = container.labels
      working_dir = labels.get("com.docker.compose.project.working_dir")
      if working_dir:
        if tree.composes.get(working_dir) is None:
          tree.composes[working_dir] = Compose(working_dir)
        tree.composes[working_dir].containers.append(SimpleContainer.from_container(container))
      else:
        tree.containers.append(SimpleContainer.from_container(container))
    return tree

  @property
  def tree(self) -> Tree:
    return self._tree

  def reload(self, compose_path: Path):
    pass