[AWS] From GCP Startup Scripts to AWS cloud-init

TIME_ZONE="Asia/Taipei"
timedatectl set-timezone ${TIME_ZONE}

echo 'export TMUX_TMPDIR="${HOME}"' >> /etc/bashrc

cat <<EOF >> /etc/bashrc
export PS1="\[\e[1;36m\]\u\[\e[34m\]@\[\e[33m\]\h\[\e[35m\][\w]\[\e[0m\]\\\\$"
EOF

cat <<'EOF' >> /etc/bashrc
export HISTTIMEFORMAT=%Y"/"%m"/"%d" "%H":"%M":"%S" "
EOF

cat <<'EOF' >> /etc/bashrc
export PROMPT_COMMAND='{ date "+[ %Y%m%d %H:%M:%S `whoami` ] `history 1 | { read x cmd; echo "$cmd"; }`"; } >> /var/log/history/historylog'
EOF

sed -i \
-e "s/^weekly/daily/g" \
-e "s/^rotate 4/rotate 31/g" \
-e "s/^#compress/compress/g" /etc/logrotate.conf

cat <<EOF > /etc/logrotate.d/history
/var/log/history/historylog
{
    daily
    rotate 180
    compress
    dateext
    create 0333 root root
}
EOF

dnf install -y https://repo.percona.com/yum/percona-release-latest.noarch.rpm
percona-release setup ps-84-lts
percona-release enable pt
dnf -y install \
git \
nc \
tmux \
percona-toolkit \
percona-server-client \
https://downloads.percona.com/downloads/percona-distribution-mysql-ps/percona-distribution-mysql-ps-8.4.7/binary/redhat/9/x86_64/percona-mysql-shell-8.4.7-1.el9.x86_64.rpm

#cloud-config
users:
  - name: yang-nian
    shell: /bin/bash
    sudo: ALL=(ALL) NOPASSWD:ALL
    lock_passwd: true
    ssh_authorized_keys:
      - ssh-rsa xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx

sudo cloud-init clean --logs
sudo cloud-init init
sudo cloud-init modules --mode=final

packages:
  - httpd

systemctl restart rsyslog
systemctl restart crond

Content-Type: multipart/mixed; boundary="//"
MIME-Version: 1.0

--//
Content-Type: text/cloud-config; charset="us-ascii"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Content-Disposition: attachment; filename="cloud-config.yaml"

#cloud-config
users:
  - name: yang-nian
    shell: /bin/bash
    sudo: ALL=(ALL) NOPASSWD:ALL
    lock_passwd: true
    ssh_authorized_keys:
      - ssh-rsa xxxxxxxxxxxxxxxxxxxxxxxxxxxx
--//
Content-Type: text/x-shellscript; charset="us-ascii"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Content-Disposition: attachment; filename="setup-script.sh"

#!/bin/bash

## Common libs
log_message() {
  local level=$1
  shift
  local msg="$@"
  local timestamp=$(date +"%Y-%m-%d %H:%M:%S")

  case $level in
    info|warn|critical)
      echo " $timestamp level=$level msg=$msg "
      ;;
    *)
      echo " $timestamp level=critical Invalid log level '$level'. Message: $msg "
      ;;
  esac
}

log_warn() {
  local msg=$1
  log_message "warn" "${msg}"
}

log_critical() {
  local msg=$1
  log_message "critical" "${msg}"
}

log_info() {
  local msg=$1
  log_message "info" "${msg}"
}

log_wrong() {
  local msg=$1
  log_message "wrong" "${msg}"

}

execute_command() {
  local level=$1
  shift
  local command="$@"
  local output
  output=$(eval "$command" 2>&1)
  local exit_status=$?

  if [ $exit_status -eq 0 ]; then
    log_info "Command '$command' executed successfully. Output : $output"
  else
    case ${level} in
      info)
        log_info "Command '$command' failed with exit status $exit_status. Output : $output"
        ;;
      warn)
        log_warn "Command '$command' failed with exit status $exit_status. Output : $output"
        ;;
      critical)
        log_critical "Command '$command' failed with exit status $exit_status. Output : $output"
        ;;
      *)
        log_wrong "'$command' failed with wrong level. $exit_status. Output : $output"
        ;;
    esac
  fi
}

