#!/bin/bash

#

# gnome-file-types-properties - 

# A zenity-script for chosing preferred applications. 

# Written by Magnus Stålnacke (jemamo at telia.com) 

# Version 0.8.4 Nov. 30 2007

#

export LANG=en_US.utf8 

ARGU="$1"

UTMP="/tmp/$(whoami)"

DATA_DIR="$HOME/.local/share/applications"

GDATA_DIR="/usr/share/applications"

G_CACHE="$GDATA_DIR/mimeinfo.cache"

U_CACHE="$DATA_DIR/mimeinfo.cache"

SUBC="/usr/share/mime/subclasses"

U_DEF="$DATA_DIR/defaults.list"

EXT="usercreated.desktop"

NONEW=1

#

# Create dirs in users home if missing. 

#

if [ -e "$DATA_DIR" ]; then :; else mkdir $DATA_DIR;fi

if [ -e "$U_DEF" ] 

then :; 

else 

  echo "[Default Applications]" > $DATA_DIR/defaults.list 

fi

#

# Check how we got started. 

#

function checkarg (){

  if [ -n "$ARGU" ]

  then

    if [ -e "$ARGU" ]

    then

      chkmtpe 

      deflist

    else

      showhelp

    fi

  else

    selfile

    dtfedit

    if [ -n "$DT_FILE" ]

    then

      update

    fi

  fi

}

#

# Mimetype check.

#

function chkmtpe (){

  MTPE=$(gnomevfs-info -s "$ARGU" | grep "MIME type" | sed 's/.*: //')

  echo "The mimetype is: $MTPE"

}

#

# Build a list with registered apps for this mimetype.

#

function deflist (){

  if [ -e "$UTMP.avail" ]; then cleanup;fi

  # Create the list in "$UTMP.avail" to be used by the zenity list-box

  if [ "$NONEW" == 1 ]

  then

   echo "FALSE Add_New_App New" > "$UTMP.avail"

  fi

  # Get all sub-mimetypes, one per row.

  SUB="$(grep -hs "^$MTPE" $SUBC $G_CACHE $U_CACHE | awk -F= '{print $1}' | sed 's/ /\n/g')"

  # Get all .desktopfiles for the above mimetypes

  MTPEs=$(grep -hs "$SUB" $U_CACHE $G_CACHE | sed 's/.*=//g;s/;$//g;s/;/\n/g;')

  if [ -n "$SUB" ]

  then

    for i in $(echo "$MTPEs" | sort -u); do

      echo "FALSE $i $(grep -hs "^Name=" $GDATA_DIR/$i $DATA_DIR/$i | \

      sed 's/Name=//;s/ /_/g')" >> "$UTMP.avail"

    done

  fi

  # Change FALSE to TRUE for the default app. 

  if grep -q "desktop" "$UTMP.avail"

  then

    DEF_APP=$(gnomevfs-info -s "$ARGU" | grep "Default app" | sed 's/.*: //')

    sed -i "s|FALSE $DEF_APP|TRUE $DEF_APP|" "$UTMP.avail"

  fi

  # Show the list

  zenity --height="350" --width="550" --list --radiolist --text \

"The list below show applications that already have an entry for the mimetype: 

$MTPE on your system, meaning they are registered to be able 

to deal with this mimetype. You can choose your preferred default application 

by checking the radiobutton. The script will adjust the mimetype settings in 

your local home directory, thus not changing anything systemwide.

If you wish you can add an application that is not in the list by checking the first 

option in list named Add_New_App 

