#!/usr/bin/python
###################################################
#
#		iTunes Python Shell Script v0.2.1
#			http://blog.jadecell.org/
#
#			inspired by the bash shell scripts on MacOSXHints
#			http://www.macosxhints.com/article.php?story=20011108211802830
#
#		README
#			- give read/execute permissions to the file (ex 'chmod 755 ./itunes')
#			- import this script into your bin directory (ex '/usr/bin' or '~/bin')
#			- alternatively you can make a symbolic link:
#				sudo ln -s /path/to/your/mypytunes/itunes.py /usr/bin/itunes

from os import system as cmd
from sys import argv as args

def pyTunes(args):
	"""Main Script function to run"""

	try: args[1]			# if no argument
	except: defHelp()		#	show help definition and exit

	if(mDefs.has_key(args[1])):	# Verify if argument is available and execute
		mounter(args)
	else:				# otherwise show list of arguments available
		print "\n** invalid argument **\n"
		showHelp()
def myException(x):
	"""Raise exception and exit out of the program"""

	print x
	raise SystemExit
def defHelp():
	"""Minimal Help function (tells user to run 'help' for list of arguments)"""

	myException(" No arguments provided - Usage %s <option>\n For a list of commands: %s help"%(args[0],args[0]))
def showHelp():
	"""Help function showing a list of commands when '<script> -h' is run"""

	helpHeader = "---------------------------------------------------"
	print helpHeader+"\n  iTunes Command Line Interface\n"+helpHeader
	print "Usage: %s <option>\n"%args[0]
	s = mDefs.keys()
	#print s
	s.sort()
	#print s
	for i in s:
		x = "  "+i+"\t"
		if len(i)<6: 	x += "\t"

		#if i[0]=='u': 	x += '-unmount '
		#else: 		x += '-connect '
		x += mDefs[i]['h']
		print x
	myException(helpHeader)
def mounter(args):
	"""iTunes execution"""
	v = mDefs[args[1]]
	try:
		cmd("osascript -e 'tell application \"iTunes\" "+v['c']+"'")
		print v['o']
	except KeyError:
		try:
			v['r']()
		except KeyError:
			print "invalid 2"
def ostream():
	"""Play streaming location"""
	try:
		print "Playing playlist %s"%(args[2])
		#cmd('osascript -e \'tell application "iTunes"\' -e "open location \"%s\"" -e "end tell"\''%(args[2]))
		cmd('osascript -e \'tell application "iTunes"\' -e \'open location "%s"\' -e \'end tell\''%(args[2]))
	except:
		myException("Please provide the stream location you want to play")
def status():
	"""Play streaming location"""
	cmd('osascript -e \'tell application "iTunes" to player state as string\' > /tmp/iTuneState')
	f = open('/tmp/iTuneState', 'r')
	s = f.readline().strip()
	print "iTunes is currently %s"%(s)
	cmd('rm /tmp/iTuneState')
	shufstat()
	if s == "playing":
		trackstat()
def trackstat():
	"""Show Playlist, Artist, and Track that is currently playing"""
	cmd('osascript -e \'tell application "iTunes" to name of current playlist as string\' > /tmp/iTunesPlaylist')
	cmd('osascript -e \'tell application "iTunes" to artist of current track as string\' >> /tmp/iTunesPlaylist')
	cmd('osascript -e \'tell application "iTunes" to name of current track as string\' >> /tmp/iTunesPlaylist')
	f = open('/tmp/iTunesPlaylist', 'r')
	s = f.read()
	x,y,z,a = s.split('\n')
	print "Current track %s: %s, on playlist \"%s\""%(y,z,x)
	cmd('rm /tmp/iTunesPlaylist')
def shufstat():
	"""Show the shuffle status (on/off)"""
	cmd("osascript -e 'tell application \"iTunes\" to get shuffle of current playlist' > /tmp/iTunesShuffle")
	f = open('/tmp/iTunesShuffle', 'r')
	s = f.readline().strip()
	print "Shuffling: %s"%(s)
	cmd('rm /tmp/iTunesShuffle')
