search.c 28.5 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
#include <stdlib.h>
#include <string.h>
26
#include <unistd.h>
Chris Allegretta's avatar
Chris Allegretta committed
27
#include <stdio.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
{
    UNSET(REGEXP_COMPILED);
    regfree(&search_regexp);
}
53
#endif
54

55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
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();
    wrefresh(bottomwin);
    if (ISSET(MARK_ISSET))
	edit_refresh_clearok();

#ifdef HAVE_REGEX_H
    if (ISSET(REGEXP_COMPILED))
	regexp_cleanup();
#endif
}

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

Chris Allegretta's avatar
Chris Allegretta committed
93
94
95
96
/* Set up the system variables for a search or replace.  Return -1 on
 * abort, 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
97
98
 * replacing = 1 if we call from do_replace(), 0 if called from
 * do_search(). */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
99
int search_init(int replacing)
Chris Allegretta's avatar
Chris Allegretta committed
100
{
101
    int i = 0;
102
    char *buf;
103
    static char *backupstring = NULL;
104
105
106
#ifdef HAVE_REGEX_H
    const char *regex_error = _("Invalid regex \"%s\"");
#endif /* HAVE_REGEX_H */
107

Chris Allegretta's avatar
Chris Allegretta committed
108
    search_init_globals();
109

110
    /* If we don't already have a backupstring, set it. */
111
    if (backupstring == NULL)
112
	backupstring = mallocstrcpy(NULL, "");
113

114
#ifndef NANO_SMALL
115
    search_history.current = (historytype *)&search_history.next;
116
#endif
117
118

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

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

131
    /* This is now one simple call.  It just does a lot */
132
    i = statusq(0, replacing ? replace_list : whereis_list, backupstring,
133
134
135
136
#ifndef NANO_SMALL
	&search_history,
#endif
	"%s%s%s%s%s%s",
Chris Allegretta's avatar
Chris Allegretta committed
137
	_("Search"),
138

139
#ifndef NANO_SMALL
140
141
	/* This string is just a modifier for the search prompt,
	   no grammar is implied */
142
143
144
	ISSET(CASE_SENSITIVE) ? _(" [Case Sensitive]") :
#endif
		"",
145
146
147
148
149

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

150
#ifndef NANO_SMALL
151
152
	/* This string is just a modifier for the search prompt,
	   no grammar is implied */
153
154
155
	ISSET(REVERSE_SEARCH) ? _(" [Backwards]") :
#endif
		"",
156

Chris Allegretta's avatar
Chris Allegretta committed
157
158
	replacing ? _(" (to replace)") : "",
	buf);
Chris Allegretta's avatar
Chris Allegretta committed
159

Chris Allegretta's avatar
Chris Allegretta committed
160
161
162
    /* Release buf now that we don't need it anymore */
    free(buf);

163
164
165
    free(backupstring);
    backupstring = NULL;

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

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

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

Chris Allegretta's avatar
Chris Allegretta committed
248
filestruct *findnextstr(int quiet, int bracket_mode,
249
			const filestruct *begin, int beginx,
250
			const char *needle, int no_sameline)
Chris Allegretta's avatar
Chris Allegretta committed
251
{
Chris Allegretta's avatar
Chris Allegretta committed
252
    filestruct *fileptr = current;
Chris Allegretta's avatar
Chris Allegretta committed
253
    const char *searchstr, *rev_start = NULL, *found = NULL;
254
    int current_x_find = 0;
Chris Allegretta's avatar
Chris Allegretta committed
255

Chris Allegretta's avatar
Chris Allegretta committed
256
    search_offscreen = 0;
257

Chris Allegretta's avatar
Chris Allegretta committed
258
    if (!ISSET(REVERSE_SEARCH)) {		/* forward search */
Chris Allegretta's avatar
Chris Allegretta committed
259
260
261
262
263
264
	/* Argh, current_x is set to -1 by nano.c:do_int_spell_fix(), and
	 * strlen returns size_t, which is unsigned. */
	assert(current_x < 0 || current_x <= strlen(fileptr->data));
	current_x_find = current_x;
	if (current_x_find < 0 || fileptr->data[current_x_find] != '\0')
	    current_x_find++;
Chris Allegretta's avatar
Chris Allegretta committed
265

Chris Allegretta's avatar
Chris Allegretta committed
266
	searchstr = &fileptr->data[current_x_find];
Chris Allegretta's avatar
Chris Allegretta committed
267

268
	/* Look for needle in searchstr.  Keep going until we find it
269
270
271
	 * and, if no_sameline is set, until it isn't on the current
	 * line.  If we don't find it, we'll end up at
	 * current[current_x] regardless of whether no_sameline is
272
	 * set. */
273
	while ((found = strstrwrapper(searchstr, needle, rev_start, current_x_find)) == NULL || (no_sameline && fileptr == current)) {
Chris Allegretta's avatar
Chris Allegretta committed
274

Chris Allegretta's avatar
Chris Allegretta committed
275
276
277
278
	    /* finished processing file, get out */
	    if (search_last_line) {
		if (!quiet)
		    not_found_msg(needle);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
279
		update_line(fileptr, current_x);
Chris Allegretta's avatar
Chris Allegretta committed
280
281
	        return NULL;
	    }
282

Chris Allegretta's avatar
Chris Allegretta committed
283
	    update_line(fileptr, 0);
284
285
286
287

	    /* reset current_x_find between lines */
	    current_x_find = 0;

Chris Allegretta's avatar
Chris Allegretta committed
288
	    fileptr = fileptr->next;
289

290
	    if (fileptr == editbot)
Chris Allegretta's avatar
Chris Allegretta committed
291
		search_offscreen = 1;
Chris Allegretta's avatar
Chris Allegretta committed
292

Chris Allegretta's avatar
Chris Allegretta committed
293
	    /* EOF reached?, wrap around once */
Chris Allegretta's avatar
Chris Allegretta committed
294
	    if (fileptr == NULL) {
295
296
		/* don't wrap if looking for bracket match */
		if (bracket_mode)
Chris Allegretta's avatar
Chris Allegretta committed
297
		    return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
298
		fileptr = fileage;
Chris Allegretta's avatar
Chris Allegretta committed
299
		search_offscreen = 1;
300
		if (!quiet)
Chris Allegretta's avatar
Chris Allegretta committed
301
		    statusbar(_("Search Wrapped"));
Chris Allegretta's avatar
Chris Allegretta committed
302
303
304
305
306
307
308
309
	    }

	    /* Original start line reached */
	    if (fileptr == begin)
		search_last_line = 1;

	    searchstr = fileptr->data;
	}
310

Chris Allegretta's avatar
Chris Allegretta committed
311
312
313
	/* We found an instance */
	current_x_find = found - fileptr->data;
	/* Ensure we haven't wrapped around again! */
Chris Allegretta's avatar
Chris Allegretta committed
314
	if (search_last_line && current_x_find > beginx) {
Chris Allegretta's avatar
Chris Allegretta committed
315
	    if (!quiet)
Chris Allegretta's avatar
Chris Allegretta committed
316
317
		not_found_msg(needle);
	    return NULL;
318
	}
319
    }
320
321
#ifndef NANO_SMALL
    else {	/* reverse search */
Chris Allegretta's avatar
Chris Allegretta committed
322
	current_x_find = current_x - 1;
Chris Allegretta's avatar
Chris Allegretta committed
323
	/* Make sure we haven't passed the beginning of the string */
Chris Allegretta's avatar
Chris Allegretta committed
324
	rev_start = &fileptr->data[current_x_find];
325
326
	searchstr = fileptr->data;

327
	/* Look for needle in searchstr.  Keep going until we find it
328
329
330
	 * and, if no_sameline is set, until it isn't on the current
	 * line.  If we don't find it, we'll end up at
	 * current[current_x] regardless of whether no_sameline is
331
	 * set. */
332
	while ((found = strstrwrapper(searchstr, needle, rev_start, current_x_find)) == NULL || (no_sameline && fileptr == current)) {
333

Chris Allegretta's avatar
Chris Allegretta committed
334
335
336
337
338
339
340
	    /* finished processing file, get out */
	    if (search_last_line) {
		if (!quiet)
		    not_found_msg(needle);
		return NULL;
	    }

Chris Allegretta's avatar
Chris Allegretta committed
341
	    update_line(fileptr, 0);
342
343
344
345

	    /* reset current_x_find between lines */
	    current_x_find = 0;

Chris Allegretta's avatar
Chris Allegretta committed
346
347
	    fileptr = fileptr->prev;

348
	    if (fileptr == edittop->prev)
Chris Allegretta's avatar
Chris Allegretta committed
349
		search_offscreen = 1;
350

351
	    /* SOF reached?, wrap around once */
Chris Allegretta's avatar
Chris Allegretta committed
352
/* ? */	    if (fileptr == NULL) {
353
354
		if (bracket_mode)
		   return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
355
		fileptr = filebot;
Chris Allegretta's avatar
Chris Allegretta committed
356
		search_offscreen = 1;
357
		if (!quiet)
Chris Allegretta's avatar
Chris Allegretta committed
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
		    statusbar(_("Search Wrapped"));
	    }
	    /* 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;
	/* Ensure we haven't wrapped around again! */
	if ((search_last_line) && (current_x_find < beginx)) {
	    if (!quiet)
		not_found_msg(needle);
	    return NULL;
	}
Chris Allegretta's avatar
Chris Allegretta committed
376
    }
Chris Allegretta's avatar
Chris Allegretta committed
377
#endif /* !NANO_SMALL */
Chris Allegretta's avatar
Chris Allegretta committed
378

Chris Allegretta's avatar
Chris Allegretta committed
379
    /* Set globals now that we are sure we found something */
380
381
382
    current = fileptr;
    current_x = current_x_find;

383
    if (!bracket_mode) {
384
	update_line(current, current_x);
385
386
387
	placewewant = xplustabs();
	reset_cursor();
    }
Chris Allegretta's avatar
Chris Allegretta committed
388
389
390
    return fileptr;
}

Chris Allegretta's avatar
Chris Allegretta committed
391
/* Search for a string. */
Chris Allegretta's avatar
Chris Allegretta committed
392
393
394
int do_search(void)
{
    int i;
395
396
    filestruct *fileptr = current, *didfind;
    int fileptr_x = current_x;
Chris Allegretta's avatar
Chris Allegretta committed
397

398
#ifndef DISABLE_WRAPPING
Chris Allegretta's avatar
Chris Allegretta committed
399
    wrap_reset();
400
#endif
401
402
403
    i = search_init(0);
    switch (i) {
    case -1:
Chris Allegretta's avatar
Chris Allegretta committed
404
405
406
	current = fileptr;
	search_abort();
	return 0;
407
    case -3:
Chris Allegretta's avatar
Chris Allegretta committed
408
409
	search_abort();
	return 0;
410
    case -2:
Chris Allegretta's avatar
Chris Allegretta committed
411
412
	do_replace();
	return 0;
413
    case 1:
Chris Allegretta's avatar
Chris Allegretta committed
414
415
416
417
	do_search();
	search_abort();
	return 1;
    }
418

Chris Allegretta's avatar
Chris Allegretta committed
419
     /* If answer is now "", copy last_search into answer... */
Chris Allegretta's avatar
Chris Allegretta committed
420
    if (answer[0] == '\0')
421
422
423
424
	answer = mallocstrcpy(answer, last_search);
    else
	last_search = mallocstrcpy(last_search, answer);

425
426
#ifndef NANO_SMALL
    /* add this search string to the search history list */
Chris Allegretta's avatar
Chris Allegretta committed
427
    if (answer[0] != '\0')
428
	update_history(&search_history, answer);
429
430
#endif	/* !NANO_SMALL */

431
    search_last_line = 0;
432
    didfind = findnextstr(FALSE, FALSE, current, current_x, answer, 0);
433

Chris Allegretta's avatar
Chris Allegretta committed
434
    if (fileptr == current && fileptr_x == current_x && didfind != NULL)
435
	statusbar(_("This is the only occurrence"));
436
    else if (current->lineno <= edittop->lineno
Chris Allegretta's avatar
Chris Allegretta committed
437
	|| current->lineno >= editbot->lineno)
438
        edit_update(current, CENTER);
439

Chris Allegretta's avatar
Chris Allegretta committed
440
    search_abort();
441

Chris Allegretta's avatar
Chris Allegretta committed
442
443
444
    return 1;
}

445
446
447
448
449
450
451
452
453
/* Search for the next string without prompting. */
int do_research(void)
{
    filestruct *fileptr = current, *didfind;
    int fileptr_x = current_x;
#ifdef HAVE_REGEX_H
    const char *regex_error = _("Invalid regex \"%s\"");
#endif /* HAVE_REGEX_H */

454
#ifndef DISABLE_WRAPPING
455
    wrap_reset();
456
#endif
457
458
459
460
461
462
463
464
465
466
467
468
469
470
    search_init_globals();

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

#ifdef HAVE_REGEX_H
	if (ISSET(USE_REGEXP))
	    if (regexp_init(last_search) == 0) {
		statusbar(regex_error, last_search);
		reset_cursor();
		return -3;
	    }
#endif

	search_last_line = 0;
471
	didfind = findnextstr(FALSE, FALSE, current, current_x, last_search, 0);
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486

	if (fileptr == current && fileptr_x == current_x && didfind != NULL)
	    statusbar(_("This is the only occurrence"));
	else if (current->lineno <= edittop->lineno
	    || current->lineno >= editbot->lineno)
	    edit_update(current, CENTER);

    } else
        statusbar(_("No current search pattern"));

    search_abort();

    return 1;
}

Chris Allegretta's avatar
Chris Allegretta committed
487
488
void replace_abort(void)
{
489
    /* Identical to search_abort, so we'll call it here.  If it
490
       does something different later, we can change it back.  For now
491
       it's just a waste to duplicate code */
492
    search_abort();
493
    placewewant = xplustabs();
494
495
}

496
#ifdef HAVE_REGEX_H
497
498
int replace_regexp(char *string, int create_flag)
{
Chris Allegretta's avatar
Chris Allegretta committed
499
    /* Split personality here - if create_flag is NULL, just calculate
500
     * the size of the replacement line (necessary because of
Chris Allegretta's avatar
Chris Allegretta committed
501
     * subexpressions like \1 \2 \3 in the replaced text). */
502
503
504

    char *c;
    int new_size = strlen(current->data) + 1;
505
    int search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so;
506
507
508

    new_size -= search_match_count;

Chris Allegretta's avatar
Chris Allegretta committed
509
510
    /* Iterate through the replacement text to handle subexpression
     * replacement using \1, \2, \3, etc. */
511
512

    c = last_replace;
513
    while (*c != '\0') {
514
515
516
517
518
519
	if (*c != '\\') {
	    if (create_flag)
		*string++ = *c;
	    c++;
	    new_size++;
	} else {
520
	    int num = (int) *(c + 1) - (int) '0';
521
522
	    if (num >= 1 && num <= 9) {

523
		int i = regmatches[num].rm_eo - regmatches[num].rm_so;
524
525
526

		if (num > search_regexp.re_nsub) {
		    /* Ugh, they specified a subexpression that doesn't
Chris Allegretta's avatar
Chris Allegretta committed
527
		     * exist. */
528
529
530
531
532
533
534
		    return -1;
		}

		/* Skip over the replacement expression */
		c += 2;

		/* But add the length of the subexpression to new_size */
535
		new_size += i;
536
537
538

		/* And if create_flag is set, append the result of the
		 * subexpression match to the new line */
539
540
541
542
543
		if (create_flag) {
		    strncpy(string, current->data + current_x +
			    regmatches[num].rm_so, i);
		    string += i;
		}
544
545
546
547
548
549
550
551

	    } else {
		if (create_flag)
		    *string++ = *c;
		c++;
		new_size++;
	    }
	}
552
553
554
    }

    if (create_flag)
Chris Allegretta's avatar
Chris Allegretta committed
555
	*string = '\0';
556
557
558

    return new_size;
}
559
#endif
560

Chris Allegretta's avatar
Chris Allegretta committed
561
char *replace_line(void)
562
563
564
565
566
567
{
    char *copy, *tmp;
    int new_line_size;
    int search_match_count;

    /* Calculate size of new line */
568
#ifdef HAVE_REGEX_H
569
    if (ISSET(USE_REGEXP)) {
570
571
572
	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
573
	 * text, return NULL, indicating an error */
574
575
	if (new_line_size < 0)
	    return NULL;
576
    } else {
577
578
579
#else
    {
#endif
580
581
582
	search_match_count = strlen(last_search);
	new_line_size = strlen(current->data) - strlen(last_search) +
	    strlen(last_replace) + 1;
583
    }
584

585
    /* Create buffer */
586
    copy = charalloc(new_line_size);
587
588
589

    /* Head of Original Line */
    strncpy(copy, current->data, current_x);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
590
    copy[current_x] = '\0';
591
592
593

    /* Replacement Text */
    if (!ISSET(USE_REGEXP))
594
	strcat(copy, last_replace);
595
#ifdef HAVE_REGEX_H
596
    else
597
	replace_regexp(copy + current_x, 1);
598
#endif
599
600

    /* The tail of the original line */
Chris Allegretta's avatar
Chris Allegretta committed
601
602
603
604
605
606

    /* 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 ;-) */
607
608
609
610
    tmp = current->data + current_x + search_match_count;
    strcat(copy, tmp);

    return copy;
Chris Allegretta's avatar
Chris Allegretta committed
611
612
}

Chris Allegretta's avatar
Chris Allegretta committed
613
614
615
/* Step through each replace word and prompt user before replacing
 * word.  Return -1 if the string to replace isn't found at all.
 * Otherwise, return the number of replacements made. */
Chris Allegretta's avatar
Chris Allegretta committed
616
617
int do_replace_loop(const char *prevanswer, const filestruct *begin,
			int *beginx, int wholewords, int *i)
Chris Allegretta's avatar
Chris Allegretta committed
618
{
Chris Allegretta's avatar
Chris Allegretta committed
619
    int replaceall = 0, numreplaced = -1;
620
#ifdef HAVE_REGEX_H
621
622
    /* The starting-line match and bol/eol regex flags. */
    int beginline = 0, bol_eol = 0;
623
#endif
624
    filestruct *fileptr = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
625

Chris Allegretta's avatar
Chris Allegretta committed
626
    switch (*i) {
627
628
629
630
631
632
633
634
635
636
637
	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. */
Chris Allegretta's avatar
Chris Allegretta committed
638
639
640
	    do_early_abort();
	    replace_abort();
	    return 0;
641
        }
Chris Allegretta's avatar
Chris Allegretta committed
642
643
    }

644
    last_replace = mallocstrcpy(last_replace, answer);
Chris Allegretta's avatar
Chris Allegretta committed
645
    while (1) {
646
647
	size_t match_len;

648
	/* Sweet optimization by Rocco here. */
Chris Allegretta's avatar
Chris Allegretta committed
649
	fileptr = findnextstr(fileptr || replaceall || search_last_line,
650
651
		FALSE, begin, *beginx, prevanswer,
#ifdef HAVE_REGEX_H
652
653
654
		/* We should find a bol and/or eol regex only once per
		 * line.  If the bol_eol flag is set, it means that the
		 * last search found one on the beginning line, so we
655
656
		 * should skip over the beginning line when doing this
		 * search. */
657
		bol_eol
658
659
660
661
662
663
#else
		0
#endif
		);

#ifdef HAVE_REGEX_H
664
	/* If the bol_eol flag is set, we've found a match on the
665
666
	 * beginning line already, and we're still on the beginning line
	 * after the search, it means that we've wrapped around, so
667
	 * we're done. */
668
	if (bol_eol && beginline && fileptr == begin)
669
	    fileptr = NULL;
670
	/* Otherwise, set the beginline flag if we've found a match on
671
	 * the beginning line, reset the bol_eol flag, and continue. */
672
	else {
673
674
	    if (fileptr == begin)
		beginline = 1;
675
	    bol_eol = 0;
676
677
	}
#endif
Chris Allegretta's avatar
Chris Allegretta committed
678

679
680
	if (current->lineno <= edittop->lineno
	    || current->lineno >= editbot->lineno)
681
	    edit_update(current, CENTER);
682

Chris Allegretta's avatar
Chris Allegretta committed
683
	/* No more matches.  Done! */
684
	if (fileptr == NULL)
Chris Allegretta's avatar
Chris Allegretta committed
685
686
	    break;

687
	/* Make sure only whole words are found. */
Chris Allegretta's avatar
Chris Allegretta committed
688
	if (wholewords && !is_whole_word(current_x, fileptr->data, prevanswer))
689
	    continue;
Chris Allegretta's avatar
Chris Allegretta committed
690

691
	/* If we're here, we've found the search string. */
Chris Allegretta's avatar
Chris Allegretta committed
692
693
694
	if (numreplaced == -1)
	    numreplaced = 0;

695
696
697
698
699
700
701
#ifdef HAVE_REGEX_H
	if (ISSET(USE_REGEXP))
	    match_len = regmatches[0].rm_eo - regmatches[0].rm_so;
	else
#endif
	    match_len = strlen(prevanswer);

702
	if (!replaceall) {
703
704
705
706
707
708
	    char *exp_word;
	    size_t xpt = xplustabs();

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

709
	    curs_set(0);
710
	    do_replace_highlight(TRUE, exp_word);
711

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
712
	    *i = do_yesno(1, _("Replace this instance?"));
Chris Allegretta's avatar
Chris Allegretta committed
713

714
715
	    do_replace_highlight(FALSE, exp_word);
	    free(exp_word);
716
	    curs_set(1);
717
718
719

	    if (*i == -1)	/* We canceled the replace. */
		break;
720
721
	}

722
#ifdef HAVE_REGEX_H
723
724
725
726
	/* Set the bol_eol flag if we're doing a bol and/or eol regex
	 * replace ("^", "$", or "^$"). */
	if (ISSET(USE_REGEXP) && regexec(&search_regexp, prevanswer, 1, NULL, REG_NOTBOL | REG_NOTEOL) == REG_NOMATCH)
	    bol_eol = 1;
727
728
#endif

Chris Allegretta's avatar
Chris Allegretta committed
729
	if (*i > 0 || replaceall) {	/* Yes, replace it!!!! */
730
731
	    char *copy;
	    int length_change;
732

Chris Allegretta's avatar
Chris Allegretta committed
733
	    if (*i == 2)
Chris Allegretta's avatar
Chris Allegretta committed
734
735
		replaceall = 1;

736
	    copy = replace_line();
737
	    if (copy == NULL) {
Jordi Mallach's avatar
   
Jordi Mallach committed
738
		statusbar(_("Replace failed: unknown subexpression!"));
739
		replace_abort();
Chris Allegretta's avatar
Chris Allegretta committed
740
		return 0;
741
	    }
Chris Allegretta's avatar
Chris Allegretta committed
742

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

745
746
747
748
749
750
751
752
#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
753

754
755
756
757
758
759
	    assert(0 <= match_len + length_change);
	    if (current == begin && current_x <= *beginx) {
		if (*beginx < current_x + match_len)
		    *beginx = current_x + match_len;
		*beginx += length_change;
	    }
760

761
	    /* Set the cursor at the last character of the replacement
762
763
	     * text, so searching will resume after the replacement
	     * text.  Note that current_x might be set to -1 here. */
764
#ifndef NANO_SMALL
765
	    if (!ISSET(REVERSE_SEARCH))
766
767
#endif
		current_x += match_len + length_change - 1;
768

769
	    /* Cleanup. */
770
771
772
	    totsize += length_change;
	    free(current->data);
	    current->data = copy;
773

Chris Allegretta's avatar
Chris Allegretta committed
774
775
776
	    edit_refresh();
	    set_modified();
	    numreplaced++;
777
	}
Chris Allegretta's avatar
Chris Allegretta committed
778
779
    }

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

Chris Allegretta's avatar
Chris Allegretta committed
784
785
786
    return numreplaced;
}

Chris Allegretta's avatar
Chris Allegretta committed
787
/* Replace a string. */
Chris Allegretta's avatar
Chris Allegretta committed
788
789
790
791
int do_replace(void)
{
    int i, numreplaced, beginx;
    filestruct *begin;
Chris Allegretta's avatar
Chris Allegretta committed
792
    char *prevanswer = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
793

Chris Allegretta's avatar
Chris Allegretta committed
794
795
796
797
798
799
    if (ISSET(VIEW_MODE)) {
	print_view_warning();
	replace_abort();
	return 0;
    }

Chris Allegretta's avatar
Chris Allegretta committed
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
    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;
    }

817
#ifndef NANO_SMALL
Chris Allegretta's avatar
Chris Allegretta committed
818
    if (answer[0] != '\0')
819
	update_history(&search_history, answer);
820
821
#endif	/* !NANO_SMALL */

822
823
    /* If answer is now == "", copy last_search into answer 
	(and prevanswer)...  */
Chris Allegretta's avatar
Chris Allegretta committed
824
    if (answer[0] == '\0')
825
	answer = mallocstrcpy(answer, last_search);
Chris Allegretta's avatar
Chris Allegretta committed
826
    else
Chris Allegretta's avatar
Chris Allegretta committed
827
	last_search = mallocstrcpy(last_search, answer);
828

Chris Allegretta's avatar
Chris Allegretta committed
829
830
    prevanswer = mallocstrcpy(prevanswer, last_search);

831
#ifndef NANO_SMALL
832
833
    replace_history.current = (historytype *)&replace_history.next;
    last_replace = mallocstrcpy(last_replace, "");
834
#endif
835
836

    i = statusq(0, replace_list_2, last_replace,
837
838
839
840
#ifndef NANO_SMALL
		&replace_history,
#endif
		_("Replace with"));
841

842
#ifndef NANO_SMALL
Chris Allegretta's avatar
Chris Allegretta committed
843
    if (i == 0 && answer[0] != '\0')
844
845
	update_history(&replace_history, answer);
#endif	/* !NANO_SMALL */
Chris Allegretta's avatar
Chris Allegretta committed
846
847

    begin = current;
848
    beginx = current_x;
Chris Allegretta's avatar
Chris Allegretta committed
849
850
851
852
853
    search_last_line = 0;

    numreplaced = do_replace_loop(prevanswer, begin, &beginx, FALSE, &i);

    /* restore where we were */
Chris Allegretta's avatar
Chris Allegretta committed
854
    current = begin;
855
    current_x = beginx;
Chris Allegretta's avatar
Chris Allegretta committed
856
    renumber_all();
857
    edit_update(current, CENTER);
Chris Allegretta's avatar
Chris Allegretta committed
858
859

    if (numreplaced >= 0)
860
	statusbar(P_("Replaced %d occurrence", "Replaced %d occurrences",
Chris Allegretta's avatar
Chris Allegretta committed
861
862
863
864
		numreplaced), numreplaced);
    else
	not_found_msg(prevanswer);

865
    free(prevanswer);
Chris Allegretta's avatar
Chris Allegretta committed
866
867
868
869
    replace_abort();
    return 1;
}

