Move createDangerfile on a separated func

danger · Feb 9, 2019 · c0b54d6 · c0b54d6
1 parent 97fcb44
commit c0b54d6
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/Sources/Runner/Commands/Edit.swift b/Sources/Runner/Commands/Edit.swift
@@ -6,24 +6,13 @@ import Logger
 import MarathonCore
 
 func editDanger(logger: Logger) throws {
-    let createDangerfile: (String) -> String = { dangerfilePath in
-        do {
-            let template = "import Danger \nlet danger = Danger()"
-            let data = template.data(using: .utf8)!
-            return try FileSystem().createFile(at: dangerfilePath, contents: data).path
-        } catch {
-            logger.logError("Could not find or generate a Dangerfile")
-            exit(1)
-        }
-    }
-
     let dangerfilePath: String
 
     if let dangerfileArgumentPath = DangerfilePathFinder.dangerfilePath() {
         dangerfilePath = dangerfileArgumentPath
 
         if !FileManager.default.fileExists(atPath: dangerfileArgumentPath) {
-            _ = createDangerfile(dangerfileArgumentPath)
+            createDangerfile(dangerfileArgumentPath)
         }
 
     } else {
@@ -70,3 +59,15 @@ func editDanger(logger: Logger) throws {
 
     try script.watch(arguments: arguments, importedFiles: importedFiles)
 }
+
+@discardableResult
+private func createDangerfile(_ dangerfilePath: String) -> String {
+    do {
+        let template = "import Danger \nlet danger = Danger()"
+        let data = template.data(using: .utf8)!
+        return try FileSystem().createFile(at: dangerfilePath, contents: data).path
+    } catch {
+        logger.logError("Could not find or generate a Dangerfile")
+        exit(1)
+    }
+}