agent: support multiple architectures

Preload `wire` with the supported architectures and it will copy its arch-specific
agent store path to the nodes.

Author: itslychee

wire/agent/default.nix

@@ -1,16 +1,26 @@
 {
-  perSystem =
-    {
-      buildRustProgram,
-      ...
-    }:
-    {
-      packages = {
-        agent = buildRustProgram {
-          name = "agent";
-          pname = "agent";
-          cargoExtraArgs = "-p agent";
-        };
-      };
-    };
+  config,
+  withSystem,
+  lib,
+  ...
+}:
+{
+
+  flake.packages =
+    # macos is not supported with the agent
+    lib.genAttrs (builtins.filter (system: !lib.hasSuffix "darwin" system) config.systems)
+
+      (
+        system:
+        withSystem system (
+          { buildRustProgram, ... }:
+          {
+            agent = buildRustProgram {
+              name = "agent";
+              pname = "agent";
+              cargoExtraArgs = "-p agent";
+            };
+          }
+        )
+      );
 }

wire/cli/default.nix

@@ -1,7 +1,9 @@
+{ self, ... }:
 {
   perSystem =
     {
       pkgs,
+      lib,
       self',
       buildRustProgram,
       ...
@@ -29,11 +31,19 @@
           nativeBuildInputs = [
             pkgs.makeWrapper
           ];
-          postBuild = ''
-            wrapProgram $out/bin/wire \
-                --set WIRE_RUNTIME ${../../runtime} \
-                --set WIRE_AGENT ${self'.packages.agent}
-          '';
+          postBuild =
+            let
+              agents = lib.mapAttrsToList (name: value: {
+                inherit name;
+                path = value.agent;
+              }) (lib.filterAttrs (_: value: value ? agent) self.packages);
+
+            in
+            ''
+              wrapProgram $out/bin/wire \
+                  --set WIRE_RUNTIME ${../../runtime} \
+                  --set WIRE_AGENT ${pkgs.linkFarm "wire-agents-farm" agents}
+            '';
           meta.mainProgram = "wire";
         };
       };

wire/lib/src/hive/key.rs

@@ -4,6 +4,7 @@ use prost::Message;
 use serde::{Deserialize, Serialize};
 use std::env;
 use std::fmt::Display;
+use std::path::Path;
 use std::pin::Pin;
 use std::process::{ExitStatus, Stdio};
 use std::str::from_utf8;
@@ -284,14 +285,22 @@ impl ExecuteStep for PushAgentStep {
     }
 
     async fn execute(&self, ctx: &mut Context<'_>) -> Result<(), HiveLibError> {
-        let agent_directory = match env::var_os("WIRE_AGENT") {
-            Some(agent) => agent.into_string().unwrap(),
-            None => panic!("WIRE_AGENT environment variable not set"),
-        };
+        let mut agent_directory = PathBuf::from(
+            env::var_os("WIRE_AGENT")
+                .expect("WIRE_AGENT environment variable not set!")
+                .into_string()
+                .unwrap(),
+        );
+
+        agent_directory.push(ctx.node.system.as_str());
+
+        let path = String::from(agent_directory.to_str().unwrap());
+
+        debug!("agent path: {path}");
 
-        push(ctx.node, ctx.name, Push::Path(&agent_directory)).await?;
+        push(ctx.node, ctx.name, Push::Path(&path)).await?;
 
-        ctx.state.agent_directory = Some(agent_directory);
+        ctx.state.agent_directory = Some(path);
 
         Ok(())
     }

wire/lib/src/hive/node.rs

@@ -62,7 +62,7 @@ pub struct Node {
     #[serde()]
     // system is the architecture of the host, this is mainly used for
     // determining the correct binary to push to the host's machine.
-    pub system: Option<String>,
+    pub system: String,
 }
 
 #[cfg(test)]
@@ -74,7 +74,7 @@ impl Default for Node {
             tags: im::HashSet::new(),
             allow_local_deployment: true,
             build_remotely: false,
-            system: None,
+            system: String::from("x86_64-linux"),
         }
     }
 }