870
int do_gotoline(int line, int save_pos)
Chris Allegretta's avatar
Chris Allegretta committed
871
{
872
    if (line <= 0) {		/* Ask for it */
Chris Allegretta's avatar
Chris Allegretta committed
873
	int st = statusq(FALSE, goto_list, line != 0 ? answer : "",
Chris Allegretta's avatar
Chris Allegretta committed
874
#ifndef NANO_SMALL
Chris Allegretta's avatar
Chris Allegretta committed
875
			NULL,
Chris Allegretta's avatar
Chris Allegretta committed
876
#endif
Chris Allegretta's avatar
Chris Allegretta committed
877
878
879
880
			_("Enter line number"));

	/* Cancel, or Enter with blank string. */
	if (st == -1 || st == -2)
Chris Allegretta's avatar
Chris Allegretta committed
881
	    statusbar(_("Aborted"));
Chris Allegretta's avatar
Chris Allegretta committed
882
	if (st != 0) {
Chris Allegretta's avatar
Chris Allegretta committed
883
	    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
884
885
	    return 0;
	}
886
887
888
889
890
891

	line = atoi(answer);

	/* Bounds check */
	if (line <= 0) {
	    statusbar(_("Come on, be reasonable"));
Chris Allegretta's avatar
Chris Allegretta committed
892
	    display_main_list();
893
	    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
894
895
896
	}
    }

897
    for (current = fileage; current->next != NULL && line > 1; line--)
898
	current = current->next;
Chris Allegretta's avatar
Chris Allegretta committed
899

900
    current_x = 0;
901

902
    /* if save_pos is nonzero, don't change the cursor position when
903
904
905
906
907
       updating the edit window */
    if (save_pos)
    	edit_update(current, NONE);
    else
	edit_update(current, CENTER);
908
    placewewant = 0;
Chris Allegretta's avatar
Chris Allegretta committed
909
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
910
911
912
913
914
    return 1;
}

