1 files changed, 70 insertions, 0 deletions
diff --git a/examples/todo/commands/configure.rs b/examples/todo/commands/configure.rs
new file mode 100644
index 0000000..42ffafd
--- /dev/null
+++ b/examples/todo/commands/configure.rs
@@ -0,0 +1,70 @@
+use crate::TodoConfig;
+use anyhow::Result;
+use notion::models::search::NotionSearch;
+use notion::models::{Database, DatabaseId};
+use notion::{AsIdentifier, NotionApi};
+use skim::{Skim, SkimItem, SkimItemReceiver, SkimItemSender, SkimOptions};
+use std::borrow::Cow;
+use std::ops::Deref;
+use std::sync::Arc;
+
+fn skim_select_database(databases: Vec<Database>) -> Result<DatabaseId> {
+    let options = SkimOptions::default();
+
+    let (sender, receiver): (SkimItemSender, SkimItemReceiver) = crossbeam_channel::bounded(500);
+
+    struct SkimDB {
+        db: Database,
+    }
+
+    impl SkimItem for SkimDB {
+        fn text(&self) -> Cow<str> {
+            Cow::Owned(self.db.title_plain_text())
+        }
+    }
+
+    for db in databases {
+        sender.send(Arc::new(SkimDB { db }))?;
+    }
+
+    // `run_with` would read and show items from the stream
+    let selected_items = Skim::run_with(&options, Some(receiver))
+        .filter(|out| !out.is_abort)
+        .map(|out| out.selected_items)
+        .unwrap_or_else(|| Vec::new());
+
+    let db = selected_items
+        .first()
+        .expect("No database selected, aborting...")
+        .clone();
+    let db: &SkimDB = db
+        .deref()
+        .as_any()
+        .downcast_ref()
+        .expect("Couldn't cast back to SkimDB");
+
+    let database_id = db.db.id();
+
+    Ok(database_id)
+}
+
+pub async fn configure(notion_api: NotionApi) -> Result<()> {
+    let databases: Vec<Database> = notion_api
+        .search(NotionSearch::filter_by_databases())
+        .await?
+        .only_databases()
+        .results;
+
+    let database_id = skim_select_database(databases)?;
+
+    println!("Selected database's id: {}", database_id);
+
+    let bytes = toml::to_vec(&TodoConfig {
+        api_token: None,
+        task_database_id: Some(database_id),
+    })?;
+
+    std::fs::write("../todo_config.toml", bytes)?;
+
+    Ok(())
+}