#!/usr/bin/python2

"""Menu Generator for Fluxbox

Generates a menu for Fluxbox using the freedesktop.org standards

Usage: fluxbox-fdo-menugen.py [options]

Options:

  -l ..., --lang=...      create the menu using a language. Default = $LANG

  -h, --help              show this help

  -f ..., --file=...      output the menu into a file. Default = ~/.fluxbox/menu

  --with-icons         do not put icons for applications in the menu

  --stdout                output the menu to standard output

"""

__author__ = "Rudolf Kastl , Antonio Gomes"

__version__ = "$Revision: 1.0 $"

__date__ = "$Date: 2005/04/09 17:46:19 $"

__license__ = "GPL"

import os,re,sys,glob,getopt

import xdg.Menu,xdg.DesktopEntry,xdg.IconTheme

def usage():

	print __doc__

def header(wm="fluxbox"):

	return """

[begin] (Fluxbox)

	[exec] (Web Browser) {htmlview}

	[exec] (Email) {evolution}

	[exec] (Terminal) {gnome-terminal}

	[exec] (Irc) {xchat}

	[separator]\n"""

def footer(wm="fluxbox"):

	return """

	[submenu] (Fluxbox Menu)

		[config] (Configure)

		[submenu] (System Styles) {Choose a style...}

			[stylesdir] (/usr/share/fluxbox/styles)

			[stylesdir] (/usr/share/commonbox/styles/)

		[end]

		[submenu] (User Styles) {Choose a style...}

			[stylesdir] (~/.fluxbox/styles)

		[end]

		[workspaces]   (Workspace List)

		[submenu] (Tools)

			[exec] (Window name) {xprop WM_CLASS|cut -d \" -f 2|xmessage -file - -center}

			[exec] (Screenshot - JPG) {import screenshot.jpg && display -resize 50% screenshot.jpg}

			[exec] (Screenshot - PNG) {import screenshot.png && display -resize 50% screenshot.png}

			[exec] (Run) {fbrun }

			[exec] (Regen Menu) {fluxbox-generate_menu --with-icons}

		[end]

		[submenu] (Window)

			[restart] (kde) {startkde}

			[restart] (openbox) {openbox}

			[restart] (gnome) {gnome-session}

		[end]

		[exec] (Lock screen) {xscreensaver-command -lock}

		[commanddialog] (Fluxbox Command)

		[reconfig] (Reload config)

		[restart] (Restart)

		[separator]

		[exit] (Exit)

	[end]

[end]\n"""	

def checkWm(entry, wm="fluxbox"):

	if entry.DesktopEntry.getOnlyShowIn() != []:

		entry.Show = False

	if entry.DesktopEntry.getNotShowIn() != []:

		if isinstance(entry, xdg.Menu.MenuEntry):

			if wm in entry.DesktopEntry.getNotShowIn():

				entry.Show = False

			else:

				entry.Show = True 

#def findIcon(icon):

#	"""Finds the path and filename for the given icon name

#		e.g. gaim --> /usr/share/pixmaps/gaim.png

#		e.g. fart.png --> /usr/share/pixmaps/fart.png

#	"""

#	if os.path.isfile(icon): return icon

#	paths = ["~/.icons","/usr/share/pixmaps","/usr/share/icons"]

#	for path in paths:

#		for dirpath , dirnames, filenames in os.walk(os.path.expanduser(path)):

#			for filename in filenames:

#				pattern = '^%s\.(png|jpg|gif|xpm)' % icon # matches <filename>.<something>

#				if icon == filename or re.search(pattern, filename):

#					return os.path.join(dirpath,filename)

def findIcon(icon):

	"""Finds the path and filename for the given icon name

		e.g. gaim --> /usr/share/pixmaps/gaim.png

		e.g. fart.png --> /usr/share/pixmaps/fart.png

	"""

	global paths

	global locs

	# Case Icon=/path/to/file.extension

	if os.path.isfile(icon): return icon

	# Case Icon=file.extension	

	if re.search("^\w+.*\.\w{3}$",icon):

		names=[icon]

	# Case Icon=file

	elif re.search("^\w+.*$",icon):

		names=[icon+".png",icon+".xpm",icon+".jpg"]

	else:

		names=[]

	for path in paths:	

		for name in names:

			if (os.path.isfile(path+name)):

				locs.append(path)

				return (path+name).encode('utf8')

