add 'take'

2 files changed, 36 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 765c408..0bb6df1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,6 +5,7 @@ use std::process::exit;
 
 mod fetch;
 mod sort;
+mod take;
 
 #[derive(Parser)]
 struct Args {
@@ -24,7 +25,10 @@ enum Commands {
 
         #[arg(short, long)]
         order: Option<Order>
-    }
+    },
+
+    /// Gets the first N items
+    Take { n: usize }
 }
 
 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
@@ -66,7 +70,7 @@ async fn main() {
                             if order == Order::Desc {
                                 sorted.items.reverse();
                             }
-                            
+
                             sorted.pretty_write_to(std::io::stdout(), b' ', 2).unwrap();
                         }, 
                         Err(e) => {
@@ -80,6 +84,26 @@ async fn main() {
                     exit(1);
                 }
             }
+        },
+
+        Commands::Take { n } => {
+            match Channel::read_from(std::io::stdin().lock()) {
+                Ok(c) => {
+                    match take::run(c, n) {
+                        Ok(c) => {
+                            c.pretty_write_to(std::io::stdout(), b' ', 2).unwrap();
+                        }, 
+                        Err(e) => {
+                            eprintln!("Error with 'sort': {:?}", e);
+                            exit(1);
+                        }
+                    }
+                },
+                Err(e) => {
+                    eprintln!("Could not parse RSS: {:?}", e);
+                    exit(1);
+                }
+            }
         }
     }
 }
diff --git a/src/take.rs b/src/take.rs
new file mode 100644
index 0000000..eed7ad1
--- /dev/null
+++ b/src/take.rs
@@ -0,0 +1,10 @@
+use anyhow::Result;
+use rss::Channel;
+
+pub fn run(channel: Channel, n: usize) -> Result<Channel> {
+    let mut new_channel = channel.clone();
+
+    new_channel.items.truncate(n);
+
+    Ok(new_channel)
+}
\ No newline at end of file