get_column_names(): Get table name from argument

Remove the hard-coded table name.

2 files changed, 14 insertions, 6 deletions

diff --git a/src/main.rs b/src/main.rs
index 629ecbc..b212c46 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,7 +11,7 @@ fn main() {
 
     let mut dbconn = rusqlite::Connection::open("./test.db").unwrap();
 
-    let table_columns = yaqlite::sqlite::get_column_names(&dbconn);
+    let table_columns = yaqlite::sqlite::get_column_names(&dbconn, "people");
     dbg!(&table_columns);
 
     let text_data = std::fs::read_to_string("test2.yml").unwrap();
diff --git a/src/sqlite.rs b/src/sqlite.rs
index 15c813c..d241053 100644
--- a/src/sqlite.rs
+++ b/src/sqlite.rs
@@ -40,13 +40,21 @@ pub fn affinity(type_name: &str) -> rusqlite::types::Type {
 pub struct Zero {}
 
 
-pub fn get_column_names(dbconn: &rusqlite::Connection) -> HashMap<String, Zero> {
+pub fn get_column_names(
+    dbconn: &rusqlite::Connection,
+    table_name: &str,
+) -> HashMap<String, Zero> {
     let mut column_names = HashMap::new();
 
-    let mut stmt = dbconn.prepare(r#"
-        SELECT "name"
-        FROM pragma_table_info("people");
-    "#).unwrap();
+    let mut stmt = dbconn.prepare(
+        &format!(
+            r#"
+                SELECT "name"
+                FROM pragma_table_info("{}");
+            "#,
+            table_name,
+        ),
+    ).unwrap();
 
     let rows = stmt.query_map(
         [],