search.c 23 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-2002 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>
Chris Allegretta's avatar
Chris Allegretta committed
29
30
#include "proto.h"
#include "nano.h"
Chris Allegretta's avatar
Chris Allegretta committed
31

Chris Allegretta's avatar
Chris Allegretta committed
32
33
34
35
36
37
38
#ifndef NANO_SMALL
#include <libintl.h>
#define _(string) gettext(string)
#else
#define _(string) (string)
#endif

39
40
/* Regular expression helper functions */

41
#ifdef HAVE_REGEX_H
42
43
44
45
46
47
void regexp_init(const char *regexp)
{
    regcomp(&search_regexp, regexp, ISSET(CASE_SENSITIVE) ? 0 : REG_ICASE);
    SET(REGEXP_COMPILED);
}

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

Chris Allegretta's avatar
Chris Allegretta committed
55
56
57
void search_init_globals(void)
{
    if (last_search == NULL) {
58
	last_search = charalloc(1);
Chris Allegretta's avatar
Chris Allegretta committed
59
60
61
	last_search[0] = 0;
    }
    if (last_replace == NULL) {
62
	last_replace = charalloc(1);
Chris Allegretta's avatar
Chris Allegretta committed
63
64
65
66
	last_replace[0] = 0;
    }
}

Chris Allegretta's avatar
Chris Allegretta committed
67
68
/* Set up the system variables for a search or replace.  Returns -1 on
   abort, 0 on success, and 1 on rerun calling program 
69
   Return -2 to run opposite program (search -> replace, replace -> search)
Chris Allegretta's avatar
Chris Allegretta committed
70
71
72
73
74

   replacing = 1 if we call from do_replace, 0 if called from do_search func.
*/
int search_init(int replacing)
{
75
    int i = 0;
76
    char *buf;
77
    static char *backupstring = NULL;
78
#ifndef NANO_SMALL
79
    toggle *t;
80
#endif
81

Chris Allegretta's avatar
Chris Allegretta committed
82
    search_init_globals();
83

84
    buf = charalloc(strlen(last_search) + 5);
Chris Allegretta's avatar
Chris Allegretta committed
85
    buf[0] = 0;
86

Chris Allegretta's avatar
Chris Allegretta committed
87
88
89
90
91
92

    /* Clear the backupstring if we've changed from Pico mode to regular
	mode */
    if (ISSET(CLEAR_BACKUPSTRING)) {
	free(backupstring);
	backupstring = NULL;
93
	UNSET(CLEAR_BACKUPSTRING);
Chris Allegretta's avatar
Chris Allegretta committed
94
95
    }
	
96
97
     /* Okay, fun time.  backupstring is our holder for what is being 
	returned from the statusq call.  Using answer for this would be tricky.
98
	Here, if we're using PICO_MODE, we only want nano to put the
99
100
101
102
103
	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.  */

104
    if (ISSET(PICO_MODE)) {
105
106
107
108
109
110
	if (backupstring == NULL || !strcmp(backupstring, last_search))
	    backupstring = mallocstrcpy(backupstring, "");
    }
    else if (backupstring == NULL)
	backupstring = mallocstrcpy(backupstring, last_search);

Chris Allegretta's avatar
Chris Allegretta committed
111
    /* If using Pico messages, we do things the old fashioned way... */
112
    if (ISSET(PICO_MODE)) {
113
114
115
116
117
118
119
120
121
122
123
	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';
	}
Chris Allegretta's avatar
Chris Allegretta committed
124
    }
125
126
    else
	strcpy(buf, "");
Chris Allegretta's avatar
Chris Allegretta committed
127

128
    /* This is now one simple call.  It just does a lot */
