rofi/source/run-dialog.c

246 lines
6.9 KiB
C
Raw Normal View History

/**
2014-03-01 16:27:52 +00:00
* rofi
*
* MIT/X11 License
* Copyright 2013-2014 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.
*
*/
#define _GNU_SOURCE
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 "history.h"
#include "run-dialog.h"
2014-01-20 23:20:09 +00:00
#ifdef TIMING
#include <time.h>
#endif
2014-03-22 20:04:19 +00:00
#define RUN_CACHE_FILE "rofi-2.runcache"
2014-01-25 22:37:37 +00:00
2014-03-22 20:04:19 +00:00
static inline int execsh ( const char *cmd, int run_in_term )
{
// use sh for args parsing
2014-06-04 19:29:23 +00:00
if ( run_in_term ) {
2014-03-22 20:04:19 +00:00
return execlp ( config.terminal_emulator, config.terminal_emulator, "-e", "sh", "-c", cmd, NULL );
}
2014-03-22 20:04:19 +00:00
return execlp ( "/bin/sh", "sh", "-c", cmd, NULL );
}
2014-03-08 20:35:27 +00:00
// execute sub-process
2014-03-22 20:04:19 +00:00
static pid_t exec_cmd ( const char *cmd, int run_in_term )
{
2014-06-04 19:29:23 +00:00
if ( !cmd || !cmd[0] ) {
2014-03-22 20:04:19 +00:00
return -1;
}
2014-03-22 20:04:19 +00:00
signal ( SIGCHLD, catch_exit );
pid_t pid = fork ();
2014-06-04 19:29:23 +00:00
if ( !pid ) {
2014-03-22 20:04:19 +00:00
setsid ();
execsh ( cmd, run_in_term );
exit ( EXIT_FAILURE );
}
/**
* This happens in non-critical time (After launching app)
* It is allowed to be a bit slower.
*/
2014-05-22 07:33:32 +00:00
char *path = NULL;
2014-06-04 19:29:23 +00:00
if ( asprintf ( &path, "%s/%s", cache_dir, RUN_CACHE_FILE ) == -1 ) {
2014-04-23 06:47:46 +00:00
return -1;
}
2014-05-22 07:33:32 +00:00
history_set ( path, cmd );
2014-03-22 20:04:19 +00:00
free ( path );
return pid;
}
// 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.
*/
2014-05-22 07:33:32 +00:00
char *path = NULL;
2014-06-04 19:29:23 +00:00
if ( asprintf ( &path, "%s/%s", cache_dir, RUN_CACHE_FILE ) == -1 ) {
2014-04-23 06:47:46 +00:00
return;
2014-05-22 07:33:32 +00:00
}
history_remove ( path, cmd );
2014-03-22 20:04:19 +00:00
free ( path );
}
2014-01-21 09:01:55 +00:00
static int sort_func ( const void *a, const void *b )
{
2014-03-22 20:04:19 +00:00
const char *astr = *( const char * const * ) a;
const char *bstr = *( const char * const * ) b;
return strcasecmp ( astr, bstr );
}
static char ** get_apps ( unsigned int *length )
{
2014-03-22 20:04:19 +00:00
unsigned int num_favorites = 0;
char *path;
char **retv = NULL;
#ifdef TIMING
struct timespec start, stop;
2014-03-22 20:04:19 +00:00
clock_gettime ( CLOCK_REALTIME, &start );
#endif
2014-06-04 19:29:23 +00:00
if ( getenv ( "PATH" ) == NULL ) {
2014-03-22 20:04:19 +00:00
return NULL;
}
2014-06-04 19:29:23 +00:00
if ( asprintf ( &path, "%s/%s", cache_dir, RUN_CACHE_FILE ) > 0 ) {
retv = history_get_list ( path, length );
2014-04-23 06:47:46 +00:00
free ( path );
// Keep track of how many where loaded as favorite.
num_favorites = ( *length );
}
2014-03-22 20:04:19 +00:00
path = strdup ( getenv ( "PATH" ) );
2014-04-23 06:47:46 +00:00
for ( const char *dirname = strtok ( path, ":" );
2014-05-22 07:33:32 +00:00
dirname != NULL;
2014-06-04 19:29:23 +00:00
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 ) {
if ( dent->d_type != DT_REG &&
dent->d_type != DT_LNK &&
2014-06-04 19:29:23 +00:00
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;
}
char ** tr = realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
2014-06-04 19:29:23 +00:00
if ( tr != NULL ) {
retv = tr;
retv[( *length )] = strdup ( dent->d_name );
retv[( *length ) + 1] = NULL;
( *length )++;
}
}
2014-03-22 20:04:19 +00:00
closedir ( dir );
}
}
// TODO: check this is still fast enough. (takes 1ms on laptop.)
if ( ( *length ) > num_favorites ) {
qsort ( &retv[num_favorites], ( *length ) - num_favorites, sizeof ( char* ), sort_func );
2014-03-11 19:16:44 +00:00
}
2014-03-22 20:04:19 +00:00
free ( path );
#ifdef TIMING
2014-03-22 20:04:19 +00:00
clock_gettime ( CLOCK_REALTIME, &stop );
2014-06-04 19:29:23 +00:00
if ( stop.tv_sec != start.tv_sec ) {
2014-03-22 20:04:19 +00:00
stop.tv_nsec += ( stop.tv_sec - start.tv_sec ) * 1e9;
}
2014-03-22 20:04:19 +00:00
long diff = stop.tv_nsec - start.tv_nsec;
printf ( "Time elapsed: %ld us\n", diff / 1000 );
#endif
return retv;
}
SwitcherMode run_switcher_dialog ( char **input, void *data )
{
2014-03-22 20:04:19 +00:00
int shift = 0;
int selected_line = 0;
SwitcherMode retv = MODE_EXIT;
// act as a launcher
unsigned int cmd_list_length = 0;
char **cmd_list = get_apps ( &cmd_list_length );
2014-06-04 19:29:23 +00:00
if ( cmd_list == NULL ) {
cmd_list = malloc ( 2 * sizeof ( char * ) );
2014-03-22 20:04:19 +00:00
cmd_list[0] = strdup ( "No applications found" );
cmd_list[1] = NULL;
}
2014-08-03 15:05:06 +00:00
int mretv = menu ( cmd_list, cmd_list_length, input, "run:",
NULL, &shift, token_match, NULL, &selected_line, config.levenshtein_sort );
2014-06-04 19:29:23 +00:00
if ( mretv == MENU_NEXT ) {
2014-01-22 08:24:31 +00:00
retv = NEXT_DIALOG;
2014-03-22 20:04:19 +00:00
}
else if ( mretv == MENU_QUICK_SWITCH ) {
retv = selected_line;
}
2014-06-04 19:29:23 +00:00
else if ( mretv == MENU_OK && cmd_list[selected_line] != NULL ) {
2014-03-22 20:04:19 +00:00
exec_cmd ( cmd_list[selected_line], shift );
}
2014-06-04 19:29:23 +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 );
}
2014-06-04 19:29:23 +00:00
else if ( mretv == MENU_ENTRY_DELETE && cmd_list[selected_line] ) {
delete_entry ( cmd_list[selected_line] );
retv = RELOAD_DIALOG;
}
2014-06-04 19:29:23 +00:00
for ( int i = 0; cmd_list != NULL && cmd_list[i] != NULL; i++ ) {
2014-03-22 20:04:19 +00:00
free ( cmd_list[i] );
}
2014-06-04 19:29:23 +00:00
if ( cmd_list != NULL ) {
2014-03-22 20:04:19 +00:00
free ( cmd_list );
}
return retv;
}