# Authors:
  # https://github.com/AlexBio
  # https://github.com/dbb
  # https://github.com/Mappleconfusers
  #
  # Debian-related zsh aliases and functions for zsh
  
  # Use aptitude if installed, or apt-get if not.
  # You can just set apt_pref='apt-get' to override it.
  if [[ -e $( which -p aptitude 2>&1 ) ]]; then
      apt_pref='aptitude'

      apt_upgr='safe-upgrade'

  else
      apt_pref='apt-get'

      apt_upgr='upgrade'

  fi
  
  # Use sudo by default if it's installed
  if [[ -e $( which -p sudo 2>&1 ) ]]; then
      use_sudo=1
  fi
  
  # Aliases ###################################################################
  # These are for more obscure uses of apt-get and aptitude that aren't covered
  # below.
  alias age='apt-get'
  alias api='aptitude'
  
  # Some self-explanatory aliases
  alias acs="apt-cache search"
  alias aps='aptitude search'
  alias as="aptitude -F \"* %p -> %d 
  (%v/%V)\" \
  		--no-gui --disable-columns search"	# search package
  
  # apt-file
  alias afs='apt-file search --regexp'
  
  
  # These are apt-get only
  alias asrc='apt-get source'
  alias app='apt-cache policy'
  
  # superuser operations ######################################################
  if [[ $use_sudo -eq 1 ]]; then
  # commands using sudo #######
      alias aac='sudo $apt_pref autoclean'
      alias abd='sudo $apt_pref build-dep'
      alias ac='sudo $apt_pref clean'
      alias ad='sudo $apt_pref update'

      alias adg='sudo $apt_pref update && sudo $apt_pref $apt_upgr'

      alias adu='sudo $apt_pref update && sudo $apt_pref dist-upgrade'
      alias afu='sudo apt-file update'

      alias ag='sudo $apt_pref $apt_upgr'

      alias ai='sudo $apt_pref install'
      # Install all packages given on the command line while using only the first word of each line:
      # acs ... | ail
      alias ail="sed -e 's/  */ /g' -e 's/ *//' | cut -s -d ' ' -f 1 | "' xargs sudo $apt_pref install'
      alias ap='sudo $apt_pref purge'
      alias ar='sudo $apt_pref remove'
  
      # apt-get only
      alias ads='sudo apt-get dselect-upgrade'
  
      # Install all .deb files in the current directory.
      # Warning: you will need to put the glob in single quotes if you use:
      # glob_subst
      alias dia='sudo dpkg -i ./*.deb'
      alias di='sudo dpkg -i'
  
      # Remove ALL kernel images and headers EXCEPT the one in use
      alias kclean='sudo aptitude remove -P ?and(~i~nlinux-(ima|hea) \
          ?not(~n`uname -r`))'
  
  
  # commands using su #########
  else
      alias aac='su -ls \'$apt_pref autoclean\' root'
      abd() {
          cmd="su -lc '$apt_pref build-dep $@' root"
          print "$cmd"
          eval "$cmd"
      }
      alias ac='su -ls \'$apt_pref clean\' root'
      alias ad='su -lc \'$apt_pref update\' root'

      alias adg='su -lc \'$apt_pref update && aptitude $apt_upgr\' root'

      alias adu='su -lc \'$apt_pref update && aptitude dist-upgrade\' root'
      alias afu='su -lc "apt-file update"'

      alias ag='su -lc \'$apt_pref $apt_upgr\' root'

      ai() {
          cmd="su -lc 'aptitude -P install $@' root"
          print "$cmd"
          eval "$cmd"
      }
      ap() {
          cmd="su -lc '$apt_pref -P purge $@' root"
          print "$cmd"
          eval "$cmd"
      }
      ar() {
          cmd="su -lc '$apt_pref -P remove $@' root"
          print "$cmd"
          eval "$cmd"
      }
  
      # Install all .deb files in the current directory
      # Assumes glob_subst is off
      alias dia='su -lc "dpkg -i ./*.deb" root'
      alias di='su -lc "dpkg -i" root'
  
      # Remove ALL kernel images and headers EXCEPT the one in use
      alias kclean='su -lc '\''aptitude remove -P ?and(~i~nlinux-(ima|hea) \
          ?not(~n`uname -r`))'\'' root'
  fi
  
  # Completion ################################################################
  
  #
  # Registers a compdef for $1 that calls $apt_pref with the commands $2
  # To do that it creates a new completion function called _apt_pref_$2
  #
  apt_pref_compdef() {
      local f fb
      f="_apt_pref_${2}"
  
      eval "function ${f}() {
          shift words; 
  	service=\"\$apt_pref\"; 
  	words=(\"\$apt_pref\" '$2' \$words); 
  	((CURRENT++))
  	test \"\${apt_pref}\" = 'aptitude' && _aptitude || _apt
      }"
  
      compdef "$f" "$1"
  }
  
  apt_pref_compdef aac "autoclean"
  apt_pref_compdef abd "build-dep"
  apt_pref_compdef ac  "clean"
  apt_pref_compdef ad  "update"
  apt_pref_compdef afu "update"

  apt_pref_compdef ag  "$apt_upgr"

  apt_pref_compdef ai  "install"
  apt_pref_compdef ail "install"
  apt_pref_compdef ap  "purge"
  apt_pref_compdef ar  "remove"
  apt_pref_compdef ads "dselect-upgrade"
  
  # Misc. #####################################################################
  # print all installed packages
  alias allpkgs='aptitude search -F "%p" --disable-columns ~i'
  
  # Create a basic .deb package
  alias mydeb='time dpkg-buildpackage -rfakeroot -us -uc'
  
  
  # Functions #################################################################
  # create a simple script that can be used to 'duplicate' a system
  apt-copy() {
      print '#!/bin/sh'"
  " > apt-copy.sh
  
      cmd='$apt_pref install'
  
      for p in ${(f)"$(aptitude search -F "%p" --disable-columns \~i)"}; {
          cmd="${cmd} ${p}"
      }
  
      print $cmd "
  " >> apt-copy.sh
  
      chmod +x apt-copy.sh
  }
  
  # Prints apt history
  # Usage:
  #   apt-history install
  #   apt-history upgrade
  #   apt-history remove
  #   apt-history rollback
  #   apt-history list
  # Based On: http://linuxcommando.blogspot.com/2008/08/how-to-show-apt-log-history.html
  apt-history () {
    case "$1" in
      install)
        zgrep --no-filename 'install ' $(ls -rt /var/log/dpkg*)
        ;;
      upgrade|remove)
        zgrep --no-filename $1 $(ls -rt /var/log/dpkg*)
        ;;
      rollback)
        zgrep --no-filename upgrade $(ls -rt /var/log/dpkg*) | \
          grep "$2" -A10000000 | \
          grep "$3" -B10000000 | \
          awk '{print $4"="$5}'
        ;;
      list)
        zcat $(ls -rt /var/log/dpkg*)
        ;;
      *)
        echo "Parameters:"
        echo " install - Lists all packages that have been installed."
        echo " upgrade - Lists all packages that have been upgraded."
        echo " remove - Lists all packages that have been removed."
        echo " rollback - Lists rollback information."
        echo " list - Lists all contains of dpkg logs."
        ;;
    esac
  }
  
  # Kernel-package building shortcut
  kerndeb () {
      # temporarily unset MAKEFLAGS ( '-j3' will fail )
      MAKEFLAGS=$( print - $MAKEFLAGS | perl -pe 's/-j\s*[\d]+//g' )
      print '$MAKEFLAGS set to '"'$MAKEFLAGS'"
  	appendage='-custom' # this shows up in $ (uname -r )
      revision=$(date +"%Y%m%d") # this shows up in the .deb file name
  
      make-kpkg clean
  
      time fakeroot make-kpkg --append-to-version "$appendage" --revision \
          "$revision" kernel_image kernel_headers
  }
  
  # List packages by size
  function apt-list-packages {
      dpkg-query -W --showformat='${Installed-Size} ${Package} ${Status}
  ' | \
      grep -v deinstall | \
      sort -n | \
      awk '{print $1" "$2}'
  }