io.github.cdimascio.dotenv.internal.ClassPathHelper.kt Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) Carmine DiMascio 2017 - 2018
 * License: MIT
 */
package io.github.cdimascio.dotenv.internal

import io.github.cdimascio.dotenv.DotEnvException
import java.io.IOException
import java.io.InputStream
import java.util.*
import java.util.stream.Stream

internal object ClasspathHelper {
    /**
     * Loads the contents of a file at the specified location and returns
     * the contents of that file as a Stream of Strings
     * @return A Stream contain the contents of the file at location
     */
    fun loadFileFromClasspath(location: String): Stream {
        val loader = ClasspathHelper::class.java
        val inputStream: InputStream? =
                loader.getResourceAsStream(location)
                        ?: loader.getResourceAsStream(location)
                        ?: ClassLoader.getSystemResourceAsStream(location)
        if (inputStream != null) {
            try {
                val scanner = Scanner(inputStream, "utf-8")
                val lines = mutableListOf()
                while (scanner.hasNext()) {
                    lines.add(scanner.nextLine())
                }
                return lines.stream()
            } catch (e: IOException) {
                throw DotEnvException("Could not parse $location from the classpath")
            }
        }
        throw DotEnvException("Could not find $location on the classpath")
    }
}