preload

#!/bin/bash

#this generates a yad-friendly app list. This is run every time the gui script is executed.
#if this script detects nothing has changed since last run, then it will echo back the app list that was generated last time.

if [ -z "$DIRECTORY" ];then
  DIRECTORY="$(readlink -f "$(dirname "$0")")"
fi

mktimestamps() {
  #these directories are checked for changes
  checkfiles="${DIRECTORY}/apps
${DIRECTORY}/data/settings
${DIRECTORY}/data/status
${DIRECTORY}/etc
${DIRECTORY}/icons/categories
${DIRECTORY}/apps
${DIRECTORY}/preload
${DIRECTORY}/api
${DIRECTORY}/data/category-overrides"
  local IFS=$'\n'
  for i in $checkfiles
  do
    printf "$i $(stat -c %Y "${i}" 2>/dev/null) "
    
    if [ -d "$i" ];then
      find "$i" -type f -printf '%T@ %p\n' | sort -rn | head -n1
    else
      printf '\n'
    fi
  done
}

[ "$1" == source ] && return 0

function error {
  echo -e "\e[91m$1\e[39m" 1>&2
  exit 1
}

#yad or xlunch format
guimode="$1"

if [ -z "$guimode" ];then
  format=yad
elif [[ "$guimode" = xlunch* ]];then
  format=xlunch
elif [[ "$guimode" = yad* ]];then
  format=yad
elif [ "$guimode" != 'yad' ] && [ "$guimode" != 'xlunch' ];then
  error "Unknown list format '$format'!"
fi

#specifies a category to preload (can be left empty to load main page)
prefix="$2"

if [ "$prefix" == '/' ];then
  prefix=
fi

