diff options
Diffstat (limited to 'source/script-dialog.c')
-rw-r--r-- | source/script-dialog.c | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/source/script-dialog.c b/source/script-dialog.c new file mode 100644 index 00000000..a43234e1 --- /dev/null +++ b/source/script-dialog.c @@ -0,0 +1,201 @@ +/** + * 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 +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <strings.h> +#include <string.h> +#include <ctype.h> +#include <assert.h> +#include "rofi.h" +#include "script-dialog.h" + + + +pid_t execute_generator ( char * cmd ) +{ + int filedes[2]; + pid_t pid; + + + if ( -1 == pipe ( filedes ) ) { + perror ( "pipe failed" ); + return 0; + } + + switch ( pid = fork () ) + { + case -1: + perror ( "Failed to fork, executing generator failed" ); + pid = 0; + break; + + case 0: /* child */ + close ( 1 ); + dup ( filedes[1] ); + close ( filedes[1] ); + execlp ( "/bin/sh", "sh", "-c", cmd, NULL ); + perror ( cmd ); + break; + + default: /* parent */ + close ( 0 ); + dup ( filedes[0] ); + close ( filedes[0] ); + close ( filedes[1] ); + break; + } + + return pid; +} + + +static char **get_script_output ( char *command, unsigned int *length ) +{ + char buffer[1024]; + char **retv = NULL; + + *length = 0; + execute_generator ( command ); + while ( fgets ( buffer, 1024, stdin ) != NULL ) { + char **tr = realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) ); + if ( tr == NULL ) { + return retv; + } + retv = tr; + retv[( *length )] = strdup ( buffer ); + retv[( *length ) + 1] = NULL; + + // Filter out line-end. + if ( retv[( *length )][strlen ( buffer ) - 1] == '\n' ) { + retv[( *length )][strlen ( buffer ) - 1] = '\0'; + } + + ( *length )++; + } + + return retv; +} + +char **execute_executor ( ScriptOptions *options, const char *result, unsigned int *length ) +{ + char **retv = NULL; + char *command; + if ( asprintf ( &command, "%s %s", options->script_path, result ) > 0 ) { + retv = get_script_output ( command, length ); + free ( command ); + } + return retv; +} + +SwitcherMode script_switcher_dialog ( char **input, void *data ) +{ + ScriptOptions *options = (ScriptOptions *) data; + assert ( options != NULL ); + int selected_line = 0; + SwitcherMode retv = MODE_EXIT; + unsigned int length = 0; + char **list = get_script_output ( options->script_path, &length ); + + + do { + unsigned int new_length = 0; + char **new_list = NULL; + int mretv = menu ( list, length, input, options->name, NULL, NULL, + token_match, NULL, &selected_line ); + + if ( mretv == MENU_NEXT ) { + retv = NEXT_DIALOG; + } + else if ( mretv == MENU_OK && list[selected_line] != NULL ) { + new_list = execute_executor ( options, list[selected_line], &new_length ); + } + else if ( mretv == MENU_CUSTOM_INPUT && *input != NULL && *input[0] != '\0' ) { + new_list = execute_executor ( options, *input, &new_length ); + } + + // Free old list. + for ( unsigned int i = 0; i < length; i++ ) { + free ( list[i] ); + } + + if ( list != NULL ) { + free ( list ); + list = NULL; + } + // If a new list was generated, use that an loop around. + if ( new_list != NULL ) { + list = new_list; + length = new_length; + free ( *input ); + *input = NULL; + } + } while ( list != NULL ); + + return retv; +} + +void script_switcher_free_options ( ScriptOptions *sw ) +{ + if ( sw == NULL ) { + return; + } + free ( sw->name ); + free ( sw->script_path ); + free ( sw ); +} + + +ScriptOptions *script_switcher_parse_setup ( const char *str ) +{ + ScriptOptions *sw = calloc ( 1, sizeof ( *sw ) ); + char *endp = NULL; + char *parse = strdup ( str ); + unsigned int index = 0; + // TODO: This is naive and can be improved. + for ( char *token = strtok_r ( parse, ":", &endp ); token != NULL; token = strtok_r ( NULL, ":", &endp ) ) { + if ( index == 0 ) { + sw->name = strdup ( token ); + } + else if ( index == 1 ) { + sw->script_path = strdup ( token ); + } + index++; + } + free ( parse ); + if ( index == 2 ) { + return sw; + } + fprintf ( stderr, "The script command '%s' has %d options, but needs 2: <name>:<script>.\n", + str, index ); + script_switcher_free_options ( sw ); + return NULL; +} + |