import json

from collections import defaultdict

from app import storage as store
from endpoints.api import (resource, nickname, require_repo_read, RepositoryParamResource,
                           format_date, NotFound)
from data import model
from util.cache import cache_control_flask_restful


def image_view(image):
  extended_props = image
  if image.storage and image.storage.id:
    extended_props = image.storage

  command = extended_props.command
  return {
    'id': image.docker_image_id,
    'created': format_date(extended_props.created),
    'comment': extended_props.comment,
    'command': json.loads(command) if command else None,
    'ancestors': image.ancestors,
    'dbid': image.id,
    'size': extended_props.image_size,
  }


@resource('/v1/repository/<repopath:repository>/image/')
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. """
    all_images = model.get_repository_images(namespace, repository)
    all_tags = model.list_repository_tags(namespace, repository)

    tags_by_image_id = defaultdict(list)
    for tag in all_tags:
      tags_by_image_id[tag.image.docker_image_id].append(tag.name)


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


    return {
      'images': [add_tags(image_view(image)) for image in all_images]
    }


@resource('/v1/repository/<repopath:repository>/image/<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.get_repo_image(namespace, repository, image_id)
    if not image:
      raise NotFound()

    return image_view(image)


@resource('/v1/repository/<repopath:repository>/image/<image_id>/changes')
class RepositoryImageChanges(RepositoryParamResource):
  """ Resource for handling repository image change lists. """

  @cache_control_flask_restful(max_age=60*60)  # Cache for one hour
  @require_repo_read
  @nickname('getImageChanges')
  def get(self, namespace, repository, image_id):
    """ Get the list of changes for the specified image. """
    image = model.get_repo_image(namespace, repository, image_id)

    if not image:
      raise NotFound()

    uuid = image.storage and image.storage.uuid
    diffs_path = store.image_file_diffs_path(namespace, repository, image_id, uuid)

    try:
      response_json = json.loads(store.get_content(diffs_path))
      return response_json
    except IOError:
      raise NotFound()