Please select" \

  --column "Pick" --column ".desktop file" --column "Application" \

  $(cat "$UTMP.avail") > "$UTMP.chosen" 

  retval=$?

  choice=$(cat "$UTMP.chosen")

  if grep -q "$MTPE=$choice" $U_DEF

  then

    retval=2

  fi

  case $retval in

  0)

    DEF_CH=$(grep -s "$choice" "$UTMP.avail" | awk '{print $2}')

    DEF_NME=$(grep -s "$choice" "$UTMP.avail" | awk '{print $3}')

    if [ "$DEF_CH" = Add_New_App ]

    then

      newmime

    fi

    if [ -z "$NOCH" ]

    then

      if grep -q "$MTPE" $U_DEF

      then

        # Set the users chosen default app for the existing mime.

        sed -i "s|$MTPE=.*|$MTPE=$DEF_CH|" $U_DEF

      else

        # If no user default for this mimetype, add it.

         echo "$MTPE=$DEF_CH" >> $U_DEF

      fi

     update

     zenity --info --text "Default application to handle $MTPE is set to $DEF_NME"

    fi;;

  1)

    echo "Cancel pressed.";;

  -1)

    echo "ESC pressed.";;

  2)

    zenity --info --text "Default was not changed.";;

  esac

  cleanup

}

#

# Show an input box for new app to handle the mimetype.

#

function newmime (){

  NEWAPP=$(zenity --entry --text \

"Fill in the Name and command for the application you want to handle the 

mimetype: $MTPE separate them with a colon (:)

Separate Name and Command with a colon. If you omit the command, or 

type something that is not in your path, you will get the chance to 

provide the command with full path by a file chooser dialog.")

  case $? in 

  0)

   if echo "$NEWAPP" | grep -q ":"

   then

     NAME=$(echo "$NEWAPP" | awk -F: '{print $1}')

     CMD=$(echo "$NEWAPP" | awk -F: '{print $2}')

     if [ -n "$NAME" ]

     then

       check

       usercreate

     else

       zenity --error --text "Application name is missing"

       newmime

     fi

   else

     zenity --error --text "Separator missing"

     newmime

   fi

   ;;

  1)

   deflist

   ;;

  -1)

   deflist

   ;;

  esac

  NOCH=yes

}

#

# Check if the command exists in path, if not give the 

# user a filechooser.

#

function check (){

  if which $CMD >&/dev/null

  then

    CMD=`which $CMD`

  else

    CMD=$(zenity --file-selection --filename=/ --title="Chose application")

    if [ -z $CMD ]

    then

      newmime

    fi

  fi

}

#

# Create a new *.desktop file for the added app that can handle 

# the mimetype in question.

#

function usercreate(){

  APP=$(basename "$CMD")

  USER_APP="$DATA_DIR/$APP-$EXT"

  if cat $G_CACHE $U_CACHE | grep "$MTPE" | grep -q "$APP" 

  then

    zenity --error --text "The mime type: $MTPE is alredy assigned to: $APP"

    NOCH=yes

  else

    if [ -e "$USER_APP" ]

    then

      sed -i "s|MimeType=|MimeType=$MTPE;|g" $USER_APP

      update

      zenity --info --text \

      "The mime type: $MTPE is now registered to be handled by: $APP"

    else

      cat <<EOF >$USER_APP 

[Desktop Entry]

Encoding=UTF-8

Name=$NAME

Exec=$CMD

Type=Application

Terminal=false

NoDisplay=true

MimeType=$MTPE;

EOF

    fi

  fi

  if [ -z "$NOCH" ]

  then

    zenity --question --text \

"The created $(basename "$USER_APP") can now be edited manually, to add, remove or change options, do you want to do this now? 

This file should work as is, so you can safely cancel this operation, but advanced users may want to edit options to their liking. But due to a bug this editor cannot handle '%' charachters and replaces them with garbage." 

    VAL=$?

    if [ "$VAL" == 0 ]

    then 

      DT_FILE="$APP"

      dtfedit

      update

    else 

      update

    fi

    zenity --info --text "$APP is now registered to handle: $MTPE"

  fi

  NONEW=2

  deflist

}

#

# Update the users mime chache and set the defaults 

# first from the defaults.list file

#

