Archlinux.fr [Forums]

Code : Tout sélectionner

#!/usr/bin/env lua
--      This program is free software; you can redistribute it and/or modify
--      it under the terms of the GNU General Public License as published by
--      the Free Software Foundation version 3 (GPLv3)
--      
--      This program is distributed in the hope that it will be useful,
--      but WITHOUT ANY WARRANTY; without even the implied warranty of
--      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--      GNU General Public License for more details.
--      
--      You should have received a copy of the GNU General Public License
--      along with this program; if not, write to the Free Software
--      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
--      MA 02110-1301, USA.
require ("lfs")

-- Localisation désirée pour le nom, commentaire et nom générique.
local locale="fr"

-- Gestionnaire de fenêtres courant (utilisé pour OnlyShowIn).
local current_wm="openbox"

-- programme utilisé pour lancer les applications consoles.
local terminal="sakura -e"

-- Où sera stocké le menu.
local menu = {}


--- returnCategory
-- permet de donner un nom à une liste de catégories.
-- TODO: retourner aussi une image pour la catégorie.
function returnCategory(desktop)
	if desktop["categories"] == nil then
		return nil;
	end
	
	if string.find(desktop["categories"],"AudioVideo") then
		return "Audio/Video"
	end
	if string.find(desktop["categories"],"Education") then
		return "Éducation"
	end
	if string.find(desktop["categories"],"Development") then
		return "Développement"
	end
	if string.find(desktop["categories"],"Game") then
		return "Jeux"
	end
	if string.find(desktop["categories"],"Graphics") then
		return "Graphiques"
	end
	if string.find(desktop["categories"],"Network") then
		return "Réseau"
	end
	if string.find(desktop["categories"],"Office") then
		return "Bureautique"
	end
	if string.find(desktop["categories"],"Settings") then
		return "Réglages"
	end
	if string.find(desktop["categories"],"System") then
		return "Système"
	end
	if string.find(desktop["categories"],"Utility") then
		return "Utilitaires"
	end

	return "Autres"
end


--- loadDesktop
-- retourne une liste de propriétés pour un fichier .deskop.
-- arg: nom complet du fichier
-- ret: structure (terminal, exec, name, genericname, comment, icon,hidden)
--
-- name, genericname et comment sont localisés si possible. 
-- Pour avoir la version originale mettre la variable locale à c (de préférence)
function loadDesktop(filename)
	local desktop_file
	local line
	local pos
	local program={
			terminal=false,
			exec,
			name,
			genericname,
			comment,
			icon,
			hidden=false
		}
	
	if string.sub(filename,-7) ~= "desktop" then
		return nil
	end

	desktop_file=io.open(filename,"r")
	if desktop_file == nil then
		print ("Erreur : impossible de charger fichier", filename)
		return nil
	end
	
	line = desktop_file:read()
	while line ~= nil do
		pos = string.find(line,"=")
		if pos ~= nil then
			local key=string.lower(string.sub(line,1,pos-1))
			local value=string.sub(line,pos+1)
			
			if key == "type" then
				program.type=string.lower(value)
			
			elseif key == "terminal" and string.lower(value) == "true" then
				program.terminal=true
				
			elseif key == "name" and program.name == nil then
				program.name=value
			elseif key =="name["..locale.."]" then
				program.name=value
--[[				
			elseif key == "genericname" and program.genericname == nil then
				program.name=value
			elseif key == "genericname["..locale.."]" then
				program.genericname=value
				
			elseif key == "comment" and program.comment == nil then
				program.comment=value
			elseif key == "comment["..locale.."]" then
				program.comment=value
]]				
			elseif key == "icon" then
				program.icon=value
				
			elseif key == "categories" then
				program.categories=value
				
			elseif key == "exec" then
				pos = string.find(value," %%[fFuUdDnNickvm]")
				if pos ~= nil then 
					program.exec = string.sub(value,1,pos-1)
				else
					program.exec=value
				end
				
			elseif key=="onlyshowin" and current_wm then
				if not string.find(value,current_wm) then
					program.hidden=true
				end
			end
		end
		line=desktop_file:read() 
	end
	desktop_file:close()
	
	if program.terminal then
		program.exec = terminal .." ".. program.exec
	end
	
	return program
end


--- readDirectory(appdir)
-- lit non récursivement les fichiers .desktop du répertoire appdir.
function readDirectory(appdir)
	local appinfo
	local category
	local file
	local entry
	
	for file in lfs.dir(appdir) do
		if file ~= "." and file ~= ".." then
			appinfo = loadDesktop(appdir.."/"..file)
			if appinfo then
				if appinfo.type == "application" and not appinfo.hidden then
					category = returnCategory(appinfo)
					if category ~= nil then
						if menu[category] == nil then menu[category] = {} end
						menu[category][appinfo.name]={exec=appinfo.exec,icon=appinfo.icon}
					else
						menu[appinfo.name]={exec=appinfo.exec,icon=appinfo.icon}
					end
				end
			end
		end
	end
