script-record.sh

#!/bin/bash
###############################################################################
# linux script recording & replay wrapper
# written by George Liu <https://centminmod.com>
###############################################################################
VER=0.2
DEBUG='n'
CPUS=$(nproc)

if [ -f /usr/bin/yum ] && [[ ! -f /usr/bin/scriptreplay || ! -f /usr/bin/script ]]; then
  sudo yum -q -y install util-linux
elif [ -f /usr/bin/apt ] && [[ ! -f /usr/bin/scriptreplay || ! -f /usr/bin/script ]]; then
  sudo apt install util-linux
fi

if [[ -f /usr/bin/yum && ! -f /usr/bin/tree ]]; then
  sudo yum -q -y install tree
elif [[ -f /usr/bin/apt && ! -f /usr/bin/tree ]]; then
  sudo apt install tree
fi

if [[ -f /usr/bin/yum && ! -f /usr/bin/pigz && ! -f /usr/local/bin/pigz ]]; then
  sudo yum -q -y install pigz
elif [[ -f /usr/bin/apt && ! -f /usr/bin/pigz && ! -f /usr/local/bin/pigz ]]; then
  sudo apt install pigz
fi

if [[ -f /usr/local/bin/pigz && "$CPUS" -gt '1' ]]; then
  COMPRESSBIN='/usr/local/bin/pigz'
elif [[ -f /usr/bin/pigz && "$CPUS" -gt '1' ]]; then
  COMPRESSBIN='/usr/bin/pigz'
else
  COMPRESSBIN='/usr/bin/gzip'
fi

if [ -f /usr/bin/pzcat ]; then
  ZCATBIN='/usr/bin/pzcat'
elif [ -f /usr/bin/zcat ]; then
  ZCATBIN='/usr/bin/zcat'
fi

disable_motd() {
  if [ -f /etc/profile.d/dmotd.sh ]; then
    # temp disable Centmin Mod dmotd.sh
    mv /etc/profile.d/dmotd.sh /etc/profile.d/dmotd.sh-disabled
    if [[ "$DEBUG" = [yY] ]]; then
      ls -lAh /etc/profile.d/dmotd.sh*
    fi
  fi
}

reenable_motd() {
  if [ -f /etc/profile.d/dmotd.sh-disabled ]; then
    # temp disable Centmin Mod dmotd.sh
    mv /etc/profile.d/dmotd.sh-disabled /etc/profile.d/dmotd.sh
    if [[ "$DEBUG" = [yY] ]]; then
      ls -lAh /etc/profile.d/dmotd.sh*
    fi
  fi
}

record() {
  SESSION_NAME="$1"
  FILEPREFIX="$HOME/.script/$(date '+%Y-%m-%d')/$(date '+%Y-%m-%d_%H-%M-%S')-${SESSION_NAME:-session}"
  mkdir -p $FILEPREFIX
  disable_motd
  /usr/bin/script -t -f -q 2>"${FILEPREFIX}/time.txt" "${FILEPREFIX}/cmds"
  $COMPRESSBIN "${FILEPREFIX}/time.txt"
  $COMPRESSBIN "${FILEPREFIX}/cmds"
  echo
  echo "files saved:"
  echo -e "${FILEPREFIX}/cmds.gz\n${FILEPREFIX}/time.txt.gz\n"
  echo "to replay 1x speed:"
  echo "$0 play ${FILEPREFIX}/cmds.gz ${FILEPREFIX}/time.txt.gz"
  echo
  echo "to replay 2x speed:"
  echo "$0 play ${FILEPREFIX}/cmds.gz ${FILEPREFIX}/time.txt.gz 2"
}

replay() {
  CMD_FILE="$1"
  TIME_FILE="$2"
  SPEED="$3"
  if [[ -f "$CMD_FILE" && -f "$TIME_FILE" ]]; then
    if [ "$3" ]; then
      scriptreplay -t <("$ZCATBIN" "$TIME_FILE") <("$ZCATBIN" "$CMD_FILE") "$SPEED"
    else
      scriptreplay -t <("$ZCATBIN" "$TIME_FILE") <("$ZCATBIN" "$CMD_FILE")
    fi
  else
    echo
    echo "error: required file path(s) to cmds.gz or time.txt.gz do not exist"
    echo
    echo "existing files detected:"
    /usr/bin/tree --charset utf8 --sort=ctime -n -f "$HOME/.script/"
    echo
    help
  fi
}

replay_nogz() {
  CMD_FILE="$1"
  TIME_FILE="$2"
  SPEED="$3"
  if [[ -f "$CMD_FILE" && -f "$TIME_FILE" ]]; then
    if [ "$3" ]; then
      scriptreplay -t "$TIME_FILE" "$CMD_FILE" "$SPEED"
    else
      scriptreplay -t "$TIME_FILE" "$CMD_FILE"
    fi
  else
    echo
    echo "error: required file path(s) to cmds.gz or time.txt.gz do not exist"
    echo
    echo "existing files detected:"
    /usr/bin/tree --charset utf8 --sort=ctime -n -f "$HOME/.script/"
    echo
    help
  fi
}

list_files() {
  echo "saved files listing:"
  echo
  /usr/bin/tree --charset utf8 --sort=ctime -n -f "$HOME/.script/"
  echo
}

help() {
  echo
  echo "Usage:"
  echo
  echo "$0 rec SESSION_NAME"
  echo "$0 play /path/to/cmds.gz /path/to/time.txt.gz"
  echo "$0 play /path/to/cmds.gz /path/to/time.txt.gz 2"
  echo "$0 play-nogz /path/to/cmds /path/to/time.txt"
  echo "$0 play-nogz /path/to/cmds /path/to/time.txt 2"
  echo "$0 list"
  echo
}

trap reenable_motd EXIT

case "$1" in
  rec )
    record "$2"
    ;;
  play )
    replay "$2" "$3" "$4"
    ;;
  play-nogz )
    replay_nogz "$2" "$3" "$4"
    ;;
  list )
    list_files
    ;;
  * )
    help
    ;;
esac