Mirrors/rofi
2
0
Fork 0
mirror of https://github.com/lbonn/rofi synced 2024-11-26 05:40:23 +00:00
Code Issues Projects Releases Packages Wiki Activity

[Dmenu][Script] Share some code between the two modi

Browse source
This commit is contained in:
Dave Davenport 2019-04-02 13:51:53 +02:00
parent 88d987b054
commit 011d92075c
4 changed files with 44 additions and 56 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Makefile.am
View file

@ -125,6 +125,7 @@ SOURCES=\
include/dialogs/window.h\ include/dialogs/window.h\
include/dialogs/dialogs.h\ include/dialogs/dialogs.h\
include/dialogs/help-keys.h\ include/dialogs/help-keys.h\
include/dialogs/dmenuscriptshared.h\
resources/resources.c\ resources/resources.c\
resources/resources.h resources/resources.h

1
meson.build
View file

@ -193,6 +193,7 @@ rofi_sources = files(
'include/dialogs/window.h', 'include/dialogs/window.h',
'include/dialogs/dialogs.h', 'include/dialogs/dialogs.h',
'include/dialogs/help-keys.h', 'include/dialogs/help-keys.h',
'include/dialogs/dmenuscriptshared.h',
) )
theme_lexer_sources = files('lexer/theme-lexer.l') theme_lexer_sources = files('lexer/theme-lexer.l')
theme_parser_sources = files('lexer/theme-parser.y') theme_parser_sources = files('lexer/theme-parser.y')

62
source/dialogs/dmenu.c
View file

