import random
import string

import pytest

from Crypto.PublicKey import RSA
from jwkest.jwk import RSAKey

from test.registry.protocols import Image, layer_bytes_for_contents
from test.registry.protocol_v1 import V1Protocol
from test.registry.protocol_v2 import V2Protocol


@pytest.fixture(scope="session")
def basic_images():
  """ Returns basic images for push and pull testing. """
  # Note: order is from base layer down to leaf.
  parent_bytes = layer_bytes_for_contents('parent contents')
  image_bytes = layer_bytes_for_contents('some contents')
  return [
    Image(id='parentid', bytes=parent_bytes, parent_id=None),
    Image(id='someid', bytes=image_bytes, parent_id='parentid'),
  ]


@pytest.fixture(scope="session")
def sized_images():
  """ Returns basic images (with sizes) for push and pull testing. """
  # Note: order is from base layer down to leaf.
  parent_bytes = layer_bytes_for_contents('parent contents', mode='')
  image_bytes = layer_bytes_for_contents('some contents', mode='')
  return [
    Image(id='parentid', bytes=parent_bytes, parent_id=None, size=len(parent_bytes),
          config={'foo': 'bar'}),
    Image(id='someid', bytes=image_bytes, parent_id='parentid', size=len(image_bytes),
          config={'foo': 'childbar'}),
  ]


@pytest.fixture(scope="session")
def multi_layer_images():
  """ Returns complex images (with sizes) for push and pull testing. """
  # Note: order is from base layer down to leaf.
  layer1_bytes = layer_bytes_for_contents('layer 1 contents', mode='', other_files={
    'file1': 'from-layer-1',
  })

  layer2_bytes = layer_bytes_for_contents('layer 2 contents', mode='', other_files={
    'file2': 'from-layer-2',
  })

  layer3_bytes = layer_bytes_for_contents('layer 3 contents', mode='', other_files={
    'file1': 'from-layer-3',
    'file3': 'from-layer-3',
  })

  layer4_bytes = layer_bytes_for_contents('layer 4 contents', mode='', other_files={
    'file3': 'from-layer-4',
  })

  layer5_bytes = layer_bytes_for_contents('layer 5 contents', mode='', other_files={
    'file4': 'from-layer-5',
  })

  return [
    Image(id='layer1', bytes=layer1_bytes, parent_id=None, size=len(layer1_bytes),
          config={'internal_id': 'layer1'}),
    Image(id='layer2', bytes=layer2_bytes, parent_id='layer1', size=len(layer2_bytes),
          config={'internal_id': 'layer2'}),
    Image(id='layer3', bytes=layer3_bytes, parent_id='layer2', size=len(layer3_bytes),
          config={'internal_id': 'layer3'}),
    Image(id='layer4', bytes=layer4_bytes, parent_id='layer3', size=len(layer4_bytes),
          config={'internal_id': 'layer4'}),
    Image(id='someid', bytes=layer5_bytes, parent_id='layer4', size=len(layer5_bytes),
          config={'internal_id': 'layer5'}),
  ]


@pytest.fixture(scope="session")
def jwk():
  return RSAKey(key=RSA.generate(2048))


@pytest.fixture(params=[V2Protocol])
def v2_protocol(request, jwk):
  return request.param(jwk)


@pytest.fixture(params=[V1Protocol])
def v1_protocol(request, jwk):
  return request.param(jwk)


@pytest.fixture(params=[V2Protocol])
def manifest_protocol(request, jwk):
  return request.param(jwk)


@pytest.fixture(params=[V1Protocol, V2Protocol])
def loginer(request, jwk):
  return request.param(jwk)


@pytest.fixture(params=[V1Protocol, V2Protocol])
def pusher(request, jwk):
  return request.param(jwk)


@pytest.fixture(params=[V1Protocol, V2Protocol])
def puller(request, jwk):
  return request.param(jwk)


@pytest.fixture(scope="session")
def random_layer_data():
  size = 4096
  contents = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(size))
  return layer_bytes_for_contents(contents)