129
    i = statusq(0, replacing ? replace_list : whereis_list, backupstring,
130
	"%s%s%s%s%s%s", 
Chris Allegretta's avatar
Chris Allegretta committed
131
	_("Search"),
132
133
134
135
136
137
138
139
140
141
142
143
144

	/* 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]") : "",

Chris Allegretta's avatar
Chris Allegretta committed
145
146
	replacing ? _(" (to replace)") : "",
	buf);
Chris Allegretta's avatar
Chris Allegretta committed
147

Chris Allegretta's avatar
Chris Allegretta committed
148
149
150
    /* Release buf now that we don't need it anymore */
    free(buf);

Chris Allegretta's avatar
Chris Allegretta committed
151
152
153
154
    /* Cancel any search, or just return with no previous search */
    if ((i == -1) || (i < 0 && !last_search[0])) {
	statusbar(_("Search Cancelled"));
	reset_cursor();
155
156
	free(backupstring);
	backupstring = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
157
	return -1;
158
159
160
161
    } else 
    switch (i) {

    case -2:	/* Same string */
162
#ifdef HAVE_REGEX_H
163
164
165
166
167
168
169
170
	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);
	}
171
#endif
172
173
	break;
    case 0:		/* They entered something new */
174
#ifdef HAVE_REGEX_H
175
176
	if (ISSET(USE_REGEXP))
	    regexp_init(answer);
177
#endif
178
179
	free(backupstring);
	backupstring = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
180
	last_replace[0] = '\0';
181
182
183
184
185
186
	break;
    case TOGGLE_CASE_KEY:
    case TOGGLE_BACKWARDS_KEY:
#ifdef HAVE_REGEX_H
    case TOGGLE_REGEXP_KEY:
#endif
187
188
189
190
	free(backupstring);
	backupstring = NULL;
	backupstring = mallocstrcpy(backupstring, answer);

191
#ifndef NANO_SMALL
192
193
194
	for (t = toggles; t != NULL; t = t->next)
	    if (i == t->val)
		TOGGLE(t->flag);
195
#endif
Chris Allegretta's avatar
Chris Allegretta committed
196
197

	return 1;
198
    case NANO_OTHERSEARCH_KEY:
199
	backupstring = mallocstrcpy(backupstring, answer);
Chris Allegretta's avatar
Chris Allegretta committed
200
	return -2;		/* Call the opposite search function */
201
    case NANO_FROMSEARCHTOGOTO_KEY:
202
203
	free(backupstring);
	backupstring = NULL;
204
205
	do_gotoline_void();
	return -3;
206
    default:
Chris Allegretta's avatar
Chris Allegretta committed
207
	do_early_abort();
208
209
	free(backupstring);
	backupstring = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
210
211
212
213
214
215
	return -3;
    }

    return 0;
}

216
217
void not_found_msg(char *str)
{
218
    if (strlen(str) <= COLS / 2)
219
220
	statusbar(_("\"%s\" not found"), str);
    else {
221
222
223
	char *foo = NULL;

	foo = mallocstrcpy(foo, str);
224
225
	foo[COLS / 2] = 0;
	statusbar(_("\"%s...\" not found"), foo);
226
227

	free(foo);
228
229
230
    }
}

231
232
233
234
235
236
237
238
239
240
241
242
243
244
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;
}

245
int past_editbuff;	/* findnextstr() is now searching lines not displayed */
246
247

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

255
256
    past_editbuff = 0;