## Timezone

TIME_ZONE="Asia/Taipei"

execute_command critical "timedatectl set-timezone \"${TIME_ZONE}\""

## Bashrc

BASHRC_FILE="/etc/bashrc"
HISTORY_LOG_DIR="/var/log/history"

cat ${BASHRC_FILE} | grep "export TMUX_TMPDIR"
if [ $? -eq 1 ]; then
    execute_command critical "echo 'export TMUX_TMPDIR=\"\${HOME}\"' >> ${BASHRC_FILE}"
else
    log_info "tmux setting is exists."
fi

cat ${BASHRC_FILE} | grep "export PS1"
if [ $? -eq 1 ]; then
    execute_command critical "cat <<EOF >> ${BASHRC_FILE}
export PS1=\"\[\e[1;36m\]\u\[\e[34m\]@\[\e[33m\]\h\[\e[35m\][\w]\[\e[0m\]$\"
EOF"
else
    log_info "bash color is exists."
fi

cat ${BASHRC_FILE} | grep "export HISTTIMEFORMAT"
if [ $? -eq 1 ]; then
    execute_command critical "cat <<'EOF' >> ${BASHRC_FILE}
export HISTTIMEFORMAT=%Y\"/\"%m\"/\"%d\" \"%H\":\"%M\":\"%S\" \"
EOF"
else
    log_info "history format datetime is exists."
fi

if [ -d ${HISTORY_LOG_DIR} ]; then
    log_info "${HISTORY_LOG_DIR} is exists."
    log_info "[Skip] mkdir "${HISTORY_LOG_DIR}
else
    execute_command critical mkdir ${HISTORY_LOG_DIR}
fi

execute_command critical touch ${HISTORY_LOG_DIR}/historylog
execute_command critical chmod 333 ${HISTORY_LOG_DIR}/historylog

cat ${BASHRC_FILE} | grep "export PROMPT_COMMAND"
if [ $? -eq 1 ]; then
    cat <<'EOF' >> ${BASHRC_FILE}
export PROMPT_COMMAND='{ date "+[ %Y%m%d %H:%M:%S `whoami` ] `history 1 | { read x cmd; echo "$cmd"; }`"; } >> /var/log/history/historylog'
EOF
else
    log_info "PROMPT_COMMAND datetime is exists."
fi

## Logrotate

LOGROTATE_CONF="/etc/logrotate.conf"
LOGROTATE_HISTORY="/etc/logrotate.d/history"

execute_command critical "sed -i \
-e \"s/^weekly/daily/g\" \
-e \"s/^rotate 4/rotate 31/g\" \
-e \"s/^#compress/compress/g\" $LOGROTATE_CONF"

execute_command critical "cat <<EOF > ${LOGROTATE_HISTORY}
${HISTORY_LOG_DIR}/historylog
{
    daily
    rotate 180
    compress
    dateext
    create 0333 root root
}
EOF"

execute_command critical dnf install -y https://repo.percona.com/yum/percona-release-latest.noarch.rpm
execute_command info percona-release show
execute_command warn percona-release setup ps-84-lts
execute_command warn percona-release enable pt
execute_command info percona-release show

execute_command critical dnf -y install \
git \
nc \
tmux \
percona-toolkit \
percona-server-client \
https://downloads.percona.com/downloads/percona-distribution-mysql-ps/percona-distribution-mysql-ps-8.4.7/binary/redhat/9/x86_64/percona-mysql-shell-8.4.7-1.el9.x86_64.rpm

## Common Task

DBA_DIR="/usr/local/dba"

execute_command warn mkdir ${DBA_DIR}

--//--