Fix #65, Fix #63 Allow user to select list of modi's

    * Add script_dialog, so user can add dialogs using a script.
    * Add a way to disable modi's
    * Add -show <modi> option.

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;
+}
+