int do_gotoline_void(void)
{
915
    return do_gotoline(0, 0);
Chris Allegretta's avatar
Chris Allegretta committed
916
}
917

918
#if defined(ENABLE_MULTIBUFFER) || !defined(DISABLE_SPELLER)
919
void do_gotopos(int line, int pos_x, int pos_y, int pos_placewewant)
920
921
922
923
924
{
    /* 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);
925

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
926
927
928
    /* make sure that the x-coordinate is sane here */
    if (pos_x > strlen(current->data))
	pos_x = strlen(current->data);
929
930

    /* set the rest of the coordinates up */
931
932
933
934
935
    current_x = pos_x;
    placewewant = pos_placewewant;
    update_line(current, pos_x);
}
#endif
936
937
938
939
940

#if !defined(NANO_SMALL) && defined(HAVE_REGEX_H)
int do_find_bracket(void)
{
    char ch_under_cursor, wanted_ch;
941
    const char *pos, *brackets = "([{<>}])";
942
    char regexp_pat[] = "[  ]";
Chris Allegretta's avatar
Chris Allegretta committed
943
    int offset, have_search_offscreen = 0, flagsave, current_x_save, count = 1;
944
945
946
    filestruct *current_save;

    ch_under_cursor = current->data[current_x];
947
948
949

/*    if ((!(pos = strchr(brackets, ch_under_cursor))) || (!((offset = pos - brackets) < 8))) { */

950
    if (((pos = strchr(brackets, ch_under_cursor)) == NULL) || (((offset = pos - brackets) < 8) == 0)) {
951
952
953
954
955
956
957
958
959
960
961
962
963
	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);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
964
/* apparent near redundancy with regexp_pat[] here is needed, [][] works, [[]] doesn't */
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979

    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;
980
	if (findnextstr(TRUE, TRUE, current, current_x, regexp_pat, 0) != NULL) {
Chris Allegretta's avatar
Chris Allegretta committed
981
982
983
984
	    have_search_offscreen |= search_offscreen;

	    /* found identical bracket */
	    if (current->data[current_x] == ch_under_cursor)
985
		count++;
Chris Allegretta's avatar
Chris Allegretta committed
986
987
988
	    else {

		/* found complementary bracket */
989
		if (!(--count)) {
Chris Allegretta's avatar
Chris Allegretta committed
990
		    if (have_search_offscreen)
991
992
993
994
995
			edit_update(current, CENTER);
		    else
			update_line(current, current_x);
		    placewewant = xplustabs();
		    reset_cursor();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
996
		    break;
997
998
		}
	    }
Chris Allegretta's avatar
Chris Allegretta committed
999
1000
1001
	} else {

	    /* didn't find either left or right bracket */
1002
1003
1004
	    statusbar(_("No matching bracket"));
	    current_x = current_x_save;
	    current = current_save;
Chris Allegretta's avatar
Chris Allegretta committed
1005
	    update_line(current, current_x);
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
	    break;
	}
    }

    if (ISSET(REGEXP_COMPILED))
	regexp_cleanup();
    flags = flagsave;
    return 0;
}
#endif
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042

