[birka-cli] add 'tags' command.

1 files changed, 12 insertions, 7 deletions

diff --git a/src/database.rs b/src/database.rs
index 5591d6a..c7c84a0 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -357,18 +357,23 @@ impl TagDatabase {
             .collect())
     }
 
-    pub fn top_tags(&self, n: i64) -> Result<Vec<(i64, String)>> {
+    pub fn top_tags(&self, n: Option<i64>) -> Result<Vec<(i64, String)>> {
         Ok(self
             .0
-            .prepare(
+            .prepare(&format!(
                 "SELECT tags.name, COUNT(mapping.tag) AS tag_count
                  FROM mapping
                  INNER JOIN tags ON tags.id = mapping.tag
                  GROUP BY tag
                  ORDER BY tag_count DESC
-                 LIMIT ?",
-            )?
-            .query_map(params![n], |row| -> rusqlite::Result<(i64, String)> {
+                 {limit}",
+                limit = if let Some(count) = n {
+                    format!("LIMIT {}", count)
+                } else {
+                    "".into()
+                }
+            ))?
+            .query_map(params![], |row| -> rusqlite::Result<(i64, String)> {
                 Ok((row.get(1)?, row.get(0)?))
             })?
             .filter_map(|s| s.ok())
@@ -599,7 +604,7 @@ mod tests {
         tb.tag_file(ids[1].id, &vec!["2"]).unwrap();
         tb.tag_file(ids[2].id, &vec!["1"]).unwrap();
         tb.tag_file(ids[2].id, &vec!["2"]).unwrap();
-        assert_eq!(tb.top_tags(2).unwrap().len(), 2);
-        assert_eq!(tb.top_tags(2).unwrap()[0].0, 2);
+        assert_eq!(tb.top_tags(Some(2)).unwrap().len(), 2);
+        assert_eq!(tb.top_tags(Some(2)).unwrap()[0].0, 2);
     }
 }