feat: Implement local driver

.devcontainer/Dockerfile

@@ -26,9 +26,6 @@ USER ${REMOTE_USER}
 # Install Rustup and set default toolchains
 RUN rustup default nightly
 
-# Install the wasm32-wasi target for nightly
-RUN rustup target add wasm32-wasip1 --toolchain nightly
-
 # Install binstall
 RUN curl -L --proto '=https' --tlsv1.2 -sSf https://raw.githubusercontent.com/cargo-bins/cargo-binstall/main/install-from-binstall-release.sh | bash
 

.github/docker/entrypoint.sh

@@ -1,17 +1,18 @@
 #!/bin/sh
 
-# Check if the environment variable PTERODACTYL is set to true
-if [ "$PTERODACTYL" = "true" ]; then
-    # Define the URL and the destination folder
-    WASM_URL="https://github.com/HttpRafa/atomic-cloud/releases/latest/download/pterodactyl-driver.wasm"
-    DEST_FOLDER="drivers/wasm"
-    WASM_FILE="$DEST_FOLDER/pterodactyl.wasm"
+# Define the destination folder
+DEST_FOLDER="drivers/wasm"
 
-    # Create the destination folder if it does not exist
-    if [ ! -d "$DEST_FOLDER" ]; then
-        echo "Creating directory $DEST_FOLDER..."
-        mkdir -p "$DEST_FOLDER"
-    fi
+# Create the destination folder if it does not exist
+if [ ! -d "$DEST_FOLDER" ]; then
+    echo "Creating directory $DEST_FOLDER..."
+    mkdir -p "$DEST_FOLDER"
+fi
+
+# Function to download the .wasm file
+download_wasm() {
+    local WASM_URL=$1
+    local WASM_FILE=$2
 
     # Check if the .wasm file already exists
     if [ -f "$WASM_FILE" ]; then
@@ -25,9 +26,21 @@ if [ "$PTERODACTYL" = "true" ]; then
 
         echo "Download complete."
     fi
-else
-    echo "PTERODACTYL is not set to true. Skipping download."
+}
+
+# Check if the environment variable PTERODACTYL is set to true
+if [ "$PTERODACTYL" = "true" ]; then
+    WASM_URL="https://github.com/HttpRafa/atomic-cloud/releases/latest/download/pterodactyl-driver.wasm"
+    WASM_FILE="$DEST_FOLDER/pterodactyl.wasm"
+    download_wasm $WASM_URL $WASM_FILE
+fi
+
+# Check if the environment variable LOCAL is set to true
+if [ "$LOCAL" = "true" ]; then
+    WASM_URL="https://github.com/HttpRafa/atomic-cloud/releases/latest/download/local-driver.wasm"
+    WASM_FILE="$DEST_FOLDER/local.wasm"
+    download_wasm $WASM_URL $WASM_FILE
 fi
 
 # Run the main command
-./controller "$@"
+exec ./controller "$@"

.github/workflows/ci.yml

@@ -69,7 +69,7 @@ jobs:
           CURRENT_BUILD: ${{ github.run_number }}
 
       - name: Run Tests
-        run: cargo test --workspace --exclude pterodactyl --all-features --verbose
+        run: cargo test --workspace --exclude pterodactyl --exclude local --all-features --verbose
 
       - name: Build & Test JVM Client
         working-directory: clients/jvm

.github/workflows/release.yml

@@ -72,11 +72,12 @@ jobs:
           git push -u origin $BRANCH
           sed -i 's/client_version=0\.0\.0-nightly/client_version=\${REF:11}/g' clients/jvm/gradle.properties
           sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' drivers/pterodactyl/Cargo.toml
+          sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' drivers/local/Cargo.toml
           sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' clients/wrapper/Cargo.toml
           sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' controller/Cargo.toml
           sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' common/Cargo.toml
           sed -i 's/version = "0\.0\.0-nightly"/version = "\${REF:11}"/g' cli/Cargo.toml
