Package-Recompiler.sh

trap 'echo Got SIGINT, ignoring.' 2

# Recompile Packages Automatically

#Paths and Filename-Definitions
db_folder='/var/db/arch-recompiler'
deps_db='deps.db'
expl_db='explicit.db'
igno_db='ignore.list'

#Definitions
default_igno_list='linux
linux-lts
linux-headers
linux-lts-headers'

#In-Memory Databases
declare -A deps_indb
declare -A expl_indb
igno_indb=()
deps_worklist=()
expl_worklist=()

#Flags
create_deps_db=0
create_expl_db=0
create_igno_db=0
fix_filerights=0

echo 'Init completed.'

if [ ! -d "$db_folder" ]; then
  echo 'db-folder does not exist, creating new database-folder...'
  mkdir -p "$db_folder" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "can't create dir, trying with sudo"
    sudo mkdir -p "$db_folder" >/dev/null 2>&1
    if test $? -ne 0; then
      echo "mkdir failed.";exit 1
    fi
    echo "changing rights for the folder with sudo..."
    sudo chown $(whoami): "$db_folder"
    if test $? -ne 0; then
      echo "chown failed.";exit 1
    fi
    sudo chmod 700 "$db_folder"
    if test $? -ne 0; then
      echo "chmod failed.";exit 1
    fi
  fi
fi

echo 'Checking Database...'
if [ ! -f "$db_folder/$deps_db" ]; then
  echo 'depency-packages database-file does not exist, going to create a new one...'
  create_deps_db=1
fi
if [ ! -f "$db_folder/$expl_db" ]; then
  echo 'explicit-packages database-file does not exist, going to create a new one...'
  create_expl_db=1
fi
if [ ! -f "$db_folder/$igno_db" ]; then
  echo 'ignore-package list-file does not exist, going to create a new one with default-list...'
  create_igno_db=1
fi

echo 'Checking file-rights...'
touch "$db_folder/$deps_db" >/dev/null 2>&1
if test $? -ne 0; then
  fix_filerights=1
fi
touch "$db_folder/$expl_db" >/dev/null 2>&1
if test $? -ne 0; then
  fix_filerights=1
fi
touch "$db_folder/$igno_db" >/dev/null 2>&1
if test $? -ne 0; then
  fix_filerights=1
fi

if [ "$fix_filerights" -eq "1" ]; then
  echo "insufficient file-rights, try to fix this with sudo..."
  sudo chown $(whoami): -R "$db_folder"
  if test $? -ne 0; then
    echo "chown failed.";exit 1
  fi
  sudo chmod 600 -R "$db_folder"
  if test $? -ne 0; then
    echo "chmod failed.";exit 1
  fi
  sudo chmod 700 "$db_folder"
  if test $? -ne 0; then
    echo "chmod failed.";exit 1
  fi
fi
unset fix_filerights

if [ "$create_deps_db" -eq "1" -o "$create_expl_db" -eq "1" -o "$create_igno_db" -eq "1" ]; then
  echo "Creating databases ..."
  if [ "$create_deps_db" -eq "1" ]; then
    echo "  Depency-packages database..."
    echo "" > "$db_folder/$deps_db" 2>&1
    if test $? -ne 0; then
      echo "failed.";exit 1
    fi
  fi
  if [ "$create_expl_db" -eq "1" ]; then
    echo "  Explicit-packages database..."
    echo "" > "$db_folder/$expl_db" 2>&1
    if test $? -ne 0; then
      echo "failed.";exit 1
    fi
  fi
  if [ "$create_igno_db" -eq "1" ]; then
    echo "  Ignore-packages list..."
    echo "$default_igno_list" > "$db_folder/$igno_db" 2>&1
    if test $? -ne 0; then
      echo "failed.";exit 1
    fi
  fi
fi

echo "Locking databases..."
if [ -f "$db_folder/$deps_db.lock" ]; then
  echo "ERROR: Depency-packages database locked.";exit 1
