Apply NixOS flakes from EC2 user data

user-data.sh.tftpl

@@ -21,7 +21,7 @@ write_bootstrap_config() {
     nixosConfigurations.ami-builder-bootstrap = nixpkgs.lib.nixosSystem {
       system = "${system}";
       modules = [
-        ({ modulesPath, pkgs, ... }: {
+        ({ config, lib, modulesPath, pkgs, ... }: {
           imports = [ "$${modulesPath}/virtualisation/amazon-image.nix" ];
 
           networking.hostName = "nixos-zfs-ec2-builder";
@@ -67,7 +67,7 @@ write_final_config() {
     nixosConfigurations.nixos-zfs-ec2 = nixpkgs.lib.nixosSystem {
       system = "${system}";
       modules = [
-        ({ modulesPath, pkgs, ... }: {
+        ({ config, lib, modulesPath, pkgs, ... }: {
           imports = [ "$${modulesPath}/virtualisation/amazon-image.nix" ];
 
           networking.hostName = "nixos-zfs-ec2";
@@ -101,6 +101,43 @@ write_final_config() {
           ];
 
           services.amazon-ssm-agent.enable = true;
+          systemd.services.amazon-init.enable = lib.mkForce false;
+
+          systemd.services.nixos-user-data-flake = {
+            description = "Apply NixOS flake from EC2 user data";
+            wantedBy = [ "multi-user.target" ];
+            after = [ "fetch-ec2-metadata.service" "network-online.target" "nix-daemon.socket" ];
+            wants = [ "fetch-ec2-metadata.service" "network-online.target" "nix-daemon.socket" ];
+            path = [ pkgs.coreutils pkgs.git ];
+            serviceConfig = {
+              Type = "oneshot";
+              RemainAfterExit = true;
+            };
+            script = ''
+              set -eu
+
+              user_data=/etc/ec2-metadata/user-data
+              state_dir=/var/lib/nixos-user-data-flake
+              applied_hash_file="$state_dir/applied-sha256"
+
+              if [ ! -s "$user_data" ]; then
+                echo "No EC2 user-data flake found at $user_data" >&2
+                exit 1
+              fi
+
+              mkdir -p "$state_dir" /etc/nixos
+              new_hash="$(sha256sum "$user_data" | cut -d ' ' -f1)"
+              old_hash="$(cat "$applied_hash_file" 2>/dev/null || true)"
+
+              if [ "$new_hash" = "$old_hash" ]; then
+                exit 0
+              fi
+
+              install -m 0644 "$user_data" /etc/nixos/flake.nix
+              ${config.system.build.nixos-rebuild}/bin/nixos-rebuild switch --flake /etc/nixos
+              printf '%s\n' "$new_hash" > "$applied_hash_file"
+            '';
+          };
 
           users.users.builder-admin = {
             isNormalUser = true;