#!/bin/bash

testpath1=/faronear/fon/sysconfig/nixhome
testpath2=~/faronear/fon/sysconfig/nixhome
testpath3=~/faronear/fon/sysconfig.git/nixhome
testpath4=`pwd`/nixhome

if [ "$1" ]
then
  SourcePath=$1
elif [ -d $testpath1 ]
then 
  SourcePath=$testpath1
elif [ -d $testpath2 ]
then
  SourcePath=$testpath2
elif [ -d $testpath3 ]
then
  SourcePath=$testpath3
elif [ -d $testpath4 ]
then
  SourcePath=$testpath4
else
  echo "=== Enter [target path] or leave [blank] to exit"
  read -p ">>> " SourcePath
  echo ""
  if [ ! -d "$SourcePath" ]
  then 
    echo "*** Source path [$SourcePath] not available! Exit now. ***"
    exit
  fi
fi

# $2 应当为用户名。为空则默认是 root
if [ "$2" ]
then
  HomePath=/home/$2
else
  HomePath=/root
fi

if [ -d "$HomePath" ]
then
  pushd $HomePath
  homescriptlist=".emacs .emacs.lisp .bashrc .bash_profile .gitignore"
  echo
  echo "=== Copy or link scripts? <l> for link, <anything else> for copy:"
  read -p ">>> " CopyOrLinkScripts
  for homescript in $homescriptlist
  do
    if [ -e "$homescript" ]
    then
      mv $homescript $homescript.backup-$(date +%Y%m%d%H%M%S)
    fi
    if [ "$CopyOrLinkScripts" = 'l' ]
    then
      echo "--- Linking $SourcePath/$homescript to $HomePath/$homescript ..."
      ln -s $SourcePath/$homescript ./
    else
      echo "--- Copying $SourcePath/$homescript to $HomePath/$homescript ..."
      cp -r $SourcePath/$homescript ./
    fi
  done
  echo

  if [ $2 ] # 如果 $2 不存在，则默认为是 root 用户，不需要设置 .ssh
  then 

    mkdir -p $HomePath/.ssh
    chmod 700 $HomePath/.ssh
    # 注意，由 root 为新用户创建的配置文件的 owner 是 root，而不是新用户，导致新用户无法读取该文件而密钥登录失败，因此要重设 owner。
    chown $2:$2 $HomePath
    if [ -e '$HomePath/.ssh/authorized_keys' ]
    then
      mv $HomePath/.ssh/authorized_keys $HomePath/.ssh/authorized_keys.backup-$(date +%Y%m%d%H%M%S)
    fi

    echo "=== Append or link or omit [$HomePath/.ssh/authorized_keys] to config ssh server? <a> for append, <l> for link, <<anything else>> for omit:"
    read -p ">>> " CopyOrLinkOrOmitAuthorizedKeys

    if [ "$CopyOrLinkOrOmitAuthorizedKeys" = 'l' ]
    then
      echo "--- Linking $SourcePath/authorized_keys to $HomePath/.ssh/authorized_keys ..."
      ln -s $SourcePath/.ssh/authorized_keys $HomePath/.ssh/authorized_keys
    elif [ "$CopyOrLinkOrOmitAuthorizedKeys" = 'a' ]
    then
      echo "--- Copying $SourcePath/authorized_keys to $HomePath/.ssh/authorized_keys ..."
      cat $SourcePath/.ssh/authorized_keys >> $HomePath/.ssh/authorized_keys
      chmod 600 $HomePath/.ssh/authorized_keys
    fi
    echo

  fi

  popd
else
  echo "!!! Not existing $HomePath, please try again."
fi