Chris Allegretta's avatar
Chris Allegretta committed
257
    if (!ISSET(REVERSE_SEARCH)) {		/* forward search */
Chris Allegretta's avatar
Chris Allegretta committed
258

Chris Allegretta's avatar
Chris Allegretta committed
259
	current_x_find = current_x + 1;
260
261
#if 0
	/* Are we now back to the place where the search started) */
262
	if ((fileptr == begin) && (beginx > current_x_find))
Chris Allegretta's avatar
Chris Allegretta committed
263
	    search_last_line = 1;
264
#endif
Chris Allegretta's avatar
Chris Allegretta committed
265
266
267
	/* Make sure we haven't passed the end of the string */
	if (strlen(fileptr->data) < current_x_find)
	    current_x_find--;
Chris Allegretta's avatar
Chris Allegretta committed
268

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

Chris Allegretta's avatar
Chris Allegretta committed
271
272
	/* Look for needle in searchstr */
	while ((found = strstrwrapper(searchstr, needle, rev_start)) == NULL) {
Chris Allegretta's avatar
Chris Allegretta committed
273

Chris Allegretta's avatar
Chris Allegretta committed
274
275
276
277
278
279
	    /* finished processing file, get out */
	    if (search_last_line) {
		if (!quiet)
		    not_found_msg(needle);
	        return NULL;
	    }
280

Chris Allegretta's avatar
Chris Allegretta committed
281
	    fileptr = fileptr->next;
282

283
	    if (fileptr == editbot)
284
		past_editbuff = 1;
Chris Allegretta's avatar
Chris Allegretta committed
285
286
287

	    /* EOF reached ?, wrap around once */
	    if (fileptr == NULL) {
288
289
		if (bracket_mode)		/* don't wrap if looking for bracket match */
		   return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
290
		fileptr = fileage;
291
		past_editbuff = 1;
292
		if (!quiet) {
293
		   statusbar(_("Search Wrapped"));
294
295
		SET(DISABLE_CURPOS);
		}
Chris Allegretta's avatar
Chris Allegretta committed
296
297
298
299
300
301
302
303
	    }

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

	    searchstr = fileptr->data;
	}
304

Chris Allegretta's avatar
Chris Allegretta committed
305
306
	/* We found an instance */
	current_x_find = found - fileptr->data;
307
#if 1
Chris Allegretta's avatar
Chris Allegretta committed
308
	/* Ensure we haven't wrapped around again! */
309
	if ((search_last_line) && (current_x_find > beginx)) {
Chris Allegretta's avatar
Chris Allegretta committed
310
	    if (!quiet)
Chris Allegretta's avatar
Chris Allegretta committed
311
312
		not_found_msg(needle);
	    return NULL;
313
	}
314
#endif
315
316
317
    } 
#ifndef NANO_SMALL
    else {	/* reverse search */
Chris Allegretta's avatar
Chris Allegretta committed
318
319

	current_x_find = current_x - 1;
320
#if 0
321
	/* Are we now back to the place where the search started) */
Chris Allegretta's avatar
Chris Allegretta committed
322
	if ((fileptr == begin) && (current_x_find > beginx))
323
	    search_last_line = 1;
324
#endif
Chris Allegretta's avatar
Chris Allegretta committed
325
	/* Make sure we haven't passed the begining of the string */
326
#if 0	/* Is this required here ? */
Chris Allegretta's avatar
Chris Allegretta committed
327
328
329
330
	if (!(&fileptr->data[current_x_find] - fileptr->data))      
	    current_x_find++;
#endif
	rev_start = &fileptr->data[current_x_find];
331
332
	searchstr = fileptr->data;

Chris Allegretta's avatar
Chris Allegretta committed
333
334
335
336
337
338
339
340
341
342
343
344
	/* Look for needle in searchstr */
	while ((found = strstrwrapper(searchstr, needle, rev_start)) == NULL) {

	    /* finished processing file, get out */
	    if (search_last_line) {
		if (!quiet)
		    not_found_msg(needle);
		return NULL;
	    }

	    fileptr = fileptr->prev;

345
	    if (fileptr == edittop->prev)
346
		past_editbuff = 1;
347

Chris Allegretta's avatar
Chris Allegretta committed
348
349
	    /* SOF reached ?, wrap around once */
/* ? */	    if (fileptr == NULL) {
350
351
		if (bracket_mode)
		   return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
352
		fileptr = filebot;
353
		past_editbuff = 1;
354
		if (!quiet) {
Chris Allegretta's avatar
Chris Allegretta committed
355
		    statusbar(_("Search Wrapped"));
356
357
		    SET(DISABLE_CURPOS);
		}
Chris Allegretta's avatar
Chris Allegretta committed
358
359
360
361
362
363
364
365
366
367
368
369
	    }

	    /* 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;
370
#if 1
Chris Allegretta's avatar
Chris Allegretta committed
371
372
373
374
375
376
	/* Ensure we haven't wrapped around again! */
	if ((search_last_line) && (current_x_find < beginx)) {
	    if (!quiet)
		not_found_msg(needle);
	    return NULL;
	}
377
#endif
Chris Allegretta's avatar
Chris Allegretta committed
378
    }
379
#endif /* NANO_SMALL */
Chris Allegretta's avatar
Chris Allegretta committed
380

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

385
386
387
388
389
    if (!bracket_mode) {
	if (past_editbuff)
	   edit_update(fileptr, CENTER);
	else
	   update_line(current, current_x);
390

391
392
393
	placewewant = xplustabs();
	reset_cursor();
    }
394

Chris Allegretta's avatar
Chris Allegretta committed
395
396
397
398
399
400
401
402
    return fileptr;
}

