select(): Column name interface and default primary key column interface

Add a new function that allows using a specified column name for
selection. The `select()` function will default to using the table's
primary key column.

Still need to update the references to the "id" column in the function.

1 files changed, 14 insertions, 0 deletions

diff --git a/src/select.rs b/src/select.rs
index 1405347..71a096e 100644
--- a/src/select.rs
+++ b/src/select.rs
@@ -14,6 +14,20 @@ pub fn select(
     table_name: &str,
     record_id: &str,
 ) -> Result<yaml_rust::Yaml, crate::Error> {
+    select_by_column(
+        dbconn,
+        table_name,
+        &crate::sqlite::table_primary_key_column(dbconn, table_name)?,
+        record_id,
+    )
+}
+
+pub fn select_by_column(
+    dbconn: &rusqlite::Connection,
+    table_name: &str,
+    primary_key_column: &str,
+    record_id: &str,
+) -> Result<yaml_rust::Yaml, crate::Error> {
     use crate::yaml::Yaml;
 
     let mut stmt = dbconn.prepare(