rofi/source/dialogs/run.c

362 lines
11 KiB
C
Raw Normal View History

/**
2014-03-01 16:27:52 +00:00
* rofi
*
* MIT/X11 License
2015-12-31 23:27:00 +00:00
* Copyright 2013-2016 Qball Cow <qball@gmpclient.org>
*
* 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 <config.h>
2014-01-20 23:20:09 +00:00
#include <stdlib.h>
#include <stdio.h>
#include <X11/X.h>
#include <unistd.h>
2014-03-08 20:35:27 +00:00
#include <limits.h>
#include <signal.h>
#include <sys/types.h>
#include <dirent.h>
#include <strings.h>
#include <string.h>
2014-01-21 13:56:25 +00:00
#include <errno.h>
2014-03-12 07:41:38 +00:00
#include "rofi.h"
#include "helper.h"
#include "history.h"
2015-03-25 07:36:19 +00:00
#include "dialogs/run.h"
2014-03-22 20:04:19 +00:00
#define RUN_CACHE_FILE "rofi-2.runcache"
2014-01-25 22:37:37 +00:00
static inline int execsh ( const char *cmd, int run_in_term )
{
int retv = TRUE;
char **args = NULL;
int argc = 0;
2014-06-04 19:29:23 +00:00
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 );
2014-03-22 20:04:19 +00:00
}
2014-08-27 17:44:15 +00:00
GError *error = NULL;
2015-09-19 18:59:50 +00:00
g_spawn_async ( NULL, args, NULL, G_SPAWN_SEARCH_PATH, NULL, NULL, NULL, &error );
if ( error != NULL ) {
2015-09-19 18:59:50 +00:00
char *msg = g_strdup_printf ( "Failed to execute: '%s'\nError: '%s'", cmd, error->message );
2015-06-08 21:22:35 +00:00
error_dialog ( msg, FALSE );
g_free ( msg );
2014-08-27 17:44:15 +00:00
// print error.
g_error_free ( error );
retv = FALSE;
2014-08-27 17:44:15 +00:00
}
// Free the args list.
g_strfreev ( args );
return retv;
}
2014-03-08 20:35:27 +00:00
// execute sub-process
static void exec_cmd ( const char *cmd, int run_in_term )
{
2014-06-04 19:29:23 +00:00
if ( !cmd || !cmd[0] ) {
return;
2014-03-22 20:04:19 +00:00
}
if ( execsh ( cmd, run_in_term ) ) {
/**
* This happens in non-critical time (After launching app)
* It is allowed to be a bit slower.
*/
char *path = g_strdup_printf ( "%s/%s", cache_dir, RUN_CACHE_FILE );
history_set ( path, cmd );
g_free ( path );
}
}
// execute sub-process
2014-03-22 20:04:19 +00:00
static void delete_entry ( const char *cmd )
{
/**
* This happens in non-critical time (After launching app)
* It is allowed to be a bit slower.
*/
char *path = g_strdup_printf ( "%s/%s", cache_dir, RUN_CACHE_FILE );
2014-05-22 07:33:32 +00:00
history_remove ( path, cmd );
g_free ( path );
}
2015-07-05 07:47:55 +00:00
static int sort_func ( const void *a, const void *b, void *data __attribute__( ( unused ) ) )
{
2014-03-22 20:04:19 +00:00
const char *astr = *( const char * const * ) a;
const char *bstr = *( const char * const * ) b;
2014-11-06 16:44:41 +00:00
if ( astr == NULL && bstr == NULL ) {
return 0;
}
else if ( astr == NULL ) {
return 1;
}
else if ( bstr == NULL ) {
return -1;
}
2014-03-22 20:04:19 +00:00
return strcasecmp ( astr, bstr );
}
2015-01-05 20:53:50 +00:00
/**
* External spider to get list of executables.
*/
2015-01-07 14:53:41 +00:00
static char ** get_apps_external ( char **retv, unsigned int *length, unsigned int num_favorites )
2015-01-05 20:53:50 +00:00
{
int fd = execute_generator ( config.run_list_command );
if ( fd >= 0 ) {
FILE *inp = fdopen ( fd, "r" );
if ( inp ) {
char buffer[1024];
while ( fgets ( buffer, 1024, inp ) != NULL ) {
2015-01-07 14:53:41 +00:00
int found = 0;
// Filter out line-end.
if ( buffer[strlen ( buffer ) - 1] == '\n' ) {
buffer[strlen ( buffer ) - 1] = '\0';
}
// This is a nice little penalty, but doable? time will tell.
// given num_favorites is max 25.
for ( unsigned int j = 0; found == 0 && j < num_favorites; j++ ) {
if ( strcasecmp ( buffer, retv[j] ) == 0 ) {
found = 1;
}
}
if ( found == 1 ) {
continue;
}
// No duplicate, add it.
2015-03-27 19:28:53 +00:00
retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
retv[( *length )] = g_strdup ( buffer );
2015-01-05 20:53:50 +00:00
( *length )++;
}
if ( fclose ( inp ) != 0 ) {
2015-09-19 10:21:30 +00:00
fprintf ( stderr, "Failed to close stdout off executor script: '%s'\n",
strerror ( errno ) );
}
2015-01-05 20:53:50 +00:00
}
}
2015-03-27 19:28:53 +00:00
retv[( *length ) ] = NULL;
2015-01-05 20:53:50 +00:00
return retv;
}
/**
* Internal spider used to get list of executables.
*/
2015-03-27 19:28:53 +00:00
static char ** get_apps ( unsigned int *length )
{
2015-03-27 19:28:53 +00:00
char **retv = NULL;
2015-02-19 12:22:10 +00:00
unsigned int num_favorites = 0;
char *path;
2015-05-14 17:45:57 +00:00
if ( g_getenv ( "PATH" ) == NULL ) {
2014-03-22 20:04:19 +00:00
return NULL;
}
path = g_strdup_printf ( "%s/%s", cache_dir, RUN_CACHE_FILE );
retv = history_get_list ( path, length );
g_free ( path );
// Keep track of how many where loaded as favorite.
num_favorites = ( *length );
path = g_strdup ( getenv ( "PATH" ) );
2015-09-19 18:59:50 +00:00
for ( const char *dirname = strtok ( path, ":" ); dirname != NULL; dirname = strtok ( NULL, ":" ) ) {
2014-03-22 20:04:19 +00:00
DIR *dir = opendir ( dirname );
2014-06-04 19:29:23 +00:00
if ( dir != NULL ) {
struct dirent *dent;
2014-06-04 19:29:23 +00:00
while ( ( dent = readdir ( dir ) ) != NULL ) {
2015-09-19 18:59:50 +00:00
if ( dent->d_type != DT_REG && dent->d_type != DT_LNK && dent->d_type != DT_UNKNOWN ) {
continue;
}
2014-06-07 09:00:21 +00:00
// Skip dot files.
if ( dent->d_name[0] == '.' ) {
2014-06-07 09:00:21 +00:00
continue;
}
int found = 0;
// This is a nice little penalty, but doable? time will tell.
// given num_favorites is max 25.
2014-06-04 19:29:23 +00:00
for ( unsigned int j = 0; found == 0 && j < num_favorites; j++ ) {
if ( strcasecmp ( dent->d_name, retv[j] ) == 0 ) {
2014-03-22 20:04:19 +00:00
found = 1;
}
}
2014-06-04 19:29:23 +00:00
if ( found == 1 ) {
2014-03-22 20:04:19 +00:00
continue;
}
retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
retv[( *length )] = g_strdup ( dent->d_name );
retv[( *length ) + 1] = NULL;
( *length )++;
}
2014-03-22 20:04:19 +00:00
closedir ( dir );
}
}
2015-01-07 14:53:41 +00:00
// Get external apps.
2015-01-05 21:16:13 +00:00
if ( config.run_list_command != NULL && config.run_list_command[0] != '\0' ) {
2015-01-07 14:53:41 +00:00
retv = get_apps_external ( retv, length, num_favorites );
2015-01-05 21:16:13 +00:00
}
// No sorting needed.
2015-01-07 14:53:41 +00:00
if ( ( *length ) == 0 ) {
return retv;
}
// TODO: check this is still fast enough. (takes 1ms on laptop.)
if ( ( *length ) > num_favorites ) {
2015-09-19 18:59:50 +00:00
g_qsort_with_data ( &retv[num_favorites], ( *length ) - num_favorites, sizeof ( char* ), sort_func, NULL );
2014-03-11 19:16:44 +00:00
}
g_free ( path );
2014-11-06 16:44:41 +00:00
unsigned int removed = 0;
for ( unsigned int index = num_favorites; index < ( ( *length ) - 1 ); index++ ) {
2015-05-14 17:45:57 +00:00
if ( g_strcmp0 ( retv[index], retv[index + 1] ) == 0 ) {
2014-11-06 16:44:41 +00:00
g_free ( retv[index] );
retv[index] = NULL;
removed++;
}
}
if ( ( *length ) > num_favorites ) {
2015-09-19 10:21:30 +00:00
g_qsort_with_data ( &retv[num_favorites], ( *length ) - num_favorites, sizeof ( char* ),
sort_func,
NULL );
}
2014-11-06 16:44:41 +00:00
// Reduce array length;
( *length ) -= removed;
return retv;
}
2015-03-27 19:28:53 +00:00
typedef struct _RunModePrivateData
{
unsigned int id;
char **cmd_list;
unsigned int cmd_list_length;
} RunModePrivateData;
static void run_mode_init ( Mode *sw )
{
2015-03-27 19:28:53 +00:00
if ( sw->private_data == NULL ) {
RunModePrivateData *pd = g_malloc0 ( sizeof ( *pd ) );
sw->private_data = (void *) pd;
pd->cmd_list = get_apps ( &( pd->cmd_list_length ) );
}
2015-03-27 19:28:53 +00:00
}
static unsigned int run_mode_get_num_entries ( const Mode *sw )
2015-03-27 19:28:53 +00:00
{
2015-11-23 20:52:55 +00:00
const RunModePrivateData *rmpd = (const RunModePrivateData *) sw->private_data;
return rmpd->cmd_list_length;
2015-03-27 19:28:53 +00:00
}
static ModeMode run_mode_result ( Mode *sw, int mretv, char **input, unsigned int selected_line )
2015-03-27 19:28:53 +00:00
{
RunModePrivateData *rmpd = (RunModePrivateData *) sw->private_data;
ModeMode retv = MODE_EXIT;
2015-03-27 19:28:53 +00:00
int shift = ( ( mretv & MENU_SHIFT ) == MENU_SHIFT );
if ( mretv & MENU_NEXT ) {
2014-01-22 08:24:31 +00:00
retv = NEXT_DIALOG;
2014-03-22 20:04:19 +00:00
}
2015-03-27 19:28:53 +00:00
else if ( mretv & MENU_PREVIOUS ) {
2014-11-11 20:50:16 +00:00
retv = PREVIOUS_DIALOG;
}
2015-03-27 19:28:53 +00:00
else if ( mretv & MENU_QUICK_SWITCH ) {
retv = ( mretv & MENU_LOWER_MASK );
}
2015-03-27 19:28:53 +00:00
else if ( ( mretv & MENU_OK ) && rmpd->cmd_list[selected_line] != NULL ) {
exec_cmd ( rmpd->cmd_list[selected_line], shift );
2014-03-22 20:04:19 +00:00
}
2015-03-27 19:28:53 +00:00
else if ( ( mretv & MENU_CUSTOM_INPUT ) && *input != NULL && *input[0] != '\0' ) {
2014-03-22 20:04:19 +00:00
exec_cmd ( *input, shift );
}
2015-03-27 19:28:53 +00:00
else if ( ( mretv & MENU_ENTRY_DELETE ) && rmpd->cmd_list[selected_line] ) {
delete_entry ( rmpd->cmd_list[selected_line] );
g_free ( rmpd->cmd_list );
// Clear the list.
rmpd->cmd_list = NULL;
rmpd->cmd_list_length = 0;
retv = RELOAD_DIALOG;
}
2015-03-27 19:28:53 +00:00
return retv;
}
static void run_mode_destroy ( Mode *sw )
2015-03-27 19:28:53 +00:00
{
RunModePrivateData *rmpd = (RunModePrivateData *) sw->private_data;
if ( rmpd != NULL ) {
g_strfreev ( rmpd->cmd_list );
g_free ( rmpd );
sw->private_data = NULL;
}
}
2015-03-27 19:28:53 +00:00
static char *mgrv ( const Mode *sw, unsigned int selected_line, G_GNUC_UNUSED int *state, int get_entry )
{
const RunModePrivateData *rmpd = (const RunModePrivateData *) sw->private_data;
return get_entry ? g_strdup ( rmpd->cmd_list[selected_line] ) : NULL;
}
static int run_token_match ( const Mode *sw, char **tokens, int not_ascii, int case_sensitive, unsigned int index )
{
const RunModePrivateData *rmpd = (const RunModePrivateData *) sw->private_data;
return token_match ( tokens, rmpd->cmd_list[index], not_ascii, case_sensitive );
}
static int run_is_not_ascii ( const Mode *sw, unsigned int index )
{
const RunModePrivateData *rmpd = (const RunModePrivateData *) sw->private_data;
return !g_str_is_ascii ( rmpd->cmd_list[index] );
}
Mode run_mode =
2015-03-27 19:28:53 +00:00
{
.name = "run",
.keycfg = NULL,
.keystr = NULL,
.modmask = AnyModifier,
.init = run_mode_init,
.get_num_entries = run_mode_get_num_entries,
.result = run_mode_result,
.destroy = run_mode_destroy,
.token_match = run_token_match,
.mgrv = mgrv,
.get_completion = NULL,
.is_not_ascii = run_is_not_ascii,
.private_data = NULL,
.free = NULL
2015-03-27 19:28:53 +00:00
};