log count

src/args.rs

@@ -33,6 +33,7 @@ pub enum HazeArgs {
     Clean,
     Logs {
         filter: Option<String>,
+        count: Option<usize>,
     },
     Open {
         filter: Option<String>,
@@ -103,7 +104,10 @@ impl HazeArgs {
             }),
             HazeCommand::Db => Ok(HazeArgs::Db { filter }),
             HazeCommand::Clean => Ok(HazeArgs::Clean),
-            HazeCommand::Logs => Ok(HazeArgs::Logs { filter }),
+            HazeCommand::Logs => Ok(HazeArgs::Logs {
+                filter,
+                count: args.next().map(|arg| arg.as_ref().parse()).transpose()?,
+            }),
             HazeCommand::Open => Ok(HazeArgs::Open { filter }),
         }
     }

src/cloud.rs

@@ -272,13 +272,14 @@ impl Cloud {
         Ok(())
     }
 
-    pub async fn logs(&self, docker: &mut Docker) -> Result<Vec<String>> {
+    pub async fn logs(&self, docker: &mut Docker, count: usize) -> Result<Vec<String>> {
         let mut logs = Vec::new();
         let mut stream = docker.logs::<String>(
             &self.id,
             Some(LogsOptions {
                 stdout: true,
                 stderr: true,
+                tail: format!("{}", count),
                 ..Default::default()
             }),
         );

src/main.rs

@@ -57,9 +57,9 @@ async fn main() -> Result<()> {
             let cloud = Cloud::get_by_filter(&mut docker, filter, &config).await?;
             cloud.destroy(&mut docker).await?;
         }
-        HazeArgs::Logs { filter } => {
+        HazeArgs::Logs { filter, count } => {
             let cloud = Cloud::get_by_filter(&mut docker, filter, &config).await?;
-            let logs = cloud.logs(&mut docker).await?;
+            let logs = cloud.logs(&mut docker, count.unwrap_or(20)).await?;
             for log in logs {
                 print!("{}", log);
             }