@ -51,6 +51,8 @@
#include "view.h" #include "view.h"
#include "rofi-icon-fetcher.h" #include "rofi-icon-fetcher.h"
#include "dialogs/dmenuscriptshared.h"
static inline unsigned int bitget ( uint32_t *array, unsigned int index ) static inline unsigned int bitget ( uint32_t *array, unsigned int index )
{ {
uint32_t bit = index % 32; uint32_t bit = index % 32;
@ -65,11 +67,6 @@ static inline void bittoggle ( uint32_t *array, unsigned int index )
*v ^= 1 << bit; *v ^= 1 << bit;
} }
typedef struct {
char *string;
char *icon_name;
uint32_t icon_fetch_uid;
} DmenuModeEntry ;
typedef struct typedef struct
{ {
/** Settings */ /** Settings */
@ -87,7 +84,7 @@ typedef struct
unsigned int num_selected_list; unsigned int num_selected_list;
unsigned int do_markup; unsigned int do_markup;
// List with entries. // List with entries.
DmenuModeEntry *cmd_list; DmenuScriptEntry *cmd_list;
unsigned int cmd_list_real_length; unsigned int cmd_list_real_length;
unsigned int cmd_list_length; unsigned int cmd_list_length;
unsigned int only_selected; unsigned int only_selected;
@ -114,7 +111,7 @@ static void read_add ( DmenuModePrivateData * pd, char *data, gsize len )
gsize data_len = len; gsize data_len = len;
if ( ( pd->cmd_list_length + 2 ) > pd->cmd_list_real_length ) { if ( ( pd->cmd_list_length + 2 ) > pd->cmd_list_real_length ) {
pd->cmd_list_real_length = MAX ( pd->cmd_list_real_length * 2, 512 ); pd->cmd_list_real_length = MAX ( pd->cmd_list_real_length * 2, 512 );
pd->cmd_list = g_realloc ( pd->cmd_list, ( pd->cmd_list_real_length ) * sizeof ( DmenuModeEntry ) ); pd->cmd_list = g_realloc ( pd->cmd_list, ( pd->cmd_list_real_length ) * sizeof ( DmenuScriptEntry ) );
} }
// Init. // Init.
pd->cmd_list[pd->cmd_list_length].icon_fetch_uid = 0; pd->cmd_list[pd->cmd_list_length].icon_fetch_uid = 0;
@ -122,22 +119,11 @@ static void read_add ( DmenuModePrivateData * pd, char *data, gsize len )
char *end = strchr(data, '\0'); char *end = strchr(data, '\0');
if ( end != NULL ) { if ( end != NULL ) {
data_len = end-data; data_len = end-data;
// Go into extra parsing. dmenuscript_parse_entry_extras ( NULL, &(pd->cmd_list[pd->cmd_list_length]), end+1, len-data_len);
gsize length_key = data_len+1;
while ( length_key <= len && data [length_key] != '\x1f' ) {
length_key++;
}
if ( length_key < len ) {
data[length_key] ='\0';
char * value = data + length_key+1;
if ( strcasecmp(&data[data_len+1], "icon" ) == 0 ) {
pd->cmd_list[pd->cmd_list_length].icon_name = g_strdup(value);
}
}
} }
char *utfstr = rofi_force_utf8 ( data, data_len ); char *utfstr = rofi_force_utf8 ( data, data_len );
pd->cmd_list[pd->cmd_list_length].string = utfstr; pd->cmd_list[pd->cmd_list_length].entry = utfstr;
pd->cmd_list[pd->cmd_list_length + 1].string = NULL; pd->cmd_list[pd->cmd_list_length + 1].entry = NULL;
pd->cmd_list_length++; pd->cmd_list_length++;
} }
@ -259,7 +245,7 @@ static char *get_display_data ( const Mode *data, unsigned int index, int *state
{ {
Mode *sw = (Mode *) data; Mode *sw = (Mode *) data;
DmenuModePrivateData *pd = (DmenuModePrivateData *) mode_get_private_data ( sw ); DmenuModePrivateData *pd = (DmenuModePrivateData *) mode_get_private_data ( sw );
DmenuModeEntry *retv = (DmenuModeEntry *) pd->cmd_list; DmenuScriptEntry *retv = (DmenuScriptEntry *) pd->cmd_list;
for ( unsigned int i = 0; i < pd->num_active_list; i++ ) { for ( unsigned int i = 0; i < pd->num_active_list; i++ ) {
if ( index >= pd->active_list[i].start && index <= pd->active_list[i].stop ) { if ( index >= pd->active_list[i].start && index <= pd->active_list[i].stop ) {
*state |= ACTIVE; *state |= ACTIVE;
@ -276,7 +262,7 @@ static char *get_display_data ( const Mode *data, unsigned int index, int *state
if ( pd->do_markup ) { if ( pd->do_markup ) {
*state |= MARKUP; *state |= MARKUP;
} }
return get_entry ? dmenu_format_output_string ( pd, retv[index].string ) : NULL; return get_entry ? dmenu_format_output_string ( pd, retv[index].entry ) : NULL;
} }
static void dmenu_mode_free ( Mode *sw ) static void dmenu_mode_free ( Mode *sw )
@ -302,8 +288,8 @@ static void dmenu_mode_free ( Mode *sw )
} }
for ( size_t i = 0; i < pd->cmd_list_length; i++ ) { for ( size_t i = 0; i < pd->cmd_list_length; i++ ) {
if ( pd->cmd_list[i].string ) { if ( pd->cmd_list[i].entry ) {
g_free ( pd->cmd_list[i].string ); g_free ( pd->cmd_list[i].entry );
g_free ( pd->cmd_list[i].icon_name ); g_free ( pd->cmd_list[i].icon_name );
} }
} }
@ -400,7 +386,7 @@ static int dmenu_mode_init ( Mode *sw )
static int dmenu_token_match ( const Mode *sw, rofi_int_matcher **tokens, unsigned int index ) static int dmenu_token_match ( const Mode *sw, rofi_int_matcher **tokens, unsigned int index )
{ {
DmenuModePrivateData *rmpd = (DmenuModePrivateData *) mode_get_private_data ( sw ); DmenuModePrivateData *rmpd = (DmenuModePrivateData *) mode_get_private_data ( sw );
return helper_token_match ( tokens, rmpd->cmd_list[index].string ); return helper_token_match ( tokens, rmpd->cmd_list[index].entry );
} }
static char *dmenu_get_message ( const Mode *sw ) static char *dmenu_get_message ( const Mode *sw )
{ {
@ -414,7 +400,7 @@ static cairo_surface_t *dmenu_get_icon ( const Mode *sw, unsigned int selected_l
{ {
DmenuModePrivateData *pd = (DmenuModePrivateData *) mode_get_private_data ( sw ); DmenuModePrivateData *pd = (DmenuModePrivateData *) mode_get_private_data ( sw );
g_return_val_if_fail ( pd->cmd_list != NULL, NULL ); g_return_val_if_fail ( pd->cmd_list != NULL, NULL );
DmenuModeEntry *dr = &( pd->cmd_list[selected_line] ); DmenuScriptEntry *dr = &( pd->cmd_list[selected_line] );
if ( dr->icon_name == NULL ) { if ( dr->icon_name == NULL ) {
return NULL; return NULL;
} }
@ -464,20 +450,20 @@ static void dmenu_finish ( RofiViewState *state, int retv )
static void dmenu_print_results ( DmenuModePrivateData *pd, const char *input ) static void dmenu_print_results ( DmenuModePrivateData *pd, const char *input )
{ {
DmenuModeEntry *cmd_list = pd->cmd_list; DmenuScriptEntry *cmd_list = pd->cmd_list;
int seen = FALSE; int seen = FALSE;
if ( pd->selected_list != NULL ) { if ( pd->selected_list != NULL ) {
for ( unsigned int st = 0; st < pd->cmd_list_length; st++ ) { for ( unsigned int st = 0; st < pd->cmd_list_length; st++ ) {
if ( bitget ( pd->selected_list, st ) ) { if ( bitget ( pd->selected_list, st ) ) {
seen = TRUE; seen = TRUE;
rofi_output_formatted_line ( pd->format, cmd_list[st].string, st, input ); rofi_output_formatted_line ( pd->format, cmd_list[st].entry , st, input );
} }
} }
} }
if ( !seen ) { if ( !seen ) {
const char *cmd = input; const char *cmd = input;
if ( pd->selected_line != UINT32_MAX ) { if ( pd->selected_line != UINT32_MAX ) {
cmd = cmd_list[pd->selected_line].string; cmd = cmd_list[pd->selected_line].entry ;
} }
rofi_output_formatted_line ( pd->format, cmd, pd->selected_line, input ); rofi_output_formatted_line ( pd->format, cmd, pd->selected_line, input );
} }
@ -488,7 +474,7 @@ static void dmenu_finalize ( RofiViewState *state )
int retv = FALSE; int retv = FALSE;
DmenuModePrivateData *pd = (DmenuModePrivateData *) rofi_view_get_mode ( state )->private_data; DmenuModePrivateData *pd = (DmenuModePrivateData *) rofi_view_get_mode ( state )->private_data;
unsigned int cmd_list_length = pd->cmd_list_length; unsigned int cmd_list_length = pd->cmd_list_length;
DmenuModeEntry *cmd_list = pd->cmd_list; DmenuScriptEntry *cmd_list = pd->cmd_list;
char *input = g_strdup ( rofi_view_get_user_input ( state ) ); char *input = g_strdup ( rofi_view_get_user_input ( state ) );
pd->selected_line = rofi_view_get_selected_line ( state );; pd->selected_line = rofi_view_get_selected_line ( state );;
@ -525,7 +511,7 @@ static void dmenu_finalize ( RofiViewState *state )
rofi_view_set_overlay ( state, NULL ); rofi_view_set_overlay ( state, NULL );
} }
} }
else if ( ( mretv & ( MENU_OK | MENU_QUICK_SWITCH ) ) && cmd_list[pd->selected_line].string != NULL ) { else if ( ( mretv & ( MENU_OK | MENU_QUICK_SWITCH ) ) && cmd_list[pd->selected_line].entry != NULL ) {
dmenu_print_results ( pd, input ); dmenu_print_results ( pd, input );
retv = TRUE; retv = TRUE;
if ( ( mretv & MENU_QUICK_SWITCH ) ) { if ( ( mretv & MENU_QUICK_SWITCH ) ) {
@ -550,7 +536,7 @@ static void dmenu_finalize ( RofiViewState *state )
// We normally do not want to restart the loop. // We normally do not want to restart the loop.
restart = FALSE; restart = FALSE;
// Normal mode // Normal mode
if ( ( mretv & MENU_OK ) && pd->selected_line != UINT32_MAX && cmd_list[pd->selected_line].string != NULL ) { if ( ( mretv & MENU_OK ) && pd->selected_line != UINT32_MAX && cmd_list[pd->selected_line].entry != NULL ) {
if ( ( mretv & MENU_CUSTOM_ACTION ) && pd->multi_select ) { if ( ( mretv & MENU_CUSTOM_ACTION ) && pd->multi_select ) {
restart = TRUE; restart = TRUE;
if ( pd->selected_list == NULL ) { if ( pd->selected_list == NULL ) {
@ -623,7 +609,7 @@ int dmenu_switcher_dialog ( void )
} }
char *input = NULL; char *input = NULL;
unsigned int cmd_list_length = pd->cmd_list_length; unsigned int cmd_list_length = pd->cmd_list_length;
DmenuModeEntry *cmd_list = pd->cmd_list; DmenuScriptEntry *cmd_list = pd->cmd_list;
pd->only_selected = FALSE; pd->only_selected = FALSE;
pd->multi_select = FALSE; pd->multi_select = FALSE;
@ -641,7 +627,7 @@ int dmenu_switcher_dialog ( void )
} }
} }
if ( config.auto_select && cmd_list_length == 1 ) { if ( config.auto_select && cmd_list_length == 1 ) {
rofi_output_formatted_line ( pd->format, cmd_list[0].string, 0, config.filter ); rofi_output_formatted_line ( pd->format, cmd_list[0].entry , 0, config.filter );
return TRUE; return TRUE;
} }
if ( find_arg ( "-password" ) >= 0 ) { if ( find_arg ( "-password" ) >= 0 ) {
@ -656,7 +642,7 @@ int dmenu_switcher_dialog ( void )
rofi_int_matcher **tokens = helper_tokenize ( select, config.case_sensitive ); rofi_int_matcher **tokens = helper_tokenize ( select, config.case_sensitive );
unsigned int i = 0; unsigned int i = 0;
for ( i = 0; i < cmd_list_length; i++ ) { for ( i = 0; i < cmd_list_length; i++ ) {
if ( helper_token_match ( tokens, cmd_list[i].string ) ) { if ( helper_token_match ( tokens, cmd_list[i].entry ) ) {
pd->selected_line = i; pd->selected_line = i;
break; break;
} }
@ -667,8 +653,8 @@ int dmenu_switcher_dialog ( void )
rofi_int_matcher **tokens = helper_tokenize ( config.filter ? config.filter : "", config.case_sensitive ); rofi_int_matcher **tokens = helper_tokenize ( config.filter ? config.filter : "", config.case_sensitive );
unsigned int i = 0; unsigned int i = 0;
for ( i = 0; i < cmd_list_length; i++ ) { for ( i = 0; i < cmd_list_length; i++ ) {
if ( tokens == NULL || helper_token_match ( tokens, cmd_list[i].string ) ) { if ( tokens == NULL || helper_token_match ( tokens, cmd_list[i].entry ) ) {
rofi_output_formatted_line ( pd->format, cmd_list[i].string, i, config.filter ); rofi_output_formatted_line ( pd->format, cmd_list[i].entry , i, config.filter );
} }
} }
helper_tokenize_free ( tokens ); helper_tokenize_free ( tokens );

36
source/dialogs/script.c
View file

@ -48,19 +48,15 @@
#include "rofi-icon-fetcher.h" #include "rofi-icon-fetcher.h"
typedef struct { #include "dialogs/dmenuscriptshared.h"
char *entry;
char *icon_name;
uint32_t icon_fetch_uid;
} ScriptEntry;
typedef struct typedef struct
{ {
/** ID of the current script. */ /** ID of the current script. */
unsigned int id; unsigned int id;
/** List of visible items. */ /** List of visible items. */
ScriptEntry *cmd_list; DmenuScriptEntry *cmd_list;
/** length list of visible items. */ /** length list of visible items. */
unsigned int cmd_list_length; unsigned int cmd_list_length;
@ -76,7 +72,10 @@ typedef struct
gboolean do_markup; gboolean do_markup;
} ScriptModePrivateData; } ScriptModePrivateData;
static void parse_entry_extras ( G_GNUC_UNUSED Mode *sw, ScriptEntry *entry, char *buffer, size_t length ) /**
* Shared function between DMENU and Script mode.
*/
void dmenuscript_parse_entry_extras ( G_GNUC_UNUSED Mode *sw, DmenuScriptEntry *entry, char *buffer, size_t length )
{ {
size_t length_key = 0;//strlen ( line ); size_t length_key = 0;//strlen ( line );
while ( length_key <= length && buffer[length_key] != '\x1f' ) { while ( length_key <= length && buffer[length_key] != '\x1f' ) {
@ -89,11 +88,12 @@ static void parse_entry_extras ( G_GNUC_UNUSED Mode *sw, ScriptEntry *entry, cha
entry->icon_name = g_strdup(value); entry->icon_name = g_strdup(value);
} }
} }
} }
/**
* End of shared functions.
*/
static void parse_header_entry ( Mode *sw, char *line, ssize_t length ) static void parse_header_entry ( Mode *sw, char *line, ssize_t length )
{ {
ScriptModePrivateData *pd = (ScriptModePrivateData *) sw->private_data; ScriptModePrivateData *pd = (ScriptModePrivateData *) sw->private_data;
@ -126,11 +126,11 @@ static void parse_header_entry ( Mode *sw, char *line, ssize_t length )
} }
} }
static ScriptEntry *get_script_output ( Mode *sw, char *command, char *arg, unsigned int *length ) static DmenuScriptEntry *get_script_output ( Mode *sw, char *command, char *arg, unsigned int *length )
{ {
int fd = -1; int fd = -1;
GError *error = NULL; GError *error = NULL;
ScriptEntry *retv = NULL; DmenuScriptEntry *retv = NULL;
char **argv = NULL; char **argv = NULL;
int argc = 0; int argc = 0;
*length = 0; *length = 0;
@ -166,14 +166,14 @@ static ScriptEntry *get_script_output ( Mode *sw, char *command, char *arg, unsi
else { else {
if ( actual_size < ( ( *length ) + 2 ) ) { if ( actual_size < ( ( *length ) + 2 ) ) {
actual_size += 256; actual_size += 256;
retv = g_realloc ( retv, ( actual_size ) * sizeof ( ScriptEntry ) ); retv = g_realloc ( retv, ( actual_size ) * sizeof ( DmenuScriptEntry ) );
} }
size_t buf_length = strlen(buffer)+1; size_t buf_length = strlen(buffer)+1;
retv[( *length )].entry = g_memdup ( buffer, buf_length); retv[( *length )].entry = g_memdup ( buffer, buf_length);
retv[( *length )].icon_name = NULL; retv[( *length )].icon_name = NULL;
retv[(*length)].icon_fetch_uid = 0; retv[(*length)].icon_fetch_uid = 0;
if ( buf_length > 0 && (read_length > (ssize_t)buf_length) ) { if ( buf_length > 0 && (read_length > (ssize_t)buf_length) ) {
parse_entry_extras ( sw, &(retv[(*length)]), buffer+buf_length, read_length-buf_length); dmenuscript_parse_entry_extras ( sw, &(retv[(*length)]), buffer+buf_length, read_length-buf_length);
} }
retv[( *length ) + 1].entry = NULL; retv[( *length ) + 1].entry = NULL;
( *length )++; ( *length )++;
@ -191,9 +191,9 @@ static ScriptEntry *get_script_output ( Mode *sw, char *command, char *arg, unsi
return retv; return retv;
} }
static ScriptEntry *execute_executor ( Mode *sw, char *result, unsigned int *length ) static DmenuScriptEntry *execute_executor ( Mode *sw, char *result, unsigned int *length )
{ {
ScriptEntry *retv = get_script_output ( sw, sw->ed, result, length ); DmenuScriptEntry *retv = get_script_output ( sw, sw->ed, result, length );
return retv; return retv;
} }
@ -238,7 +238,7 @@ static ModeMode script_mode_result ( Mode *sw, int mretv, char **input, unsigned
{ {
ScriptModePrivateData *rmpd = (ScriptModePrivateData *) sw->private_data; ScriptModePrivateData *rmpd = (ScriptModePrivateData *) sw->private_data;
ModeMode retv = MODE_EXIT; ModeMode retv = MODE_EXIT;
ScriptEntry *new_list = NULL; DmenuScriptEntry *new_list = NULL;
unsigned int new_length = 0; unsigned int new_length = 0;
if ( ( mretv & MENU_NEXT ) ) { if ( ( mretv & MENU_NEXT ) ) {
@ -324,7 +324,7 @@ static cairo_surface_t *script_get_icon ( const Mode *sw, unsigned int selected_
{ {
ScriptModePrivateData *pd = (ScriptModePrivateData *) mode_get_private_data ( sw ); ScriptModePrivateData *pd = (ScriptModePrivateData *) mode_get_private_data ( sw );
g_return_val_if_fail ( pd->cmd_list != NULL, NULL ); g_return_val_if_fail ( pd->cmd_list != NULL, NULL );
ScriptEntry *dr = &( pd->cmd_list[selected_line] ); DmenuScriptEntry *dr = &( pd->cmd_list[selected_line] );
if ( dr->icon_name == NULL ) { if ( dr->icon_name == NULL ) {
return NULL; return NULL;
} }