""" List and lookup repository images. """

import json

from collections import defaultdict
from endpoints.api import (resource, nickname, require_repo_read, RepositoryParamResource,
                           format_date, path_param)
from endpoints.exception import NotFound
from data import model


def image_view(image, image_map, include_ancestors=True):
  command = image.command

  def docker_id(aid):
    if aid not in image_map:
      return ''

    return image_map[aid].docker_image_id

  image_data = {
    'id': image.docker_image_id,
    'created': format_date(image.created),
    'comment': image.comment,
    'command': json.loads(command) if command else None,
    'size': image.storage.image_size,
    'uploading': image.storage.uploading,
    'sort_index': len(image.ancestors),
  }

  if include_ancestors:
    # Calculate the ancestors string, with the DBID's replaced with the docker IDs.
    ancestors = [docker_id(a) for a in image.ancestor_id_list()]
    image_data['ancestors'] = '/{0}/'.format('/'.join(ancestors))

  return image_data


def historical_image_view(image, image_map):
  ancestors = [image_map[a] for a in image.ancestor_id_list()]
  normal_view = image_view(image, image_map)
  normal_view['history'] = [image_view(parent, image_map, False) for parent in ancestors]
  return normal_view


@resource('/v1/repository/<apirepopath:repository>/image/')
@path_param('repository', 'The full path of the repository. e.g. namespace/name')
class RepositoryImageList(RepositoryParamResource):
  """ Resource for listing repository images. """
  @require_repo_read
  @nickname('listRepositoryImages')
  def get(self, namespace, repository):
    """ List the images for the specified repository. """
    repo = model.repository.get_repository(namespace, repository)
    if not repo:
      raise NotFound()

    all_images = model.image.get_repository_images_without_placements(repo)
    all_tags = model.tag.list_repository_tags(namespace, repository)

    tags_by_docker_id = defaultdict(list)
    found_image_ids = set()

    for tag in all_tags:
      tags_by_docker_id[tag.image.docker_image_id].append(tag.name)
      found_image_ids.add(tag.image.id)
      found_image_ids.update(tag.image.ancestor_id_list())

    image_map = {}
    filtered_images = []
    for image in all_images:
      if image.id in found_image_ids:
        image_map[image.id] = image
        filtered_images.append(image)

    def add_tags(image_json):
      image_json['tags'] = tags_by_docker_id[image_json['id']]
      return image_json

    return {
      'images': [add_tags(image_view(image, image_map)) for image in filtered_images]
    }


@resource('/v1/repository/<apirepopath:repository>/image/<image_id>')
@path_param('repository', 'The full path of the repository. e.g. namespace/name')
@path_param('image_id', 'The Docker image ID')
class RepositoryImage(RepositoryParamResource):
  """ Resource for handling repository images. """
  @require_repo_read
  @nickname('getImage')
  def get(self, namespace, repository, image_id):
    """ Get the information available for the specified image. """
    image = model.image.get_repo_image_extended(namespace, repository, image_id)
    if not image:
      raise NotFound()

    # Lookup all the ancestor images for the image.
    image_map = {}
    for current_image in model.image.get_parent_images(namespace, repository, image):
      image_map[current_image.id] = current_image

    return historical_image_view(image, image_map)