void search_abort(void)
{
    UNSET(KEEP_CUTBUFFER);
    display_main_list();
    wrefresh(bottomwin);
403
    if (ISSET(MARK_ISSET))
404
	edit_refresh_clearok();
405

406
#ifdef HAVE_REGEX_H
407
    if (ISSET(REGEXP_COMPILED))
408
	regexp_cleanup();
409
#endif
Chris Allegretta's avatar
Chris Allegretta committed
410
411
412
413
414
415
}

/* Search for a string */
int do_search(void)
{
    int i;
416
417
    filestruct *fileptr = current, *didfind;
    int fileptr_x = current_x;
Chris Allegretta's avatar
Chris Allegretta committed
418
419

    wrap_reset();
420
421
422
    i = search_init(0);
    switch (i) {
    case -1:
Chris Allegretta's avatar
Chris Allegretta committed
423
424
425
	current = fileptr;
	search_abort();
	return 0;
426
    case -3:
Chris Allegretta's avatar
Chris Allegretta committed
427
428
	search_abort();
	return 0;
429
    case -2:
Chris Allegretta's avatar
Chris Allegretta committed
430
431
	do_replace();
	return 0;
432
    case 1:
Chris Allegretta's avatar
Chris Allegretta committed
433
434
435
436
	do_search();
	search_abort();
	return 1;
    }
437
438

    /* The sneaky user deleted the previous search string */
439
    if (!ISSET(PICO_MODE) && !strcmp(answer, "")) {
Jordi Mallach's avatar
   
Jordi Mallach committed
440
	statusbar(_("Search Cancelled"));
441
442
443
444
	search_abort();
	return 0;
    }

445
     /* If answer is now == "", then PICO_MODE is set.  So, copy
446
447
448
449
450
451
452
	last_search into answer... */

    if (!strcmp(answer, ""))
	answer = mallocstrcpy(answer, last_search);
    else
	last_search = mallocstrcpy(last_search, answer);

453
    search_last_line = 0;
454
455
456
457
458
459
    didfind = findnextstr(FALSE, FALSE, current, current_x, answer);

    if ((fileptr == current) && (fileptr_x == current_x) &&
	didfind != NULL)
	statusbar(_("This is the only occurrence"));

Chris Allegretta's avatar
Chris Allegretta committed
460
    search_abort();
461

Chris Allegretta's avatar
Chris Allegretta committed
462
463
464
465
466
467
    return 1;
}

void print_replaced(int num)
{
    if (num > 1)
Jordi Mallach's avatar
   
Jordi Mallach committed
468
	statusbar(_("Replaced %d occurrences"), num);
Chris Allegretta's avatar
Chris Allegretta committed
469
    else if (num == 1)
Jordi Mallach's avatar
   
Jordi Mallach committed
470
	statusbar(_("Replaced 1 occurrence"));
Chris Allegretta's avatar
Chris Allegretta committed
471
472
473
474
}

void replace_abort(void)
{
475
    /* Identical to search_abort, so we'll call it here.  If it
476
       does something different later, we can change it back.  For now
477
       it's just a waste to duplicate code */
478
    search_abort();
479
    placewewant = xplustabs();
480
481
}

482
#ifdef HAVE_REGEX_H
483
484
485
486
487
488
489
490
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;
491
    int search_match_count = regmatches[0].rm_eo - regmatches[0].rm_so;
