allow line specifications in logs

vapor · Dec 13, 2018 · 9fbaaa7 · 9fbaaa7
1 parent e38b249
commit 9fbaaa7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Sources/CloudCommands/CloudGroup.swift b/Sources/CloudCommands/CloudGroup.swift
@@ -63,6 +63,7 @@ struct Logs: Command {
     var options: [CommandOption] = [
         .app,
         .env,
+        .lines,
         .showTimestamps,
     ]
 
@@ -113,7 +114,8 @@ struct LogsRunner: AuthorizedRunner {
                 // pod -- a specific pod
                 // timestamps -- whether to include timestamps
                 let timestamps = self.ctx.flag(.showTimestamps)
-                let query = "lines=200&timestamps=\(timestamps.description)"
+                let lines = self.ctx.options.value(.lines) ?? "200"
+                let query = "lines=\(lines)&timestamps=\(timestamps.description)"
                 let list = logs.list(query: query)
                 return list.map { list in
                     for log in list {

diff --git a/Sources/CloudCommands/CommandOption.swift b/Sources/CloudCommands/CommandOption.swift
@@ -52,6 +52,14 @@ extension CommandOption {
             "the slug associated with your app."
         ]
     )
+
+    static let lines: CommandOption = .value(
+        name: "lines",
+        short: "l",
+        default: "200",
+        help: ["if passed, should show timestamps"]
+    )
+
     static let showTimestamps: CommandOption = .flag(
         name: "show-timestamps",
         short: "t",