X-Git-Url: https://iankelling.org/git/?a=blobdiff_plain;f=rootsshsync;h=a4db227fd511828dc666ff421af0e9308652b93f;hb=93db0c16393fe75efef3167c1d2b332f50387c23;hp=aef668a7b61b9bb9e80bf8af26b76f8705028b84;hpb=be6cf346ed72fc65178006bf81a1e760e19627ec;p=distro-setup

diff --git a/rootsshsync b/rootsshsync
index aef668a..a4db227 100755
--- a/rootsshsync
+++ b/rootsshsync
@@ -1,4 +1,4 @@
-#!/bin/bash -l
+#!/bin/bash
 # Copyright (C) 2016 Ian Kelling
 
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -13,10 +13,32 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-if s test -e /q/root/.ssh; then
-    s lnf /q/root/.ssh /root
+set -eE -o pipefail
+trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR
+
+[[ $EUID == 0 ]] || exec sudo "$BASH_SOURCE" "$@"
+
+if test -e /q/root/.ssh; then
+  dest=/q/root/.ssh
+  /a/exe/lnf $dest /root
 else
-    mkdir /root/.ssh
+  dest=/root/.ssh
+  mkdir -p $dest
+  chmod 700 $dest
+fi
+
+user_ssh_dir=$(eval echo ~${SUDO_USER:-$USER})/.ssh
+
+# remove broken links, or else rsync has error about them.
+find $user_ssh_dir -xtype l -exec rm '{}' \;
+# -t times, so it won't rewrite the file every time,
+# -L resolve links
+rsync -rtL --delete $user_ssh_dir/ $dest
+chown -R root:root /root/.ssh
+d=/etc/initramfs-tools
+if [[ -e $d ]] && ! diff -q /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys &>/dev/null; then
+  mkdir -p /etc/initramfs-tools/root/.ssh
+  chmod 700 $d/root $d/root/.ssh
+  cp -p /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys
+  update-initramfs -u -k all
 fi
-s cp -rL $(eval echo ~${SUDO_USER:-$USER})/.ssh/* /root/.ssh
-s chown -R root:root /root/.ssh