function update (){

  update-desktop-database $DATA_DIR

  for i in $(sed '/\[.*\]/d' $U_DEF); do

    MTYPE=$(echo "$i" | awk -F= '{print $1}')

    M_FIL=$(echo "$i" | awk -F= '{print $2}')

    if grep -q "$MTYPE" $U_CACHE

    then

     sed -i "\|$MTYPE|s|$M_FIL||;s|$MTYPE=|$i;|;s|;;|;|" $U_CACHE

    else

     echo "$i" >> $U_CACHE

    fi

  done

  # If there are rows with empty mimetypes, delete them.

  sed -i '/.*=$/d' $U_CACHE

}

#

# Give the user a helping hand. 

#

function showhelp (){

echo

echo "Usage: $(basename $0) [ARGUMENT]"

echo '''

  -h, -?, --help  Show this help and exit. 

DESCRIPTION

  This script is dependent on zenity for creating its GUI. It is 

  to be used as a tool for choosing the default application or to 

  run for different type of files (mimetypes). It modifies or add 

  files in "$HOME/.local/share/applications" 

  When passed a valid filename it checks its mimetype as interpreted  

  by gnomevfs, it will show a list of already registered applications 

  able to handle the mimetype in question, the user can change default 

  application in this list. The list also contains an option to add 

  the Name and command of a program that is not registered to be able 

  to open files of the mimetype. After any changes this script will 

  run the update-desktop-database to update the users mime cache. 

  When executed without any argument, the script will list all 

  applications associated with a mimetype. When selecting an item from 

  this list, the user will be able to edit the options for the selected 

  application. The edited file will be saved in the above path and the 

  update-desktop-database is executed.

  When given an invalid argument or any of the options -h -? --help, 

  this text is shown. 

'''

}

#

# Remove tempfiles.

#

function cleanup (){

  rm -f "$UTMP.avail"

  rm -f "$UTMP.chosen"

}

#

# The .desktop file selector

#

function selfile (){

  DT_FILE=$(for i in $GDATA_DIR/*.desktop $DATA_DIR/*.desktop; do

    if grep -q "MimeType" $i 

    then

      BNME=$(basename $i | sed 's/\.desktop//;s/-usercreated//')

      if [ -e $DATA_DIR/$BNME-$EXT ] 

      then 

        i="$DATA_DIR/$BNME-$EXT"

      fi

      CMD=$(grep -m 1 "^Exec" $i | sed "s/^Exec=//")

      NME=$(grep -m 1 "^Name=" $i | sed 's/^Name=//')

      echo "$BNME---$NME---\"$CMD\""

    fi

  done | sort -u | sed 's/---/\n/g' |  zenity --list --width="550" --height="400" \

  --column="Base name" --column="Program name" --column="Program command" \

  --print-column="1" --text "Select item below. 

By selecting an item in this list and klick OK, you will get another window where 

you can edit the file that contains the options of the selected application. The 

edited file will end up in your: $DATA_DIR 

Pressing "Esc", click Cancel or use the window close (X) button will close without 

any changes in this or the following editor window. Be aware that the editor do 

have problems with the (%)percent charachter and may replace it by garbage.")

}

#

# The .desktop file editor

#

function dtfedit (){ 

  USER_DTF="$DT_FILE-$EXT"

  if [ -n "$DT_FILE" ]

  then 

    # Chose to edit users own file if exist

    if [ -e $DATA_DIR/$USER_DTF ]

    then

      TO_EDIT="$DATA_DIR/$USER_DTF"

    else

      TO_EDIT="$GDATA_DIR/$DT_FILE.desktop"

    fi

    cat "$TO_EDIT" | zenity --height="350" --width="550" --text-info --editable \

    --title "Edititng the file: $USER_DTF" > $UTMP.$USER_DTF 

    if [ -s $UTMP.$USER_DTF ]

    then 

      mv $UTMP.$USER_DTF $DATA_DIR/$USER_DTF 

    else 

      rm -f $UTMP.$USER_DTF 

      DT_FILE=""

    fi

  fi

} 

#

# Check if we have zenity installed then start this mess.

#

if which zenity >&/dev/null 

then

 checkarg 

else 

 echo "Error: zenity not found!"

fi