Add static func on Environment

ml-archive · Oct 23, 2018 · 815e3da · 815e3da
1 parent 8becbd4
commit 815e3da
Showing 1 changed file with 8 additions and 4 deletions.
diff --git 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)
 }