search.c 27.9 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
/**************************************************************************
 *   search.c                                                             *
 *                                                                        *
5
 *   Copyright (C) 2000-2004 Chris Allegretta                             *
Chris Allegretta's avatar
Chris Allegretta committed
6
7
 *   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 *
8
 *   the Free Software Foundation; either version 2, or (at your option)  *
Chris Allegretta's avatar
Chris Allegretta committed
9
10
11
12
13
14
15
16
17
18
19
20
21
 *   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.            *
 *                                                                        *
 **************************************************************************/

22
23
#include "config.h"

Chris Allegretta's avatar
Chris Allegretta committed
24
25
26
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
27
#include <unistd.h>
Chris Allegretta's avatar
Chris Allegretta committed
28
#include <ctype.h>
29
#include <errno.h>
Chris Allegretta's avatar
Chris Allegretta committed
30
#include <assert.h>
Chris Allegretta's avatar
Chris Allegretta committed
31
32
#include "proto.h"
#include "nano.h"
Chris Allegretta's avatar
Chris Allegretta committed
33

34
35
/* Regular expression helper functions */

36
#ifdef HAVE_REGEX_H
37
int regexp_init(const char *regexp)
38
{
39
40
41
42
43
    /* Hmm, perhaps we should check for whether regcomp returns successfully */
    if (regcomp(&search_regexp, regexp, (ISSET(CASE_SENSITIVE) ? 0 : REG_ICASE) 
		| REG_EXTENDED) != 0)
	return 0;

44
    SET(REGEXP_COMPILED);
45
    return 1;
46
47
}

48
void regexp_cleanup(void)
49
{
50
51
52
53
    if (ISSET(REGEXP_COMPILED)) {
	UNSET(REGEXP_COMPILED);
	regfree(&search_regexp);
    }
54
}
55
#endif
56

57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
void not_found_msg(const char *str)
{
    if (strlen(str) <= COLS / 2)
	statusbar(_("\"%s\" not found"), str);
    else {
	char *foo = mallocstrcpy(NULL, str);

	foo[COLS / 2] = '\0';
	statusbar(_("\"%s...\" not found"), foo);
	free(foo);
    }
}

void search_abort(void)
{
    display_main_list();
73
#ifndef NANO_SMALL
74
    if (ISSET(MARK_ISSET))
75
76
	edit_refresh();
#endif
77
#ifdef HAVE_REGEX_H
78
    regexp_cleanup();
79
80
81
#endif
}

