/** Try reading the specified function mapping */ static wint_t input_try_mapping( input_mapping_t *m) { int j, k; wint_t c=0; /* Check if the actual function code of this mapping is on the stack */ c = input_common_readch( 0 ); if( c == input_function_get_code( m->command ) ) { return input_exec_binding( m, m->seq ); } input_unreadch( c ); if( m->seq != 0 ) { for( j=0; m->seq[j] != L'\0' && m->seq[j] == (c=input_common_readch( j>0 )); j++ ) ; if( m->seq[j] == L'\0' ) { return input_exec_binding( m, m->seq ); } else { /* Return the read characters */ input_unreadch(c); for( k=j-1; k>=0; k-- ) { input_unreadch( m->seq[k] ); } } } return 0; }
/** The commandline builtin. It is used for specifying a new value for the commandline. */ static int builtin_commandline( wchar_t **argv ) { int buffer_part=0; int cut_at_cursor=0; int argc = builtin_count_args( argv ); int append_mode=0; int function_mode = 0; int tokenize = 0; int cursor_mode = 0; int line_mode = 0; int search_mode = 0; wchar_t *begin, *end; current_buffer = (wchar_t *)builtin_complete_get_temporary_buffer(); if( current_buffer ) { current_cursor_pos = wcslen( current_buffer ); } else { current_buffer = reader_get_buffer(); current_cursor_pos = reader_get_cursor_pos(); } if( !get_buffer() ) { if (is_interactive_session) { /* Prompt change requested while we don't have a prompt, most probably while reading the init files. Just ignore it. */ return 1; } sb_append( sb_err, argv[0], L": Can not set commandline in non-interactive mode\n", (void *)0 ); builtin_print_help( argv[0], sb_err ); return 1; } woptind=0; while( 1 ) { static const struct woption long_options[] = { { L"append", no_argument, 0, 'a' } , { L"insert", no_argument, 0, 'i' } , { L"replace", no_argument, 0, 'r' } , { L"current-job", no_argument, 0, 'j' } , { L"current-process", no_argument, 0, 'p' } , { L"current-token", no_argument, 0, 't' } , { L"current-buffer", no_argument, 0, 'b' } , { L"cut-at-cursor", no_argument, 0, 'c' } , { L"function", no_argument, 0, 'f' } , { L"tokenize", no_argument, 0, 'o' } , { L"help", no_argument, 0, 'h' } , { L"input", required_argument, 0, 'I' } , { L"cursor", no_argument, 0, 'C' } , { L"line", no_argument, 0, 'L' } , { L"search-mode", no_argument, 0, 'S' } , { 0, 0, 0, 0 } } ; int opt_index = 0; int opt = wgetopt_long( argc, argv, L"abijpctwforhI:CLS", long_options, &opt_index ); if( opt == -1 ) break; switch( opt ) { case 0: if(long_options[opt_index].flag != 0) break; sb_printf( sb_err, BUILTIN_ERR_UNKNOWN, argv[0], long_options[opt_index].name ); builtin_print_help( argv[0], sb_err ); return 1; case L'a': append_mode = APPEND_MODE; break; case L'b': buffer_part = STRING_MODE; break; case L'i': append_mode = INSERT_MODE; break; case L'r': append_mode = REPLACE_MODE; break; case 'c': cut_at_cursor=1; break; case 't': buffer_part = TOKEN_MODE; break; case 'j': buffer_part = JOB_MODE; break; case 'p': buffer_part = PROCESS_MODE; break; case 'f': function_mode=1; break; case 'o': tokenize=1; break; case 'I': current_buffer = woptarg; current_cursor_pos = wcslen( woptarg ); break; case 'C': cursor_mode = 1; break; case 'L': line_mode = 1; break; case 'S': search_mode = 1; break; case 'h': builtin_print_help( argv[0], sb_out ); return 0; case L'?': builtin_unknown_option( argv[0], argv[woptind-1] ); return 1; } } if( function_mode ) { int i; /* Check for invalid switch combinations */ if( buffer_part || cut_at_cursor || append_mode || tokenize || cursor_mode || line_mode || search_mode ) { sb_printf(sb_err, BUILTIN_ERR_COMBO, argv[0] ); builtin_print_help( argv[0], sb_err ); return 1; } if( argc == woptind ) { sb_printf( sb_err, BUILTIN_ERR_MISSING, argv[0] ); builtin_print_help( argv[0], sb_err ); return 1; } for( i=woptind; i<argc; i++ ) { wint_t c = input_function_get_code( argv[i] ); if( c != -1 ) { /* input_unreadch inserts the specified keypress or readline function at the top of the stack of unused keypresses */ input_unreadch(c); } else { sb_printf( sb_err, _(L"%ls: Unknown input function '%ls'\n"), argv[0], argv[i] ); builtin_print_help( argv[0], sb_err ); return 1; } } return 0; } /* Check for invalid switch combinations */ if( (search_mode || line_mode || cursor_mode) && (argc-woptind > 1) ) { sb_append( sb_err, argv[0], L": Too many arguments\n", (void *)0 ); builtin_print_help( argv[0], sb_err ); return 1; } if( (buffer_part || tokenize || cut_at_cursor) && (cursor_mode || line_mode || search_mode) ) { sb_printf( sb_err, BUILTIN_ERR_COMBO, argv[0] ); builtin_print_help( argv[0], sb_err ); return 1; } if( (tokenize || cut_at_cursor) && (argc-woptind) ) { sb_printf( sb_err, BUILTIN_ERR_COMBO2, argv[0], L"--cut-at-cursor and --tokenize can not be used when setting the commandline" ); builtin_print_help( argv[0], sb_err ); return 1; } if( append_mode && !(argc-woptind) ) { sb_printf( sb_err, BUILTIN_ERR_COMBO2, argv[0], L"insertion mode switches can not be used when not in insertion mode" ); builtin_print_help( argv[0], sb_err ); return 1; } /* Set default modes */ if( !append_mode ) { append_mode = REPLACE_MODE; } if( !buffer_part ) { buffer_part = STRING_MODE; } if( cursor_mode ) { if( argc-woptind ) { wchar_t *endptr; int new_pos; errno = 0; new_pos = wcstol( argv[woptind], &endptr, 10 ); if( *endptr || errno ) { sb_printf( sb_err, BUILTIN_ERR_NOT_NUMBER, argv[0], argv[woptind] ); builtin_print_help( argv[0], sb_err ); } current_buffer = reader_get_buffer(); new_pos = maxi( 0, mini( new_pos, wcslen( current_buffer ) ) ); reader_set_buffer( current_buffer, new_pos ); return 0; } else { sb_printf( sb_out, L"%d\n", reader_get_cursor_pos() ); return 0; } } if( line_mode ) { int pos = reader_get_cursor_pos(); wchar_t *buff = reader_get_buffer(); sb_printf( sb_out, L"%d\n", parse_util_lineno( buff, pos ) ); return 0; } if( search_mode ) { return !reader_search_mode(); } switch( buffer_part ) { case STRING_MODE: { begin = get_buffer(); end = begin+wcslen(begin); break; } case PROCESS_MODE: { parse_util_process_extent( get_buffer(), get_cursor_pos(), &begin, &end ); break; } case JOB_MODE: { parse_util_job_extent( get_buffer(), get_cursor_pos(), &begin, &end ); break; } case TOKEN_MODE: { parse_util_token_extent( get_buffer(), get_cursor_pos(), &begin, &end, 0, 0 ); break; } } switch(argc-woptind) { case 0: { write_part( begin, end, cut_at_cursor, tokenize ); break; } case 1: { replace_part( begin, end, argv[woptind], append_mode ); break; } default: { string_buffer_t sb; int i; sb_init( &sb ); sb_append( &sb, argv[woptind] ); for( i=woptind+1; i<argc; i++ ) { sb_append( &sb, L"\n" ); sb_append( &sb, argv[i] ); } replace_part( begin, end, (wchar_t *)sb.buff, append_mode ); sb_destroy( &sb ); break; } } return 0; }
/** The commandline builtin. It is used for specifying a new value for the commandline. */ static int builtin_commandline(parser_t &parser, wchar_t **argv) { int buffer_part=0; int cut_at_cursor=0; int argc = builtin_count_args(argv); int append_mode=0; int function_mode = 0; int selection_mode = 0; int tokenize = 0; int cursor_mode = 0; int line_mode = 0; int search_mode = 0; int paging_mode = 0; const wchar_t *begin = NULL, *end = NULL; scoped_push<const wchar_t *> saved_current_buffer(¤t_buffer); scoped_push<size_t> saved_current_cursor_pos(¤t_cursor_pos); wcstring transient_commandline; if (get_top_transient(&transient_commandline)) { current_buffer = transient_commandline.c_str(); current_cursor_pos = transient_commandline.size(); } else { current_buffer = reader_get_buffer(); current_cursor_pos = reader_get_cursor_pos(); } if (!get_buffer()) { if (is_interactive_session) { /* Prompt change requested while we don't have a prompt, most probably while reading the init files. Just ignore it. */ return 1; } stderr_buffer.append(argv[0]); stderr_buffer.append(L": Can not set commandline in non-interactive mode\n"); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } woptind=0; while (1) { static const struct woption long_options[] = { { L"append", no_argument, 0, 'a' }, { L"insert", no_argument, 0, 'i' }, { L"replace", no_argument, 0, 'r' }, { L"current-job", no_argument, 0, 'j' }, { L"current-process", no_argument, 0, 'p' }, { L"current-token", no_argument, 0, 't' }, { L"current-buffer", no_argument, 0, 'b' }, { L"cut-at-cursor", no_argument, 0, 'c' }, { L"function", no_argument, 0, 'f' }, { L"tokenize", no_argument, 0, 'o' }, { L"help", no_argument, 0, 'h' }, { L"input", required_argument, 0, 'I' }, { L"cursor", no_argument, 0, 'C' }, { L"line", no_argument, 0, 'L' }, { L"search-mode", no_argument, 0, 'S' }, { L"selection", no_argument, 0, 's' }, { L"paging-mode", no_argument, 0, 'P' }, { 0, 0, 0, 0 } }; int opt_index = 0; int opt = wgetopt_long(argc, argv, L"abijpctwforhI:CLSsP", long_options, &opt_index); if (opt == -1) break; switch (opt) { case 0: if (long_options[opt_index].flag != 0) break; append_format(stderr_buffer, BUILTIN_ERR_UNKNOWN, argv[0], long_options[opt_index].name); builtin_print_help(parser, argv[0], stderr_buffer); return 1; case L'a': append_mode = APPEND_MODE; break; case L'b': buffer_part = STRING_MODE; break; case L'i': append_mode = INSERT_MODE; break; case L'r': append_mode = REPLACE_MODE; break; case 'c': cut_at_cursor=1; break; case 't': buffer_part = TOKEN_MODE; break; case 'j': buffer_part = JOB_MODE; break; case 'p': buffer_part = PROCESS_MODE; break; case 'f': function_mode=1; break; case 'o': tokenize=1; break; case 'I': current_buffer = woptarg; current_cursor_pos = wcslen(woptarg); break; case 'C': cursor_mode = 1; break; case 'L': line_mode = 1; break; case 'S': search_mode = 1; break; case 's': selection_mode = 1; break; case 'P': paging_mode = 1; break; case 'h': builtin_print_help(parser, argv[0], stdout_buffer); return 0; case L'?': builtin_unknown_option(parser, argv[0], argv[woptind-1]); return 1; } } if (function_mode) { int i; /* Check for invalid switch combinations */ if (buffer_part || cut_at_cursor || append_mode || tokenize || cursor_mode || line_mode || search_mode || paging_mode) { append_format(stderr_buffer, BUILTIN_ERR_COMBO, argv[0]); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } if (argc == woptind) { append_format(stderr_buffer, BUILTIN_ERR_MISSING, argv[0]); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } for (i=woptind; i<argc; i++) { wchar_t c = input_function_get_code(argv[i]); if (c != (wchar_t)(-1)) { /* input_unreadch inserts the specified keypress or readline function at the top of the stack of unused keypresses */ input_unreadch(c); } else { append_format(stderr_buffer, _(L"%ls: Unknown input function '%ls'\n"), argv[0], argv[i]); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } } return 0; } if (selection_mode) { size_t start, len; const wchar_t *buffer = reader_get_buffer(); if (reader_get_selection(&start, &len)) { wchar_t *selection = new wchar_t[len + 1]; selection[len] = L'\0'; selection = wcsncpy(selection, buffer + start, len); append_format(stdout_buffer, selection); delete selection; } else { append_format(stdout_buffer, L""); } return 0; } /* Check for invalid switch combinations */ if ((search_mode || line_mode || cursor_mode || paging_mode) && (argc-woptind > 1)) { append_format(stderr_buffer, argv[0], L": Too many arguments\n", NULL); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } if ((buffer_part || tokenize || cut_at_cursor) && (cursor_mode || line_mode || search_mode || paging_mode)) { append_format(stderr_buffer, BUILTIN_ERR_COMBO, argv[0]); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } if ((tokenize || cut_at_cursor) && (argc-woptind)) { append_format(stderr_buffer, BUILTIN_ERR_COMBO2, argv[0], L"--cut-at-cursor and --tokenize can not be used when setting the commandline"); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } if (append_mode && !(argc-woptind)) { append_format(stderr_buffer, BUILTIN_ERR_COMBO2, argv[0], L"insertion mode switches can not be used when not in insertion mode"); builtin_print_help(parser, argv[0], stderr_buffer); return 1; } /* Set default modes */ if (!append_mode) { append_mode = REPLACE_MODE; } if (!buffer_part) { buffer_part = STRING_MODE; } if (cursor_mode) { if (argc-woptind) { wchar_t *endptr; long new_pos; errno = 0; new_pos = wcstol(argv[woptind], &endptr, 10); if (*endptr || errno) { append_format(stderr_buffer, BUILTIN_ERR_NOT_NUMBER, argv[0], argv[woptind]); builtin_print_help(parser, argv[0], stderr_buffer); } current_buffer = reader_get_buffer(); new_pos = maxi(0L, mini(new_pos, (long)wcslen(current_buffer))); reader_set_buffer(current_buffer, (size_t)new_pos); return 0; } else { append_format(stdout_buffer, L"%lu\n", (unsigned long)reader_get_cursor_pos()); return 0; } } if (line_mode) { size_t pos = reader_get_cursor_pos(); const wchar_t *buff = reader_get_buffer(); append_format(stdout_buffer, L"%lu\n", (unsigned long)parse_util_lineno(buff, pos)); return 0; } if (search_mode) { return ! reader_search_mode(); } if (paging_mode) { return ! reader_has_pager_contents(); } switch (buffer_part) { case STRING_MODE: { begin = get_buffer(); end = begin+wcslen(begin); break; } case PROCESS_MODE: { parse_util_process_extent(get_buffer(), get_cursor_pos(), &begin, &end); break; } case JOB_MODE: { parse_util_job_extent(get_buffer(), get_cursor_pos(), &begin, &end); break; } case TOKEN_MODE: { parse_util_token_extent(get_buffer(), get_cursor_pos(), &begin, &end, 0, 0); break; } } switch (argc-woptind) { case 0: { write_part(begin, end, cut_at_cursor, tokenize); break; } case 1: { replace_part(begin, end, argv[woptind], append_mode); break; } default: { wcstring sb = argv[woptind]; int i; for (i=woptind+1; i<argc; i++) { sb.push_back(L'\n'); sb.append(argv[i]); } replace_part(begin, end, sb.c_str(), append_mode); break; } } return 0; }