From b160b4a1e102f92eb992199a207d342ddcb726e5 Mon Sep 17 00:00:00 2001
From: hygienic-books <hygienic-books@tentic.net>
Date: Fri, 3 Nov 2023 01:38:56 +0100
Subject: [PATCH] feat(zbm): Prepare SSH in ZBM (#6)

---
 setup.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/setup.sh b/setup.sh
index f7b74ed..5d6da21 100644
--- a/setup.sh
+++ b/setup.sh
@@ -841,9 +841,16 @@ function we_want_ssh () {
 
 function configure_ssh_in_zbm () {
     #2.11
+    get_dropbear_hooks
+    customize_dropbear_hooks
     paru_install 'mkinitcpio-nfs-utils' 'dropbear' 'mkinitcpio-dropbear'
     in_file_in_array_insert_n_at_the_end '/etc/zfsbootmenu/mkinitcpio.conf' 'HOOKS' 'net'
     in_file_in_array_insert_n_at_the_end '/etc/zfsbootmenu/mkinitcpio.conf' 'HOOKS' 'dropbear'
+    for key_type in 'dss' 'ecdsa' 'ed25519' 'rsa'; do
+        dropbearkey -t "${key_type}" -f '/etc/dropbear/dropbear_'"${key_type}"'_host_key'
+    done
+    set_pub_keys '/etc/dropbear/root_key' "${ARCHZBM_SSH_AUTH_KEYS}"
+    ensure_ip_in_kcl
 }
 
 function add_syslinux_pacman_hook () {