# -*- coding: utf-8 -*-

import logging

import aiodocker

from playground.control.connectors.docker.base import PlaygroundDockerContext
from playground.control.utils import mktar_from_docker_context


logger = logging.getLogger(__name__)


class PlaygroundDockerImages(PlaygroundDockerContext):

    async def build(
            self,
            build_from: str,
            image_tag: str) -> bool:
        # returns True/False indicating success
        image_tag = self._image_tag(image_tag)
        build_from = self._image_tag(build_from)
        logger.info(
            f'Building image: {image_tag} from {build_from}')
        try:
            await self._build(build_from, image_tag)
        except aiodocker.DockerError as e:
            logger.error(
                f'Failed building image: {image_tag} from {build_from} \n {e}')
            return False
        try:
            await self.docker.images.inspect(name=image_tag)
        except aiodocker.DockerError as e:
            logger.error(
                f'Failed inspecting built image: {image_tag} \n {e}')
            return False
        return True

    async def exists(self, image_tag: str) -> bool:
        image_tag = self._image_tag(image_tag)
        logger.debug(f'Checking for image ({image_tag})')
        try:
            # todo: get filter working
            images = await self.docker.images.list()
            return bool([
                _result
                for _result
                in images
                if image_tag in (_result['RepoTags'] or [])])
        except aiodocker.DockerError as e:
            logger.error(
                f'Failed checking for image ({image_tag}): {e}')
            return False

    async def pull(self, image_tag: str, force: bool = False) -> bool:
        # returns True/False indicating whether image is present after call
        image_tag = self._image_tag(image_tag)
        if not force and await self.exists(image_tag):
            return True
        logger.info(f'Pulling image {image_tag}')
        try:
            await self.docker.images.pull(image_tag)
        except aiodocker.DockerError as e:
            logger.error(f'Failed pulling image: {image_tag} {e}')
            return False
        return True

    def _image_tag(self, image_tag):
        # appends :latest if image has no tag
        return (
            image_tag
            if ":" in image_tag
            else f"{image_tag}:latest")

    async def _build(self, build_from, image_tag):
        tar_obj = mktar_from_docker_context('context')
        try:
            await self.docker.images.build(
                fileobj=tar_obj,
                encoding="gzip",
                buildargs=dict(BUILD_FROM=build_from),
                tag=image_tag)
        finally:
            tar_obj.close()
