move sqlite connection function to database.rs

src/database.rs

@@ -0,0 +1,19 @@
+use std::env;
+
+use sqlite::Connection;
+use tracing::error;
+
+pub fn initialise_sqlite_connection() -> Connection {
+    let path = match env::var("DBPATH") {
+        Ok(env) => env,
+        Err(_) => "sqlite.db".to_owned(),
+    };
+    let conn = match sqlite::open(path) {
+        Ok(conn) => conn,
+        Err(e) => {
+            error!("error establishing sqlite db connection: {e}");
+            panic!();
+        }
+    };
+    return conn;
+}

src/main.rs

@@ -1,3 +1,4 @@
+mod database;
 mod routes;
 mod setup;
 use axum::Router;

src/routes/splash.rs

@@ -1,4 +1,4 @@
-use crate::setup;
+use crate::database;
 use axum::{
     extract::{Path, Query},
     http::StatusCode,
@@ -39,7 +39,7 @@ struct SplashGetParams {
 }
 
 async fn splash(Query(params): Query<SplashGetParams>) -> Response {
-    let conn = setup::initialise_sqlite_connection();
+    let conn = database::initialise_sqlite_connection();
     let query = "SELECT * FROM splashes ORDER BY RANDOM() LIMIT 1";
     let mut statement = conn.prepare(query).unwrap();
 
@@ -69,7 +69,7 @@ async fn splash(Query(params): Query<SplashGetParams>) -> Response {
 }
 
 async fn splash_by_id(Path(id): Path<String>) -> Response {
-    let conn = setup::initialise_sqlite_connection();
+    let conn = database::initialise_sqlite_connection();
     let query = "SELECT * FROM splashes WHERE id = :id";
     let mut statement = conn.prepare(query).unwrap();
     statement.bind((":id", id.as_str())).unwrap();
@@ -91,7 +91,7 @@ async fn splash_by_id(Path(id): Path<String>) -> Response {
 }
 
 async fn splashes() -> Response {
-    let conn = setup::initialise_sqlite_connection();
+    let conn = database::initialise_sqlite_connection();
     let query = "SELECT * FROM splashes LIMIT :limit";
     let mut statement = conn.prepare(query).unwrap();
     statement.bind((":limit", 200)).unwrap();

src/setup.rs

@@ -1,11 +1,11 @@
-use sqlite::Connection;
-use std::env;
 use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
 use tokio::net::TcpListener;
 use tracing::Level;
 use tracing::{error, info, trace};
 use tracing_subscriber::FmtSubscriber;
 
+use crate::database;
+
 pub fn initialise_logging() {
     let subscriber = FmtSubscriber::builder()
         .with_max_level(Level::TRACE)
@@ -27,23 +27,8 @@ pub fn initialise_dotenv() {
     };
 }
 
-pub fn initialise_sqlite_connection() -> Connection {
-    let path = match env::var("DBPATH") {
-        Ok(env) => env,
-        Err(_) => "sqlite.db".to_owned(),
-    };
-    let conn = match sqlite::open(path) {
-        Ok(conn) => conn,
-        Err(e) => {
-            error!("error establishing sqlite db connection: {e}");
-            panic!();
-        }
-    };
-    return conn;
-}
-
 pub fn initialise_sqlite_db_tables() {
-    let conn = initialise_sqlite_connection();
+    let conn = database::initialise_sqlite_connection();
     conn.execute(
         "CREATE TABLE IF NOT EXISTS splashes (
             id      TEXT NOT NULL UNIQUE PRIMARY KEY,