import okio.ByteString

/**
 * Basic I/O for `PublicSuffixDatabase.list`
 */
internal interface PublicSuffixList {
  fun ensureLoaded()

  val bytes: ByteString
  val exceptionBytes: ByteString

  companion object
}

    return object : ForwardingSource(source) {
      @Throws(IOException::class)
      override fun read(
        sink: Buffer,
        byteCount: Long,
      ): Long {
        val buffer = Buffer()
        val read = delegate.read(buffer, byteCount)
        if (read != -1L) {
          sink.write(buffer.readByteString().toAsciiUppercase())
        }
        return read
      }
    }
  }

  }

  private val bytesOut = Buffer()
  private val hpackWriter = Hpack.Writer(out = bytesOut)

  @ParameterizedTest
  @ArgumentsSource(StoriesTestProvider::class)
  fun testRoundTrip(story: Story) {
    assumeFalse(
      story === Story.MISSING,
      "Test stories missing, checkout git submodule",
    )

    val newCases = mutableListOf<Case>()
    for (case in story.cases) {

            val objectExpr = mce.objectExpression
            checkIndirectOutputContains(objectExpr, mce)
        }
    }

    private
    fun checkOutputContains(call: MethodCallExpression) {
        val receiver = call.receiver!!
        if (receiver is PropertyExpression) {
            if (receiver.propertyAsString == "output") {
                val objectExpr = receiver.objectExpression!!

  compilerOptions {
    jvmTarget.set(JvmTarget.JVM_17)
  }
}

tasks.withType<JavaCompile> {
  sourceCompatibility = JvmTarget.JVM_17.target
  targetCompatibility = JvmTarget.JVM_17.target
}

val copyResourcesTemplates =
  tasks.register<Copy>("copyResourcesTemplates") {
    from("src/main/resources-templates")
    into(layout.buildDirectory.dir("generated/resources-templates"))

import okio.Path.Companion.toPath
import okio.Source

internal class ResourcePublicSuffixList(
  override val path: Path = PUBLIC_SUFFIX_RESOURCE,
  val fileSystem: FileSystem = FileSystem.Companion.RESOURCES,
) : BasePublicSuffixList() {
  override fun listSource(): Source = fileSystem.source(path)

  companion object {
    @JvmField
    val PUBLIC_SUFFIX_RESOURCE =

  block: () -> T,
): T {
  var startNs = -1L
  val loggingEnabled = isLoggable(Level.FINE)
  if (loggingEnabled) {
    startNs = queue.taskRunner.backend.nanoTime()
    log(task, queue, "starting")
  }

  var completedNormally = false
  try {
    val result = block()
    completedNormally = true
    return result
  } finally {
    if (loggingEnabled) {
      val elapsedNs = queue.taskRunner.backend.nanoTime() - startNs

/** This extends [X509TrustManager] for Android to disable verification for a set of hosts. */
internal class InsecureAndroidTrustManager(
  private val delegate: X509TrustManager,
  private val insecureHosts: List<String>,
) : X509TrustManager {
  private val checkServerTrustedMethod: Method? =
    try {
      delegate::class.java.getMethod(
        "checkServerTrusted",
        Array<X509Certificate>::class.java,

    /**
     * Decodes a small integer (1 byte) from NDR format.
     *
     * @return the decoded small integer value
     */
    public int dec_ndr_small() {
        final int val = this.buf[this.index] & 0xFF;
        advance(1);
        return val;
    }

    /**
     * Encodes a short integer (2 bytes) in NDR format.
     *
     * @param s the short integer value to encode
     */

        return doGet(userProperties != null ? userProperties.get(key) : null, def);
    }

    private static boolean doGet(Object val, boolean def) {
        if (val instanceof Boolean bool) {
            return bool;
        } else if (val != null) {
            return Boolean.parseBoolean(val.toString());
        } else {
            return def;
        }
    }