-          git add clients/jvm/gradle.properties drivers/pterodactyl/Cargo.toml clients/wrapper/Cargo.toml controller/Cargo.toml common/Cargo.toml cli/Cargo.toml
+          git add clients/jvm/gradle.properties drivers/pterodactyl/Cargo.toml drivers/local/Cargo.toml clients/wrapper/Cargo.toml controller/Cargo.toml common/Cargo.toml cli/Cargo.toml
           git commit -m "ci(release): bump version"
           git push
 
@@ -112,6 +113,7 @@ jobs:
           cp ./target/x86_64-pc-windows-gnu/release/wrapper.exe wrapper-windows-x86_64.exe
 
           cp ./target/wasm32-wasip2/release/pterodactyl.wasm pterodactyl-driver.wasm
+          cp ./target/wasm32-wasip2/release/local.wasm local-driver.wasm
 
           cp $(find ./clients/jvm/paper/build -name "*-all.jar") paper-client.jar
 
@@ -131,4 +133,5 @@ jobs:
             wrapper-linux-x86_64
             wrapper-windows-x86_64.exe
             pterodactyl-driver.wasm
+            local-driver.wasm
             paper-client.jar

Cargo.toml

@@ -9,6 +9,7 @@ members = [
 
     # Drivers
     "drivers/pterodactyl",
+    "drivers/local",
 
     # Clients
     "cli",

Makefile

@@ -3,7 +3,6 @@
 # Configuration
 WASM_RUSTFLAGS = -Z wasi-exec-model=reactor
 WASM_TARGET = wasm32-wasip2
-WASM_COMPONENT = target/wasm32-wasip2/release/pterodactyl.wasm
 
 # Directories
 RUN_DIR = run
@@ -72,6 +71,7 @@ build-wrapper:
 build-drivers:
 	$(SETENV) RUSTFLAGS="$(WASM_RUSTFLAGS)"
 	cargo build -p pterodactyl --target $(WASM_TARGET) --release
+	cargo build -p local --target $(WASM_TARGET) --release
 
 # Create driver directory if it doesn't exist
 $(DRIVER_DIR):

README.md

@@ -2,6 +2,12 @@
 
 Atomic Cloud is a cloud that was primarily developed for the game Minecraft, but it is possible to integrate other games with Atomic Cloud.
 
+## Disclaimer
+
+This project is a work in progress and a way for me to learn Rust. Since it’s one of my first Rust apps, the code might not be perfect. You might notice things like too many locks or overusing Mutex, and I’m still figuring out some of Rust’s trickier features like lifetimes.
+
+If you like the project, I’d love to hear your suggestions or corrections! Just keep it friendly and helpful. Instead of saying, "everything is bad" let me know what can be improved and how. Over time, the code will get better.
+
 ## Documentation
 
 For detailed information, please visit the [Atomic Cloud Documentation](https://httprafa.github.io/atomic-cloud/).

cli/Cargo.toml

@@ -12,7 +12,7 @@ simplelog = { version = "0.12.2", features = ["paris"] }
 loading = "0.4.0"
 
 # File management
-dirs = "5.0.1"
+dirs = "6.0.0"
 
 # Input parsing
 inquire = "0.7.5"

cli/src/application/profile.rs

@@ -86,7 +86,7 @@ impl Profiles {
             profiles.add_profile(profile);
         }
 
-        progress.success(format!("Loaded {} profiles(s)", profiles.profiles.len()));
+        progress.success(format!("Loaded {} profile(s)", profiles.profiles.len()));
         progress.end();
         profiles
     }
@@ -177,7 +177,7 @@ impl Profile {
             authorization: self.authorization.clone(),
             url: self.url.clone(),
         };
-        stored_profile.save_to_file(&Storage::get_profile_file(&self.id))
+        stored_profile.save_to_file(&Storage::get_profile_file(&self.id), true)
     }
 
     pub fn compute_id(name: &str) -> String {

clients/wrapper/Cargo.toml

@@ -17,7 +17,7 @@ anyhow = "1.0.95"
 ctrlc = "3.4.5"
 
 # User system
-uuid = { version = "1.11.1", features = ["v4"] }
+uuid = { version = "1.12.0", features = ["v4"] }
 
 # Command line arguments
 clap = { version = "4.5.26", features = ["derive"] }

common/src/allocator.rs

@@ -0,0 +1,47 @@
+use std::{
+    collections::{BTreeSet, HashSet},
+    hash::Hash,
+    ops::{AddAssign, Range},
+};
+
+pub struct NumberAllocator<T> {
+    next: T,
+    max: T,
+    available: BTreeSet<T>,
+    active: HashSet<T>,
+}
+
+impl<T> NumberAllocator<T>
+where
+    T: Copy + Ord + Hash + AddAssign + From<u8>,
+{
+    pub fn new(range: Range<T>) -> Self {
+        Self {
+            next: range.start,
+            max: range.end,
+            available: BTreeSet::new(),
+            active: HashSet::new(),
+        }
+    }
+
+    pub fn allocate(&mut self) -> Option<T> {
+        if let Some(&id) = self.available.iter().next() {
+            self.available.remove(&id);
+            self.active.insert(id);
+            Some(id)
+        } else if self.next < self.max {
+            let id = self.next;
+            self.next += T::from(1);
+            self.active.insert(id);
+            Some(id)
+        } else {
+            None
+        }
+    }
+
+    pub fn release(&mut self, value: T) {
+        if self.active.remove(&value) {
+            self.available.insert(value);
+        }
+    }
+}

common/src/config.rs

@@ -4,9 +4,11 @@ use anyhow::Result;
 use serde::{de::DeserializeOwned, Serialize};
 
 pub trait SaveToTomlFile: Serialize {
-    fn save_to_file(&self, path: &Path) -> Result<()> {
-        if let Some(parent) = path.parent() {
-            fs::create_dir_all(parent)?;
+    fn save_to_file(&self, path: &Path, create_parent: bool) -> Result<()> {
+        if create_parent {
+            if let Some(parent) = path.parent() {
+                fs::create_dir_all(parent)?;
+            }
         }
         fs::write(path, toml::to_string(self)?)?;
         Ok(())

common/src/lib.rs

@@ -1,3 +1,6 @@
+pub mod allocator;
 pub mod config;
 pub mod init;
+pub mod name;
+pub mod tick;
 pub mod version;

common/src/name.rs

@@ -0,0 +1,45 @@
+use std::time::{SystemTime, UNIX_EPOCH};
+
+#[derive(Clone)]
+pub struct TimedName {
+    raw_name: String,
+    name: String,
+}
+
+impl TimedName {
+    pub fn new(cloud_identifier: &str, name: &str, permanent: bool) -> Self {
+        Self {
+            raw_name: name.to_string(),
+            name: Self::generate(Some(cloud_identifier.to_string()), name, permanent),
+        }
+    }
+    pub fn new_no_identifier(name: &str, permanent: bool) -> Self {
+        Self {
+            raw_name: name.to_string(),
+            name: Self::generate(None, name, permanent),
+        }
+    }
+
+    fn generate(cloud_identifier: Option<String>, name: &str, permanent: bool) -> String {
+        let timestamp = SystemTime::now()
+            .duration_since(UNIX_EPOCH)
+            .unwrap()
+            .as_secs();
+        match (cloud_identifier, permanent) {
+            (Some(identifier), true) => format!("{}@{}", name, identifier),
+            (None, true) => name.to_string(),
+            (Some(identifier), false) => format!("{}@{}#{}", name, identifier, timestamp),
+            (None, false) => format!("{}#{}", name, timestamp),
+        }
+    }
+
+    pub fn get_name(&self) -> &str {
+        &self.name
+    }
+    pub fn get_name_cloned(&self) -> String {
+        self.name.clone()
+    }
+    pub fn get_raw_name(&self) -> &str {
+        &self.raw_name
+    }
+}

common/src/tick.rs

@@ -0,0 +1,6 @@
+#[derive(PartialEq)]
+pub enum TickResult {
+    Ok,
+    Drop,
+    Stop,
+}

controller/Cargo.toml

@@ -17,7 +17,7 @@ anyhow = "1.0.95"
 ctrlc = "3.4.5"
 
 # Unit system
-uuid = { version = "1.11.1", features = ["v4"] }
+uuid = { version = "1.12.0", features = ["v4"] }
 
 # Command line arguments
 clap = { version = "4.5.26", features = ["derive"] }
@@ -39,8 +39,8 @@ prost = "0.13.4"
 tonic = "0.12.3"
 
 # Drivers
-wasmtime = { version = "28.0.0", default-features = false, features = ["runtime", "component-model", "cranelift", "parallel-compilation", "cache"], optional = true }
-wasmtime-wasi = { version = "28.0.0", optional = true }
+wasmtime = { version = "28.0.1", default-features = false, features = ["runtime", "component-model", "cranelift", "parallel-compilation", "cache"], optional = true }
+wasmtime-wasi = { version = "28.0.1", optional = true }
 minreq = { version = "2.13.0", features = ["https-rustls"], optional = true }
 
 [build-dependencies]

controller/src/application.rs

@@ -113,6 +113,10 @@ impl Controller {
         info!("<red>Stopping</> all units...");
         self.units.stop_all_instant();
 
+        // Let the drivers cleanup there messes
+        info!("Letting the drivers <red>cleanup</>...");
+        self.drivers.cleanup();
+
         // Stop network stack
         info!("<red>Stopping</> network stack...");
         network_handle.shutdown();
@@ -179,6 +183,12 @@ impl Controller {
     }
 
     fn tick(&self) {
+        // Tick all drivers
+        self.drivers.tick();
+
+        // Tick all driver cloudlets
+        self.lock_cloudlets().tick();
+
         // Check if all deployments have started there units etc..
         self.lock_deployments().tick(&self.units);
 

controller/src/application/auth.rs

@@ -176,7 +176,7 @@ impl Auth {
             token: token.to_string(),
         };
         let user_path = Storage::get_user_file(username);
-        if stored_user.save_to_file(&user_path).is_err() {
+        if stored_user.save_to_file(&user_path, true).is_err() {
             error!(
                 "<red>Failed</> to save user to file: <red>{}</>",
                 &user_path.display()

controller/src/application/cloudlet.rs

@@ -116,6 +116,17 @@ impl Cloudlets {
         cloudlets
     }
 
+    pub fn tick(&self) {
+        for cloudlet in self.cloudlets.values() {
+            if let Err(error) = cloudlet.get_inner().tick() {
+                error!(
+                    "<red>Failed</> to tick cloudlet <blue>{}</>: <red>{}</>",
+                    cloudlet.name, error
+                );
+            }
+        }
+    }
+
     pub fn get_amount(&self) -> usize {
         self.cloudlets.len()
     }
@@ -215,7 +226,7 @@ impl Cloudlets {
 
         match self.add_cloudlet(cloudlet) {
             Ok(_) => {
-                stored_cloudlet.save_to_file(&Storage::get_cloudlet_file(name))?;
+                stored_cloudlet.save_to_file(&Storage::get_cloudlet_file(name), true)?;
                 info!("<green>Created</> cloudlet <blue>{}</>", name);
                 Ok(CreationResult::Created)
             }
@@ -393,7 +404,7 @@ impl Cloudlet {
             status: self.status.read().unwrap().clone(),
             controller: self.controller.clone(),
         };
-        stored_cloudlet.save_to_file(&Storage::get_cloudlet_file(&self.name))
+        stored_cloudlet.save_to_file(&Storage::get_cloudlet_file(&self.name), true)
     }
 }