From 82f6ce1b1f003903c0e8ffceafe17682ecd7fb88 Mon Sep 17 00:00:00 2001
From: "Matthias P. Braendli" <matthias.braendli@mpb.li>
Date: Tue, 23 Jan 2024 09:33:44 +0100
Subject: Create database if it doesn't exist

---
 src/db.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src')

diff --git a/src/db.rs b/src/db.rs
index 5d0f627..d5285ba 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -1,4 +1,5 @@
 use std::time::{SystemTime, UNIX_EPOCH};
+use std::io;
 
 use log::debug;
 use sqlx::SqlitePool;
@@ -18,6 +19,19 @@ pub struct Packet {
 
 impl Database {
     pub async fn new() -> Self {
+        {
+            // Ensure the database file exists
+            match std::fs::OpenOptions::new().write(true)
+                .create_new(true)
+                .open("cats-radio-node.db") {
+                    Ok(_f) => (),
+                    Err(e) if e.kind() == io::ErrorKind::AlreadyExists => (),
+                    Err(e) => {
+                        panic!("Failed to ensure DB exists: {e}");
+                    },
+                }
+        }
+
         let pool = SqlitePool::connect("sqlite:cats-radio-node.db").await.unwrap();
         let mut conn = pool.acquire().await.unwrap();
 
-- 
cgit v1.2.3