#!/bin/sh

FLASH_LOCK="$HOME/.mdv-folders-flash"
FLASH_ROOT=/live/media
[ -e "$FLASH_ROOT/My Documents" ] && FLASH_ALA_WINDOWS=1

[ -e $FLASH_LOCK ] && exit

while pidof -x xdg-user-dirs-update; do
    [ "$SECONDS" -le 60 ] || return
    sleep 1
done

function create_link() {
	NAME=$1
	SRC=$2
	[ -z "$SRC" ] && SRC=$($NAME)
	XDG_DIR=$(xdg-user-dir $NAME)
	FLASH_DIR="$FLASH_ROOT/$SRC"
	mkdir -p "$FLASH_DIR"
	mv $XDG_DIR/{.??,}* "$FLASH_DIR"
	rmdir $XDG_DIR && ln -sf "$FLASH_DIR" "$XDG_DIR"
}

if [ -n "$FLASH_ALA_WINDOWS" ]; then
	create_link "DOCUMENTS" "My Documents"
	create_link "DOWNLOAD"  "My Documents/My Downloads"
	create_link "MUSIC"     "My Documents/My Music"
	create_link "PICTURES"  "My Documents/My Images"
	create_link "VIDEOS"     "My Documents/My Video"
else
    for d in DOWNLOAD DOCUMENTS MUSIC PICTURES VIDEOS; do
	create_link $d $(echo $d | tr A-Z a-z)
    done
fi

touch $FLASH_LOCK
