feat: image generation based off language

src/vmm/src/service.rs

@@ -1,5 +1,5 @@
 use self::vmmorchestrator::{
-    vmm_service_server::VmmService as VmmServiceTrait, RunVmmRequest, RunVmmResponse,
+    vmm_service_server::VmmService as VmmServiceTrait, Language, RunVmmRequest, RunVmmResponse
 };
 use crate::core::vmm::VMM;
 use crate::VmmErrors;
@@ -36,7 +36,7 @@ pub struct VmmService;
 impl VmmServiceTrait for VmmService {
     async fn run(
         &self,
-        _request: Request<RunVmmRequest>,
+        request: Request<RunVmmRequest>,
     ) -> Result<Response<RunVmmResponse>, Status> {
         let response = vmmorchestrator::RunVmmResponse {};
 
@@ -77,10 +77,21 @@ impl VmmServiceTrait for VmmService {
         let initramfs_exists = Path::new(&initramfs_entire_file_path).try_exists().expect(&format!("Could not access folder {:?}", &initramfs_entire_file_path));
         if !initramfs_exists
         {
+            // get request with the language
+            let req: RunVmmRequest = request.into_inner();
+            let language: Language = Language::from_i32(req.language).expect("Unknown language");
+
+            let image = match language {
+                Language::Rust    => "rust:alpine",
+                Language::Python  => "python:alpine",
+                Language::Node    => "node:alpine",
+            };
+
             info!("Initramfs not found, building initramfs");
             // Execute the script using sh and capture output and error streams
             let output = Command::new("sh")
                 .arg("./tools/rootfs/mkrootfs.sh")
+                .arg(image)
                 .stdout(Stdio::piped())
                 .stderr(Stdio::piped())
                 .output()

tools/rootfs/mkrootfs.sh

@@ -8,7 +8,7 @@ fi
 
 if [ -d fs-gen ]
 then
-    cargo run --bin fs-gen -- alpine:latest ./fs-gen/test -o ../tools/rootfs/initramfs.img
+    cargo run --bin fs-gen -- $1 ./fs-gen/test -o ../tools/rootfs/initramfs.img
 else
     echo "Module fs-gen not found"
 fi