/* $Id$ */ /************************************************************************** * search.c * * * * Copyright (C) 2000-2002 Chris Allegretta * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2, or (at your option) * * any later version. * * * * This program is distributed in the hope that it will be useful, * * but WITHOUT ANY WARRANTY; without even the implied warranty of * * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * * GNU General Public License for more details. * * * * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the Free Software * * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. * * * **************************************************************************/ #include "config.h" #include #include #include #include #include #include "proto.h" #include "nano.h" #ifdef ENABLE_NLS #include #define _(string) gettext(string) #else #define _(string) (string) #endif /* Regular expression helper functions */ #ifdef HAVE_REGEX_H void regexp_init(const char *regexp) { regcomp(&search_regexp, regexp, ISSET(CASE_SENSITIVE) ? 0 : REG_ICASE); SET(REGEXP_COMPILED); } void regexp_cleanup(void) { UNSET(REGEXP_COMPILED); regfree(&search_regexp); } #endif void search_init_globals(void) { if (last_search == NULL) { last_search = charalloc(1); last_search[0] = '\0'; } if (last_replace == NULL) { last_replace = charalloc(1); last_replace[0] = '\0'; } } /* Set up the system variables for a search or replace. Returns -1 on abort, 0 on success, and 1 on rerun calling program Return -2 to run opposite program (search -> replace, replace -> search) replacing = 1 if we call from do_replace, 0 if called from do_search func. */ int search_init(int replacing) { int i = 0; char *buf; static char *backupstring = NULL; search_init_globals(); buf = charalloc(strlen(last_search) + 5); buf[0] = '\0'; /* Clear the backupstring if we've changed from Pico mode to regular mode */ if (ISSET(CLEAR_BACKUPSTRING)) { free(backupstring); backupstring = NULL; UNSET(CLEAR_BACKUPSTRING); } /* Okay, fun time. backupstring is our holder for what is being returned from the statusq call. Using answer for this would be tricky. Here, if we're using PICO_MODE, we only want nano to put the old string back up as editable if it's not the same as last_search. Otherwise, if we don't already have a backupstring, set it to last_search. */ if (ISSET(PICO_MODE)) { if (backupstring == NULL || !strcmp(backupstring, last_search)) backupstring = mallocstrcpy(backupstring, ""); } else if (backupstring == NULL) backupstring = mallocstrcpy(backupstring, last_search); /* If using Pico messages, we do things the old fashioned way... */ if (ISSET(PICO_MODE)) { if (last_search[0]) { /* We use COLS / 3 here because we need to see more on the line */ if (strlen(last_search) > COLS / 3) { snprintf(buf, COLS / 3 + 3, " [%s", last_search); sprintf(&buf[COLS / 3 + 2], "...]"); } else sprintf(buf, " [%s]", last_search); } else { buf[0] = '\0'; } } else buf[0] = '\0'; /* This is now one simple call. It just does a lot */ i = statusq(0, replacing ? replace_list : whereis_list, backupstring, "%s%s%s%s%s%s", _("Search"), /* This string is just a modifier for the search prompt, no grammar is implied */ ISSET(CASE_SENSITIVE) ? _(" [Case Sensitive]") : "", /* This string is just a modifier for the search prompt, no grammar is implied */ ISSET(USE_REGEXP) ? _(" [Regexp]") : "", /* This string is just a modifier for the search prompt, no grammar is implied */ ISSET(REVERSE_SEARCH) ? _(" [Backwards]") : "", replacing ? _(" (to replace)") : "", buf); /* Release buf now that we don't need it anymore */ free(buf); /* Cancel any search, or just return with no previous search */ if ((i == -1) || (i < 0 && !last_search[0])) { statusbar(_("Search Cancelled")); reset_cursor(); free(backupstring); backupstring = NULL; return -1; } else switch (i) { case -2: /* Same string */ #ifdef HAVE_REGEX_H if (ISSET(USE_REGEXP)) { /* If we're in pico mode, answer is "", use last_search! */ if (ISSET(PICO_MODE)) regexp_init(last_search); else regexp_init(answer); } #endif break; case 0: /* They entered something new */ #ifdef HAVE_REGEX_H if (ISSET(USE_REGEXP)) regexp_init(answer); #endif free(backupstring); backupstring = NULL; last_replace[0] = '\0'; break; case TOGGLE_CASE_KEY: TOGGLE(CASE_SENSITIVE); goto string_reinit; case TOGGLE_BACKWARDS_KEY: TOGGLE(REVERSE_SEARCH); goto string_reinit; #ifdef HAVE_REGEX_H case TOGGLE_REGEXP_KEY: TOGGLE(USE_REGEXP); #endif string_reinit: free(backupstring); backupstring = NULL; backupstring = mallocstrcpy(backupstring, answer); return 1; case NANO_OTHERSEARCH_KEY: backupstring = mallocstrcpy(backupstring, answer); return -2; /* Call the opposite search function */ case NANO_FROMSEARCHTOGOTO_KEY: free(backupstring); backupstring = NULL; do_gotoline_void(); return -3; default: do_early_abort(); free(backupstring); backupstring = NULL; return -3; } return 0; } void not_found_msg(char *str) { if (strlen(str) <= COLS / 2) statusbar(_("\"%s\" not found"), str); else { char *foo = NULL; foo = mallocstrcpy(foo, str); foo[COLS / 2] = '\0'; statusbar(_("\"%s...\" not found"), foo); free(foo); } } int is_whole_word(int curr_pos, filestruct *fileptr, char *searchword) { /* start of line or previous character not a letter */ if ((curr_pos < 1) || (!isalpha((int) fileptr->data[curr_pos-1]))) /* end of line or next character not a letter */ if (((curr_pos + strlen(searchword)) == strlen(fileptr->data)) || (!isalpha((int) fileptr->data[curr_pos + strlen(searchword)]))) return TRUE; return FALSE; } int past_editbuff; /* findnextstr() is now searching lines not displayed */ filestruct *findnextstr(int quiet, int bracket_mode, filestruct * begin, int beginx, char *needle) { filestruct *fileptr = current; char *searchstr, *rev_start = NULL, *found = NULL; int current_x_find = 0; past_editbuff = 0; if (!ISSET(REVERSE_SEARCH)) { /* forward search */ current_x_find = current_x + 1; #if 0 /* Are we now back to the place where the search started) */ if ((fileptr == begin) && (beginx > current_x_find)) search_last_line = 1; #endif /* Make sure we haven't passed the end of the string */ if (strlen(fileptr->data) < current_x_find) current_x_find--; searchstr = &fileptr->data[current_x_find]; /* Look for needle in searchstr */ while ((found = strstrwrapper(searchstr, needle, rev_start, current_x_find)) == NULL) { /* finished processing file, get out */ if (search_last_line) { if (!quiet) not_found_msg(needle); update_line(fileptr, current_x); return NULL; } update_line(fileptr, 0); /* reset current_x_find between lines */ current_x_find = 0; fileptr = fileptr->next; if (fileptr == editbot) past_editbuff = 1; /* EOF reached ?, wrap around once */ if (fileptr == NULL) { /* don't wrap if looking for bracket match */ if (bracket_mode) return NULL; fileptr = fileage; past_editbuff = 1; if (!quiet) { statusbar(_("Search Wrapped")); SET(DISABLE_CURPOS); } } /* Original start line reached */ if (fileptr == begin) search_last_line = 1; searchstr = fileptr->data; } /* We found an instance */ current_x_find = found - fileptr->data; #if 1 /* Ensure we haven't wrapped around again! */ if ((search_last_line) && (current_x_find > beginx)) { if (!quiet) not_found_msg(needle); return NULL; } #endif } #ifndef NANO_SMALL else { /* reverse search */ current_x_find = current_x - 1; #if 0 /* Are we now back to the place where the search started) */ if ((fileptr == begin) && (current_x_find > beginx)) search_last_line = 1; #endif /* Make sure we haven't passed the begining of the string */ #if 0 /* Is this required here ? */ if (!(&fileptr->data[current_x_find] - fileptr->data)) current_x_find++; #endif rev_start = &fileptr->data[current_x_find]; searchstr = fileptr->data; /* Look for needle in searchstr */ while ((found = strstrwrapper(searchstr, needle, rev_start, current_x_find)) == NULL) { /* finished processing file, get out */ if (search_last_line) { if (!quiet) not_found_msg(needle); return NULL; } update_line(fileptr, 0); /* reset current_x_find between lines */ current_x_find = 0; fileptr = fileptr->prev; if (fileptr == edittop->prev) past_editbuff = 1; /* SOF reached ?, wrap around once */ /* ? */ if (fileptr == NULL) { if (bracket_mode) return NULL; fileptr = filebot; past_editbuff = 1; if (!quiet) { statusbar(_("Search Wrapped")); SET(DISABLE_CURPOS); } } /* Original start line reached */ if (fileptr == begin) search_last_line = 1; searchstr = fileptr->data; rev_start = fileptr->data + strlen(fileptr->data); } /* We found an instance */ current_x_find = found - fileptr->data; #if 1 /* Ensure we haven't wrapped around again! */ if ((search_last_line) && (current_x_find < beginx)) { if (!quiet) not_found_msg(needle); return NULL; } #endif } #endif /* NANO_SMALL */ /* Set globals now that we are sure we found something */ current = fileptr; current_x = current_x_find; if (!bracket_mode) { if (past_editbuff) edit_update(fileptr, CENTER); else update_line(current, current_x); placewewant = xplustabs(); reset_cursor(); } return fileptr; } void search_abort(void) { UNSET(KEEP_CUTBUFFER); display_main_list(); wrefresh(bottomwin); if (ISSET(MARK_ISSET)) edit_refresh_clearok(); #ifdef HAVE_REGEX_H if (ISSET(REGEXP_COMPILED)) regexp_cleanup(); #endif } /* Search for a string */ int do_search(void) { int i; filestruct *fileptr = current, *didfind; int fileptr_x = current_x; wrap_reset(); i = search_init(0); switch (i) { case -1: current = fileptr; search_abort(); return 0; case -3: search_abort(); return 0; case -2: do_replace(); return 0; case 1: do_search(); search_abort(); return 1; } /* The sneaky user deleted the previous search string */ if (!ISSET(PICO_MODE) && answer[0] == '\0') { statusbar(_("Search Cancelled")); search_abort(); return 0; } /* If answer is now == "", then PICO_MODE is set. So, copy last_search into answer... */ if (answer[0] == '\0') answer = mallocstrcpy(answer, last_search); else last_search = mallocstrcpy(last_search, answer); search_last_line = 0; didfind = findnextstr(FALSE, FALSE, current, current_x, answer); if ((fileptr == current) && (fileptr_x == current_x) && didfind != NULL) statusbar(_("This is the only occurrence")); search_abort(); return 1; } void print_replaced(int num) { if (num > 1) statusbar(_("Replaced %d occurrences"), num); else if (num == 1) statusbar(_("Replaced 1 occurrence")); } void replace_abort(void) { /* Identical to search_abort, so we'll call it here. If it does something different later, we can change it back. For now it's just a waste to duplicate code */ search_abort(); placewewant = xplustabs(); } #ifdef HAVE_REGEX_H int replace_regexp(char *string, int create_flag) { /* split personality here - if create_flag is null, just calculate * the size of the replacement line (necessary because of * subexpressions like \1 \2 \3 in the replaced text) */ char *c; int new_size = strlen(current->data) + 1; int search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so; new_size -= search_match_count; /* Iterate through the replacement text to handle * subexpression replacement using \1, \2, \3, etc */ c = last_replace; while (*c) { if (*c != '\\') { if (create_flag) *string++ = *c; c++; new_size++; } else { int num = (int) *(c + 1) - (int) '0'; if (num >= 1 && num <= 9) { int i = regmatches[num].rm_so; if (num > search_regexp.re_nsub) { /* Ugh, they specified a subexpression that doesn't exist. */ return -1; } /* Skip over the replacement expression */ c += 2; /* But add the length of the subexpression to new_size */ new_size += regmatches[num].rm_eo - regmatches[num].rm_so; /* And if create_flag is set, append the result of the * subexpression match to the new line */ while (create_flag && i < regmatches[num].rm_eo) *string++ = *(current->data + i++); } else { if (create_flag) *string++ = *c; c++; new_size++; } } } if (create_flag) *string = 0; return new_size; } #endif char *replace_line(void) { char *copy, *tmp; int new_line_size; int search_match_count; /* Calculate size of new line */ #ifdef HAVE_REGEX_H if (ISSET(USE_REGEXP)) { search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so; new_line_size = replace_regexp(NULL, 0); /* If they specified an invalid subexpression in the replace * text, return NULL, indicating an error */ if (new_line_size < 0) return NULL; } else { #else { #endif search_match_count = strlen(last_search); new_line_size = strlen(current->data) - strlen(last_search) + strlen(last_replace) + 1; } /* Create buffer */ copy = charalloc(new_line_size); /* Head of Original Line */ strncpy(copy, current->data, current_x); copy[current_x] = '\0'; /* Replacement Text */ if (!ISSET(USE_REGEXP)) strcat(copy, last_replace); #ifdef HAVE_REGEX_H else (void) replace_regexp(copy + current_x, 1); #endif /* The tail of the original line */ /* This may expose other bugs, because it no longer goes through each character in the string and tests for string goodness. But because we can assume the invariant that current->data is less than current_x + strlen(last_search) long, this should be safe. Or it will expose bugs ;-) */ tmp = current->data + current_x + search_match_count; strcat(copy, tmp); return copy; } /* step through each replace word and prompt user before replacing word */ int do_replace_loop(char *prevanswer, filestruct *begin, int *beginx, int wholewords, int *i) { int replaceall = 0, numreplaced = 0; filestruct *fileptr = NULL; char *copy; switch (*i) { case -1: /* Aborted enter */ if (last_replace[0] != '\0') answer = mallocstrcpy(answer, last_replace); statusbar(_("Replace Cancelled")); replace_abort(); return 0; case 0: /* They actually entered something */ break; default: if (*i != -2) { /* First page, last page, for example could get here */ do_early_abort(); replace_abort(); return 0; } } if (ISSET(PICO_MODE) && answer[0] == '\0') answer = mallocstrcpy(answer, last_replace); last_replace = mallocstrcpy(last_replace, answer); while (1) { /* Sweet optimization by Rocco here */ #if 0 fileptr = findnextstr(replaceall, FALSE, begin, *beginx, prevanswer); #else if (fileptr != 0) fileptr = findnextstr(1, FALSE, begin, *beginx, prevanswer); else fileptr = findnextstr(replaceall || (search_last_line ? 1 : 0), FALSE, begin, *beginx, prevanswer); #endif /* No more matches. Done! */ if (!fileptr) break; /* Make sure only whole words are found */ if ((wholewords) && (!is_whole_word(current_x, fileptr, prevanswer))) continue; /* If we're here, we've found the search string */ if (!replaceall) { curs_set(0); do_replace_highlight(TRUE, prevanswer); *i = do_yesno(1, 1, _("Replace this instance?")); do_replace_highlight(FALSE, prevanswer); curs_set(1); } if (*i > 0 || replaceall) { /* Yes, replace it!!!! */ if (*i == 2) replaceall = 1; copy = replace_line(); if (!copy) { statusbar(_("Replace failed: unknown subexpression!")); replace_abort(); return 0; } /* Cleanup */ totsize -= strlen(current->data); free(current->data); current->data = copy; totsize += strlen(current->data); if (!ISSET(REVERSE_SEARCH)) { /* Stop bug where we replace a substring of the replacement text */ current_x += strlen(last_replace) - 1; /* Adjust the original cursor position - COULD BE IMPROVED */ if (search_last_line) { *beginx += strlen(last_replace) - strlen(last_search); /* For strings that cross the search start/end boundary */ /* Don't go outside of allocated memory */ if (*beginx < 1) *beginx = 1; } } else { if (current_x > 1) current_x--; if (search_last_line) { *beginx += strlen(last_replace) - strlen(last_search); if (*beginx > strlen(current->data)) *beginx = strlen(current->data); } } edit_refresh(); set_modified(); numreplaced++; } else if (*i == -1) /* Abort, else do nothing and continue loop */ break; } return numreplaced; } /* Replace a string */ int do_replace(void) { int i, numreplaced, beginx; filestruct *begin; char *prevanswer = NULL, *buf = NULL; if (ISSET(VIEW_MODE)) { print_view_warning(); replace_abort(); return 0; } i = search_init(1); switch (i) { case -1: statusbar(_("Replace Cancelled")); replace_abort(); return 0; case 1: do_replace(); return 1; case -2: do_search(); return 0; case -3: replace_abort(); return 0; } /* Again, there was a previous string, but they deleted it and hit enter */ if (!ISSET(PICO_MODE) && answer[0] == '\0') { statusbar(_("Replace Cancelled")); replace_abort(); return 0; } /* If answer is now == "", then PICO_MODE is set. So, copy last_search into answer (and prevanswer)... */ if (answer[0] == '\0') { answer = mallocstrcpy(answer, last_search); prevanswer = mallocstrcpy(prevanswer, last_search); } else { last_search = mallocstrcpy(last_search, answer); prevanswer = mallocstrcpy(prevanswer, answer); } if (ISSET(PICO_MODE)) { buf = charalloc(strlen(last_replace) + 5); if (last_replace[0] != '\0') { if (strlen(last_replace) > (COLS / 3)) { strncpy(buf, last_replace, COLS / 3); sprintf(&buf[COLS / 3 - 1], "..."); } else sprintf(buf, "%s", last_replace); i = statusq(0, replace_list_2, "", _("Replace with [%s]"), buf); } else i = statusq(0, replace_list_2, "", _("Replace with")); } else i = statusq(0, replace_list_2, last_replace, _("Replace with")); /* save where we are */ begin = current; #if 0 /* why + 1 ? isn't this taken care of in findnextstr() ? */ beginx = current_x + 1; #else beginx = current_x; #endif search_last_line = 0; numreplaced = do_replace_loop(prevanswer, begin, &beginx, FALSE, &i); /* restore where we were */ current = begin; #if 0 current_x = beginx - 1; #else current_x = beginx; #endif renumber_all(); edit_update(current, CENTER); print_replaced(numreplaced); replace_abort(); return 1; } void goto_abort(void) { UNSET(KEEP_CUTBUFFER); display_main_list(); } int do_gotoline(int line, int save_pos) { if (line <= 0) { /* Ask for it */ if (statusq(0, goto_list, "", _("Enter line number"))) { statusbar(_("Aborted")); goto_abort(); return 0; } line = atoi(answer); /* Bounds check */ if (line <= 0) { statusbar(_("Come on, be reasonable")); goto_abort(); return 0; } } for (current = fileage; current->next != NULL && line > 1; line--) current = current->next; current_x = 0; /* if save_pos is non-zero, don't change the cursor position when updating the edit window */ if (save_pos) edit_update(current, NONE); else edit_update(current, CENTER); placewewant = 0; goto_abort(); return 1; } int do_gotoline_void(void) { return do_gotoline(0, 0); } #if defined(ENABLE_MULTIBUFFER) || !defined(DISABLE_SPELLER) void do_gotopos(int line, int pos_x, int pos_y, int pos_placewewant) { /* since do_gotoline() resets the x-coordinate but not the y-coordinate, set the coordinates up this way */ current_y = pos_y; do_gotoline(line, 1); /* make sure that the x-coordinate is sane here */ if (pos_x > strlen(current->data)) pos_x = strlen(current->data); /* set the rest of the coordinates up */ current_x = pos_x; placewewant = pos_placewewant; update_line(current, pos_x); } #endif #if !defined(NANO_SMALL) && defined(HAVE_REGEX_H) int do_find_bracket(void) { char ch_under_cursor, wanted_ch; char *pos, *brackets = "([{<>}])"; char regexp_pat[] = "[ ]"; int offset, have_past_editbuff = 0, flagsave, current_x_save, count = 1; filestruct *current_save; ch_under_cursor = current->data[current_x]; if ((!(pos = strchr(brackets, ch_under_cursor))) || (!((offset = pos - brackets) < 8))) { statusbar(_("Not a bracket")); return 1; } blank_statusbar_refresh(); wanted_ch = *(brackets + ((strlen(brackets) - (offset + 1)))); current_x_save = current_x; current_save = current; flagsave = flags; SET(USE_REGEXP); /* apparent near redundancy with regexp_pat[] here is needed, [][] works, [[]] doesn't */ if (offset < (strlen(brackets) / 2)) { /* on a left bracket */ regexp_pat[1] = wanted_ch; regexp_pat[2] = ch_under_cursor; UNSET(REVERSE_SEARCH); } else { /* on a right bracket */ regexp_pat[1] = ch_under_cursor; regexp_pat[2] = wanted_ch; SET(REVERSE_SEARCH); } regexp_init(regexp_pat); while (1) { search_last_line = 0; if (findnextstr(1, 1, current, current_x, regexp_pat)) { have_past_editbuff |= past_editbuff; if (current->data[current_x] == ch_under_cursor) /* found identical bracket */ count++; else { /* found complementary bracket */ if (!(--count)) { if (have_past_editbuff) edit_update(current, CENTER); else update_line(current, current_x); placewewant = xplustabs(); reset_cursor(); break ; } } } else { /* didn't find either left or right bracket */ statusbar(_("No matching bracket")); current_x = current_x_save; current = current_save; break; } } if (ISSET(REGEXP_COMPILED)) regexp_cleanup(); flags = flagsave; return 0; } #endif