feat: provide database access for pre-commit

Author: bitfl0wer

.github/workflows/pre-commit.yaml

@@ -15,7 +15,23 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+      - uses: Swatinem/rust-cache@9d47c6ad4b02e050fd481d890b2ea34778fd09d6
+        with:
+          cache-all-crates: "true"
+      - name: Setup PostgreSQL for Linux/macOS/Windows
+        uses: ikalnytskyi/action-setup-postgres@v7
+        id: postgres
+        with:
+          postgres-version: 17
+      - name: "Install dependencies, prepare database"
+        env:
+          DATABASE_URL: ${{ steps.postgres.outputs.connection-uri }}
+        run: |
+          cargo install -f -q sqlx-cli
+          cargo sqlx migrate run
       - uses: actions/setup-python@v5
         with:
           python-version: "3.13"
       - uses: pre-commit/action@576ff52938d158a24ac7e009dfa94b1455e7df99
+        env:
+          DATABASE_URL: ${{ steps.postgres.outputs.connection-uri }}

src/main.rs

@@ -15,6 +15,7 @@ use std::str::FromStr;
 
 use clap::Parser;
 use log::{LevelFilter, debug, error, info, trace};
+use polyproto::key;
 use sqlx::query_scalar;
 
 /// Module housing the HTTP API routes and functionality
@@ -105,16 +106,16 @@ async fn main() -> StdResult<()> {
     };
     let keys_in_table =
         query_scalar!("SELECT COUNT(*) FROM api_keys").fetch_one(&database.pool).await?;
-    if let Some(number) = keys_in_table
-        && number >= 1
-    {
-    } else {
-        let api_key =
-            api_keys::add_api_key_to_database(&ApiKey::new_random(&mut rand::rng()), &database)
-                .await?;
-        info!("Added an API key to the database, since none were available: {api_key}");
-        info!("Save this API key, as it will not be shown again on future starts.");
-    }
+    match keys_in_table {
+        Some(0) | None => {
+            let api_key =
+                api_keys::add_api_key_to_database(&ApiKey::new_random(&mut rand::rng()), &database)
+                    .await?;
+            info!("Added an API key to the database, since none were available: {api_key}");
+            info!("Save this API key, as it will not be shown again on future starts.");
+        }
+        _ => (),
+    };
 
     Ok(())
 }