fi
touch "$db_folder/$deps_db.lock" >/dev/null 2>&1
if test $? -ne 0; then
  echo "ERROR: Can't lock depency-packages database.";exit 1
fi
if [ -f "$db_folder/$expl_db.lock" ]; then
  echo "ERROR: Explicit-packages database locked.";exit 1
fi
touch "$db_folder/$expl_db.lock" >/dev/null 2>&1
if test $? -ne 0; then
  echo "ERROR: Can't lock explicit-packages database.";exit 1
fi
if [ -f "$db_folder/$igno_db.lock" ]; then
  echo "ERROR: Ignore-packages database locked.";exit 1
fi
touch "$db_folder/$igno_db.lock" >/dev/null 2>&1
if test $? -ne 0; then
  echo "ERROR: Can't lock ignore-packages database.";exit 1
fi

echo "Loading databases..."

tmp1=""
tmp2=""
echo "  Depency-packages database..."
while read e; do
  tmp1=$(echo $e | cut -d' ' -f1)
  tmp2=$(echo $e | cut -d' ' -f2)
  if [ ! -z "$tmp1" -a ! -z "$tmp2" ]; then
    deps_indb["$tmp1"]="$tmp2"
  fi
done <"$db_folder/$deps_db"

tmp1=""
tmp2=""
echo "  Explicit-packages database..."
while read e; do
  tmp1=$(echo $e | cut -d' ' -f1)
  tmp2=$(echo $e | cut -d' ' -f2)
  if [ ! -z "$tmp1" -a ! -z "$tmp2" ]; then
    expl_indb["$tmp1"]="$tmp2"
  fi
done <"$db_folder/$expl_db"

unset tmp1 tmp2

echo "  Ignore-packages database..."
while read e; do
  if [ ! -z "$e" ]; then
    igno_indb+=("$e")
  fi
done <"$db_folder/$igno_db"

unset e

echo "Cleaning database for uninstalled packages..."
echo "  Depency-packages database..."
for e in "${!deps_indb[@]}"; do
  yaourt -Qidn "$e" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "Delete $e from database, its no longer installed (as depency)."
    unset -v deps_indb[$e]
    grep -v "^$e " "$db_folder/$deps_db" > "$db_folder/$deps_db.bak" && mv "$db_folder/$deps_db.bak" "$db_folder/$deps_db"
    if test $? -ne 0; then
      echo "ERROR: Can't update depency-packages database-file."; exit 1
    fi
  fi
done

echo "  Explicit-packages database..."
for e in "${!expl_indb[@]}"; do
  yaourt -Qien "$e" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "Delete $e from database, its no longer (explicit) installed."
    unset -v expl_indb[$e]
    grep -v "^$e " "$db_folder/$expl_db" > "$db_folder/$expl_db.bak" && mv "$db_folder/$expl_db.bak" "$db_folder/$expl_db"
    if test $? -ne 0; then
      echo "ERROR: Can't update explicit-packages database."; exit 1
    fi
  fi
done

echo "Calculating worklist..."
tmp1=""
tmp2=""
echo "  Depency-packages..."
oldifs="$IFS"
IFS=$'\n'
for e in `yaourt -Qdn`; do
  tmp1=$(echo $e | cut -d' ' -f1)
  tmp2=$(echo $e | cut -d' ' -f2)
  if [ ! -z "$tmp1" -a ! -z "$tmp2" ]; then
    if [ "${deps_indb[$tmp1]}" = "$tmp2" ]; then
      continue
    fi
    if [[ " ${igno_indb[@]} " =~ " ${tmp1} " ]]; then
      continue
    fi
    deps_worklist+=("$tmp1")
  fi
done
tmp1=""
tmp2=""
echo "  Explicit-packages..."
for e in `yaourt -Qen`; do
  tmp1=$(echo $e | cut -d' ' -f1)
  tmp2=$(echo $e | cut -d' ' -f2)
  if [ ! -z "$tmp1" -a ! -z "$tmp2" ]; then
  if [ "${expl_indb[$tmp1]}" = "$tmp2" ]; then
      continue
    fi
    if [[ " ${igno_indb[@]} " =~ " ${tmp1} " ]]; then
      continue
    fi
    expl_worklist+=("$tmp1")
  fi
