1 files changed, 19 insertions, 0 deletions

diff --git a/src/db.rs b/src/db.rs
new file mode 100644
index 0000000..a5c0f85
--- /dev/null
+++ b/src/db.rs
@@ -0,0 +1,19 @@
+use anyhow::Result;
+use rusqlite::{Connection, OpenFlags, NO_PARAMS};
+
+use std::path::Path;
+
+pub fn init_db<P: AsRef<Path>>(p: P) -> Result<Connection> {
+    let conn = Connection::open_with_flags(
+        p,
+        OpenFlags::SQLITE_OPEN_CREATE | OpenFlags::SQLITE_OPEN_READ_WRITE,
+    )?;
+    conn.execute(
+        "CREATE TABLE IF NOT EXISTS urls (
+            link TEXT PRIMARY KEY,
+            shortlink TEXT NOT NULL
+        )",
+        NO_PARAMS,
+    )?;
+    return Ok(conn);
+}