dockolor.plugin.zsh

# utils to check if awk is installed
dockolor_has_awk() {
  if command -v awk >/dev/null 2>&1; then
    echo true
  else
    echo false
  fi
}

# todo: merge both cmd
dockolor_colorize() {
  # check if the user want json format or table
  if [[ "$*" == *"--format json"* ]]; then
    # we got json so we need to check for eg State":"exited" in each line and colorize in function
    command docker "$@" | awk '
    BEGIN {
      RED = "\033[31m"
      GREEN = "\033[32m"
      YELLOW = "\033[33m"
      RESET = "\033[0m"
    }
    # set the color in fc of the status
    {
      line = $0
      lower = tolower(line)
      if (lower ~ /"state":"up"/ || lower ~ /"state":"running"/) {
        color = GREEN
      } else if (lower ~ /"state":"paused"/) {
        color = YELLOW
      } else if (lower ~ /"state":"exited"/ || lower ~ /"state":"dead"/) {
        color = RED
      } else {
        color = RESET
      }
      printf "%s%s%s\n", color, line, RESET
    }'
  else
    command docker "$@" | awk '
    BEGIN {
      RED = "\033[31m"
      GREEN = "\033[32m"
      YELLOW = "\033[33m"
      RESET = "\033[0m"
    }
    # skip the first line because it containt header
    # reminder, NR > built-in var in awk
    NR == 1 {
      print $0
      next
    }
    # set the color in fc of the status
    {
      line = $0
      lower = tolower(line)
      # paused has to be check first since the string is likely up since x (paused)
      if (lower ~ /paused/) {
        color = YELLOW
      } else if (lower ~ /up/ || lower ~ /running/) {
        color = GREEN
      } else if (lower ~ /exited/ || lower ~ /dead/) {
        color = RED
      } else {
        color = RESET
      }
      printf "%s%s%s\n", color, line, RESET
    }'
  fi
}

# main function
# check awk and colorize
dockolor() {
  if [ "$(dockolor_has_awk)" = false ]; then
    echo "You have installed dockolor but you don't have awk installed"
    docker "$@"
  else
    dockolor_colorize "$@"
  fi
}

# ### ###
# remap each alias or cmd
# ### ###

# if the user have the docker plugin loaded
# we remove the alias
if alias dps >/dev/null 2>&1; then
  unalias dps
fi

if alias dpsa >/dev/null 2>&1; then
  unalias dpsa
fi

dps() {
  dockolor ps "$@"
}

dpsa() {
  dockolor ps -a "$@"
}

# intercept docker command to colorize ps output
# or pass through
docker() {
  if [[ "$1" == "ps" || "$1" == "container" && "$2" == "ps" ]]; then
    dockolor "$@"
  else
    command docker "$@"
  fi
}