end


--- generateMenuJWM
-- génère un menu dynamique pour JWM à intégrer dans .jwmrc avec
-- <Include>exec:lua /path/nomfichier.lua</Include>
function generateMenuJWM()
	print ("<JWM>")
	for i,k in pairs(menu) do
		if k.exec == nil then
			print("<Menu label=\""..i.."\">")
			for name,data in pairs(k) do
				print("  <Program icon=\"".. data.icon .."\" label=\"".. name .."\">".. data.exec .."</Program>")
			end
			print("</Menu>")
		else 
			print("<Program icon=\"".. k.icon .."\" label=\"".. i .."\">".. k.exec .."</Program>")
			print("</item>")
		end
	end
	print("</JWM>")
end


--- generateMenuOpenBox
-- génère un menu dynamique pour OpenBox à intégrer dans menu.xml avec
-- <menu id="id" label="label" execute="lua /path/nomfichier.lua" />
-- et
-- <menu id="id" /> dans le menu "root-menu"
function generateMenuOpenBox()
	print ([[<openbox_pipe_menu xmlns="http://openbox.org/"
    	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    	xsi:schemaLocation="http://openbox.org/
    	file:///usr/share/openbox/menu.xsd">]])
	for i,k in pairs(menu) do
		if k.exec == nil then
			print("<menu id=\"gnome-menu-".. i .."\" label=\"".. i .."\">")
			for name,data in pairs(k) do
				print("  <item label=\"".. name .."\">")
				print("    <action name=\"Execute\"><execute>".. data.exec .."</execute></action>")
				print("  </item>")
			end
			print("</menu>")
		else 
			print("<item label=\"".. i .."\">")
			print("  <action name=\"Execute\"><execute>".. k.exec .."</execute></action>")
			print("</item>")
		end
	end
	print("</openbox_pipe_menu>")
end

readDirectory("/usr/share/applications")
-- readDirectory("/home/mimas/.local/share/applications")

-- *** Choisir generateMenuJWM() ou generateMenuOpenBox ***
generateMenuOpenBox()

Code : Tout sélectionner

<openbox_menu xmlns="http://openbox.org/3.4/menu">
	<menu id="bookmarks-menu" label="Bookmarks" execute="lua ~/.config/openbox/bookmarks.lua" />
[..]
<menu id="root-menu" label="Openbox 3">
  	[...]
  	<menu id="bookmarks-menu" />
  	<separator />
[...]

Code : Tout sélectionner

local bookmarks=io.open(os.getenv("HOME").."/.gtk-bookmarks","r")

-- simple vérification
if bookmarks == nil then
	os.exit(1)
end

print ([[<openbox_pipe_menu xmlns="http://openbox.org/"
    	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    	xsi:schemaLocation="http://openbox.org/
    	file:///usr/share/openbox/menu.xsd">]])  	
while true do
	local bookmark=bookmarks:read()
	if bookmark == nil then break end
	
	for location, name in string.gmatch(bookmark, "(.+)%s(.+)") do
		print("<item label=\"".. name .."\">")
		print("  <action name=\"Execute\"><execute>thunar ".. location .."</execute></action>")
		print("</item>")
  end
end 
print("</openbox_pipe_menu>")

bookmarks:close()

Code : Tout sélectionner

 #!/usr/bin/env lua ...

... return program 
end 

Code : Tout sélectionner

-- Où sera stocké le menu.
local menu = {}

Code : Tout sélectionner

--- generateMenuOpenBox ....

-- *** Choisir generateMenuJWM() ou generateMenuOpenBox ***
generateMenuOpenBox(

Code : Tout sélectionner

using GLib;

/* Répertoires où il va lire les .desktop */
public const string[] directories = {
	"/usr/share/applications/",
	"/home/mimas/.local/share/applications/"
};

public
enum Category {
	AUDIOVIDEO,
	DEVELOPMENT,
	EDUCATION,
	GAME,
	GRAPHICS,
	MISC,
	NETWORK,
	SETTINGS,
	NONE,
	OFFICE,
	SYSTEM,
	UTILITY
}

public
struct DesktopEntry
{
	public string name;
	public string exec;
	public uint category;
}

private
string sanitize_execname(string exec)
{
	string new_exec = null;

	if (exec.contains("%"))
	{
		if (exec.contains("%f"))
			new_exec = exec.replace("%f","");
		else if (exec.contains("%F"))
			new_exec = exec.replace("%F","");
		else if (exec.contains("%u"))
			new_exec = exec.replace("%u","");
		else if (exec.contains("%U"))
			new_exec = exec.replace("%U","");
		else if (exec.contains("%d"))
			new_exec = exec.replace("%d","");
		else if (exec.contains("%D"))
			new_exec = exec.replace("%D","");
		else if (exec.contains("%n"))
			new_exec = exec.replace("%n","");
		else if (exec.contains("%N"))
			new_exec = exec.replace("%N","");
		else if (exec.contains("%i"))
			new_exec = exec.replace("%i","");
		else if (exec.contains("%c"))
			new_exec = exec.replace("%c","");
		else if (exec.contains("%k"))
			new_exec = exec.replace("%k","");
		else if (exec.contains("%v"))
			new_exec = exec.replace("%v","");
		else if (exec.contains("%m"))
			new_exec = exec.replace("%m","");

		return new_exec.chomp();
	}
	return exec;
}

private
void populate_menu_with_desktop(
				ref SList<DesktopEntry?> list,
				string filename)
{
	DesktopEntry d = DesktopEntry();
	string category;

	GLib.KeyFile desktop = new KeyFile();

	try
	{
		if( desktop.load_from_file(filename, GLib.KeyFileFlags.NONE))
		{
			d.name = desktop.get_locale_string("Desktop Entry", "Name", null);

			if (d.name==null)
				return;

			d.exec = sanitize_execname(desktop.get_string("Desktop Entry", "Exec"));

			if (desktop.has_key("Desktop Entry", "Categories"))
			{
				category = desktop.get_string("Desktop Entry", "Categories");

				if (category.contains("AudioVideo"))
					d.category = Category.AUDIOVIDEO;
				else if (category.contains("Development"))
					d.category = Category.DEVELOPMENT;
				else if (category.contains("Education"))
					d.category = Category.EDUCATION;
				else if (category.contains("Game"))
					d.category = Category.GAME;
				else if (category.contains("Graphics"))
					d.category = Category.GRAPHICS;
				else if (category.contains("Network"))
					d.category = Category.NETWORK;
				else if (category.contains("Settings"))
					d.category = Category.SETTINGS;
				else if (category.contains("Office"))
					d.category = Category.OFFICE;
				else if (category.contains("Utility"))
					d.category = Category.UTILITY;
				else if (category.contains("System"))
					d.category = Category.SYSTEM;
				else
					d.category = Category.MISC;
			}
			else
				d.category = Category.NONE;

			list.append(d);
			return;
		}
	}
	catch (Error e)
	{
		//stdout.printf("%s - %s\n", filename, e.message);
	}
	return;
}

private
void write_openbox_menu(
				SList<DesktopEntry?> list,
				string name,
				uint category)
{
	bool display_menu = true;
	bool in_menu = false;

	foreach (DesktopEntry de in list)
	{
		if (de.category == category)
		{
			if (display_menu == true && name != "") {
				stdout.printf("<menu id=\"menu-%s\" label=\"%s\">\n", name, name);
				display_menu = false;
				in_menu = true;
			}
			stdout.printf("  <item label=\"%s\">", de.name);
			stdout.printf("<action name=\"Execute\"><execute>%s</execute></action>", de.exec);
			stdout.printf("</item>\n");
		}
	}
	if (in_menu == true)
		stdout.printf("</menu>\n");
}

public
int main(string[] argv)
{
	string entry;
	SList<DesktopEntry?> menulist = new SList<DesktopEntry?>();

	foreach (string directory in directories)
	{
		try
		{
			Dir dir = GLib.Dir.open(directory);
			while ( (entry = dir.read_name()) != null )
				populate_menu_with_desktop(ref menulist, Path.build_filename(directory, entry));
		}
		catch (Error e)
		{
			stdout.printf("%s\n", e.message);
		}
	}

	stdout.printf("<openbox_pipe_menu xmlns=\"http://openbox.org/\"");
	stdout.printf(" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"");
	stdout.printf("	xsi:schemaLocation=\"http://openbox.org/ file:///usr/share/openbox/menu.xsd\">\n");
	write_openbox_menu(menulist, "", Category.NONE);
	stdout.printf("<separator />");
	write_openbox_menu(menulist, "Developpement", Category.DEVELOPMENT);
	write_openbox_menu(menulist, "Jeux", Category.GAME);
	write_openbox_menu(menulist, "Audio/Vidéo", Category.AUDIOVIDEO);
	write_openbox_menu(menulist, "Éducation", Category.EDUCATION);
	write_openbox_menu(menulist, "Graphisme", Category.GRAPHICS);
	write_openbox_menu(menulist, "Bureautique", Category.OFFICE);
	write_openbox_menu(menulist, "Utilitaire", Category.UTILITY);
	write_openbox_menu(menulist, "Internet", Category.NETWORK);
	write_openbox_menu(menulist, "Système", Category.SYSTEM);
	write_openbox_menu(menulist, "réglages", Category.SETTINGS);
	write_openbox_menu(menulist, "Autres", Category.MISC);
	stdout.printf("</openbox_pipe_menu>");

	return 0;
}