URIs.kt

package org.javacs.kt.util

import java.net.URI
import java.net.URLEncoder
import java.net.URLDecoder
import java.nio.charset.StandardCharsets
import java.nio.file.Path
import java.nio.file.Paths

/**
 * Parse a possibly-percent-encoded URI string.
 * Decoding is necessary since some language clients
 * (including VSCode) invalidly percent-encode colons.
 */
fun parseURI(uri: String): URI =
    URI.create(runCatching {
        // val decoded = URLDecoder.decode(uri, StandardCharsets.UTF_8)

        // Don't encode the protocol
        val protocol = uri.substring(0, uri.indexOf('/'))
        val path = uri.substringAfter('/')

        val parts = path.split("/")
        val encodedParts = parts.map { URLEncoder.encode(it, StandardCharsets.UTF_8) }
        val encoded = protocol + '/' + encodedParts.joinToString(separator = "/")

        // URLEncoder uses '+' instead of '%20' fpr spaces
        encoded.replace("+", "%20")
    }.getOrDefault(uri))

val URI.filePath: Path? get() = runCatching { Paths.get(this) }.getOrNull()

/** Fetches the file extension WITHOUT the dot. */
val URI.fileExtension: String?
    get() {
        val str = toString()
        val dotOffset = str.lastIndexOf(".")
        val queryStart = str.indexOf("?")
        val end = if (queryStart != -1) queryStart else str.length
        return if (dotOffset < 0) null else str.substring(dotOffset + 1, end)
    }

fun describeURIs(uris: Collection<URI>): String =
    if (uris.isEmpty()) "0 files"
    else if (uris.size > 5) "${uris.size} files"
    else uris.map(::describeURI).joinToString(", ")

fun describeURI(uri: String): String = describeURI(parseURI(uri))

fun describeURI(uri: URI): String =
    uri.path?.let {
        val (parent, fileName) = it.partitionAroundLast("/")
        ".../" + parent.substringAfterLast("/") + fileName
    } ?: uri.toString()