ci: add tests and update CI file

.env

@@ -1 +1,2 @@
-DATABASE_URL=sqlite:data/bot.db
+DATABASE_URL=sqlite:data/bot.db
+CONFIG_PATH=data/config.yaml

.github/workflows/ci.yml

@@ -9,7 +9,8 @@ on:
 
 env:
   CARGO_TERM_COLOR: always
-  DATABASE_URL: sqlite:data/bot.db
+  DATABASE_URL: sqlite:/tmp/bot.db
+  CONFIG_PATH: /tmp/config.yaml
   REGISTRY: ghcr.io
   IMAGE_NAME: ${{ github.repository }}
 
@@ -28,12 +29,12 @@ jobs:
       - name: Cache dependencies
         uses: Swatinem/rust-cache@v2
 
-      - name: Create data directory
-        run: mkdir -p data
+      - name: Copy config
+        run: cp config.example.yaml /tmp/config.yaml
 
       - name: Install sqlx-cli
         run: cargo install sqlx-cli
-        
+
       - name: Create database
         run: |
           sqlx database create
@@ -59,15 +60,26 @@ jobs:
 
       - name: Cache dependencies
         uses: Swatinem/rust-cache@v2
+
+      - name: Copy config
+        run: cp config.example.yaml /tmp/config.yaml
+
+      - name: Install sqlx-cli
+        run: cargo install sqlx-cli
+
+      - name: Create database
+        run: |
+          sqlx database create
+          sqlx migrate run
       
       - name: Build
         run: cargo build --release
 
       - name: Upload artifacts
         uses: actions/upload-artifact@v4
         with:
-          name: invitationbot
-          path: target/release/invitationbot 
+          name: InvitationBot
+          path: target/release/InvitationBot 
 
   release:
     name: Release
@@ -82,20 +94,17 @@ jobs:
       - name: Download artifact
         uses: actions/download-artifact@v4
         with:
-          name: invitationbot
+          name: InvitationBot
           path: ./
 
       - name: Make binary executable
-        run: chmod +x invitationbot
+        run: chmod +x InvitationBot
 
       - name: Create Release
         uses: softprops/action-gh-release@v1
         with:
           files: |
-            invitationbot
-            LICENSE
-            README.md
-            config.example.yaml
+            InvitationBot
           generate_release_notes: true
 
   docker:

Cargo.toml

@@ -15,4 +15,10 @@ axum = "0.8.1"
 tower-http = { version = "0.6.2", features = ["cors"] }
 chrono = { version = "0.4", features = ["serde"] }
 once_cell = "1.20.2"
-rust-embed = "8.0"
+rust-embed = "8.0"
+
+[dev-dependencies]
+tempfile = "3.9"
+tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }
+uuid = { version = "1.0", features = ["v4"] }
+tower = { version = "0.4", features = ["util"] }

Dockerfile

@@ -1,4 +1,4 @@
-FROM rust:1.75-slim-bookworm as builder
+FROM rust:1.83-slim-bookworm as builder
 
 WORKDIR /usr/src/app
 COPY . .
@@ -8,18 +8,24 @@ RUN apt-get update && apt-get install -y \
     libssl-dev \
     && rm -rf /var/lib/apt/lists/*
 
+ENV DATABASE_URL=sqlite:/tmp/bot.db
+RUN cargo install sqlx-cli && \
+    sqlx database create && \
+    sqlx migrate run
+
 RUN cargo build --release
 
 # Use distroless as runtime image
 FROM gcr.io/distroless/cc-debian12
 
 WORKDIR /app
 
-COPY --from=builder /usr/src/app/target/release/invitationbot /app/
+COPY --from=builder /usr/src/app/target/release/InvitationBot /app/
 COPY --from=builder /usr/src/app/migrations /app/migrations
 
 ENV DATABASE_URL=sqlite:data/bot.db
+ENV CONFIG_PATH=data/config.yaml
 
 VOLUME ["/app/data"]
 USER nonroot
-ENTRYPOINT ["/app/invitationbot"] 
+ENTRYPOINT ["/app/InvitationBot"] 

README.md

@@ -35,6 +35,7 @@ cp config.example.yaml data/config.yaml
 
 # Set up the database
 echo "DATABASE_URL=sqlite:data/bot.db" > .env
+echo "CONFIG_PATH=data/config.yaml" >> .env
 cargo install sqlx-cli
 sqlx database create
 sqlx migrate run

src/handlers/mod.rs

@@ -1,5 +1,5 @@
-use poise::serenity_prelude::{self as serenity};
 use crate::Data;
+use poise::serenity_prelude::{self as serenity};
 
 pub async fn handle_guild_member_addition(
     ctx: &serenity::Context,
@@ -11,7 +11,9 @@ pub async fn handle_guild_member_addition(
     if let Ok(invites) = guild_id.invites(&ctx.http).await {
         for invite in invites {
             // Check if this is our invite code
-            if let Ok(Some(invite_id)) = crate::utils::db::find_invite_by_code(&data.db, &invite.code).await {
+            if let Ok(Some(invite_id)) =
+                crate::utils::db::find_invite_by_code(&data.db, &invite.code).await
+            {
                 // Check if this invite is set to 2 uses, has been used once, and was created by us
                 // If these conditions are met, we can assume this invite was used by the new member
                 if invite.max_uses == 2 && invite.uses == 1 {
@@ -20,13 +22,14 @@ pub async fn handle_guild_member_addition(
                         &data.db,
                         &invite_id,
                         &new_member.user.id.to_string(),
-                    ).await;
-
+                    )
+                    .await;
+
                     // Delete the invite link
                     let _ = invite.delete(&ctx.http).await;
                     break;
                 }
             }
         }
     }
-} 
+}