def parseMenu(menu,depth=1):

	global wm

	global use_icons

	if use_icons:

		print "%s[submenu] (%s) <%s> " % ( (depth*"\t"), menu.getName().encode('utf8'), findIcon(menu.getIcon()) )

	else:

		print "%s[submenu] (%s) " % ( (depth*"\t"), menu.getName().encode('utf8'), )

	depth += 1

	for entry in menu.getEntries():

		if isinstance(entry, xdg.Menu.Menu):

			parseMenu(entry,depth)

		elif isinstance(entry, xdg.Menu.MenuEntry):

			checkWm(entry,wm)

			if entry.Show == False: continue

			if use_icons:

				print "%s[exec] (%s) {%s} <%s> " % ( (depth*"\t"), entry.DesktopEntry.getName().encode("utf8"), entry.DesktopEntry.getExec().split()[0], findIcon(entry.DesktopEntry.getIcon()) )

			else:

				print "%s[exec] (%s) {%s} " % ( (depth*"\t"), entry.DesktopEntry.getName().encode("utf8"), entry.DesktopEntry.getExec().split()[0] )

		elif isinstance(entry,xdg.Menu.Separator):

			print "%s[separator]" % (depth*"\t")

		elif isinstance(entry.xdg.Menu.Header):

			print "%s%s" % ( (depth*"\t"), entry.Name )

	depth -= 1

	print "%s[end]" % (depth*"\t")

def main(argv):

# Setting the default values

	global locs

	locs=[]

	global wm

	wm = "fluxbox"

	global file

	file = "~/.fluxbox/menu"

	global use_icons

	use_icons = False

	lang = os.getenv("LANG","C")

	file = os.path.expanduser("~/.fluxbox/menu")

	try:

		opts, args = getopt.getopt(argv, "hf:dl:d", ["help","lang=","file=","with-icons","stdout"])

	except getopt.GetoptError:

		usage()

		sys.exit(2)

	for opt, arg in opts:

		if opt in ("-h", "--help"):

			usage()

			sys.exit()

		elif opt in ("-l", "--lang"):

			lang = arg

		elif opt in ("-f", "--file"):

			file = os.path.expanduser(arg)

		elif opt == '--with-icons':

			use_icons = True		

			global paths

			locations = \

			['/usr/share/pixmaps/', \

			'/usr/share/icons/hicolor/24x24/apps/', \

			'/usr/share/icons/hicolor/32x32/apps/', \

			'/usr/share/icons/hicolor/48x48/apps/', \

			'/usr/share/icons/hicolor/24x24/stock/generic/', \

			'/usr/share/icons/HighContrastLargePrintInverse/48x48/apps/', \

			'/usr/share/icons/crystalsvg/16x16/apps/', \

			'/usr/share/icons/crystalsvg/32x32/apps/', \

			'/usr/share/icons/crystalsvg/48x48/apps/', \

			'/usr/share/icons/crystalsvg/48x48/devices/', \

			'/usr/share/icons/', \

			'/usr/share/icons/hicolor/32x32/apps/', \

			'/usr/share/icons/Bluecurve/24x24/stock/', \

			'/usr/share/icons/Bluecurve/48x48/filesystems/']

			paths=locations

#			for path in locations:

#				paths+=glob.glob(path+"/*/")+glob.glob(path+"/*/*/")

#			print paths

		elif opt == '--stdout':

			file = sys.stdout

	fsock = open(file,'w')

	saveout = sys.stdout

	sys.stdout = fsock

	menu=xdg.Menu.parse()

# is done automatically now

#	menu.setLocale(lang)

	print header()

	parseMenu(menu)

	print footer()

	sys.stdout = saveout

#	print menu

if __name__ == "__main__":

	main(sys.argv[1:])

#WindowMaker Reminder

#>>> p=re.compile("(\).*),(\s*\))")

#>>> p.search("((ola),(ola),)").groups()

#('),(ola)', ')')