#For systems with older versions of yad, the text color column cannot be left blank. This python script determines the default text color from GTK bindings.
if [ -z "${text_color+x}" ] && [ "$format" == yad ];then
  #0400 is the latest version
  yad_version="$(zcat /usr/share/doc/yad/NEWS.gz | head -n 1 | tr -cd '0123456789\n')"
  if [ $yad_version -lt 0400 ]; then
    text_color=$(echo "import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
from gi.repository import Gdk
win = Gtk.Window()
vb = Gtk.VBox()
tv = Gtk.TextView()
vb.add(tv)
win.show_all()
style = tv.get_style_context()
textcolor = style.get_color(Gtk.StateType.NORMAL)
print(Gdk.RGBA.to_string(textcolor))
" | python -)
  else
    text_color=""
  fi
fi
export text_color

timestampfile="${DIRECTORY}/data/preload/timestamps-$(echo "$prefix" | tr -d '/')"
listfile="${DIRECTORY}/data/preload/LIST-$(echo "$prefix" | tr -d '/')"

mkdir -p "${DIRECTORY}/data/preload"

reloadlist=0

if [ -f "$timestampfile" ];then
  #get modified timestamps for files/directories in the pi-apps folder
  timestamps="$(mktimestamps)"
  
  if [ "$timestamps" == "$(cat "$timestampfile")" ];then
    #if current timestamps and saved timestamps match, then don't reload the list
    reloadlist=0
    echo "Timestamps match." 1>&2
  else
    #timestamps don't match, so reload the list
    reloadlist=1
    echo "Timestamps don't match" 1>&2
    #echo -e "original file: $(cat "$timestampfile")\nnew timestamp: $timestamps" 1>&2
  fi
else
  #timestamp file not found
  reloadlist=1
fi

if [ ! -f "$listfile" ] || [ -z "$(cat "$listfile")" ];then
  echo "list file for $prefix does not exist." 1>&2
  reloadlist=1
fi

#If updates available, show special Updates category (returned separately to avoid re-preloading after update-check finishes)
if [ -z "$prefix" ] && [ "$format" == yad ] && ([ -s "${DIRECTORY}/data/update-status/updatable-files" ] || [ -s "${DIRECTORY}/data/update-status/updatable-apps" ]);then
  #yad format
  echo "${DIRECTORY}/icons/categories/Updates.png
Updates
Updates/
Pi-Apps updates are available. Click here to update your apps.
$text_color"
elif [ -z "$prefix" ] && [ "$format" == xlunch ] && ([ -s "${DIRECTORY}/data/update-status/updatable-files" ] || [ -s "${DIRECTORY}/data/update-status/updatable-apps" ]);then
  #xlunch format
  echo "Updates;${DIRECTORY}/icons/categories/Updates-64.png;:exec "\""echo Updates/"\"""
fi

if [ $reloadlist == 1 ];then
  echo "Generating list for '$prefix'..." 1>&2
  
  #for app_categories() and app_status() functions
  if ! command -v app_categories >/dev/null ;then
    source "${DIRECTORY}/api"
  fi
  
  if [ ! -z "$prefix" ];then
    echo "Showing apps within $prefix/" 1>&2
    vfiles="$(app_categories | grep . | sort -f | uniq | grep "^$prefix/" | sed "s+$prefix/++g")" #generate a virtual file system with apps in folders represented as subdirectories
  else
    vfiles="$(app_categories | grep . | sort -f | uniq)"
  fi
  
  #remove apps within categories - show this layer of stuff only.
  vfiles="$(sed 's+/.*+/+g' <<<"$vfiles" | sort -f | uniq)"
  
  #get list of apps - excluding folders and apps that are incompatible with CPU architecture
  APPS="$(grep -v '/' <<<"$vfiles" | list_intersect "$(list_apps cpu_installable)")"
  
  #get list of folders - excluding apps - and hide the hidden folder.
  DIRS="$(grep '/' <<<"$vfiles" | tr -d '/' | grep -vFx "hidden")"
  
  #shuffle the list if enabled
  if [ "$(cat "${DIRECTORY}/data/settings/Shuffle App list")" == 'Yes' ];then
    APPS="$(echo "$APPS" | shuf)"
    DIRS="$(echo "$DIRS" | shuf)"
  fi
  
  if [ "$format" == yad ];then
    IFS=$'\n'
    
    #initial value of listfile: if within a prefix, start with a Back button
    if [ ! -z "$prefix" ];then
      echo "${DIRECTORY}/icons/back.png
Back
$(dirname "$prefix" | sed 's+^\.$++g')/
Return to the previous location
$text_color" | tee "$listfile"
    else
      echo -n '' > "$listfile"
    fi
    
    for dir in $DIRS
    do
      if [ -f "${DIRECTORY}/icons/categories/${dir}.png" ];then
        diricon="${DIRECTORY}/icons/categories/${dir}.png"
      else
        diricon="${DIRECTORY}/icons/categories/default.png"
      fi
      declare -A dir_lookup
      dir_lookup=( ["Browsers"]="Internet browsers." ["All Apps"]="All Pi-Apps Applications in one long list." ["Appearance"]="Applications and Themes which modify the look and feel of your OS." ["System Management"]="Apps that help you keep track of system resources and general system management." ["Games"]="Games and Emulators" ["Installed"]="All Pi-Apps Apps that you have installed." ["Internet"]="Browsers, Chat Clients, Email Clients, and so much more." ["Multimedia"]="Video playback and creation, audio playback and creation, and streaming alternatives." ["Packages"]="Simple Apps that install directly from APT repos." ["Tools"]="An assortment of helpful programs that don't already fit into another category." ["Terminals"]="Alternative terminal programs built for the modern age as well as to replicate your old vintage computer." ["Programming"]="Code editors, IDEs, and other applications to help you write and make other programs." ["Creative Arts"]="Drawing, Painting, and Photo and Movie Editors" ["Engineering"]="3D Printing slicers, CAD/modeling, and general design software" ["Office"]="Office suites (document and slideshow editors), and other office tools." ["Emulation"]="Applications that help you run non-ARM or non-Linux software." ["Communication"]="Internet messaging, calling, video chatting, and email clients.")

      if [ ! -z "$prefix" ];then
        add="$diricon
$dir
$prefix/$dir/
"${dir_lookup["$dir"]}"
$text_color"
      else
        add="$diricon
$dir
$dir/
"${dir_lookup["$dir"]}"
$text_color"
      fi
      echo "$add"
    done | sed 's/&/&amp;/g' | tee -a "$listfile"
    #finished preloading categories
    
    #preload apps
    for app in $APPS
    do
      #get installation status of app
      status="$(cat "${DIRECTORY}/data/status/${app}" 2>/dev/null)"
      
      #determine app icon
      if [ -f "${DIRECTORY}/apps/${app}/icon-24.png" ];then
        add="${DIRECTORY}/apps/${app}/icon-24.png"$'\n'
      else
        add="${DIRECTORY}/apps/${app}/none-24.png"$'\n'
      fi
      
      #rest of list attributes
      if [ -z "$status" ];then
        add+="$app
$prefix/$app
$(head -n1 "${DIRECTORY}/apps/${app}/description" || echo "Description unavailable")"$'\n'
      else
        add+="$app
$prefix/$app
"\("$status"\)" $(head -n1 "${DIRECTORY}/apps/${app}/description" || echo "Description unavailable")"$'\n'
      fi
      
      #determine status text-color for app name (green for installed, red for uninstalled, yellow for corrupted)
      if [ -z "$status" ];then
        add+="$text_color"
      elif [ "$status" == installed ];then
        add+="#00AA00"
      elif [ "$status" == uninstalled ];then
        add+="#CC3333"
      elif [ "$status" == corrupted ];then
        add+="#888800"
      elif [ "$status" == disabled ];then
        add+="#FF0000"
      else
        # fallback incase unexpected status
        add+="$text_color"
      fi
      #output finished app lines
      echo "$add"
    done | sed 's/&/&amp;/g' | tee -a "$listfile"
    #finished preloading apps
    
  elif [ "$format" == xlunch ];then
    #XLUNCH list format
    
    #initial value of listfile: if within a prefix, start with a Back button, or if on main page start with Updates
    if [ ! -z "$prefix" ];then
      echo "Back;${DIRECTORY}/icons/back-64.png;:exec "\""echo /"\""" | tee "$listfile"
    else
      echo -n '' > "$listfile"
    fi
    
    IFS=$'\n'
    for dir in $DIRS
    do
      if [ -f "${DIRECTORY}/icons/categories/${dir}-64.png" ];then
        diricon="${DIRECTORY}/icons/categories/${dir}-64.png"
      else
        diricon="${DIRECTORY}/icons/categories/default-64.png"
      fi
      
      echo "${dir};${diricon};:exec "\""echo '${prefix}${dir}/'"\"""
    done | tee -a "$listfile"
    
    for app in $APPS
    do
      if [ -f "${DIRECTORY}/data/status/${app}" ];then
        echo "${app} ($(cat "${DIRECTORY}/data/status/${app}" 2>/dev/null));${DIRECTORY}/apps/${app}/icon-64.png;:exec "\""echo '$prefix/${app}'"\"""
      else
        #status file missing - app has never been installed
        echo "${app};${DIRECTORY}/apps/${app}/icon-64.png;:exec "\""echo '$prefix/${app}'"\"""
      fi
    done | tee -a "$listfile"
    
  fi
  
  #save timestamps to file too
  if [ -z "$timestamps" ];then
    timestamps="$(mktimestamps)"
  fi
  echo "$timestamps" > "$timestampfile"
else
  echo "Reading list file for '$prefix'..." 1>&2
  cat "$listfile"
fi

#preload all categories in background
"${DIRECTORY}/etc/preload-daemon" "$format" &>/dev/null &