com.lightningkite.lightningserver.files.FileSystem.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of server-core Show documentation

A set of tools to fill in/replace what Ktor is lacking in.

The newest version!

package com.lightningkite.lightningserver.files

import com.lightningkite.lightningserver.core.ContentType
import com.lightningkite.lightningserver.core.Disconnectable
import com.lightningkite.lightningserver.http.HttpContent
import com.lightningkite.lightningserver.serverhealth.HealthCheckable
import com.lightningkite.lightningserver.serverhealth.HealthStatus

/**
 * An abstracted model for reading and writing files in a storage solution.
 * Every implementation will handle how to resolve FileObjects in their own system.
 */
interface FileSystem : HealthCheckable, Disconnectable {
    val root: FileObject
    val rootUrls: List get() = listOf(root.url)

    companion object {
        fun register(system: FileSystem) {
            system.rootUrls.forEach {
                roots.add(it to system)
            }
        }

        val urlRoots get() = roots.map { it.first }
        private val roots = ArrayList>()
        fun resolve(url: String): FileObject? {
            val (root, sys) = roots.find { url.startsWith(it.first) } ?: return null
            return sys.root.resolve(url.removePrefix(root).substringBefore('?'))
        }
    }

    override suspend fun healthCheck(): HealthStatus {
        try {
            val testFile = root.resolve("health-check/test-file.txt")
            val testContent = "Test Content"
            testFile.put(HttpContent.Text(testContent, ContentType.Text.Plain))
            if (testFile.head()?.type != ContentType.Text.Plain) return HealthStatus(
                HealthStatus.Level.ERROR,
                additionalMessage = "Test write resulted in file of incorrect MIME type"
            )
            if (testFile.get()!!.text() != testContent) return HealthStatus(
                HealthStatus.Level.ERROR,
                additionalMessage = "Test content did not match"
            )
            return HealthStatus(HealthStatus.Level.OK)
        } catch (e: Exception) {
            return HealthStatus(HealthStatus.Level.ERROR, additionalMessage = e.message)
        }
    }
}