From 6f404ddc8e59c3ca3aa20d1bb0b40a0904df1747 Mon Sep 17 00:00:00 2001 From: Dave Davenport Date: Tue, 3 Nov 2015 08:31:38 +0100 Subject: [PATCH] Initial import desktop file runner. --- Makefile.am | 2 + include/dialogs/dialogs.h | 1 + include/dialogs/drun.h | 5 + source/dialogs/drun.c | 229 ++++++++++++++++++++++++++++++++++++++ source/rofi.c | 4 + 5 files changed, 241 insertions(+) create mode 100644 include/dialogs/drun.h create mode 100644 source/dialogs/drun.c diff --git a/Makefile.am b/Makefile.am index adbc86b0..8a49572e 100644 --- a/Makefile.am +++ b/Makefile.am @@ -41,6 +41,7 @@ rofi_SOURCES=\ source/x11-helper.c\ source/dialogs/run.c\ source/dialogs/ssh.c\ + source/dialogs/drun.c\ source/dialogs/dmenu.c\ source/dialogs/combi.c\ source/dialogs/window.c\ @@ -56,6 +57,7 @@ rofi_SOURCES=\ include/x11-helper.h\ include/dialogs/ssh.h\ include/dialogs/run.h\ + include/dialogs/drun.h\ include/dialogs/dmenu.h\ include/dialogs/combi.h\ include/dialogs/script.h\ diff --git a/include/dialogs/dialogs.h b/include/dialogs/dialogs.h index 7600d16a..382784d5 100644 --- a/include/dialogs/dialogs.h +++ b/include/dialogs/dialogs.h @@ -7,6 +7,7 @@ #include "dialogs/run.h" #include "dialogs/ssh.h" +#include "dialogs/drun.h" #include "dialogs/dmenu.h" #include "dialogs/script.h" #include "dialogs/window.h" diff --git a/include/dialogs/drun.h b/include/dialogs/drun.h new file mode 100644 index 00000000..2c50211f --- /dev/null +++ b/include/dialogs/drun.h @@ -0,0 +1,5 @@ +#ifndef __DRUN_DIALOG_H__ +#define __DRUN_DIALOG_H__ + +extern Switcher drun_mode; +#endif diff --git a/source/dialogs/drun.c b/source/dialogs/drun.c new file mode 100644 index 00000000..39705865 --- /dev/null +++ b/source/dialogs/drun.c @@ -0,0 +1,229 @@ +/** + * rofi + * + * MIT/X11 License + * Copyright 2013-2015 Qball Cow + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS + * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + * + */ +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include + +#include "rofi.h" +#include "helper.h" +#include "dialogs/drun.h" + +#define RUN_CACHE_FILE "rofi-2.runcache" + +static inline int execsh ( const char *cmd, int run_in_term ) +{ + int retv = TRUE; + char **args = NULL; + int argc = 0; + if ( run_in_term ) { + helper_parse_setup ( config.run_shell_command, &args, &argc, "{cmd}", cmd, NULL ); + } + else { + helper_parse_setup ( config.run_command, &args, &argc, "{cmd}", cmd, NULL ); + } + GError *error = NULL; + g_spawn_async ( NULL, args, NULL, G_SPAWN_SEARCH_PATH, NULL, NULL, NULL, &error ); + if ( error != NULL ) { + char *msg = g_strdup_printf ( "Failed to execute: '%s'\nError: '%s'", cmd, error->message ); + error_dialog ( msg, FALSE ); + g_free ( msg ); + // print error. + g_error_free ( error ); + retv = FALSE; + } + + // Free the args list. + g_strfreev ( args ); + return retv; +} + +// execute sub-process +static void exec_cmd ( const char *cmd, int run_in_term ) +{ + if ( !cmd || !cmd[0] ) { + return; + } + + execsh ( cmd, run_in_term ); +} +/** + * Internal spider used to get list of executables. + */ +static char** get_apps ( unsigned int *length ) +{ + char **retv = NULL; + + const char * const * dr = g_get_system_data_dirs(); + while(dr != NULL && *dr != NULL ) { + gchar *bp = g_build_filename(*dr, "applications", NULL); + DIR *dir = opendir ( bp ); + + if ( dir != NULL ) { + struct dirent *dent; + + while ( ( dent = readdir ( dir ) ) != NULL ) { + if ( dent->d_type != DT_REG && dent->d_type != DT_LNK && dent->d_type != DT_UNKNOWN ) { + continue; + } + // Skip dot files. + if ( dent->d_name[0] == '.' ) { + continue; + } + gchar *path = g_build_filename(bp, dent->d_name, NULL); + GKeyFile *kf = g_key_file_new(); + GError *error = NULL; + // TODO: check what flags to set; + g_key_file_load_from_file(kf, path, 0, NULL); + if ( error == NULL ) { + retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( *retv ) ); + if( g_key_file_has_key(kf, "Desktop Entry", "Name", NULL) ) { + gchar *n = NULL; + gchar *gn = NULL; + n = g_key_file_get_string(kf, "Desktop Entry", "Name", NULL); + gn = g_key_file_get_string(kf, "Desktop Entry", "GenericName", NULL); + retv[( *length )] = g_strdup_printf("%-30s\t%s", n, gn?gn:""); + g_free(n); g_free(gn); + } else { + retv[(*length)] = g_strdup(dent->d_name); + } + retv[( *length ) + 1] = NULL; + ( *length )++; + } else { + g_error_free(error); + } + g_key_file_free(kf); + g_free(path); + } + + closedir ( dir ); + } + dr++; + g_free(bp); + } + + // No sorting needed. + if ( ( *length ) == 0 ) { + return retv; + } + + return retv; +} + +typedef struct _DRunModePrivateData +{ + unsigned int id; + char **cmd_list; + unsigned int cmd_list_length; +} DRunModePrivateData; + +static void drun_mode_init ( Switcher *sw ) +{ + if ( sw->private_data == NULL ) { + DRunModePrivateData *pd = g_malloc0 ( sizeof ( *pd ) ); + sw->private_data = (void *) pd; + } +} + +static char ** drun_mode_get_data ( unsigned int *length, Switcher *sw ) +{ + DRunModePrivateData *rmpd = (DRunModePrivateData *) sw->private_data; + if ( rmpd->cmd_list == NULL ) { + rmpd->cmd_list_length = 0; + rmpd->cmd_list = get_apps ( &( rmpd->cmd_list_length ) ); + } + if ( length != NULL ) { + *length = rmpd->cmd_list_length; + } + return rmpd->cmd_list; +} + +static SwitcherMode drun_mode_result ( int mretv, char **input, unsigned int selected_line, + Switcher *sw ) +{ + DRunModePrivateData *rmpd = (DRunModePrivateData *) sw->private_data; + SwitcherMode retv = MODE_EXIT; + + int shift = ( ( mretv & MENU_SHIFT ) == MENU_SHIFT ); + + if ( mretv & MENU_NEXT ) { + retv = NEXT_DIALOG; + } + else if ( mretv & MENU_PREVIOUS ) { + retv = PREVIOUS_DIALOG; + } + else if ( mretv & MENU_QUICK_SWITCH ) { + retv = ( mretv & MENU_LOWER_MASK ); + } + else if ( ( mretv & MENU_OK ) && rmpd->cmd_list[selected_line] != NULL ) { + exec_cmd ( rmpd->cmd_list[selected_line], shift ); + } + else if ( ( mretv & MENU_CUSTOM_INPUT ) && *input != NULL && *input[0] != '\0' ) { + exec_cmd ( *input, shift ); + } + return retv; +} + +static void drun_mode_destroy ( Switcher *sw ) +{ + DRunModePrivateData *rmpd = (DRunModePrivateData *) sw->private_data; + if ( rmpd != NULL ) { + g_strfreev(rmpd->cmd_list); + g_free ( rmpd ); + sw->private_data = NULL; + } +} + +static const char *mgrv ( unsigned int selected_line, void *sw, G_GNUC_UNUSED int *state ) +{ + return drun_mode_get_data ( NULL, sw )[selected_line]; +} + +Switcher drun_mode = +{ + .name = "drun", + .keycfg = NULL, + .keystr = NULL, + .modmask = AnyModifier, + .init = drun_mode_init, + .get_data = drun_mode_get_data, + .result = drun_mode_result, + .destroy = drun_mode_destroy, + .token_match = token_match, + .mgrv = mgrv, + .private_data = NULL, + .free = NULL +}; diff --git a/source/rofi.c b/source/rofi.c index 2765f1e9..fde01d92 100644 --- a/source/rofi.c +++ b/source/rofi.c @@ -1839,6 +1839,10 @@ static void setup_switchers ( void ) switchers[num_switchers].sw = &run_mode; num_switchers++; } + else if ( strcasecmp ( token, "drun" ) == 0 ) { + switchers[num_switchers].sw = &drun_mode; + num_switchers++; + } // combi dialog else if ( strcasecmp ( token, "combi" ) == 0 ) { switchers[num_switchers].sw = &combi_mode;