1 files changed, 20 insertions, 2 deletions

diff --git a/flake.nix b/flake.nix
index bb3c45c5..768f19bd 100644
--- a/flake.nix
+++ b/flake.nix
@@ -22,9 +22,16 @@
       ref = "master";
       inputs.nixpkgs.follows = "nixpkgs";
     };
+    deploy-rs = {
+      type = "github";
+      owner = "serokell";
+      repo = "deploy-rs";
+      ref = "master";
+      inputs.nixpkgs.follows = "nixpkgs";
+    };
   };
 
-  outputs = { self, nixpkgs, home-manager, sops-nix, ... }@inputs:
+  outputs = { self, nixpkgs, home-manager, sops-nix, deploy-rs, ... }@inputs:
     let
       inherit (builtins) attrNames attrValues elemAt toJSON isNull pathExists;
       inherit (nixpkgs) lib;
@@ -157,11 +164,22 @@
 
         apps = activateNixosConfigurations;
 
-        devShell = forAllSystems (system: systemPkgs: import ./shell.nix { pkgs = self.legacyPackages.${system}; });
+        devShell = forAllSystems (system: systemPkgs: import ./shell.nix { pkgs = self.legacyPackages.${system}; deploy-rs = deploy-rs.defaultPackage.${system}; });
 
         defaultTemplate = {
           path = ./.;
           description = "GKleen's flakey nixos configuration";
         };
+
+        deploy.nodes = mapAttrs (hostname: _: {
+          inherit hostname;
+          sshUser = "root";
+            
+          profiles.system = {
+            path = deploy-rs.lib.${self.nixosConfigurations.${hostname}.config.nixpkgs.system}.activate.nixos self.nixosConfigurations.${hostname};
+          };
+        }) (nixImport { dir = ./hosts; _import = (_path: name: name); });
+
+        checks = builtins.mapAttrs (system: deployLib: deployLib.deployChecks self.deploy) deploy-rs.lib;
       };
 }