Chris Allegretta's avatar
Chris Allegretta committed
82
83
84
void search_init_globals(void)
{
    if (last_search == NULL) {
85
	last_search = charalloc(1);
Chris Allegretta's avatar
Chris Allegretta committed
86
	last_search[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
87
88
    }
    if (last_replace == NULL) {
89
	last_replace = charalloc(1);
Chris Allegretta's avatar
Chris Allegretta committed
90
	last_replace[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
91
92
93
    }
}

94
95
96
97
98
/* Set up the system variables for a search or replace.  Return -1 if
 * the search should be cancelled (due to Cancel, Go to Line, or a
 * failed regcomp()).  Return 0 on success, and 1 on rerun calling
 * program.  Return -2 to run opposite program (search -> replace,
 * replace -> search).
Chris Allegretta's avatar
Chris Allegretta committed
99
 *
Chris Allegretta's avatar
Chris Allegretta committed
100
101
 * replacing = 1 if we call from do_replace(), 0 if called from
 * do_search(). */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
102
int search_init(int replacing)
Chris Allegretta's avatar
Chris Allegretta committed
103
{
104
    int i = 0;
105
    char *buf;
106
    static char *backupstring = NULL;
107
108
109
110
111
112

    /* We display the search prompt below.  If the user types a partial
     * search string and then Replace or a toggle, we will return to
     * do_search() or do_replace() and be called again.  In that case,
     * we should put the same search string back up.  backupstring holds
     * this string. */
113

Chris Allegretta's avatar
Chris Allegretta committed
114
    search_init_globals();
115

116
    /* If we don't already have a backupstring, set it. */
117
    if (backupstring == NULL)
118
	backupstring = mallocstrcpy(NULL, "");
119

120
#ifndef NANO_SMALL
121
    search_history.current = (historytype *)&search_history.next;
122
#endif
123
124

    if (last_search[0] != '\0') {
125
126
	char *disp = display_string(last_search, 0, COLS / 3);

Chris Allegretta's avatar
Chris Allegretta committed
127
	buf = charalloc(COLS / 3 + 7);
128
129
	/* We use COLS / 3 here because we need to see more on the
	 * line. */
130
131
132
	sprintf(buf, " [%s%s]", disp,
		strlenpt(last_search) > COLS / 3 ? "..." : "");
	free(disp);
Chris Allegretta's avatar
Chris Allegretta committed
133
134
    } else {
	buf = charalloc(1);
Chris Allegretta's avatar
Chris Allegretta committed
135
	buf[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
136
    }
Chris Allegretta's avatar
Chris Allegretta committed
137

138
    /* This is now one simple call.  It just does a lot. */
139
    i = statusq(0, replacing ? replace_list : whereis_list, backupstring,
140
141
142
143
#ifndef NANO_SMALL
	&search_history,
#endif
	"%s%s%s%s%s%s",
Chris Allegretta's avatar
Chris Allegretta committed
144
	_("Search"),
145

146
#ifndef NANO_SMALL
147
148
	/* This string is just a modifier for the search prompt; no
	 * grammar is implied. */
149
150
151
	ISSET(CASE_SENSITIVE) ? _(" [Case Sensitive]") :
#endif
		"",
152

153
154
	/* This string is just a modifier for the search prompt; no
	 * grammar is implied. */
155
156
	ISSET(USE_REGEXP) ? _(" [Regexp]") : "",

157
#ifndef NANO_SMALL
158
159
	/* This string is just a modifier for the search prompt; no
	 * grammar is implied. */
160
161
162
	ISSET(REVERSE_SEARCH) ? _(" [Backwards]") :
#endif
		"",
163

Chris Allegretta's avatar
Chris Allegretta committed
164
165
	replacing ? _(" (to replace)") : "",
	buf);
Chris Allegretta's avatar
Chris Allegretta committed
166

167
    /* Release buf now that we don't need it anymore. */
Chris Allegretta's avatar
Chris Allegretta committed
168
169
    free(buf);

170
171
172
    free(backupstring);
    backupstring = NULL;

173
174
175
    /* Cancel any search, or just return with no previous search. */
    if (i == -1 || (i < 0 && last_search[0] == '\0') ||
	    (!replacing && i == 0 && answer[0] == '\0')) {
Chris Allegretta's avatar
Chris Allegretta committed
176
177
	statusbar(_("Search Cancelled"));
	reset_cursor();
178
179
180
#ifndef NANO_SMALL
	search_history.current = search_history.next;
#endif
Chris Allegretta's avatar
Chris Allegretta committed
181
	return -1;
Chris Allegretta's avatar
Chris Allegretta committed
182
183
    } else {
	switch (i) {
184
	case -2:	/* It's the same string. */
185
#ifdef HAVE_REGEX_H
186
187
188
189
190
191
	    /* Since answer is "", use last_search! */
	    if (ISSET(USE_REGEXP) && regexp_init(last_search) == 0) {
		statusbar(_("Invalid regex \"%s\""), last_search);
		reset_cursor();
		return -1;
	    }
192
#endif
Chris Allegretta's avatar
Chris Allegretta committed
193
	    break;
194
	case 0:		/* They entered something new. */
195
	    last_replace[0] = '\0';
196
#ifdef HAVE_REGEX_H
197
198
199
200
201
	    if (ISSET(USE_REGEXP) && regexp_init(answer) == 0) {
		statusbar(_("Invalid regex \"%s\""), answer);
		reset_cursor();
		return -1;
	    }
202
#endif
Chris Allegretta's avatar
Chris Allegretta committed
203
204
205
206
207
208
209
210
211
212
	    break;
#ifndef NANO_SMALL
	case TOGGLE_CASE_KEY:
	    TOGGLE(CASE_SENSITIVE);
	    backupstring = mallocstrcpy(backupstring, answer);
	    return 1;
	case TOGGLE_BACKWARDS_KEY:
	    TOGGLE(REVERSE_SEARCH);
	    backupstring = mallocstrcpy(backupstring, answer);
	    return 1;
213
#ifdef HAVE_REGEX_H
Chris Allegretta's avatar
Chris Allegretta committed
214
215
216
217
	case TOGGLE_REGEXP_KEY:
	    TOGGLE(USE_REGEXP);
	    backupstring = mallocstrcpy(backupstring, answer);
	    return 1;
218
#endif
Chris Allegretta's avatar
Chris Allegretta committed
219
220
221
#endif /* !NANO_SMALL */
	case NANO_OTHERSEARCH_KEY:
	    backupstring = mallocstrcpy(backupstring, answer);
222
	    return -2;	/* Call the opposite search function. */
Chris Allegretta's avatar
Chris Allegretta committed
223
	case NANO_FROMSEARCHTOGOTO_KEY:
224
225
226
#ifndef NANO_SMALL
	    search_history.current = search_history.next;
#endif
227
	    i = (int)strtol(answer, &buf, 10);	/* Just testing answer here. */
228
229
230
231
	    if (!(errno == ERANGE || *answer == '\0' || *buf != '\0'))
		do_gotoline(-1, 0);
	    else
		do_gotoline_void();
232
	    /* Fall through. */
Chris Allegretta's avatar
Chris Allegretta committed
233
	default:
234
	    return -1;
Chris Allegretta's avatar
Chris Allegretta committed
235
	}
Chris Allegretta's avatar
Chris Allegretta committed
236
237
238
239
    }
    return 0;
}

240
241
int is_whole_word(int curr_pos, const char *datastr, const char
	*searchword)
242
{
Chris Allegretta's avatar
Chris Allegretta committed
243
    size_t sln = curr_pos + strlen(searchword);
244

245
246
    /* Start of line or previous character is not a letter and end of
     * line or next character is not a letter. */
Chris Allegretta's avatar
Chris Allegretta committed
247
248
    return (curr_pos < 1 || !isalpha((int)datastr[curr_pos - 1])) &&
	(sln == strlen(datastr) || !isalpha((int)datastr[sln]));
249
250
}

251
252
253
254
255
256
257
258
/* Look for needle, starting at current, column current_x.  If
 * no_sameline is nonzero, skip over begin when looking for needle.
 * begin is the line where we first started searching, at column beginx.
 * If can_display_wrap is nonzero, we put messages on the statusbar, and
 * wrap around the file boundaries.  The return value specifies whether
 * we found anything. */
int findnextstr(int can_display_wrap, int wholeword, const filestruct
	*begin, size_t beginx, const char *needle, int no_sameline)
Chris Allegretta's avatar
Chris Allegretta committed
259
{
Chris Allegretta's avatar
Chris Allegretta committed
260
    filestruct *fileptr = current;
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
    const char *rev_start = NULL, *found = NULL;
    size_t current_x_find = 0;
	/* Where needle was found. */

    /* rev_start might end up 1 character before the start or after the
     * end of the line.  This won't be a problem because strstrwrapper()
     * will return immediately and say that no match was found, and
     * rev_start will be properly set when the search continues on the
     * previous or next line. */
#ifndef NANO_SMALL
    if (ISSET(REVERSE_SEARCH))
	rev_start = fileptr->data + (current_x - 1);
    else
#endif
	rev_start = fileptr->data + (current_x + 1);
Chris Allegretta's avatar
Chris Allegretta committed
276

277
278
279
    /* Look for needle in searchstr. */
    while (1) {
	found = strstrwrapper(fileptr->data, needle, rev_start);
Chris Allegretta's avatar
Chris Allegretta committed
280

281
282
283
284
	if (found != NULL && (!wholeword || is_whole_word(found -
		fileptr->data, fileptr->data, needle))) {
	    if (!no_sameline || fileptr != current)
		break;
Chris Allegretta's avatar
Chris Allegretta committed
285
	}
286

287
288
289
	/* Finished processing file, get out. */
	if (search_last_line) {
	    if (can_display_wrap)
Chris Allegretta's avatar
Chris Allegretta committed
290
		not_found_msg(needle);
291
	    return 0;
292
	}
293
	fileptr =
294
#ifndef NANO_SMALL
295
296
297
		ISSET(REVERSE_SEARCH) ? fileptr->prev :
#endif
		fileptr->next;
298

299
300
301
302
303
304
305
306
307
308
309
310
	/* Start or end of buffer reached; wrap around. */
	if (fileptr == NULL) {
	    if (!can_display_wrap)
		return 0;
	    fileptr =
#ifndef NANO_SMALL
		ISSET(REVERSE_SEARCH) ? filebot :
#endif
		fileage;
	    if (can_display_wrap)
		statusbar(_("Search Wrapped"));
	}
Chris Allegretta's avatar
Chris Allegretta committed
311

312
313
314
315
316
317
318
319
320
	/* Original start line reached. */
	if (fileptr == begin)
	    search_last_line = 1;
	rev_start = fileptr->data;
#ifndef NANO_SMALL
	if (ISSET(REVERSE_SEARCH))
	    rev_start += strlen(fileptr->data);
#endif
    }
321

322
323
    /* We found an instance. */
    current_x_find = found - fileptr->data;
Chris Allegretta's avatar
Chris Allegretta committed
324

325
326
327
328
329
330
331
332
333
    /* Ensure we haven't wrapped around again! */
    if (search_last_line &&
#ifndef NANO_SMALL
	((!ISSET(REVERSE_SEARCH) && current_x_find > beginx) ||
	(ISSET(REVERSE_SEARCH) && current_x_find < beginx))
#else
	current_x_find > beginx
#endif
	) {
Chris Allegretta's avatar
Chris Allegretta committed
334

335
336
337
	if (can_display_wrap)
	    not_found_msg(needle);
	return 0;
Chris Allegretta's avatar
Chris Allegretta committed
338
339
    }

340
    /* Set globals now that we are sure we found something. */
341
342
343
    current = fileptr;
    current_x = current_x_find;

344
    return 1;
Chris Allegretta's avatar
Chris Allegretta committed
345
346
}

Chris Allegretta's avatar
Chris Allegretta committed
347
/* Search for a string. */
Chris Allegretta's avatar
Chris Allegretta committed
348
349
350
int do_search(void)
{
    int i;
351
352
    filestruct *fileptr = current;
    int fileptr_x = current_x, didfind;
Chris Allegretta's avatar
Chris Allegretta committed
353

354
#ifndef DISABLE_WRAPPING
Chris Allegretta's avatar
Chris Allegretta committed
355
    wrap_reset();
356
#endif
357
    i = search_init(0);
358
359
    if (i == -1)	/* Cancel, Go to Line, blank search string, or
			 * regcomp() failed. */
Chris Allegretta's avatar
Chris Allegretta committed
360
	search_abort();
361
    else if (i == -2)	/* Replace. */
Chris Allegretta's avatar
Chris Allegretta committed
362
	do_replace();
363
364
365
#ifndef NANO_SMALL
    else if (i == 1)	/* Case Sensitive, Backwards, or Regexp search
			 * toggle. */
Chris Allegretta's avatar
Chris Allegretta committed
366
	do_search();
367
#endif
368

369
370
371
372
    if (i != 0)
	return 0;

    /* If answer is now "", copy last_search into answer. */
Chris Allegretta's avatar
Chris Allegretta committed
373
    if (answer[0] == '\0')
374
375
376
377
	answer = mallocstrcpy(answer, last_search);
    else
	last_search = mallocstrcpy(last_search, answer);

378
#ifndef NANO_SMALL
379
380
    /* If answer is not "", add this search string to the search history
     * list. */
Chris Allegretta's avatar
Chris Allegretta committed
381
    if (answer[0] != '\0')
382
	update_history(&search_history, answer);
383
#endif
384

385
    search_last_line = 0;
386
387
388
    didfind = findnextstr(TRUE, FALSE, current, current_x, answer, FALSE);
    edit_refresh();
    placewewant = xplustabs();
389

390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
    /* Check to see if there's only one occurrence of the string and
     * we're on it now. */
    if (fileptr == current && fileptr_x == current_x && didfind) {
#ifdef HAVE_REGEX_H
	/* Do the search again, skipping over the current line, if we're
	 * doing a bol and/or eol regex search ("^", "$", or "^$"), so
	 * that we find one only once per line.  We should only end up
	 * back at the same position if the string isn't found again, in
	 * which case it's the only occurrence. */
	if (ISSET(USE_REGEXP) && regexp_bol_or_eol(&search_regexp, last_search)) {
	    didfind = findnextstr(TRUE, FALSE, current, current_x, answer, TRUE);
	    if (fileptr == current && fileptr_x == current_x && !didfind)
		statusbar(_("This is the only occurrence"));
	} else {
#endif
	    statusbar(_("This is the only occurrence"));
#ifdef HAVE_REGEX_H
	}
#endif
    }
410

Chris Allegretta's avatar
Chris Allegretta committed
411
    search_abort();
412

Chris Allegretta's avatar
Chris Allegretta committed
413
414
415
    return 1;
}

416
417
418
/* Search for the next string without prompting. */
int do_research(void)
{
419
420
    filestruct *fileptr = current;
    int fileptr_x = current_x, didfind;
421

422
#ifndef DISABLE_WRAPPING
423
    wrap_reset();
424
#endif
425
426
427
428
429
    search_init_globals();

    if (last_search[0] != '\0') {

#ifdef HAVE_REGEX_H
430
431
432
	/* Since answer is "", use last_search! */
	if (ISSET(USE_REGEXP) && regexp_init(last_search) == 0) {
		statusbar(_("Invalid regex \"%s\""), last_search);
433
		reset_cursor();
434
		return -1;
435
436
437
438
	    }
#endif

	search_last_line = 0;
439
440
441
	didfind = findnextstr(TRUE, FALSE, current, current_x, last_search, FALSE);
	edit_refresh();
	placewewant = xplustabs();
442

443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
	/* Check to see if there's only one occurrence of the string and
	 * we're on it now. */
	if (fileptr == current && fileptr_x == current_x && didfind) {
#ifdef HAVE_REGEX_H
	    /* Do the search again, skipping over the current line, if
	     * we're doing a bol and/or eol regex search ("^", "$", or
	     * "^$"), so that we find one only once per line.  We should
	     * only end up back at the same position if the string isn't
	     * found again, in which case it's the only occurrence. */
	    if (ISSET(USE_REGEXP) && regexp_bol_or_eol(&search_regexp, last_search)) {
		didfind = findnextstr(TRUE, FALSE, current, current_x, answer, TRUE);
		if (fileptr == current && fileptr_x == current_x && !didfind)
		    statusbar(_("This is the only occurrence"));
	    } else {
#endif
		statusbar(_("This is the only occurrence"));
#ifdef HAVE_REGEX_H
	    }
#endif
	}
463
464
465
466
467
468
469
470
    } else
        statusbar(_("No current search pattern"));

    search_abort();

    return 1;
}

Chris Allegretta's avatar
Chris Allegretta committed
471
472
void replace_abort(void)
{
473
474
475
    /* 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. */
476
    search_abort();
477
    placewewant = xplustabs();
478
479
}

480
#ifdef HAVE_REGEX_H
481
482
int replace_regexp(char *string, int create_flag)
{
483
    /* Split personality here - if create_flag is zero, just calculate
484
     * the size of the replacement line (necessary because of
485
     * subexpressions \1 to \9 in the replaced text). */
486

487
    const char *c = last_replace;
488
    int search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so;
489
    int new_size = strlen(current->data) + 1 - search_match_count;
490

Chris Allegretta's avatar
Chris Allegretta committed
491
492
    /* Iterate through the replacement text to handle subexpression
     * replacement using \1, \2, \3, etc. */
493
    while (*c != '\0') {
494
495
496
	int num = (int)(*(c + 1) - '0');

	if (*c != '\\' || num < 1 || num > 9 || num > search_regexp.re_nsub) {
497
498
499
500
501
	    if (create_flag)
		*string++ = *c;
	    c++;
	    new_size++;
	} else {
502
	    int i = regmatches[num].rm_eo - regmatches[num].rm_so;
503

504
505
	    /* Skip over the replacement expression. */
	    c += 2;
506

507
508
	    /* But add the length of the subexpression to new_size. */
	    new_size += i;
509

510
511
512
513
514
515
	    /* And if create_flag is nonzero, append the result of the
	     * subexpression match to the new line. */
	    if (create_flag) {
		strncpy(string, current->data + current_x +
			regmatches[num].rm_so, i);
		string += i;
516
517
	    }
	}
518
519
520
    }

    if (create_flag)
Chris Allegretta's avatar
Chris Allegretta committed
521
	*string = '\0';
522
523
524

    return new_size;
}
525
#endif
526

527
char *replace_line(const char *needle)
528
{
529
    char *copy;
530
531
532
    int new_line_size;
    int search_match_count;

533
    /* Calculate the size of the new line. */
534
#ifdef HAVE_REGEX_H
535
    if (ISSET(USE_REGEXP)) {
536
 	search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so;
537
	new_line_size = replace_regexp(NULL, 0);
538
    } else {
539
#endif
540
541
542
543
	search_match_count = strlen(needle);
	new_line_size = strlen(current->data) - search_match_count +
	    strlen(answer) + 1;
#ifdef HAVE_REGEX_H
544
    }
545
#endif
546

547
    /* Create the buffer. */
548
    copy = charalloc(new_line_size);
549

550
    /* The head of the original line. */
551
552
    strncpy(copy, current->data, current_x);

553
    /* The replacement text. */
554
#ifdef HAVE_REGEX_H
555
556
    if (ISSET(USE_REGEXP))
	replace_regexp(copy + current_x, TRUE);
557
    else
558
#endif
559
	strcpy(copy + current_x, answer);
560

561
562
563
    /* The tail of the original line. */
    assert(current_x + search_match_count <= strlen(current->data));
    strcat(copy, current->data + current_x + search_match_count);
564
565

    return copy;
Chris Allegretta's avatar
Chris Allegretta committed
566
567
}

568
569
570
571
572
573
574
575
576
/* Step through each replace word and prompt user before replacing.
 * Parameters real_current and real_current_x are needed by the internal
 * speller, to allow the cursor position to be updated when a word
 * before the cursor is replaced by a shorter word.
 *
 * needle is the string to seek.  We replace it with answer.  Return -1
 * if needle isn't found, else the number of replacements performed. */
int do_replace_loop(const char *needle, const filestruct *real_current,
	size_t *real_current_x, int wholewords)
Chris Allegretta's avatar
Chris Allegretta committed
577
{
Chris Allegretta's avatar
Chris Allegretta committed
578
    int replaceall = 0, numreplaced = -1;
579
580
    const filestruct *current_save = current;
    size_t current_x_save = current_x;
581
#ifdef HAVE_REGEX_H
582
    /* The starting-line match and bol/eol regex flags. */
583
    int begin_line = 0, bol_or_eol = 0;
584
#endif
585
586
#ifndef NANO_SMALL
    int mark_set = ISSET(MARK_ISSET);
Chris Allegretta's avatar
Chris Allegretta committed
587

588
589
590
    UNSET(MARK_ISSET);
    edit_refresh();
#endif
591

592
593
    while (findnextstr(TRUE, wholewords, current_save, current_x_save,
	needle,
594
#ifdef HAVE_REGEX_H
595
596
597
598
599
	/* We should find a bol and/or eol regex only once per line.  If
	 * the bol_or_eol flag is set, it means that the last search
	 * found one on the beginning line, so we should skip over the
	 * beginning line when doing this search. */
	bol_or_eol
600
#else
601
	FALSE
602
#endif
603
604
605
606
	) != 0) {

	int i = 0;
	size_t match_len;
607
608

#ifdef HAVE_REGEX_H
609
	/* If the bol_or_eol flag is set, we've found a match on the
610
611
	 * beginning line already, and we're still on the beginning line
	 * after the search, it means that we've wrapped around, so
612
	 * we're done. */
613
614
615
616
617
	if (bol_or_eol && begin_line && current == real_current)
	    break;
	/* Otherwise, set the begin_line flag if we've found a match on
	 * the beginning line, reset the bol_or_eol flag, and
	 * continue. */
618
	else {
619
620
621
	    if (current == real_current)
		begin_line = 1;
	    bol_or_eol = 0;
622
623
	}
#endif
Chris Allegretta's avatar
Chris Allegretta committed
624

625
	edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
626

627
628
629
630
631
#ifdef HAVE_REGEX_H
	if (ISSET(USE_REGEXP))
	    match_len = regmatches[0].rm_eo - regmatches[0].rm_so;
	else
#endif
632
633
634
635
636
	    match_len = strlen(needle);

	/* Record for the return value that we found the search string. */
	if (numreplaced == -1)
	    numreplaced = 0;
637

638
	if (!replaceall) {
639
640
641
642
643
644
	    char *exp_word;
	    size_t xpt = xplustabs();

	    exp_word = display_string(current->data, xpt,
		strnlenpt(current->data, match_len + current_x) - xpt);

645
	    curs_set(0);
646
	    do_replace_highlight(TRUE, exp_word);
647

648
	    i = do_yesno(1, _("Replace this instance?"));
Chris Allegretta's avatar
Chris Allegretta committed
649

650
651
	    do_replace_highlight(FALSE, exp_word);
	    free(exp_word);
652
	    curs_set(1);
653

654
	    if (i == -1)	/* We canceled the replace. */
655
		break;
656
657
	}

658
#ifdef HAVE_REGEX_H
659
	/* Set the bol_or_eol flag if we're doing a bol and/or eol regex
660
	 * replace ("^", "$", or "^$"). */
661
662
663
	if (ISSET(USE_REGEXP) && regexp_bol_or_eol(&search_regexp,
		needle))
	    bol_or_eol = 1;
664
665
#endif

666
	if (i > 0 || replaceall) {	/* Yes, replace it!!!! */
667
668
	    char *copy;
	    int length_change;
669

670
	    if (i == 2)
Chris Allegretta's avatar
Chris Allegretta committed
671
672
		replaceall = 1;

673
	    copy = replace_line(needle);
Chris Allegretta's avatar
Chris Allegretta committed
674

675
	    length_change = strlen(copy) - strlen(current->data);
Chris Allegretta's avatar
Chris Allegretta committed
676

677
678
679
680
681
682
683
684
#ifndef NANO_SMALL
	    if (current == mark_beginbuf && mark_beginx > current_x) {
		if (mark_beginx < current_x + match_len)
		    mark_beginx = current_x;
		else
		    mark_beginx += length_change;
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
685

686
	    assert(0 <= match_len + length_change);
687
688
689
690
	    if (current == real_current && current_x <= *real_current_x) {
		if (*real_current_x < current_x + match_len)
		    *real_current_x = current_x + match_len;
		*real_current_x += length_change;
691
	    }
692

693
	    /* Set the cursor at the last character of the replacement
694
695
	     * text, so searching will resume after the replacement
	     * text.  Note that current_x might be set to -1 here. */
696
#ifndef NANO_SMALL
697
	    if (!ISSET(REVERSE_SEARCH))
698
699
#endif
		current_x += match_len + length_change - 1;
700

701
	    /* Cleanup. */
702
703
704
	    totsize += length_change;
	    free(current->data);
	    current->data = copy;
705

Chris Allegretta's avatar
Chris Allegretta committed
706
707
708
	    edit_refresh();
	    set_modified();
	    numreplaced++;
709
	}
Chris Allegretta's avatar
Chris Allegretta committed
710
711
    }

Chris Allegretta's avatar
Chris Allegretta committed
712
713
714
715
    /* If text has been added to the magicline, make a new magicline. */
    if (filebot->data[0] != '\0')
	new_magicline();

716
717
718
719
720
#ifndef NANO_SMALL
    if (mark_set)
	SET(MARK_ISSET);
#endif

Chris Allegretta's avatar
Chris Allegretta committed
721
722
723
    return numreplaced;
}

Chris Allegretta's avatar
Chris Allegretta committed
724
/* Replace a string. */
Chris Allegretta's avatar
Chris Allegretta committed
725
726
int do_replace(void)
{
727
728
729
    int i, numreplaced;
    filestruct *edittop_save, *begin;
    size_t beginx;
Chris Allegretta's avatar
Chris Allegretta committed
730

Chris Allegretta's avatar
Chris Allegretta committed
731
732
733
734
735
736
    if (ISSET(VIEW_MODE)) {
	print_view_warning();
	replace_abort();
	return 0;
    }

Chris Allegretta's avatar
Chris Allegretta committed
737
    i = search_init(1);
738
739
    if (i == -1) {		/* Cancel, Go to Line, blank search
				 * string, or regcomp() failed. */
Chris Allegretta's avatar
Chris Allegretta committed
740
741
	replace_abort();
	return 0;
742
    } else if (i == -2) {	/* No Replace. */
Chris Allegretta's avatar
Chris Allegretta committed
743
744
	do_search();
	return 0;
745
746
747
748
749
    } else if (i == 1)		/* Case Sensitive, Backwards, or Regexp
				 * search toggle. */
	do_replace();

    if (i != 0)
Chris Allegretta's avatar
Chris Allegretta committed
750
751
	return 0;

752
753
754
    /* If answer is not "", add answer to the search history list and
     * copy answer into last_search. */
    if (answer[0] != '\0') {
755
#ifndef NANO_SMALL
756
	update_history(&search_history, answer);
757
#endif
Chris Allegretta's avatar
Chris Allegretta committed
758
	last_search = mallocstrcpy(last_search, answer);
759
    }
Chris Allegretta's avatar
Chris Allegretta committed
760

761
#ifndef NANO_SMALL
762
763
    replace_history.current = (historytype *)&replace_history.next;
    last_replace = mallocstrcpy(last_replace, "");
764
#endif
765
766

    i = statusq(0, replace_list_2, last_replace,
767
768
769
770
#ifndef NANO_SMALL
		&replace_history,
#endif
		_("Replace with"));
771

772
#ifndef NANO_SMALL
773
774
775
    /* Add this replace string to the replace history list.  i == 0
     * means that the string is not "". */
    if (i == 0)
776
	update_history(&replace_history, answer);
777
778
779
780
781
782
783
784
785
786
787
788
789
#endif

    if (i != 0 && i != -2) {
	if (i == -1) {		/* Cancel. */
	    if (last_replace[0] != '\0')
		answer = mallocstrcpy(answer, last_replace);
	    statusbar(_("Replace Cancelled"));
	}
	replace_abort();
	return 0;
    }

    last_replace = mallocstrcpy(last_replace, answer);
Chris Allegretta's avatar
Chris Allegretta committed
790

791
    /* Save where we are. */
Chris Allegretta's avatar
Chris Allegretta committed
792
    begin = current;
793
    beginx = current_x;
794
795
    edittop_save = edittop;
    search_last_line = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
796

797
    numreplaced = do_replace_loop(last_search, begin, &beginx, FALSE);
Chris Allegretta's avatar
Chris Allegretta committed
798

799
    /* Restore where we were. */
Chris Allegretta's avatar
Chris Allegretta committed
800
    current = begin;
801
    current_x = beginx;
Chris Allegretta's avatar
Chris Allegretta committed
802
    renumber_all();
803
    edit_update(edittop_save, TOP);
Chris Allegretta's avatar
Chris Allegretta committed
804
805

    if (numreplaced >= 0)
806
	statusbar(P_("Replaced %d occurrence", "Replaced %d occurrences",
Chris Allegretta's avatar
Chris Allegretta committed
807
808
		numreplaced), numreplaced);

Chris Allegretta's avatar
Chris Allegretta committed
809
810
811
812
    replace_abort();
    return 1;
}

813
int do_gotoline(int line, int save_pos)
Chris Allegretta's avatar
Chris Allegretta committed
814
{
815
    if (line <= 0) {		/* Ask for it */
Chris Allegretta's avatar
Chris Allegretta committed
816
	int st = statusq(FALSE, goto_list, line != 0 ? answer : "",
Chris Allegretta's avatar
Chris Allegretta committed
817
#ifndef NANO_SMALL
Chris Allegretta's avatar
Chris Allegretta committed
818
			NULL,
Chris Allegretta's avatar
Chris Allegretta committed
819
#endif
Chris Allegretta's avatar
Chris Allegretta committed
820
821
822
823
			_("Enter line number"));

	/* Cancel, or Enter with blank string. */
	if (st == -1 || st == -2)
Chris Allegretta's avatar
Chris Allegretta committed
824
	    statusbar(_("Aborted"));
Chris Allegretta's avatar
Chris Allegretta committed
825
	if (st != 0) {
Chris Allegretta's avatar
Chris Allegretta committed
826
	    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
827
828
	    return 0;
	}
829
830
831

	line = atoi(answer);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
832
	/* Bounds check. */
833
834
	if (line <= 0) {
	    statusbar(_("Come on, be reasonable"));
Chris Allegretta's avatar
Chris Allegretta committed
835
	    display_main_list();
836
	    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
837
838
839
	}
    }

840
    for (current = fileage; current->next != NULL && line > 1; line--)
841
	current = current->next;
Chris Allegretta's avatar
Chris Allegretta committed
842

843
    current_x = 0;
844

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
845
846
847
848
    /* If save_pos is nonzero, don't change the cursor position when
     * updating the edit window. */
    edit_update(current, save_pos ? NONE : CENTER);

849
    placewewant = 0;
Chris Allegretta's avatar
Chris Allegretta committed
850
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
851
852
853
854
855
    return 1;
}

int do_gotoline_void(void)
{
856
    return do_gotoline(0, 0);
Chris Allegretta's avatar
Chris Allegretta committed
857
}
858

859
#if defined(ENABLE_MULTIBUFFER) || !defined(DISABLE_SPELLER)
860
void do_gotopos(int line, int pos_x, int pos_y, int pos_placewewant)
861
862
863
864
865
{
    /* 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);
866

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
867
868
869
    /* make sure that the x-coordinate is sane here */
    if (pos_x > strlen(current->data))
	pos_x = strlen(current->data);
870
871

    /* set the rest of the coordinates up */
872
873
874
875
876
    current_x = pos_x;
    placewewant = pos_placewewant;
    update_line(current, pos_x);
}
#endif
877
878
879
880
881

#if !defined(NANO_SMALL) && defined(HAVE_REGEX_H)
int do_find_bracket(void)
{
    char ch_under_cursor, wanted_ch;
882
    const char *pos, *brackets = "([{<>}])";
883
    char regexp_pat[] = "[  ]";
884
    int flagsave, current_x_save, count = 1;
885
886
887
    filestruct *current_save;

    ch_under_cursor = current->data[current_x];
888

889
890
    pos = strchr(brackets, ch_under_cursor);
    if (ch_under_cursor == '\0' || pos == NULL) {
891
892
893
894
	statusbar(_("Not a bracket"));
	return 1;
    }

895
896
    assert(strlen(brackets) % 2 == 0);
    wanted_ch = brackets[(strlen(brackets) - 1) - (pos - brackets)];
897
898
899
900
901
902

    current_x_save = current_x;
    current_save = current;
    flagsave = flags;
    SET(USE_REGEXP);

903
904
    /* Apparent near redundancy with regexp_pat[] here is needed.
     * "[][]" works, "[[]]" doesn't. */
905

906
    if (pos < brackets + (strlen(brackets) / 2)) {	/* On a left bracket. */
907
908
909
	regexp_pat[1] = wanted_ch;
	regexp_pat[2] = ch_under_cursor;
	UNSET(REVERSE_SEARCH);
910
    } else {			/* On a right bracket. */
911
912
913
914
915
916
	regexp_pat[1] = ch_under_cursor;
	regexp_pat[2] = wanted_ch;
	SET(REVERSE_SEARCH);
    }

    regexp_init(regexp_pat);
917
918
    /* We constructed regexp_pat to be a valid expression. */
    assert(ISSET(REGEXP_COMPILED));
919

920
    search_last_line = 0;
921
    while (1) {
922
923
	if (findnextstr(FALSE, FALSE, current, current_x, regexp_pat, FALSE) != 0) {
	    /* Found identical bracket. */
Chris Allegretta's avatar
Chris Allegretta committed
924
	    if (current->data[current_x] == ch_under_cursor)
925
		count++;
926
927
928
929
930
	    /* Found complementary bracket. */
	    else if (--count == 0) {
		edit_refresh();
		placewewant = xplustabs();
		break;
931
	    }
Chris Allegretta's avatar
Chris Allegretta committed
932
	} else {
933
	    /* Didn't find either a left or right bracket. */
934
935
936
	    statusbar(_("No matching bracket"));
	    current_x = current_x_save;
	    current = current_save;
Chris Allegretta's avatar
Chris Allegretta committed
937
	    update_line(current, current_x);
938
939
940
941
	    break;
	}
    }

942
    regexp_cleanup();
943
944
945
946
    flags = flagsave;
    return 0;
}
#endif
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973

#ifndef NANO_SMALL
/*
 * search and replace history list support functions
 */

/* initialize search and replace history lists */
void history_init(void)
{
    search_history.next = (historytype *)&search_history.prev;
    search_history.prev = NULL;
    search_history.tail = (historytype *)&search_history.next;
    search_history.current = search_history.next;
    search_history.count = 0;
    search_history.len = 0;

    replace_history.next = (historytype *)&replace_history.prev;
    replace_history.prev = NULL;
    replace_history.tail = (historytype *)&replace_history.next;
    replace_history.current = replace_history.next;
    replace_history.count = 0;
    replace_history.len = 0;
}

/* find first node containing string *s in history list *h */
historytype *find_node(historytype *h, char *s)
{
Chris Allegretta's avatar
Chris Allegretta committed
974
    for (; h->next != NULL; h = h->next)
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
	if (strcmp(s, h->data) == 0)
	    return h;
    return NULL;
}

/* remove node *r */
void remove_node(historytype *r)
{
    r->prev->next = r->next;
    r->next->prev = r->prev;
    free(r->data);
    free(r);
}

/* add a node after node *h */
void insert_node(historytype *h, const char *s)
{
    historytype *a;

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
994
    a = (historytype *)nmalloc(sizeof(historytype));
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
    a->next = h->next;
    a->prev = h->next->prev;
    h->next->prev = a;
    h->next = a;
    a->data = mallocstrcpy(NULL, s);
}

/* update history list */
void update_history(historyheadtype *h, char *s)
{
    historytype *p;

Chris Allegretta's avatar
Chris Allegretta committed
1007
1008
1009
    if ((p = find_node(h->next, s)) != NULL) {
	if (p == h->next)		/* catch delete and re-insert of
					   same string in 1st node */
1010
	    goto up_hs;
Chris Allegretta's avatar
Chris Allegretta committed
1011
	remove_node(p);			/* delete identical older string */
1012
1013
1014
1015
1016
1017
1018
1019
	h->count--;
    }
    if (h->count == MAX_SEARCH_HISTORY) {	/* list 'full', delete oldest */
	remove_node(h->tail);
	h->count--;
    }
    insert_node((historytype *)h, s);
    h->count++;
1020
    SET(HISTORY_CHANGED);
Chris Allegretta's avatar
Chris Allegretta committed
1021
  up_hs:
1022
1023
1024
1025
1026
1027
    h->current = h->next;
}

/* return a pointer to either the next older history or NULL if no more */
char *get_history_older(historyheadtype *h)
{
Chris Allegretta's avatar
Chris Allegretta committed
1028
    if (h->current->next != NULL) {	/* any older entries? */
1029
1030
1031
1032
1033
1034
1035
1036
	h->current = h->current->next;	/* yes */
	return h->current->data;	/* return it */
    }
    return NULL;			/* end of list */
}

char *get_history_newer(historyheadtype *h)
{
Chris Allegretta's avatar
Chris Allegretta committed
1037
    if (h->current->prev != NULL) {
1038
	h->current = h->current->prev;
Chris Allegretta's avatar
Chris Allegretta committed
1039
	if (h->current->prev != NULL)
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
	    return h->current->data;
    }
    return NULL;
}

/* get a completion */
char *get_history_completion(historyheadtype *h, char *s)
{
    historytype *p;

Chris Allegretta's avatar
Chris Allegretta committed
1050
1051
    for (p = h->current->next; p->next != NULL; p = p->next) {
	if (strncmp(s, p->data, h->len) == 0 && strlen(p->data) != h->len) {
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
	    h->current = p;
	    return p->data;
	}
    }
    h->current = (historytype*)h;
    null_at(&s, h->len);
    return s;
}

/* free a history list */
void free_history(historyheadtype *h)
{
    historytype *p, *n;

Chris Allegretta's avatar
Chris Allegretta committed
1066
    for (p = h->next; (n = p->next); p = n)
1067
1068
1069
1070
1071
	remove_node(p);
}

/* end of history support functions */
#endif /* !NANO_SMALL */