feat: b64encode inputs to script

35C4n0r · 35C4n0r · commit dc391700aba8 · 2025-07-29T00:34:10.000+05:30
diff --git a/registry/coder-labs/modules/gemini/main.test.ts b/registry/coder-labs/modules/gemini/main.test.ts
@@ -74,30 +74,26 @@ describe("gemini", async () => {
     const { id } = await setup();
     await execModuleScript(id);
     await expectAgentAPIStarted(id);
-    // const resp = await readFileContainer(id, "/home/coder/.gemini-module/install.log");
-    // console.error("\n\n\nINSTALL_LOG => ", resp, "\n\n\n")
-    //
-    // const resp2 = await readFileContainer(id, "/home/coder/.gemini-module/agentapi-start.log");
-    // console.error("\n\n\nSTART_LOG => ", resp2, "\n\n\n")
   });
 
   test("install-gemini-version", async () => {
+    const version_to_install = "0.1.13";
     const { id } = await setup({
+      skipGeminiMock: true,
       moduleVariables: {
         install_gemini: "true",
-        gemini_version: "v2.5.0",
+        gemini_version: version_to_install,
       },
     });
     await execModuleScript(id);
-    // Check for version in install log or binary (customize as needed)
     const resp = await execContainer(id, [
       "bash",
       "-c",
       `cat /home/coder/.gemini-module/install.log || true`,
     ]);
-    expect(resp.stdout).toContain("v2.5.0");
+    expect(resp.stdout).toContain(version_to_install);
   });
-  //
+
   test("gemini-settings-json", async () => {
     const settings = '{"foo": "bar"}';
     const { id } = await setup({
diff --git a/registry/coder-labs/modules/gemini/main.tf b/registry/coder-labs/modules/gemini/main.tf
@@ -192,7 +192,7 @@ module "agentapi" {
      GOOGLE_GENAI_USE_VERTEXAI='${var.use_vertexai}' \
      GEMINI_MODEL='${var.gemini_model}' \
      GEMINI_START_DIRECTORY='${var.folder}' \
-     GEMINI_TASK_PROMPT='${data.coder_parameter.ai_prompt.value}' \
+     GEMINI_TASK_PROMPT='${base64encode(data.coder_parameter.ai_prompt.value)}' \
      /tmp/start.sh
    EOT
 
@@ -205,11 +205,11 @@ module "agentapi" {
     chmod +x /tmp/install.sh
     ARG_INSTALL='${var.install_gemini}' \
     ARG_GEMINI_VERSION='${var.gemini_version}' \
-    ARG_GEMINI_CONFIG='${var.gemini_settings_json}' \
-    BASE_EXTENSIONS='${replace(local.base_extensions, "'", "'\\''")}' \
-    ADDITIONAL_EXTENSIONS='${replace(var.additional_extensions != null ? var.additional_extensions : "", "'", "'\\''")}' \
+    ARG_GEMINI_CONFIG='${base64encode(var.gemini_settings_json)}' \
+    BASE_EXTENSIONS='${base64encode(replace(local.base_extensions, "'", "'\\''"))}' \
+    ADDITIONAL_EXTENSIONS='${base64encode(replace(var.additional_extensions != null ? var.additional_extensions : "", "'", "'\\''"))}' \
     GEMINI_START_DIRECTORY='${var.folder}' \
-    GEMINI_INSTRUCTION_PROMPT='${var.gemini_system_prompt}' \
+    GEMINI_INSTRUCTION_PROMPT='${base64encode(var.gemini_system_prompt)}' \
     /tmp/install.sh
   EOT
 }
diff --git a/registry/coder-labs/modules/gemini/scripts/install.sh b/registry/coder-labs/modules/gemini/scripts/install.sh
@@ -9,6 +9,11 @@ command_exists() {
 
 set -o nounset
 
+ARG_GEMINI_CONFIG=$(echo -n "$ARG_GEMINI_CONFIG" | base64 -d)
+BASE_EXTENSIONS=$(echo -n "$BASE_EXTENSIONS" | base64 -d)
+ADDITIONAL_EXTENSIONS=$(echo -n "$ADDITIONAL_EXTENSIONS" | base64 -d)
+GEMINI_INSTRUCTION_PROMPT=$(echo -n "$GEMINI_INSTRUCTION_PROMPT" | base64 -d)
+
 echo "--------------------------------"
 printf "gemini_config: %s\n" "$ARG_GEMINI_CONFIG"
 printf "install: %s\n" "$ARG_INSTALL"
diff --git a/registry/coder-labs/modules/gemini/scripts/start.sh b/registry/coder-labs/modules/gemini/scripts/start.sh
@@ -13,7 +13,9 @@ else
   export PATH="$HOME/.npm-global/bin:$PATH"
 fi
 
-printf "Version: %s\n" "$(gemini --version)\n"
+printf "Version: %s\n" "$(gemini --version)"
+
+GEMINI_TASK_PROMPT=$(echo -n "$GEMINI_TASK_PROMPT" | base64 -d)
 
 if command_exists gemini; then
     printf "Gemini is installed\n"
@@ -41,7 +43,7 @@ else
 fi
 
 if [ -n "$GEMINI_TASK_PROMPT" ]; then
-    printf "Running the task prompt\n"
+    printf "Running the task prompt %s\n" "$GEMINI_TASK_PROMPT"
     PROMPT="Every step of the way, report tasks to Coder with proper descriptions and statuses. Your task at hand: $GEMINI_TASK_PROMPT"
     GEMINI_ARGS=(--prompt-interactive "$PROMPT")
 else