done
IFS="$oldifs"
unset tmp1 tmp2 e

echo "Compiling packages..."
tmp1=""
tmp2=""
echo "  Depency-Packages"
for e in "${deps_worklist[@]}"; do
  echo "   Package: '$e'"
  yaourt -Sy > /dev/null
  tmp1=$(yaourt -Qidn "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  tmp2=$(yaourt -Si   "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  if [ ! -z "$tmp1" ]; then
    if [ -z "$tmp2" ]; then
      echo "ERROR: we seem to have no Internet-connection..."; exit 1
    fi
    if [ "$tmp1" != "$tmp2" ]; then
      echo "local / remote version missmatch, can't recompile."
      continue
    fi
  else
    echo "Warning: Package '$e' has been uninstalled."
    continue
  fi
  
  yaourt -Sb $e --asdeps --noconfirm > /dev/null 2>&1
  if test $? -ne 0; then
    echo "Warning: Package '$e' could not be compiled, maybe you want to add it to ignore list"
  fi
  tmp1=$(yaourt -Qidn "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  if [ -z "$tmp1" ]; then
    echo "Warning: Can't fetch version of '$e', ignoring."
    continue
  fi
  grep -v "^$e " "$db_folder/$deps_db" > "$db_folder/$deps_db.bak" &&
  echo "$e $tmp1" >> "$db_folder/$deps_db.bak" &&
  mv "$db_folder/$deps_db.bak" "$db_folder/$deps_db"
  if test $? -ne 0; then
    echo "ERROR: Database-File action failed."; exit 1
  fi
done
unset deps_worklist

tmp1=""
echo "  Explicit-Packages"
for e in "${expl_worklist[@]}"; do
  echo "   Package: $e"
  yaourt -Sy > /dev/null
  tmp1=$(yaourt -Qien "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  tmp2=$(yaourt -Si   "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  if [ ! -z "$tmp1" ]; then
    if [ -z "$tmp2" ]; then
      echo "ERROR: we seem to have no Internet-connection..."; exit 1
    fi
    if [ "$tmp1" != "$tmp2" ]; then
      echo "Warning: local / remote version missmatch, can't recompile."
      continue
    fi
  else
    echo "Warning: Package '$e' has been uninstalled."
    continue
  fi
  
  yaourt -Sb $e --asexplicit --noconfirm > /dev/null 2>&1
  if test $? -ne 0; then
    echo "Warning: Package '$e' could not be compiled, maybe you want to add it to ignore list"
  fi
  tmp1=$(yaourt -Qien "$e" | grep "^Version" | cut -d':' -f2 | tr -d '[:space:]')
  if [ $? -ne 0 -o -z "$tmp1" ]; then
    echo "Warning: Can't fetch version of '$e', ignoring."
    continue
  fi
  grep -v "^$e " "$db_folder/$expl_db" > "$db_folder/$expl_db.bak" &&
  echo "$e $tmp1" >> "$db_folder/$expl_db.bak" &&
  mv "$db_folder/$expl_db.bak" "$db_folder/$expl_db"
  if test $? -ne 0; then
    echo "ERROR: Database-File action failed."; exit 1
  fi
done

unset tmp1 tmp2 expl_worklist

echo "Unlocking databases..."
[ -f "$db_folder/$deps_db.lock" ] && (
  rm "$db_folder/$deps_db.lock" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "ERROR: Can't unlock depency-packages database.";exit 1
  fi
)
[ -f "$db_folder/$expl_db.lock" ] && (
  rm "$db_folder/$expl_db.lock" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "ERROR: Can't unlock explicit-packages database.";exit 1
  fi
)
[ -f "$db_folder/$igno_db.lock" ] && (
  rm "$db_folder/$igno_db.lock" >/dev/null 2>&1
  if test $? -ne 0; then
    echo "ERROR: Can't unlock ignore-packages database.";exit 1
  fi
)