From 63be98db7cb8c48c77dc634ad092d4fc077e3381 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Sat, 12 Mar 2022 21:36:53 +0100
Subject: get_column_names(): Get table name from argument

Remove the hard-coded table name.
---
 src/sqlite.rs | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

(limited to 'src/sqlite.rs')

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(
         [],
-- 
cgit v1.2.3