rofi/source/dialogs/script.c

212 lines
6.8 KiB
C
Raw Normal View History

/**
* 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>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <strings.h>
#include <string.h>
#include <ctype.h>
#include <assert.h>
#include <errno.h>
#include "rofi.h"
2015-03-25 07:36:19 +00:00
#include "dialogs/script.h"
#include "helper.h"
2016-01-07 18:47:37 +00:00
#include "mode-private.h"
2015-03-27 19:28:53 +00:00
static char **get_script_output ( const char *command, unsigned int *length )
{
char **retv = NULL;
*length = 0;
int fd = execute_generator ( command );
if ( fd >= 0 ) {
FILE *inp = fdopen ( fd, "r" );
if ( inp ) {
2016-04-10 12:30:13 +00:00
char *buffer = NULL;
size_t buffer_length = 0;
while ( getline ( &buffer, &buffer_length, inp ) > 0 ) {
retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
retv[( *length )] = g_strdup ( buffer );
retv[( *length ) + 1] = NULL;
// Filter out line-end.
if ( retv[( *length )][strlen ( buffer ) - 1] == '\n' ) {
retv[( *length )][strlen ( buffer ) - 1] = '\0';
}
( *length )++;
}
2016-04-10 12:30:13 +00:00
if ( buffer ) {
free ( buffer );
}
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 ) );
}
}
}
return retv;
}
static char **execute_executor ( Mode *sw, const char *result, unsigned int *length )
{
char **retv = NULL;
2014-11-22 21:40:53 +00:00
char *arg = g_shell_quote ( result );
2015-03-27 19:28:53 +00:00
char *command = g_strdup_printf ( "%s %s", (const char *) sw->ed, arg );
retv = get_script_output ( command, length );
g_free ( command );
g_free ( arg );
return retv;
}
static void script_switcher_free ( Mode *sw )
{
2015-03-27 19:28:53 +00:00
if ( sw == NULL ) {
return;
}
g_free ( sw->ed );
g_free ( sw );
}
typedef struct
2015-03-27 19:28:53 +00:00
{
unsigned int id;
char **cmd_list;
unsigned int cmd_list_length;
} ScriptModePrivateData;
static int script_mode_init ( Mode *sw )
2015-03-27 19:28:53 +00:00
{
if ( sw->private_data == NULL ) {
ScriptModePrivateData *pd = g_malloc0 ( sizeof ( *pd ) );
sw->private_data = (void *) pd;
pd->cmd_list = get_script_output ( (const char *) sw->ed, &( pd->cmd_list_length ) );
2015-03-27 19:28:53 +00:00
}
return TRUE;
2015-03-27 19:28:53 +00:00
}
static unsigned int script_mode_get_num_entries ( const Mode *sw )
2015-03-27 19:28:53 +00:00
{
2015-11-23 20:52:55 +00:00
const ScriptModePrivateData *rmpd = (const ScriptModePrivateData *) sw->private_data;
return rmpd->cmd_list_length;
}
static ModeMode script_mode_result ( Mode *sw, int mretv, char **input, unsigned int selected_line )
{
2015-03-27 19:28:53 +00:00
ScriptModePrivateData *rmpd = (ScriptModePrivateData *) sw->private_data;
ModeMode retv = MODE_EXIT;
2015-03-27 19:28:53 +00:00
char **new_list = NULL;
unsigned int new_length = 0;
if ( ( mretv & MENU_NEXT ) ) {
retv = NEXT_DIALOG;
}
2015-03-27 19:28:53 +00:00
else if ( ( mretv & MENU_PREVIOUS ) ) {
retv = PREVIOUS_DIALOG;
}
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 ) {
new_list = execute_executor ( sw, rmpd->cmd_list[selected_line], &new_length );
}
else if ( ( mretv & MENU_CUSTOM_INPUT ) && *input != NULL && *input[0] != '\0' ) {
new_list = execute_executor ( sw, *input, &new_length );
}
// If a new list was generated, use that an loop around.
if ( new_list != NULL ) {
g_strfreev ( rmpd->cmd_list );
rmpd->cmd_list = new_list;
rmpd->cmd_list_length = new_length;
g_free ( *input );
*input = NULL;
retv = RELOAD_DIALOG;
}
return retv;
}
static void script_mode_destroy ( Mode *sw )
2015-03-27 19:28:53 +00:00
{
ScriptModePrivateData *rmpd = (ScriptModePrivateData *) sw->private_data;
if ( rmpd != NULL ) {
g_strfreev ( rmpd->cmd_list );
g_free ( rmpd );
sw->private_data = NULL;
}
}
2016-01-07 20:27:20 +00:00
static char *_get_display_value ( const Mode *sw, unsigned int selected_line, G_GNUC_UNUSED int *state, int get_entry )
{
ScriptModePrivateData *rmpd = sw->private_data;
return get_entry ? g_strdup ( rmpd->cmd_list[selected_line] ) : NULL;
}
2016-05-22 15:47:34 +00:00
static int script_token_match ( const Mode *sw, GRegex **tokens, unsigned int index )
{
ScriptModePrivateData *rmpd = sw->private_data;
2016-05-22 15:47:34 +00:00
return token_match ( tokens, rmpd->cmd_list[index] );
}
2016-01-07 20:27:20 +00:00
#include "mode-private.h"
Mode *script_switcher_parse_setup ( const char *str )
{
Mode *sw = g_malloc0 ( sizeof ( *sw ) );
char *endp = NULL;
char *parse = g_strdup ( str );
unsigned int index = 0;
const char *const sep = ":";
for ( char *token = strtok_r ( parse, sep, &endp ); token != NULL; token = strtok_r ( NULL, sep, &endp ) ) {
if ( index == 0 ) {
2015-03-27 19:28:53 +00:00
g_strlcpy ( sw->name, token, 32 );
}
else if ( index == 1 ) {
2015-11-27 12:01:25 +00:00
sw->ed = (void *) rofi_expand_path ( token );
}
index++;
}
g_free ( parse );
if ( index == 2 ) {
2016-01-07 20:27:20 +00:00
sw->free = script_switcher_free;
sw->_init = script_mode_init;
sw->_get_num_entries = script_mode_get_num_entries;
sw->_result = script_mode_result;
sw->_destroy = script_mode_destroy;
sw->_token_match = script_token_match;
sw->_get_completion = NULL,
sw->_preprocess_input = NULL,
2016-01-07 20:27:20 +00:00
sw->_get_display_value = _get_display_value;
2015-03-27 19:28:53 +00:00
return sw;
}
2015-09-19 18:59:50 +00:00
fprintf ( stderr, "The script command '%s' has %u options, but needs 2: <name>:<script>.\n", str, index );
2015-03-27 19:28:53 +00:00
script_switcher_free ( sw );
return NULL;
}