diff --git a/Sources/Sugar/Helpers/Environment.swift b/Sources/Sugar/Helpers/Environment.swift index 244734d..08f5416 100644 --- a/Sources/Sugar/Helpers/Environment.swift +++ b/Sources/Sugar/Helpers/Environment.swift @@ -12,6 +12,13 @@ public extension Environment { public static func get(_ key: String) -> String? { return ProcessInfo.processInfo.environment[key] } + + public static func assertGet(_ key: String) throws -> String { + guard let value = Environment.get(key) else { + throw Environment.EnvironmentError.keyNotFound(key: key) + } + return value + } } public func env(_ key: String, _ fallback: String) -> String { @@ -23,8 +30,5 @@ public func env(_ key: String) -> String? { } public func assertEnv(_ key: String) throws -> String { - guard let value = Environment.get(key) else { - throw Environment.EnvironmentError.keyNotFound(key: key) - } - return value + return try Environment.assertGet(key) }