#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
1043
    for (; h->next != NULL; h = h->next)
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
	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
1063
    a = (historytype *)nmalloc(sizeof(historytype));
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
    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
1076
1077
1078
    if ((p = find_node(h->next, s)) != NULL) {
	if (p == h->next)		/* catch delete and re-insert of
					   same string in 1st node */
1079
	    goto up_hs;
Chris Allegretta's avatar
Chris Allegretta committed
1080
	remove_node(p);			/* delete identical older string */
1081
1082
1083
1084
1085
1086
1087
1088
	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++;
1089
    SET(HISTORY_CHANGED);
Chris Allegretta's avatar
Chris Allegretta committed
1090
  up_hs:
1091
1092
1093
1094
1095
1096
    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
1097
    if (h->current->next != NULL) {	/* any older entries? */
1098
1099
1100
1101
1102
1103
1104
1105
	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
1106
    if (h->current->prev != NULL) {
1107
	h->current = h->current->prev;
Chris Allegretta's avatar
Chris Allegretta committed
1108
	if (h->current->prev != NULL)
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
	    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
1119
1120
    for (p = h->current->next; p->next != NULL; p = p->next) {
	if (strncmp(s, p->data, h->len) == 0 && strlen(p->data) != h->len) {
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
	    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
1135
    for (p = h->next; (n = p->next); p = n)
1136
1137
1138
1139
1140
	remove_node(p);
}

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