4coder/4coder_search.cpp

974 lines
34 KiB
C++

/*
4coder_search.cpp - Commands that search accross buffers including word complete,
and list all locations.
*/
// TOP
//
// Search Iteration Systems
//
static void
search_key_alloc(Heap *heap, Search_Key *key, umem *size, i32 count){
count = clamp_top(count, ArrayCount(key->words));
umem min_size = 0x7FFFFFFF;
umem total_size = 0;
for (i32 i = 0; i < count; ++i){
total_size += size[i];
if (min_size > size[i]){
min_size = size[i];
}
}
if (key->base == 0){
umem max_base_size = total_size*2;
key->base = heap_array(heap, u8, (i32)max_base_size);
key->base_size = max_base_size;
}
else if (key->base_size < total_size){
umem max_base_size = total_size*2;
heap_free(heap, key->base);
key->base = heap_array(heap, u8, (i32)max_base_size);
key->base_size = max_base_size;
}
u8 *char_ptr = key->base;
for (i32 i = 0; i < count; ++i){
key->words[i].str = char_ptr;
key->words[i].size = 0;
#if 0
key->words[i].str = char_ptr;
key->words[i].size = 0;
key->words[i].memory_size = size[i];
#endif
char_ptr += size[i];
}
key->count = count;
key->min_size = min_size;
}
static void
search_iter_init(Search_Iter *iter, Search_Key key){
iter->key = key;
iter->i = 0;
iter->range_initialized = 0;
}
static void
search_set_init(Heap *heap, Search_Set *set, i32 range_count){
if (set->ranges == 0){
i32 max = range_count*2;
set->ranges = heap_array(heap, Search_Range, max);
set->max = max;
}
else if (set->max < range_count){
i32 max = range_count*2;
heap_free(heap, set->ranges);
set->ranges = heap_array(heap, Search_Range, max);
set->max = max;
}
set->count = range_count;
}
static void
search_hits_table_alloc(Heap *heap, Table *hits, i32 table_size){
void *mem = 0;
i32 mem_size = table_required_mem_size(table_size, sizeof(Offset_String));
if (hits->hash_array == 0){
mem = heap_allocate(heap, mem_size);
}
else{
heap_free(heap, hits->hash_array);
mem = heap_allocate(heap, mem_size);
}
table_init_memory(hits, mem, table_size, sizeof(Offset_String));
}
static void
search_hits_init(Heap *heap, Table *hits, String_Space *str, i32 table_size, i32 str_size){
if (hits->hash_array == 0){
search_hits_table_alloc(heap, hits, table_size);
}
else{
i32 mem_size = table_required_mem_size(table_size, sizeof(Offset_String));
heap_free(heap, hits->hash_array);
void *mem = heap_allocate(heap, mem_size);
table_init_memory(hits, mem, table_size, sizeof(Offset_String));
}
if (str->space == 0){
str->space = heap_array(heap, char, str_size);
str->max = str_size;
}
else if (str->max < str_size){
heap_free(heap, str->space);
str->space = heap_array(heap, char, str_size);
str->max = str_size;
}
str->pos = str->new_pos = 0;
table_clear(hits);
}
//
// Table Operations
//
static b32
search_hit_add(Heap *heap, Table *hits, String_Space *space, char *str, i32 len){
b32 result = false;
Assert(len != 0);
Offset_String ostring = strspace_append(space, str, len);
if (ostring.size == 0){
i32 new_size = space->max*2;
if (new_size < space->max + len){
new_size = space->max + len;
}
char *new_space = heap_array(heap, char, new_size);
memcpy(new_space, space->space, space->new_pos);
heap_free(heap, space->space);
space->space = new_space;
ostring = strspace_append(space, str, len);
}
Assert(ostring.size != 0);
if (table_at_capacity(hits)){
Table new_hits = {};
search_hits_table_alloc(heap, &new_hits, hits->max*2);
table_clear(&new_hits);
table_rehash(hits, &new_hits, space->space, tbl_offset_string_hash, tbl_offset_string_compare);
heap_free(heap, hits->hash_array);
*hits = new_hits;
}
if (!table_add(hits, &ostring, space->space, tbl_offset_string_hash, tbl_offset_string_compare)){
result = true;
strspace_keep_prev(space);
}
else{
strspace_discard_prev(space);
}
return(result);
}
static b32
search_hit_add(Heap *heap, Table *hits, String_Space *space, String_Const_u8 string){
return(search_hit_add(heap, hits, space, (char*)string.str, (i32)string.size));
}
//
// Search Key Checking
//
static void
seek_potential_match(Application_Links *app, Search_Range *range, Search_Key key, Search_Match *result, Seek_Potential_Match_Direction direction, i32 start_pos, i32 end_pos){
b32 case_insensitive = ((range->flags & SearchFlag_CaseInsensitive) != 0);
b32 forward = (direction == SeekPotentialMatch_Forward);
#define OptFlag(b,f) ((b)?(f):(0))
Buffer_Seek_String_Flags flags = 0
| OptFlag(case_insensitive, BufferSeekString_CaseInsensitive)
| OptFlag(!forward, BufferSeekString_Backward);
result->buffer = range->buffer;
i32 best_pos = -1;
if (forward){
best_pos = end_pos;
}
for (i32 i = 0; i < key.count; ++i){
String_Const_u8 word = key.words[i];
i32 new_pos = -1;
buffer_seek_string(app, result->buffer, start_pos, end_pos, range->start, word, &new_pos, flags);
if (new_pos >= 0){
if (forward){
if (new_pos < best_pos){
best_pos = new_pos;
}
}
else{
if (new_pos > best_pos){
best_pos = new_pos;
}
}
}
}
result->start = best_pos;
}
static i32
buffer_seek_alpha_numeric_end(Application_Links *app, Buffer_ID buffer_id, i32 pos){
char space[1024];
Stream_Chunk chunk = {};
if (init_stream_chunk(&chunk, app, buffer_id, pos, space, sizeof(space))){
i32 still_looping = true;
do{
for (; pos < chunk.end; ++pos){
u8 at_pos = (u8)chunk.data[pos];
if (!character_is_alpha_numeric_unicode(at_pos)) goto double_break;
}
still_looping = forward_stream_chunk(&chunk);
}while(still_looping);
}
double_break:;
return(pos);
}
static i32
match_check(Application_Links *app, Search_Range *range, i32 *pos, Search_Match *result_ptr, Search_Key key){
i32 result_code = FindResult_None;
Search_Match result = *result_ptr;
i32 end_pos = range->start + range->size;
i32 type = (range->flags & SearchFlag_MatchMask);
result.match_word_index = -1;
for (i32 i = 0; i < key.count; ++i){
String_Const_u8 word = key.words[i];
i32 found_match = FindResult_None;
switch (type){
case SearchFlag_MatchWholeWord:
{
u8 prev = ' ';
if (character_is_alpha_numeric(string_get_character(word, 0))){
prev = buffer_get_char(app, result.buffer, result.start - 1);
}
if (!character_is_alpha_numeric(prev)){
result.end = result.start + (i32)word.size;
if (result.end <= end_pos){
u8 next = ' ';
if (character_is_alpha_numeric_unicode(string_get_character(word, word.size - 1))){
next = buffer_get_char(app, result.buffer, result.end);
}
if (!character_is_alpha_numeric_unicode(next)){
result.found_match = true;
found_match = FindResult_FoundMatch;
}
}
else{
found_match = FindResult_PastEnd;
}
}
}break;
case SearchFlag_MatchWordPrefix:
{
u8 prev = buffer_get_char(app, result.buffer, result.start - 1);
if (!character_is_alpha_numeric_unicode(prev)){
result.end = buffer_seek_alpha_numeric_end(app, result.buffer, result.start);
if (result.end <= end_pos){
result.found_match = true;
found_match = FindResult_FoundMatch;
}
else{
found_match = FindResult_PastEnd;
}
}
}break;
case SearchFlag_MatchSubstring:
{
result.end = result.start + (i32)word.size;
if (result.end <= end_pos){
result.found_match = true;
found_match = FindResult_FoundMatch;
}
else{
found_match = FindResult_PastEnd;
}
}break;
}
if (found_match == FindResult_FoundMatch){
result_code = FindResult_FoundMatch;
result.match_word_index = i;
break;
}
else if (found_match == FindResult_PastEnd){
result_code = FindResult_PastEnd;
}
}
*result_ptr = result;
return(result_code);
}
//
// Find Next Match
//
static i32
search_front_to_back(Application_Links *app, Search_Range *range, Search_Key key, i32 *pos, Search_Match *result){
i32 found_match = FindResult_None;
for (;found_match == FindResult_None;){
found_match = FindResult_None;
i32 end_pos = range->start + range->size;
if (*pos + key.min_size < end_pos){
i32 start_pos = *pos;
if (start_pos < range->start){
start_pos = range->start;
}
seek_potential_match(app, range, key, result, SeekPotentialMatch_Forward, start_pos, end_pos);
if (result->start < end_pos){
*pos = result->start + 1;
found_match = match_check(app, range, pos, result, key);
if (found_match == FindResult_FoundMatch){
*pos = result->end;
}
}
else{
found_match = FindResult_PastEnd;
*pos = end_pos + 1;
}
}
else{
found_match = FindResult_PastEnd;
*pos = end_pos + 1;
}
}
return(found_match);
}
static i32
search_back_to_front(Application_Links *app, Search_Range *range, Search_Key key, i32 *pos, Search_Match *result){
i32 found_match = FindResult_None;
for (;found_match == FindResult_None;){
found_match = FindResult_None;
if (*pos > range->start){
i32 start_pos = *pos;
seek_potential_match(app, range, key, result, SeekPotentialMatch_Backward, start_pos, 0);
if (result->start >= range->start){
*pos = result->start - 1;
found_match = match_check(app, range, pos, result, key);
if (found_match == FindResult_FoundMatch){
*pos = result->start - (i32)key.words[result->match_word_index].size;
}
}
else{
found_match = FindResult_PastEnd;
}
}
else{
found_match = FindResult_PastEnd;
}
}
return(found_match);
}
static void
search_iter_next_range(Search_Iter *it){
++it->i;
it->pos = 0;
it->back_pos = 0;
it->range_initialized = 0;
}
static Search_Match
search_next_match(Application_Links *app, Search_Set *set, Search_Iter *it_ptr){
Search_Match result = {};
Search_Iter iter = *it_ptr;
i32 count = set->count;
for (;iter.i < count;){
Search_Range *range = set->ranges + iter.i;
i32 find_result = FindResult_None;
if (!iter.range_initialized){
iter.range_initialized = true;
switch (range->type){
case SearchRange_BackToFront:
{
iter.back_pos = range->start + range->size-1;
}break;
case SearchRange_Wave:
{
iter.back_pos = range->mid_start-1;
iter.pos = range->mid_start + range->mid_size;
}break;
}
}
switch (range->type){
case SearchRange_FrontToBack:
{
find_result = search_front_to_back(app, range, iter.key, &iter.pos, &result);
}break;
case SearchRange_BackToFront:
{
find_result = search_back_to_front(app, range, iter.key, &iter.back_pos, &result);
}break;
case SearchRange_Wave:
{
Search_Match forward_match = {};
Search_Match backward_match = {};
i32 forward_result = FindResult_PastEnd;
i32 backward_result = FindResult_PastEnd;
if (iter.pos < range->start + range->size){
forward_result = search_front_to_back(app, range, iter.key, &iter.pos, &forward_match);
}
if (iter.back_pos > range->start){
backward_result = search_back_to_front(app, range, iter.key, &iter.back_pos, &backward_match);
}
if (forward_result == FindResult_FoundMatch){
if (backward_result == FindResult_FoundMatch){
find_result = FindResult_FoundMatch;
i32 forward_start = range->mid_start + range->mid_size;
i32 forward_distance = (forward_match.start - forward_start);
i32 backward_distance = (range->mid_start - backward_match.end);
if (backward_distance < forward_distance){
iter.pos = forward_match.start;
result = backward_match;
}
else{
iter.back_pos = backward_match.start;
result = forward_match;
}
}
else{
find_result = FindResult_FoundMatch;
result = forward_match;
}
}
else{
if (backward_result == FindResult_FoundMatch){
find_result = FindResult_FoundMatch;
result = backward_match;
--iter.pos;
}
else{
find_result = FindResult_PastEnd;
}
}
}break;
}
if (find_result == FindResult_FoundMatch){
goto double_break;
}
else if (find_result == FindResult_PastEnd){
search_iter_next_range(&iter);
}
}
double_break:;
*it_ptr = iter;
return(result);
}
//
// Generic Search All Buffers
//
static void
initialize_generic_search_all_buffers(Application_Links *app, Heap *heap, String_Const_u8 *strings, i32 count, Search_Range_Flag match_flags, Buffer_ID *skip_buffers, i32 skip_buffer_count, Search_Set *set, Search_Iter *iter){
block_zero_struct(set);
block_zero_struct(iter);
Search_Key key = {};
umem sizes[ArrayCount(key.words)] = {};
count = clamp_top(count, ArrayCount(key.words));
for (i32 i = 0; i < count; ++i){
sizes[i] = strings[i].size;
}
// TODO(allen): Why on earth am I allocating these separately in this case? Upgrade to just use the string array on the stack!
// NOTE(allen): ?? What did that TODO mean!?
search_key_alloc(heap, &key, sizes, count);
for (i32 i = 0; i < count; ++i){
key.words[i].size = strings[i].size;
block_copy(key.words[i].str, strings[i].str, strings[i].size);
}
search_iter_init(iter, key);
i32 buffer_count = get_buffer_count(app);
search_set_init(heap, set, buffer_count);
Search_Range *ranges = set->ranges;
View_ID view = 0;
get_active_view(app, AccessProtected, &view);
Buffer_ID buffer = 0;
view_get_buffer(app, view, AccessProtected, &buffer);
i32 j = 0;
if (buffer_exists(app, buffer)){
b32 skip = false;
for (i32 i = 0; i < skip_buffer_count; ++i){
if (buffer == skip_buffers[i]){
skip = true;
break;
}
}
if (!skip){
ranges[j].type = SearchRange_FrontToBack;
ranges[j].flags = match_flags;
ranges[j].buffer = buffer;
ranges[j].start = 0;
buffer_get_size(app, buffer, &ranges[j].size);
++j;
}
}
Buffer_ID buffer_it = 0;
for (get_buffer_next(app, 0, AccessAll, &buffer_it);
buffer_exists(app, buffer_it);
get_buffer_next(app, buffer_it, AccessAll, &buffer_it)){
if (buffer_it == buffer){
continue;
}
b32 skip = false;
for (i32 i = 0; i < skip_buffer_count; ++i){
if (buffer_it == skip_buffers[i]){
skip = true;
break;
}
}
if (!skip){
if (!buffer_has_name_with_star(app, buffer_it)){
ranges[j].type = SearchRange_FrontToBack;
ranges[j].flags = match_flags;
ranges[j].buffer = buffer_it;
ranges[j].start = 0;
buffer_get_size(app, buffer_it, &ranges[j].size);
++j;
}
}
}
set->count = j;
}
////////////////////////////////
static String_Const_u8 search_name = string_u8_litexpr("*search*");
static void
list__parameters_buffer(Application_Links *app, Heap *heap,
String_Const_u8 *strings, i32 count, Search_Range_Flag match_flags,
Buffer_ID search_buffer_id){
Arena *scratch = context_get_arena(app);
// Setup the search buffer for 'init' mode
buffer_set_setting(app, search_buffer_id, BufferSetting_ReadOnly, true);
buffer_set_setting(app, search_buffer_id, BufferSetting_RecordsHistory, false);
// Initialize a generic search all buffers
Search_Set set = {};
Search_Iter iter = {};
initialize_generic_search_all_buffers(app, heap, strings, count, match_flags, &search_buffer_id, 1, &set, &iter);
// List all locations into search buffer
Temp_Memory all_temp = begin_temp(scratch);
// Setup buffered output
Cursor buffering_cursor = make_cursor(push_array(scratch, u8, KB(64)), KB(64));
Buffer_Insertion out = begin_buffer_insertion_at_buffered(app, search_buffer_id, 0, &buffering_cursor);
Buffer_ID prev_match_id = 0;
b32 no_matches = true;
for (Search_Match match = search_next_match(app, &set, &iter);
match.found_match;
match = search_next_match(app, &set, &iter)){
Partial_Cursor word_pos = {};
if (buffer_compute_cursor(app, match.buffer, seek_pos(match.start), &word_pos)){
if (prev_match_id != match.buffer){
if (prev_match_id != 0){
insertc(&out, '\n');
}
prev_match_id = match.buffer;
}
Temp_Memory line_temp = begin_temp(scratch);
String_Const_u8 file_name = push_buffer_file_name(app, scratch, match.buffer);
String_Const_u8 full_line_str = push_buffer_line(app, scratch, match.buffer, word_pos.line);
if (full_line_str.size > 0){
String_Const_u8 line_str = string_skip_chop_whitespace(full_line_str);
insertf(&out, "%.*s:%d:%d: %.*s\n", string_expand(file_name), word_pos.line, word_pos.character, string_expand(line_str));
}
end_temp(line_temp);
no_matches = false;
}
}
if (no_matches){
insert_string(&out, string_u8_litexpr("no matches\n"));
}
end_buffer_insertion(&out);
end_temp(all_temp);
// Lock *search* as the jump buffer
lock_jump_buffer(search_name);
}
static void
list__parameters(Application_Links *app, Heap *heap, String_Const_u8 *strings, i32 count, Search_Range_Flag match_flags, View_ID default_target_view){
// Open the search buffer
Buffer_ID search_buffer_id = create_or_switch_to_buffer_and_clear_by_name(app, search_name, default_target_view);
list__parameters_buffer(app, heap, strings, count, match_flags, search_buffer_id);
}
static void
list_single__parameters(Application_Links *app, Heap *heap, String_Const_u8 str, b32 substrings, b32 case_insensitive, View_ID default_target_view){
Search_Range_Flag flags = 0;
if (substrings){
flags |= SearchFlag_MatchSubstring;
}
else{
flags |= SearchFlag_MatchWholeWord;
}
if (case_insensitive){
flags |= SearchFlag_CaseInsensitive;
}
list__parameters(app, heap, &str, 1, flags, default_target_view);
}
static void
list_query__parameters(Application_Links *app, Heap *heap, b32 substrings, b32 case_insensitive, View_ID default_target_view){
char space[1024];
String_Const_u8 str = get_query_string(app, "List Locations For: ", space, sizeof(space));
if (str.size > 0){
list_single__parameters(app, heap, str, substrings, case_insensitive, default_target_view);
}
}
static void
list_identifier__parameters(Application_Links *app, Heap *heap, b32 substrings, b32 case_insensitive, View_ID default_target_view){
View_ID view = 0;
get_active_view(app, AccessProtected, &view);
Buffer_ID buffer = 0;
view_get_buffer(app, view, AccessProtected, &buffer);
if (buffer != 0){
i32 pos = 0;
view_get_cursor_pos(app, view, &pos);
Scratch_Block scratch(app);
String_Const_u8 str = get_token_or_word_under_pos(app, scratch, buffer, pos);
if (str.size > 0){
list_single__parameters(app, heap, str, substrings, case_insensitive, default_target_view);
}
}
}
static void
list_selected_range__parameters(Application_Links *app, Heap *heap, b32 substrings, b32 case_insensitive, View_ID default_target_view){
View_ID view = 0;
get_active_view(app, AccessProtected, &view);
Arena *scratch = context_get_arena(app);
Temp_Memory temp = begin_temp(scratch);
String_Const_u8 str = push_string_in_view_range(app, scratch, view);
if (str.size > 0){
list_single__parameters(app, heap, str, substrings, case_insensitive, default_target_view);
}
end_temp(temp);
}
static void
list_type_definition__parameters(Application_Links *app, Heap *heap, String_Const_u8 str, View_ID default_target_view){
Arena *scratch = context_get_arena(app);
Temp_Memory temp = begin_temp(scratch);
String_Const_u8 match_strings[9];
i32 i = 0;
match_strings[i++] = (string_u8_pushf(scratch, "struct %.*s{" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "struct %.*s\n{", string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "struct %.*s {" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "union %.*s{" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "union %.*s\n{" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "union %.*s {" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "enum %.*s{" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "enum %.*s\n{" , string_expand(str)));
match_strings[i++] = (string_u8_pushf(scratch, "enum %.*s {" , string_expand(str)));
list__parameters(app, heap, match_strings, ArrayCount(match_strings), 0, default_target_view);
end_temp(temp);
}
////////////////////////////////
CUSTOM_COMMAND_SIG(list_all_locations)
CUSTOM_DOC("Queries the user for a string and lists all exact case-sensitive matches found in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_query__parameters(app, &global_heap, false, false, target_view);
}
CUSTOM_COMMAND_SIG(list_all_substring_locations)
CUSTOM_DOC("Queries the user for a string and lists all case-sensitive substring matches found in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_query__parameters(app, &global_heap, true, false, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_case_insensitive)
CUSTOM_DOC("Queries the user for a string and lists all exact case-insensitive matches found in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_query__parameters(app, &global_heap, false, true, target_view);
}
CUSTOM_COMMAND_SIG(list_all_substring_locations_case_insensitive)
CUSTOM_DOC("Queries the user for a string and lists all case-insensitive substring matches found in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_query__parameters(app, &global_heap, true, true, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_of_identifier)
CUSTOM_DOC("Reads a token or word under the cursor and lists all exact case-sensitive mathces in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_identifier__parameters(app, &global_heap, false, false, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_of_identifier_case_insensitive)
CUSTOM_DOC("Reads a token or word under the cursor and lists all exact case-insensitive mathces in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_identifier__parameters(app, &global_heap, false, true, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_of_selection)
CUSTOM_DOC("Reads the string in the selected range and lists all exact case-sensitive mathces in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_selected_range__parameters(app, &global_heap, false, false, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_of_selection_case_insensitive)
CUSTOM_DOC("Reads the string in the selected range and lists all exact case-insensitive mathces in all open buffers.")
{
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_selected_range__parameters(app, &global_heap, false, true, target_view);
}
CUSTOM_COMMAND_SIG(list_all_locations_of_type_definition)
CUSTOM_DOC("Queries user for string, lists all locations of strings that appear to define a type whose name matches the input string.")
{
char space[1024];
String_Const_u8 str = get_query_string(app, "List Definitions For: ", space, sizeof(space));
if (str.size > 0){
View_ID target_view = get_next_view_after_active(app, AccessAll);
list_type_definition__parameters(app, &global_heap, str, target_view);
}
}
CUSTOM_COMMAND_SIG(list_all_locations_of_type_definition_of_identifier)
CUSTOM_DOC("Reads a token or word under the cursor and lists all locations of strings that appear to define a type whose name matches it.")
{
View_ID target_view = 0;
get_active_view(app, AccessProtected, &target_view);
Buffer_ID buffer = 0;
view_get_buffer(app, target_view, AccessProtected, &buffer);
i32 pos = 0;
view_get_cursor_pos(app, target_view, &pos);
Scratch_Block scratch(app);
String_Const_u8 str = get_token_or_word_under_pos(app, scratch, buffer, pos);
if (str.size > 0){
target_view = get_next_view_after_active(app, AccessAll);
list_type_definition__parameters(app, &global_heap, str, target_view);
}
}
//
// Word Complete Command
//
static Word_Complete_State complete_state = {};
CUSTOM_COMMAND_SIG(word_complete)
CUSTOM_DOC("Iteratively tries completing the word to the left of the cursor with other words in open buffers that have the same prefix string.")
{
View_ID view = 0;
get_active_view(app, AccessOpen, &view);
Buffer_ID buffer = 0;
if (view_get_buffer(app, view, AccessOpen, &buffer)){
i32 do_init = false;
Managed_Scope scope = 0;
view_get_managed_scope(app, view, &scope);
u64 rewrite = 0;
managed_variable_get(app, scope, view_rewrite_loc, &rewrite);
if (rewrite != RewriteWordComplete){
do_init = true;
}
managed_variable_set(app, scope, view_next_rewrite_loc, RewriteWordComplete);
if (!complete_state.initialized){
do_init = true;
}
i32 word_end = 0;
i32 word_start = 0;
i32 cursor_pos = 0;
umem size = 0;
if (do_init){
// NOTE(allen): Get the range where the
// partial word is written.
word_end = 0;
view_get_cursor_pos(app, view, &word_end);
word_start = word_end;
cursor_pos = word_end - 1;
char space[1024];
Stream_Chunk chunk = {};
if (init_stream_chunk(&chunk, app, buffer, cursor_pos, space, sizeof(space))){
i32 still_looping = true;
do{
for (; cursor_pos >= chunk.start; --cursor_pos){
u8 c = chunk.data[cursor_pos];
if (character_is_alpha_unicode(c)){
word_start = cursor_pos;
}
else if (!character_is_base10(c)){
goto double_break;
}
}
still_looping = backward_stream_chunk(&chunk);
}while(still_looping);
}
double_break:;
size = word_end - word_start;
if (size == 0){
complete_state.initialized = false;
return;
}
// NOTE(allen): Initialize the search iterator with the partial word.
complete_state.initialized = true;
Search_Key key = {};
search_key_alloc(&global_heap, &key, &size, 1);
buffer_read_range(app, buffer, word_start, word_end, (char*)key.words[0].str);
key.words[0].size = size;
search_iter_init(&complete_state.iter, key);
// NOTE(allen): Initialize the set of ranges to be searched.
i32 buffer_count = get_buffer_count(app);
search_set_init(&global_heap, &complete_state.set, buffer_count);
Search_Range *ranges = complete_state.set.ranges;
ranges[0].type = SearchRange_Wave;
ranges[0].flags = SearchFlag_MatchWordPrefix;
ranges[0].buffer = buffer;
ranges[0].start = 0;
buffer_get_size(app, buffer, &ranges[0].size);
ranges[0].mid_start = word_start;
ranges[0].mid_size = (i32)size;
Buffer_ID buffer_it = 0;
i32 j = 1;
for (get_buffer_next(app, 0, AccessAll, &buffer_it);
buffer_it != 0;
get_buffer_next(app, buffer_it, AccessAll, &buffer_it)){
if (buffer != buffer_it){
ranges[j].type = SearchRange_FrontToBack;
ranges[j].flags = SearchFlag_MatchWordPrefix;
ranges[j].buffer = buffer_it;
ranges[j].start = 0;
buffer_get_size(app, buffer_it, &ranges[j].size);
++j;
}
}
complete_state.set.count = j;
// NOTE(allen): Initialize the search hit table.
search_hits_init(&global_heap, &complete_state.hits, &complete_state.str, 100, (4 << 10));
String_Const_u8 word = complete_state.iter.key.words[0];
search_hit_add(&global_heap, &complete_state.hits, &complete_state.str, word);
complete_state.word_start = word_start;
complete_state.word_end = word_end;
}
else{
word_start = complete_state.word_start;
word_end = complete_state.word_end;
size = (i32)complete_state.iter.key.words[0].size;
}
// NOTE(allen): Iterate through matches.
if (size > 0){
for (;;){
i32 match_size = 0;
Search_Match match = search_next_match(app, &complete_state.set, &complete_state.iter);
if (match.found_match){
match_size = match.end - match.start;
Arena *scratch = context_get_arena(app);
Scratch_Block temp_auto_closer(scratch);
char *spare = push_array(scratch, char, match_size);
buffer_read_range(app, match.buffer, match.start, match.end, spare);
if (search_hit_add(&global_heap, &complete_state.hits, &complete_state.str, spare, match_size)){
buffer_replace_range(app, buffer, make_range(word_start, word_end), SCu8(spare, match_size));
view_set_cursor(app, view, seek_pos(word_start + match_size), true);
complete_state.word_end = word_start + match_size;
complete_state.set.ranges[0].mid_size = match_size;
break;
}
}
else{
complete_state.iter.pos = 0;
complete_state.iter.i = 0;
search_hits_init(&global_heap, &complete_state.hits, &complete_state.str, 100, (4 << 10));
String_Const_u8 word = complete_state.iter.key.words[0];
search_hit_add(&global_heap, &complete_state.hits, &complete_state.str, word);
match_size = (i32)word.size;
buffer_replace_range(app, buffer, make_range(word_start, word_end), word);
view_set_cursor(app, view, seek_pos(word_start + match_size), true);
complete_state.word_end = word_start + match_size;
complete_state.set.ranges[0].mid_size = match_size;
break;
}
}
}
}
}
// BOTTOM