#!/bin/bash
# author: grimi
# name: mime-archpkg
# require: grep, sed, rsvg-convert (librsvg), gtk-update-icon-cache, !convert (imagemagick)

declare MODE="usage"
declare MYDIR="/usr/share/mime-archpkg"
declare ICODIR="/usr/share/icons"
declare -r NAME="application-x-archpkg"
declare -i MUTE=0


add() {
   # $1 = theme (if not present, add for all), $2 = [0|1] mute
   local file theme elem suf abc one="$1"
   local -i size mute=${2:-$MUTE}
   local -a taba tab
   if [[ -d $MYDIR ]]; then
      cd "$MYDIR"
      for file in *.svgz; do
         theme="$ICODIR/${file%.svgz}"
         if [[ $one && ${file%.svgz} != $one ]]; then
            continue
         fi
         if [[ -f $theme/index.theme ]]; then
            if [[ $mute -ne 1 ]]; then
               echo ">> Adding mimetype icons for $theme... <<"
            fi
            taba=($(grep -e "\[.*mimetypes.*\]" -e "\[.*mimes.*\]" "$theme/index.theme"|sed 's/\[//; s/\]//'))
            for elem in "${taba[@]}"; do
               tab=($(echo -e ${elem/\//\\n}))
               if [[ $elem =~ "scalable" ]]; then
                  size=0
               else
                  if [[ ${tab[0]} =~ [0-9]{2,3} ]]; then
                     abc="${tab[0]}"
                  elif [[ ${tab[1]} =~ [0-9]{2,3} ]]; then
                     abc="${tab[1]}"
                  else
                     continue
                  fi
                  if [[ -h $theme/${tab[0]} ]]; then         # skip link
                     continue
                  fi
                  if [[ $elem =~ "@2x" ]]; then
                     abc=${abc/@2x/}
                     size=${abc/x*/}
                     size=$size*2
                  else
                     size=${abc/x*/}
                  fi
               fi
               if [[ $size -ne 0 ]]; then
                  if [[ -d $theme/$elem ]]; then
                     if [[ -f $theme/$elem/package-x-generic.png ]]; then
                        suf=png
                     else
                        suf=svg
                     fi
                     # convert -resize ${size}x${size} -background none "$file" /tmp/$NAME.$suf
                     rsvg-convert -o /tmp/$NAME.$suf -f $suf -w $size -h $size "$file"
                     install -m644 /tmp/$NAME.$suf "$theme/$elem/$NAME.$suf"
                     rm -f /tmp/$NAME.$suf
                  fi
               elif [[ -d $theme/$elem ]]; then
                  rsvg-convert -o /tmp/$NAME.svg -f svg "$file"
                  install -m644 /tmp/$NAME.svg "$theme/$elem/$NAME.svg"
                  rm -f /tmp/$NAME.svg
               fi
            done
            if [[ -f $theme/icon-theme.cache ]]; then
               gtk-update-icon-cache -q -f "$theme"
            fi
         fi
      done
   fi
}

remove() {
   # $1 = theme (if not present, remove from all)
   local file theme one="$1"
   if [[ -d $MYDIR ]]; then
      cd "$MYDIR"
      for file in *.svgz; do
         theme="$ICODIR/${file%.svgz}"
         if [[ $one && ${file%.svgz} != $one ]]; then
            continue
         fi
         if [[ -d $theme ]]; then
            if [[ $MUTE -ne 1 ]]; then
               echo ">> Removing installed mimetype icons from $theme... <<"
            fi
            find "$theme/" -type f -name $NAME.png -exec rm "{}" \;
            find "$theme/" -type f -name $NAME.svg -exec rm "{}" \;
            if [[ -f $theme/icon-theme.cache ]]; then
               gtk-update-icon-cache -q -f "$theme"
            fi
         fi
      done
   fi
}

update() {
   # $1 = theme (if not present, update for all)
   local file theme one="$1"
   if [[ -d $MYDIR ]]; then
      cd "$MYDIR"
      for file in *.svgz; do
         theme="$ICODIR/${file%.svgz}"
         if [[ $one && ${file%.svgz} != $one ]]; then
            continue
         fi
         if [[ -d $theme && -f $theme/index.theme ]]; then
            if [[ $MUTE -ne 1 ]]; then
               echo ">> Updating installed mimetype icons for $theme... <<"
            fi
            find "$theme/" -type f -name $NAME.png -exec rm "{}" \;
            find "$theme/" -type f -name $NAME.svg -exec rm "{}" \;
            add "${file%.svgz}" 1
         fi
      done
   fi

}

usage() {
   echo "USAGE: [-q] [-r] <-a|-u|-d> [theme]"
   echo "     -q: don't comment action"
   echo "     -r: relative path (default abs.)"
   echo "     -a: add archpkg mime icons"
   echo "     -u: update archpkg mime icons"
   echo "     -d: remove archpkg mime icons"
   echo "  theme: action only for entered theme"
}


if [[ $1 == "-q" ]]; then
   MUTE=1
   shift
fi

if [[ $1 == "-r" ]]; then
   if [[ $PWD != "/" ]]; then
      MYDIR="$PWD/${MYDIR:1}"
      ICODIR="$PWD/${ICODIR:1}"
   fi
   shift
fi

case "$1" in
   -a) MODE=add ;;
   -u) MODE=update ;;
   -d) MODE=remove ;;
esac

if [[ $MODE != "usage" && $(whoami) != "root" ]]; then
   echo "root privileges needed!"
   exit 1
fi

$MODE "$2"