import std/httpclient
import pkg/codex/contracts
import ./twonodes
import ../codex/examples
import ../contracts/time
import ./codexconfig
import ./codexclient
import ./nodeconfigs

proc findItem[T](items: seq[T], item: T): ?!T =
  for tmp in items:
    if tmp == item:
      return success tmp

  return failure("Not found")

multinodesuite "Sales":
  let salesConfig = NodeConfigs(
    clients: CodexConfigs.init(nodes = 1).some,
    providers: CodexConfigs.init(nodes = 1).some,
  )

  var host: CodexClient
  var client: CodexClient

  setup:
    host = providers()[0].client
    client = clients()[0].client

  test "node handles new storage availability", salesConfig:
    let availability1 = host.postAvailability(
      totalSize = 1.u256, duration = 2.u256, minPrice = 3.u256, maxCollateral = 4.u256
    ).get
    let availability2 = host.postAvailability(
      totalSize = 4.u256, duration = 5.u256, minPrice = 6.u256, maxCollateral = 7.u256
    ).get
    check availability1 != availability2

  test "node lists storage that is for sale", salesConfig:
    let availability = host.postAvailability(
      totalSize = 1.u256, duration = 2.u256, minPrice = 3.u256, maxCollateral = 4.u256
    ).get
    check availability in host.getAvailabilities().get

  test "updating non-existing availability", salesConfig:
    let nonExistingResponse = host.patchAvailabilityRaw(
      AvailabilityId.example,
      duration = 100.u256.some,
      minPrice = 200.u256.some,
      maxCollateral = 200.u256.some,
    )
    check nonExistingResponse.status == "404 Not Found"

  test "updating availability", salesConfig:
    let availability = host.postAvailability(
      totalSize = 140000.u256,
      duration = 200.u256,
      minPrice = 300.u256,
      maxCollateral = 300.u256,
    ).get

    host.patchAvailability(
      availability.id,
      duration = 100.u256.some,
      minPrice = 200.u256.some,
      maxCollateral = 200.u256.some,
    )

    let updatedAvailability = (host.getAvailabilities().get).findItem(availability).get
    check updatedAvailability.duration == 100
    check updatedAvailability.minPrice == 200
    check updatedAvailability.maxCollateral == 200
    check updatedAvailability.totalSize == 140000
    check updatedAvailability.freeSize == 140000

  test "updating availability - freeSize is not allowed to be changed", salesConfig:
    let availability = host.postAvailability(
      totalSize = 140000.u256,
      duration = 200.u256,
      minPrice = 300.u256,
      maxCollateral = 300.u256,
    ).get
    let freeSizeResponse =
      host.patchAvailabilityRaw(availability.id, freeSize = 110000.u256.some)
    check freeSizeResponse.status == "400 Bad Request"
    check "not allowed" in freeSizeResponse.body

  test "updating availability - updating totalSize", salesConfig:
    let availability = host.postAvailability(
      totalSize = 140000.u256,
      duration = 200.u256,
      minPrice = 300.u256,
      maxCollateral = 300.u256,
    ).get
    host.patchAvailability(availability.id, totalSize = 100000.u256.some)
    let updatedAvailability = (host.getAvailabilities().get).findItem(availability).get
    check updatedAvailability.totalSize == 100000
    check updatedAvailability.freeSize == 100000

  test "updating availability - updating totalSize does not allow bellow utilized",
    salesConfig:
    let originalSize = 0xFFFFFF.u256
    let data = await RandomChunker.example(blocks = 8)
    let availability = host.postAvailability(
      totalSize = originalSize,
      duration = 20 * 60.u256,
      minPrice = 300.u256,
      maxCollateral = 300.u256,
    ).get

    # Lets create storage request that will utilize some of the availability's space
    let cid = client.upload(data).get
    let id = client.requestStorage(
      cid,
      duration = 20 * 60.u256,
      reward = 400.u256,
      proofProbability = 3.u256,
      expiry = 10 * 60,
      collateral = 200.u256,
      nodes = 3,
      tolerance = 1,
    ).get

    check eventually(client.purchaseStateIs(id, "started"), timeout = 10 * 60 * 1000)
    let updatedAvailability = (host.getAvailabilities().get).findItem(availability).get
    check updatedAvailability.totalSize != updatedAvailability.freeSize

    let utilizedSize = updatedAvailability.totalSize - updatedAvailability.freeSize
    let totalSizeResponse = host.patchAvailabilityRaw(
      availability.id, totalSize = (utilizedSize - 1.u256).some
    )
    check totalSizeResponse.status == "400 Bad Request"
    check "totalSize must be larger then current totalSize" in totalSizeResponse.body

    host.patchAvailability(availability.id, totalSize = (originalSize + 20000).some)
    let newUpdatedAvailability =
      (host.getAvailabilities().get).findItem(availability).get
    check newUpdatedAvailability.totalSize == originalSize + 20000
    check newUpdatedAvailability.freeSize - updatedAvailability.freeSize == 20000