import sys

import os

import codecs

import xdg.Menu

from xdg.DesktopEntry import *

sys.stdout = codecs.getwriter('utf-8')(sys.stdout)

# For testing purpose. Uncomment to run this file without args...

# sys.argv.append('debian-menu')

if len(sys.argv) >= 3:

    # xdg_config_dirs hold possible paths. We use Debian one.

    fvwm_menu_name = sys.argv[1]

    sys_menu_file = sys.argv[2]

else:

    print("You forget pass NAME which is like /etc/xdg/NAME.menu")

    exit(1)

sys_menu = xdg.Menu.parse(sys_menu_file)

def parse_menu(menu, name):

    ''' Parse menu file. '''

    print 'DestroyMenu recreate %s' % name

    for entry in menu.getEntries():

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

            icon = entry.getIcon()

            subname = name+"-"+str(entry)

            print u'AddToMenu "%s" "%s%%folder.png%%" Popup "%s"' % (name, entry.getName(), subname)

            parse_menu(entry, subname)

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

            desktop = DesktopEntry(entry.DesktopEntry.getFileName())

            icon = desktop.getIcon()

            cmd = desktop.getExec().rstrip('%FUfu')

            if icon == '':

                print u'AddToMenu "%s" "%s" Exec exec %s' % (name, desktop.getName(), cmd)

            else:

                if icon.rfind('/') == -1 and icon.rfind('.') == -1:

                    icon = icon + ".xpm"

                if icon.endswith('.svg'):

                    icon = icon + ":32x32"

                print u'AddToMenu "%s" "%s%%%s%%" Exec exec %s' % (name, desktop.getName(), icon, cmd)

        else:

            pass

parse_menu(sys_menu, fvwm_menu_name)