cli + query: improvements

crates/brk_cli/src/query.rs

@@ -0,0 +1,40 @@
+use brk_computer::Computer;
+use brk_indexer::Indexer;
+use brk_query::{Index, Output, Params as QueryParams, Query, Tabled, Value};
+use tabled::settings::Style;
+
+pub fn query(indexer: Indexer, computer: Computer, params: &QueryParams) -> color_eyre::Result<()> {
+    let query = Query::build(&indexer, &computer);
+
+    let ids = params.values.iter().flat_map(|v| v.split(",")).collect::<Vec<_>>();
+
+    let index = Index::try_from(params.index.as_str())?;
+
+    let res = query.search(index, &ids, params.from, params.to, params.format)?;
+
+    if params.format.is_some() {
+        println!("{}", res);
+    } else {
+        println!(
+            "{}",
+            match res {
+                Output::Json(v) => match v {
+                    Value::Single(v) => v.to_string(),
+                    v => {
+                        let v = match v {
+                            Value::Single(_) => unreachable!("Already processed"),
+                            Value::List(v) => vec![v],
+                            Value::Matrix(v) => v,
+                        };
+                        let mut table = v.to_table(ids.iter().map(|id| id.to_string()).collect::<Vec<_>>());
+                        table.with(Style::psql());
+                        table.to_string()
+                    }
+                },
+                _ => unreachable!(),
+            }
+        );
+    }
+
+    Ok(())
+}