492
493
494
495
496
497
498
499

    new_size -= search_match_count;

    /* Iterate through the replacement text to handle
     * subexpression replacement using \1, \2, \3, etc */

    c = last_replace;
    while (*c) {
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
	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++;
	    }
	}
535
536
537
    }

    if (create_flag)
538
	*string = 0;
539
540
541

    return new_size;
}
542
#endif
543

Chris Allegretta's avatar
Chris Allegretta committed
544
char *replace_line(void)
545
546
547
548
549
550
{
    char *copy, *tmp;
    int new_line_size;
    int search_match_count;

    /* Calculate size of new line */
551
#ifdef HAVE_REGEX_H
552
    if (ISSET(USE_REGEXP)) {
553
554
555
	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
556
	 * text, return NULL, indicating an error */
557
558
	if (new_line_size < 0)
	    return NULL;
559
    } else {
560
561
562
#else
    {
#endif
563
564
565
	search_match_count = strlen(last_search);
	new_line_size = strlen(current->data) - strlen(last_search) +
	    strlen(last_replace) + 1;
566
    }
567

568
    /* Create buffer */
569
    copy = charalloc(new_line_size);
570
571
572
573
574
575
576

    /* Head of Original Line */
    strncpy(copy, current->data, current_x);
    copy[current_x] = 0;

    /* Replacement Text */
    if (!ISSET(USE_REGEXP))
577
	strcat(copy, last_replace);
578
#ifdef HAVE_REGEX_H
579
    else
580
	(void) replace_regexp(copy + current_x, 1);
581
#endif
582
583
584

    /* The tail of the original line */
    /* This may expose other bugs, because it no longer
585
       goes through each character in the string
586
587
588
589
590
591
592
593
       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;
Chris Allegretta's avatar
Chris Allegretta committed
594
595
}

596
/* step through each replace word and prompt user before replacing word */
Chris Allegretta's avatar
Chris Allegretta committed
597
598
int do_replace_loop(char *prevanswer, filestruct *begin, int *beginx,
			int wholewords, int *i)
Chris Allegretta's avatar
Chris Allegretta committed
599
{
Chris Allegretta's avatar
Chris Allegretta committed
600
    int replaceall = 0, numreplaced = 0;
601
602

    filestruct *fileptr = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
603
    char *copy;
Chris Allegretta's avatar
Chris Allegretta committed
604

Chris Allegretta's avatar
Chris Allegretta committed
605
    switch (*i) {
606
    case -1:				/* Aborted enter */
607
	if (strcmp(last_replace, ""))
608
	    answer = mallocstrcpy(answer, last_replace);
609
610
611
	statusbar(_("Replace Cancelled"));
	replace_abort();
	return 0;
612
613
614
    case 0:		/* They actually entered something */
	break;
    default:
Chris Allegretta's avatar
Chris Allegretta committed
615
        if (*i != -2) {	/* First page, last page, for example 
616
				   could get here */
Chris Allegretta's avatar
Chris Allegretta committed
617
618
619
	    do_early_abort();
	    replace_abort();
	    return 0;
620
        }
Chris Allegretta's avatar
Chris Allegretta committed
621
622
    }

623
624
625
    if (ISSET(PICO_MODE) && !strcmp(answer, ""))
	answer = mallocstrcpy(answer, last_replace);

626
    last_replace = mallocstrcpy(last_replace, answer);
Chris Allegretta's avatar
Chris Allegretta committed
627
628
    while (1) {

629
	/* Sweet optimization by Rocco here */
630
#if 0
631
	fileptr = findnextstr(replaceall, FALSE, begin, *beginx, prevanswer);
632
633
634
635
636
637
#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
Chris Allegretta's avatar
Chris Allegretta committed
638
639

	/* No more matches.  Done! */
640
	if (!fileptr)
Chris Allegretta's avatar
Chris Allegretta committed
641
642
	    break;

643
	/* Make sure only whole words are found */
644
645
	if ((wholewords) && (!is_whole_word(current_x, fileptr, prevanswer)))
	    continue;
Chris Allegretta's avatar
Chris Allegretta committed
646

Chris Allegretta's avatar
Chris Allegretta committed
647
	/* If we're here, we've found the search string */
648
649
650
651
652
	if (!replaceall) {

	    curs_set(0);
	    do_replace_highlight(TRUE, prevanswer);

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

655
656
657
658
	    do_replace_highlight(FALSE, prevanswer);
	    curs_set(1);
	}

Chris Allegretta's avatar
Chris Allegretta committed
659
660
	if (*i > 0 || replaceall) {	/* Yes, replace it!!!! */
	    if (*i == 2)
Chris Allegretta's avatar
Chris Allegretta committed
661
662
		replaceall = 1;

663
664
	    copy = replace_line();
	    if (!copy) {
Jordi Mallach's avatar
   
Jordi Mallach committed
665
		statusbar(_("Replace failed: unknown subexpression!"));
666
		replace_abort();
Chris Allegretta's avatar
Chris Allegretta committed
667
		return 0;
668
	    }
Chris Allegretta's avatar
Chris Allegretta committed
669
670

	    /* Cleanup */
671
	    totsize -= strlen(current->data);
Chris Allegretta's avatar
Chris Allegretta committed
672
673
	    free(current->data);
	    current->data = copy;
674
	    totsize += strlen(current->data);
Chris Allegretta's avatar
Chris Allegretta committed
675

676
677
678
	    if (!ISSET(REVERSE_SEARCH)) {
		/* Stop bug where we replace a substring of the replacement text */
		current_x += strlen(last_replace) - 1;
Chris Allegretta's avatar
Chris Allegretta committed
679

680
681
682
		/* Adjust the original cursor position - COULD BE IMPROVED */
		if (search_last_line) {
		    *beginx += strlen(last_replace) - strlen(last_search);
683

684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
		    /* 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);
		}
699
700
	    }

Chris Allegretta's avatar
Chris Allegretta committed
701
702
703
	    edit_refresh();
	    set_modified();
	    numreplaced++;
Chris Allegretta's avatar
Chris Allegretta committed
704
	} else if (*i == -1)	/* Abort, else do nothing and continue loop */
Chris Allegretta's avatar
Chris Allegretta committed
705
706
707
	    break;
    }

Chris Allegretta's avatar
Chris Allegretta committed
708
709
710
711
712
713
714
715
716
717
    return numreplaced;
}

/* Replace a string */
int do_replace(void)
{
    int i, numreplaced, beginx;
    filestruct *begin;
    char *prevanswer = NULL, *buf = NULL;

Chris Allegretta's avatar
Chris Allegretta committed
718
719
720
721
722
723
    if (ISSET(VIEW_MODE)) {
	print_view_warning();
	replace_abort();
	return 0;
    }

Chris Allegretta's avatar
Chris Allegretta committed
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
    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;
    }

741
    /* Again, there was a previous string, but they deleted it and hit enter */
742
    if (!ISSET(PICO_MODE) && !strcmp(answer, "")) {
Chris Allegretta's avatar
Chris Allegretta committed
743
744
745
746
747
	statusbar(_("Replace Cancelled"));
	replace_abort();
	return 0;
    }

748
     /* If answer is now == "", then PICO_MODE is set.  So, copy
749
750
751
752
753
754
755
756
	last_search into answer (and prevanswer)... */
    if (!strcmp(answer, "")) {
	answer = mallocstrcpy(answer, last_search);
	prevanswer = mallocstrcpy(prevanswer, last_search);
    } else {
	last_search = mallocstrcpy(last_search, answer);
	prevanswer = mallocstrcpy(prevanswer, answer);
    }
Chris Allegretta's avatar
Chris Allegretta committed
757

758
    if (ISSET(PICO_MODE)) {
759
	buf = charalloc(strlen(last_replace) + 5);
Chris Allegretta's avatar
Chris Allegretta committed
760
761
762
763
764
765
766
	if (strcmp(last_replace, "")) {
	    if (strlen(last_replace) > (COLS / 3)) {
		strncpy(buf, last_replace, COLS / 3);
		sprintf(&buf[COLS / 3 - 1], "...");
	    } else
		sprintf(buf, "%s", last_replace);

767
	    i = statusq(0, replace_list_2, "",
Chris Allegretta's avatar
Chris Allegretta committed
768
769
770
			_("Replace with [%s]"), buf);
	}
	else
771
	    i = statusq(0, replace_list_2, "",
Chris Allegretta's avatar
Chris Allegretta committed
772
773
774
			_("Replace with"));
    }
    else
775
	i = statusq(0, replace_list_2, last_replace, 
Chris Allegretta's avatar
Chris Allegretta committed
776
777
778
779
			_("Replace with"));

    /* save where we are */
    begin = current;
780
781
#if 0
    /* why + 1  ? isn't this taken care of in findnextstr() ? */
Chris Allegretta's avatar
Chris Allegretta committed
782
    beginx = current_x + 1;
783
784
785
#else
    beginx = current_x;
#endif
Chris Allegretta's avatar
Chris Allegretta committed
786
787
788
789
790
    search_last_line = 0;

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

    /* restore where we were */
Chris Allegretta's avatar
Chris Allegretta committed
791
    current = begin;
792
#if 0
793
    current_x = beginx - 1;
794
795
796
#else
    current_x = beginx;
#endif
Chris Allegretta's avatar
Chris Allegretta committed
797
    renumber_all();
798
    edit_update(current, CENTER);
Chris Allegretta's avatar
Chris Allegretta committed
799
800
801
802
803
804
805
806
807
808
809
    print_replaced(numreplaced);
    replace_abort();
    return 1;
}

void goto_abort(void)
{
    UNSET(KEEP_CUTBUFFER);
    display_main_list();
}

810
int do_gotoline(int line, int save_pos)
Chris Allegretta's avatar
Chris Allegretta committed
811
{
812
    int i = 1;
813
814

    if (line <= 0) {		/* Ask for it */
Chris Allegretta's avatar
Chris Allegretta committed
815

816
	int j = 0;
Chris Allegretta's avatar
Chris Allegretta committed
817

818
	j = statusq(0, goto_list, "", _("Enter line number"));
819
	if (j != 0) {
Chris Allegretta's avatar
Chris Allegretta committed
820
821
822
823
	    statusbar(_("Aborted"));
	    goto_abort();
	    return 0;
	}
824
825
826
827
828
829

	line = atoi(answer);

	/* Bounds check */
	if (line <= 0) {
	    statusbar(_("Come on, be reasonable"));
Chris Allegretta's avatar
Chris Allegretta committed
830
	    goto_abort();
831
	    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
832
833
834
	}
    }

835
836
    for (current = fileage; ((current->next != NULL) && (i < line)); i++)
	current = current->next;
Chris Allegretta's avatar
Chris Allegretta committed
837

838
    current_x = 0;
839
840
841
842
843
844
845

    /* 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);
Chris Allegretta's avatar
Chris Allegretta committed
846

847
    placewewant = xplustabs();
Chris Allegretta's avatar
Chris Allegretta committed
848
849
850
851
852
853
    goto_abort();
    return 1;
}

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

857
#if (defined ENABLE_MULTIBUFFER || !defined DISABLE_SPELLER)
858
void do_gotopos(int line, int pos_x, int pos_y, int pos_placewewant)
859
860
861
862
863
864
{

    /* 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);
865
866
867
868
869
870
871
872

    /* recalculate the x-coordinate and place we want, just in case their
       values are insane; if they aren't, they won't be changed by this */
    current_x = pos_x;
    pos_placewewant = xplustabs();
    pos_x = actual_x(current, pos_placewewant);

    /* set the rest of the coordinates up */
873
874
875
876
877
    current_x = pos_x;
    placewewant = pos_placewewant;
    update_line(current, pos_x);
}
#endif
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949

#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