Add dev/prod conditions in the nix files

api/default.nix

@@ -1,4 +1,4 @@
-{pkgs, nosocket, ...}:
+{pkgs, nosocket, appEnv, ...}:
 with pkgs; rec {
   package = mkYarnPackage {
     name = "piston";
@@ -25,11 +25,8 @@ with pkgs; rec {
       };
     };
   };
-  container = pkgs.dockerTools.buildLayeredImageWithNixDb {
-    name = "piston";
-    tag = "base-latest";
 
-    contents = with pkgs; [
+  basePackages = with pkgs; [
     package
     nosocket
     bash
@@ -42,6 +39,8 @@ with pkgs; rec {
     gnugrep
     rename
     util-linux
+  ];
+  devPackages = with pkgs; [
     nodejs-16_x
     yarn
     python3
@@ -50,7 +49,7 @@ with pkgs; rec {
     gnused
   ];
 
-    extraCommands = ''
+  baseCommands = ''
     mkdir -p piston/{jobs,runtimes} etc/nix {,var/}tmp run/lock
     echo -e "experimental-features = nix-command flakes" >> etc/nix/nix.conf
     echo "nixbld:x:30000:nixbld1,nixbld10,nixbld11,nixbld12,nixbld13,nixbld14,nixbld15,nixbld16,nixbld17,nixbld18,nixbld19,nixbld2,nixbld20,nixbld21,nixbld22,nixbld23,nixbld24,nixbld25,nixbld26,nixbld27,nixbld28,nixbld29,nixbld3,nixbld30,nixbld31,nixbld32,nixbld4,nixbld5,nixbld6,nixbld7,nixbld8,nixbld9" >> etc/group
@@ -59,12 +58,22 @@ with pkgs; rec {
       echo "nixbld$i:x:$(( $i + 30000 )):30000:Nix build user $i:/var/empty:/run/current-system/sw/bin/nologin" >> etc/passwd
     done
 
+    chmod 1777 {,var/}tmp/
+  '';
+  devCommands = ''
     mkdir -p usr/bin
     ln -s /bin/env usr/bin/env
     chmod -R 1777 usr
-      chmod 1777 {,var/}tmp/
   '';
 
+  container = pkgs.dockerTools.buildLayeredImageWithNixDb {
+    name = "piston";
+    tag = "base-latest";
+
+    contents = if appEnv == "dev" then basePackages ++ devPackages else basePackages;
+
+    extraCommands = if appEnv == "dev" then baseCommands + devCommands else baseCommands;
+
     config = {
       Cmd = ["${package}/bin/pistond"];
       Env = [

flake.nix

@@ -6,9 +6,16 @@
   let
     system = "x86_64-linux";
     pkgs = nixpkgs.legacyPackages.${system};
-    baseContainer = (import ./api {
-      inherit pkgs;
-      nosocket = self.legacyPackages."${system}".nosocket;
+    _nosocket = self.legacyPackages."${system}".nosocket;
+    _devContainer = (import ./api {
+      inherit pkgs _nosocket;
+      nosocket = _nosocket;
+      appEnv = "dev";
+    }).container;
+    _prodContainer = (import ./api {
+      inherit pkgs _nosocket;
+      nosocket = _nosocket;
+      appEnv = "prod";
     }).container;
     args = {
       inherit pkgs;
@@ -74,6 +81,7 @@
       piston = (import ./api { inherit pkgs nosocket; }).package;
     };
 
-    container = baseContainer;
+    devContainer = _devContainer;
+    prodContainer = _prodContainer;
   };
 }

piston

@@ -13,7 +13,8 @@ SUBCOMMAND="$1"
 shift
 
 build_base() {
-    CONTAINER_PATH="$(nix build ".#container" --no-link --json | jq '.[0].outputs.out' -r)"
+    container_flake_key=$([[ $1 == "dev" ]] && echo "devContainer" || echo "prodContainer")
+    CONTAINER_PATH="$(nix build ".#$container_flake_key" --no-link --json | jq '.[0].outputs.out' -r)"
     echo "The image archive was created at: $CONTAINER_PATH"
     docker load -i $CONTAINER_PATH || exit 1
     docker tag "$IMAGE_NAME_DEV:$IMAGE_TAG" "$IMAGE_NAME:$IMAGE_TAG" || exit 1
@@ -64,7 +65,7 @@ case "$SUBCOMMAND" in
         echo "Removing the Nix volume if it exists"
         docker volume rm -f $DEV_VOLUME_NAME
         echo "Building the base docker image"
-        build_base
+        build_base dev
         echo "Installing the required node modules"
         docker run \
             --rm \