def vol():
	"""Change Volume"""
	print "Changing iTunes volume level."
	cmd('osascript -e \'tell application "iTunes" to get (sound volume as integer)\' > /tmp/iTunesVol')
	f = open('/tmp/iTunesVol')
	s = int(f.readline().strip())
	try: args[2]
	except: myException("could not understand your volume command\n'up', 'down', or integer value (ie. '60')")
	if args[2] == "up":
		newvol = s+10
	elif args[2] == "down":
		newvol = s-10
	else:
		try:
			newvol = int(args[2])
		except:
			myException("could not understand your volume command\n'up', 'down', or integer value (ie. '60')")
	cmd('osascript -e \'tell application "iTunes" to set sound volume to (%i as integer)\''%(newvol))
	print "from %i to %i"%(s,newvol)
	cmd('rm /tmp/iTunesVol')
def playlist():
	"""Change playlist"""
	try:
		print "Changing iTunes plalist to %s"%(args[2])
		cmd('osascript -e \'tell application "iTunes"\' -e \'set the new_playlist to "%s" as string\' -e "play playlist new_playlist" -e "end tell"'%(args[2]))
	except:
		myException("Pleasae define the playlist you want to play")
def llist():
	"""List playlists available on iTunes"""
	print "Listing available playlists:"
	cmd('osascript -e \'tell application "iTunes"\' -e \'name of every user playlist of source "Library"\' -e "end tell" > /tmp/iTunesPlaylists')
	f = open('/tmp/iTunesPlaylists')
	s = f.read()
	s = s.replace("\n","")
	print s
	cmd('rm /tmp/iTunesPlaylists')




###
# These are the main definitions called by the program from the command line
#	All Commands
#		"h" = Help menu description
#	Generic Commands
#		"o" = Output after execution of generic commands
#		"c" = Osascript execution for iTunes
#	Method Commands
#		"r" = Run specific function
###
mDefs = {
	"status": {
		"h": "Shows iTunes' status, current artist, and track.",
		"r": status
	},
	"play": {
		"h": "Start playing iTunes.",
		"o": "Playing iTunes.",
		"c": "to play"
	},
	"stop": {
		"h": "Stop iTunes.",
		"o": "Stopping iTunes.",
		"c": "to stop"
	},
	"pause": {
		"h": "Pause iTunes.",
		"o": "Pausing iTunes.",
		"c": "to pause"
	},
	"playpause": {
		"h": "Play/Pause iTunes.",
		"o": "Play/Pause iTunes.",
		"c": "to playpause"
	},
	"next": {
		"h": "Go to the next track.",
		"o": "Going to the next track.",
		"c": "to next track"
	},
	"prev": {
		"h": "Go to the previous track.",
		"o": "Going to the previous track.",
		"c": "to previous track"
	},
	"mute": {
		"h": "Mute iTunes volume.",
		"o": "Muting iTunes volume",
		"c": "to set mute to true"
	},
	"unmute": {
		"h": "Unmute iTunes volume.",
		"o": "Unmuting iTunes volume.",
		"c": "to set mute to false"
	},
	"quit": {
		"h": "Quit iTunes.",
		"o": "Quitting iTunes.",
		"c": "to quit"
	},
	"launch": {
		"h": "Launch iTunes.",
		"o": "Launching iTunes.",
		"c": "to launch"
	},
	"shuf": {
		"h": "Turn on shuffle playlists",
		"o": "Suffling current playlist",
		"c": "to set shuffle of current playlist to 1"
	},
	"noshuf": {
		"h": "Turn off shuffle playlists",
		"o": "No longer shuffling current playlist",
		"c": "to set shuffle of current playlist to 0"
	},
	"vol": {
		"h": "---\n\t\t  up: Increase iTunes volume by 10%\n\t\t  down: Decrease iTunes volume by 10%\n\t\t  #: Set iTunes volume to # [0-100].",
		"r": vol
	},
	"track": {
		"h": "Shows current playlist, artist and track",
		"r": trackstat
	},
	"playlist": {
		"h": "---\n\t\t  @: Play iTunes playlist @",
		"r": playlist
	},
	"list": {
		"h": "List playlists",
		"r": llist
	},
	"ostream": {
		"h": "Open stream",
		"r": ostream
	},
	"shufstat": {
		"h": "Show iTunes shuffle status.",
		"r": shufstat
	},
	"help": {
		"h": "Show help menu. (this menu)",
		"o": "",
		"r": showHelp
	}	
}

if __name__ == "__main__":
	pyTunes(args)