package works.iterative.mongo

import zio.*
import zio.test.*
import zio.json.*
import Assertion.*
import java.io.File

object MongoJsonFileRepositoryIntegrationSpec extends DefaultRunnableSpec:
  case class ExampleMetadata(osobniCislo: String)
  sealed trait ExampleCriteria
  case class ByOsobniCislo(osobniCislo: String) extends ExampleCriteria

  given JsonCodec[ExampleMetadata] = DeriveJsonCodec.gen

  override def spec = suite("Mongo file repository integration spec")(
    test("repo can put and read back file")(
      for
        repo <- ZIO
          .service[MongoJsonFileRepository[ExampleMetadata, ExampleCriteria]]
        _ <- repo.put(
          "test.txt",
          "Example content".getBytes(),
          ExampleMetadata("10123")
        )
        byId <- repo.find("test.txt")
        result <- repo.matching(ByOsobniCislo("10123"))
      yield assertTrue(byId.isDefined, result.head.name == "test.txt")
    )
  ).provideCustomLayer(layer.mapError(TestFailure.fail))

  val layer =
    import org.mongodb.scala.*
    import org.mongodb.scala.model.Filters.*
    import org.bson.json.JsonObject
    import org.mongodb.scala.bson.conversions.Bson
    import org.mongodb.scala.bson.Document
    import org.mongodb.scala.gridfs.GridFSBucket
    MongoConfig.fromEnv >>> MongoClient.layer >>> (for
      client <- ZIO.service[MongoClient]
      bucket <- Task.attempt(
        GridFSBucket(client.getDatabase("test"), "testfiles")
      )
    yield new MongoJsonFileRepository[ExampleMetadata, ExampleCriteria](
      bucket, {
        _ match
          case ByOsobniCislo(osc) => equal("metadata.osobniCislo", osc)
      }
    )).toLayer