blob: a4f3f7e07271e06fbb3dbea25eaba20f5322b536 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +000011 * diff.c: code for diff'ing two, three or four buffers.
Bram Moolenaare828b762018-09-10 17:51:58 +020012 *
13 * There are three ways to diff:
14 * - Shell out to an external diff program, using files.
15 * - Use the compiled-in xdiff library.
16 * - Let 'diffexpr' do the work, using files.
Bram Moolenaar071d4272004-06-13 20:20:40 +000017 */
18
19#include "vim.h"
Bram Moolenaare828b762018-09-10 17:51:58 +020020#include "xdiff/xdiff.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000021
22#if defined(FEAT_DIFF) || defined(PROTO)
23
Bram Moolenaard2b58c02018-09-16 18:10:48 +020024static int diff_busy = FALSE; // using diff structs, don't change them
25static int diff_need_update = FALSE; // ex_diffupdate needs to be called
Bram Moolenaar071d4272004-06-13 20:20:40 +000026
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010027// flags obtained from the 'diffopt' option
Bram Moolenaar785fc652018-09-15 19:17:38 +020028#define DIFF_FILLER 0x001 // display filler lines
29#define DIFF_IBLANK 0x002 // ignore empty lines
30#define DIFF_ICASE 0x004 // ignore case
31#define DIFF_IWHITE 0x008 // ignore change in white space
32#define DIFF_IWHITEALL 0x010 // ignore all white space changes
33#define DIFF_IWHITEEOL 0x020 // ignore change in white space at EOL
34#define DIFF_HORIZONTAL 0x040 // horizontal splits
35#define DIFF_VERTICAL 0x080 // vertical splits
36#define DIFF_HIDDEN_OFF 0x100 // diffoff when hidden
37#define DIFF_INTERNAL 0x200 // use internal xdiff algorithm
Bram Moolenaarc8234772019-11-10 21:00:27 +010038#define DIFF_CLOSE_OFF 0x400 // diffoff when closing window
Bram Moolenaar4223d432021-02-10 13:18:17 +010039#define DIFF_FOLLOWWRAP 0x800 // follow the wrap option
Bram Moolenaar785fc652018-09-15 19:17:38 +020040#define ALL_WHITE_DIFF (DIFF_IWHITE | DIFF_IWHITEALL | DIFF_IWHITEEOL)
Bram Moolenaarc8234772019-11-10 21:00:27 +010041static int diff_flags = DIFF_INTERNAL | DIFF_FILLER | DIFF_CLOSE_OFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +000042
Bram Moolenaare828b762018-09-10 17:51:58 +020043static long diff_algorithm = 0;
44
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010045#define LBUFLEN 50 // length of line in diff file
Bram Moolenaar071d4272004-06-13 20:20:40 +000046
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010047static int diff_a_works = MAYBE; // TRUE when "diff -a" works, FALSE when it
48 // doesn't work, MAYBE when not checked yet
Bram Moolenaar48e330a2016-02-23 14:53:34 +010049#if defined(MSWIN)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010050static int diff_bin_works = MAYBE; // TRUE when "diff --binary" works, FALSE
51 // when it doesn't work, MAYBE when not
52 // checked yet
Bram Moolenaar071d4272004-06-13 20:20:40 +000053#endif
54
Bram Moolenaare828b762018-09-10 17:51:58 +020055// used for diff input
56typedef struct {
57 char_u *din_fname; // used for external diff
58 mmfile_t din_mmfile; // used for internal diff
59} diffin_T;
60
61// used for diff result
62typedef struct {
63 char_u *dout_fname; // used for external diff
64 garray_T dout_ga; // used for internal diff
65} diffout_T;
66
Lewis Russelld9da86e2021-12-28 13:54:41 +000067// used for recording hunks from xdiff
68typedef struct {
69 linenr_T lnum_orig;
70 long count_orig;
71 linenr_T lnum_new;
72 long count_new;
73} diffhunk_T;
74
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +010075typedef enum {
76 DIO_OUTPUT_INDICES = 0, // default
77 DIO_OUTPUT_UNIFIED = 1 // unified diff format
78} dio_outfmt_T;
79
Bram Moolenaare828b762018-09-10 17:51:58 +020080// two diff inputs and one result
81typedef struct {
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +010082 diffin_T dio_orig; // original file input
83 diffin_T dio_new; // new file input
84 diffout_T dio_diff; // diff result
85 int dio_internal; // using internal diff
86 dio_outfmt_T dio_outfmt; // internal diff output format
87 int dio_ctxlen; // unified diff context length
Bram Moolenaare828b762018-09-10 17:51:58 +020088} diffio_T;
89
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010090static int diff_buf_idx(buf_T *buf);
91static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp);
92static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, linenr_T line2, long amount, long amount_after);
93static void diff_check_unchanged(tabpage_T *tp, diff_T *dp);
94static int diff_check_sanity(tabpage_T *tp, diff_T *dp);
Bram Moolenaare828b762018-09-10 17:51:58 +020095static int check_external_diff(diffio_T *diffio);
96static int diff_file(diffio_T *diffio);
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010097static int diff_equal_entry(diff_T *dp, int idx1, int idx2);
98static int diff_cmp(char_u *s1, char_u *s2);
Bram Moolenaar071d4272004-06-13 20:20:40 +000099#ifdef FEAT_FOLDING
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100100static void diff_fold_update(diff_T *dp, int skip_idx);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000101#endif
Lewis Russelld9da86e2021-12-28 13:54:41 +0000102static void diff_read(int idx_orig, int idx_new, diffio_T *dio);
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100103static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, int idx_new);
104static diff_T *diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp);
Lewis Russelld9da86e2021-12-28 13:54:41 +0000105static int parse_diff_ed(char_u *line, diffhunk_T *hunk);
106static int parse_diff_unified(char_u *line, diffhunk_T *hunk);
Yegappan Lakshmananfa378352024-02-01 22:05:27 +0100107static int xdiff_out_indices(long start_a, long count_a, long start_b, long count_b, void *priv);
108static int xdiff_out_unified(void *priv, mmbuffer_t *mb, int nbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200110#define FOR_ALL_DIFFBLOCKS_IN_TAB(tp, dp) \
111 for ((dp) = (tp)->tp_first_diff; (dp) != NULL; (dp) = (dp)->df_next)
112
Bram Moolenaar071d4272004-06-13 20:20:40 +0000113/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000114 * Called when deleting or unloading a buffer: No longer make a diff with it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000115 */
116 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100117diff_buf_delete(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000118{
119 int i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000120 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000121
Bram Moolenaar29323592016-07-24 22:04:11 +0200122 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000123 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000124 i = diff_buf_idx_tp(buf, tp);
125 if (i != DB_COUNT)
126 {
127 tp->tp_diffbuf[i] = NULL;
128 tp->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000129 if (tp == curtab)
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100130 {
131 // don't redraw right away, more might change or buffer state
132 // is invalid right now
133 need_diff_redraw = TRUE;
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100134 redraw_later(UPD_VALID);
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100135 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000136 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000137 }
138}
139
140/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000141 * Check if the current buffer should be added to or removed from the list of
142 * diff buffers.
143 */
144 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100145diff_buf_adjust(win_T *win)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000146{
147 win_T *wp;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000148 int i;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000149
150 if (!win->w_p_diff)
151 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100152 // When there is no window showing a diff for this buffer, remove
153 // it from the diffs.
Bram Moolenaar29323592016-07-24 22:04:11 +0200154 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000155 if (wp->w_buffer == win->w_buffer && wp->w_p_diff)
156 break;
157 if (wp == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000158 {
159 i = diff_buf_idx(win->w_buffer);
160 if (i != DB_COUNT)
161 {
162 curtab->tp_diffbuf[i] = NULL;
163 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000164 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000165 }
166 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000167 }
168 else
169 diff_buf_add(win->w_buffer);
170}
171
172/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000173 * Add a buffer to make diffs for.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000174 * Call this when a new buffer is being edited in the current window where
175 * 'diff' is set.
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000176 * Marks the current buffer as being part of the diff and requiring updating.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000177 * This must be done before any autocmd, because a command may use info
178 * about the screen contents.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 */
180 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100181diff_buf_add(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000182{
183 int i;
184
185 if (diff_buf_idx(buf) != DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100186 return; // It's already there.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187
188 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000189 if (curtab->tp_diffbuf[i] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000190 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000191 curtab->tp_diffbuf[i] = buf;
192 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000193 diff_redraw(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000194 return;
195 }
196
Bram Moolenaare1242042021-12-16 20:56:57 +0000197 semsg(_(e_cannot_diff_more_than_nr_buffers), DB_COUNT);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198}
199
200/*
Bram Moolenaar25ea0542017-02-03 23:16:28 +0100201 * Remove all buffers to make diffs for.
202 */
203 static void
204diff_buf_clear(void)
205{
206 int i;
207
208 for (i = 0; i < DB_COUNT; ++i)
209 if (curtab->tp_diffbuf[i] != NULL)
210 {
211 curtab->tp_diffbuf[i] = NULL;
212 curtab->tp_diff_invalid = TRUE;
213 diff_redraw(TRUE);
214 }
215}
216
217/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000218 * Find buffer "buf" in the list of diff buffers for the current tab page.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219 * Return its index or DB_COUNT if not found.
220 */
221 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100222diff_buf_idx(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000223{
224 int idx;
225
226 for (idx = 0; idx < DB_COUNT; ++idx)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000227 if (curtab->tp_diffbuf[idx] == buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000228 break;
229 return idx;
230}
231
232/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000233 * Find buffer "buf" in the list of diff buffers for tab page "tp".
234 * Return its index or DB_COUNT if not found.
235 */
236 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100237diff_buf_idx_tp(buf_T *buf, tabpage_T *tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000238{
239 int idx;
240
241 for (idx = 0; idx < DB_COUNT; ++idx)
242 if (tp->tp_diffbuf[idx] == buf)
243 break;
244 return idx;
245}
246
247/*
248 * Mark the diff info involving buffer "buf" as invalid, it will be updated
249 * when info is requested.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000250 */
251 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100252diff_invalidate(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000253{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000254 tabpage_T *tp;
255 int i;
256
Bram Moolenaar29323592016-07-24 22:04:11 +0200257 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000258 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000259 i = diff_buf_idx_tp(buf, tp);
260 if (i != DB_COUNT)
261 {
262 tp->tp_diff_invalid = TRUE;
263 if (tp == curtab)
264 diff_redraw(TRUE);
265 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000266 }
267}
268
269/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000270 * Called by mark_adjust(): update line numbers in "curbuf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000271 */
272 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100273diff_mark_adjust(
274 linenr_T line1,
275 linenr_T line2,
276 long amount,
277 long amount_after)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000278{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000279 int idx;
280 tabpage_T *tp;
281
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100282 // Handle all tab pages that use the current buffer in a diff.
Bram Moolenaar29323592016-07-24 22:04:11 +0200283 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000284 {
285 idx = diff_buf_idx_tp(curbuf, tp);
286 if (idx != DB_COUNT)
287 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after);
288 }
289}
290
291/*
292 * Update line numbers in tab page "tp" for "curbuf" with index "idx".
293 * This attempts to update the changes as much as possible:
294 * When inserting/deleting lines outside of existing change blocks, create a
295 * new change block and update the line numbers in following blocks.
296 * When inserting/deleting lines in existing change blocks, update them.
297 */
298 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100299diff_mark_adjust_tp(
300 tabpage_T *tp,
301 int idx,
302 linenr_T line1,
303 linenr_T line2,
304 long amount,
305 long amount_after)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000306{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000307 diff_T *dp;
308 diff_T *dprev;
309 diff_T *dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 int i;
311 int inserted, deleted;
312 int n, off;
313 linenr_T last;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100314 linenr_T lnum_deleted = line1; // lnum of remaining deletion
Bram Moolenaar071d4272004-06-13 20:20:40 +0000315 int check_unchanged;
316
Bram Moolenaare3521d92018-09-16 14:10:31 +0200317 if (diff_internal())
318 {
Bram Moolenaar198fa062018-09-18 21:20:26 +0200319 // Will update diffs before redrawing. Set _invalid to update the
Bram Moolenaare3521d92018-09-16 14:10:31 +0200320 // diffs themselves, set _update to also update folds properly just
321 // before redrawing.
Bram Moolenaar5f57bdc2018-10-25 17:52:23 +0200322 // Do update marks here, it is needed for :%diffput.
Bram Moolenaare3521d92018-09-16 14:10:31 +0200323 tp->tp_diff_invalid = TRUE;
324 tp->tp_diff_update = TRUE;
Bram Moolenaare3521d92018-09-16 14:10:31 +0200325 }
326
Bram Moolenaar071d4272004-06-13 20:20:40 +0000327 if (line2 == MAXLNUM)
328 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100329 // mark_adjust(99, MAXLNUM, 9, 0): insert lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000330 inserted = amount;
331 deleted = 0;
332 }
333 else if (amount_after > 0)
334 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100335 // mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000336 inserted = amount_after;
337 deleted = 0;
338 }
339 else
340 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100341 // mark_adjust(98, 99, MAXLNUM, -2): delete lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000342 inserted = 0;
343 deleted = -amount_after;
344 }
345
346 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000347 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000348 for (;;)
349 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100350 // If the change is after the previous diff block and before the next
351 // diff block, thus not touching an existing change, create a new diff
352 // block. Don't do this when ex_diffgetput() is busy.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000353 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2
354 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1))
355 && (dprev == NULL
356 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1)
357 && !diff_busy)
358 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000359 dnext = diff_alloc_new(tp, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000360 if (dnext == NULL)
361 return;
362
363 dnext->df_lnum[idx] = line1;
364 dnext->df_count[idx] = inserted;
365 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000366 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367 {
368 if (dprev == NULL)
369 dnext->df_lnum[i] = line1;
370 else
371 dnext->df_lnum[i] = line1
372 + (dprev->df_lnum[i] + dprev->df_count[i])
373 - (dprev->df_lnum[idx] + dprev->df_count[idx]);
374 dnext->df_count[i] = deleted;
375 }
376 }
377
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100378 // if at end of the list, quit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000379 if (dp == NULL)
380 break;
381
382 /*
383 * Check for these situations:
384 * 1 2 3
385 * 1 2 3
386 * line1 2 3 4 5
387 * 2 3 4 5
388 * 2 3 4 5
389 * line2 2 3 4 5
390 * 3 5 6
391 * 3 5 6
392 */
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100393 // compute last line of this change
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394 last = dp->df_lnum[idx] + dp->df_count[idx] - 1;
395
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100396 // 1. change completely above line1: nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +0000397 if (last >= line1 - 1)
398 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100399 // 6. change below line2: only adjust for amount_after; also when
400 // "deleted" became zero when deleted all lines between two diffs
Bram Moolenaar071d4272004-06-13 20:20:40 +0000401 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2)
402 {
403 if (amount_after == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100404 break; // nothing left to change
Bram Moolenaar071d4272004-06-13 20:20:40 +0000405 dp->df_lnum[idx] += amount_after;
406 }
407 else
408 {
409 check_unchanged = FALSE;
410
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100411 // 2. 3. 4. 5.: inserted/deleted lines touching this diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412 if (deleted > 0)
413 {
Bram Moolenaarc101abf2022-06-26 16:53:34 +0100414 off = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000415 if (dp->df_lnum[idx] >= line1)
416 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000417 if (last <= line2)
418 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100419 // 4. delete all lines of diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420 if (dp->df_next != NULL
421 && dp->df_next->df_lnum[idx] - 1 <= line2)
422 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100423 // delete continues in next diff, only do
424 // lines until that one
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425 n = dp->df_next->df_lnum[idx] - lnum_deleted;
426 deleted -= n;
427 n -= dp->df_count[idx];
428 lnum_deleted = dp->df_next->df_lnum[idx];
429 }
430 else
431 n = deleted - dp->df_count[idx];
432 dp->df_count[idx] = 0;
433 }
434 else
435 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100436 // 5. delete lines at or just before top of diff
Bram Moolenaarc101abf2022-06-26 16:53:34 +0100437 off = dp->df_lnum[idx] - lnum_deleted;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000438 n = off;
439 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1;
440 check_unchanged = TRUE;
441 }
442 dp->df_lnum[idx] = line1;
443 }
444 else
445 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000446 if (last < line2)
447 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100448 // 2. delete at end of diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000449 dp->df_count[idx] -= last - lnum_deleted + 1;
450 if (dp->df_next != NULL
451 && dp->df_next->df_lnum[idx] - 1 <= line2)
452 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100453 // delete continues in next diff, only do
454 // lines until that one
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455 n = dp->df_next->df_lnum[idx] - 1 - last;
456 deleted -= dp->df_next->df_lnum[idx]
457 - lnum_deleted;
458 lnum_deleted = dp->df_next->df_lnum[idx];
459 }
460 else
461 n = line2 - last;
462 check_unchanged = TRUE;
463 }
464 else
465 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100466 // 3. delete lines inside the diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000467 n = 0;
468 dp->df_count[idx] -= deleted;
469 }
470 }
471
472 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000473 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000474 {
Bram Moolenaar4e677b92022-07-28 18:44:27 +0100475 if (dp->df_lnum[i] > off)
476 dp->df_lnum[i] -= off;
477 else
478 dp->df_lnum[i] = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000479 dp->df_count[i] += n;
480 }
481 }
482 else
483 {
484 if (dp->df_lnum[idx] <= line1)
485 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100486 // inserted lines somewhere in this diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000487 dp->df_count[idx] += inserted;
488 check_unchanged = TRUE;
489 }
490 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100491 // inserted lines somewhere above this diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000492 dp->df_lnum[idx] += inserted;
493 }
494
495 if (check_unchanged)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100496 // Check if inserted lines are equal, may reduce the
497 // size of the diff. TODO: also check for equal lines
498 // in the middle and perhaps split the block.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000499 diff_check_unchanged(tp, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500 }
501 }
502
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100503 // check if this block touches the previous one, may merge them.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000504 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx]
505 == dp->df_lnum[idx])
506 {
507 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000508 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000509 dprev->df_count[i] += dp->df_count[i];
510 dprev->df_next = dp->df_next;
511 vim_free(dp);
512 dp = dprev->df_next;
513 }
514 else
515 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100516 // Advance to next entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517 dprev = dp;
518 dp = dp->df_next;
519 }
520 }
521
522 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000523 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000524 while (dp != NULL)
525 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100526 // All counts are zero, remove this entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000527 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000528 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 break;
530 if (i == DB_COUNT)
531 {
532 dnext = dp->df_next;
533 vim_free(dp);
534 dp = dnext;
535 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000536 tp->tp_first_diff = dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000537 else
538 dprev->df_next = dnext;
539 }
540 else
541 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100542 // Advance to next entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000543 dprev = dp;
544 dp = dp->df_next;
545 }
546
547 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000548
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000549 if (tp == curtab)
550 {
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200551 // Don't redraw right away, this updates the diffs, which can be slow.
552 need_diff_redraw = TRUE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000553
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100554 // Need to recompute the scroll binding, may remove or add filler
555 // lines (e.g., when adding lines above w_topline). But it's slow when
556 // making many changes, postpone until redrawing.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000557 diff_need_scrollbind = TRUE;
558 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559}
560
561/*
562 * Allocate a new diff block and link it between "dprev" and "dp".
563 */
564 static diff_T *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100565diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566{
567 diff_T *dnew;
568
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200569 dnew = ALLOC_ONE(diff_T);
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000570 if (dnew == NULL)
571 return NULL;
572
573 dnew->df_next = dp;
574 if (dprev == NULL)
575 tp->tp_first_diff = dnew;
576 else
577 dprev->df_next = dnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000578 return dnew;
579}
580
581/*
582 * Check if the diff block "dp" can be made smaller for lines at the start and
583 * end that are equal. Called after inserting lines.
584 * This may result in a change where all buffers have zero lines, the caller
585 * must take care of removing it.
586 */
587 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100588diff_check_unchanged(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000589{
590 int i_org;
591 int i_new;
592 int off_org, off_new;
593 char_u *line_org;
594 int dir = FORWARD;
595
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100596 // Find the first buffers, use it as the original, compare the other
597 // buffer lines against this one.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000598 for (i_org = 0; i_org < DB_COUNT; ++i_org)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000599 if (tp->tp_diffbuf[i_org] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100601 if (i_org == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +0000602 return;
603
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000604 if (diff_check_sanity(tp, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 return;
606
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100607 // First check lines at the top, then at the bottom.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608 off_org = 0;
609 off_new = 0;
610 for (;;)
611 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100612 // Repeat until a line is found which is different or the number of
613 // lines has become zero.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614 while (dp->df_count[i_org] > 0)
615 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100616 // Copy the line, the next ml_get() will invalidate it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 if (dir == BACKWARD)
618 off_org = dp->df_count[i_org] - 1;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000619 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000620 dp->df_lnum[i_org] + off_org, FALSE));
621 if (line_org == NULL)
622 return;
623 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new)
624 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000625 if (tp->tp_diffbuf[i_new] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626 continue;
627 if (dir == BACKWARD)
628 off_new = dp->df_count[i_new] - 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100629 // if other buffer doesn't have this line, it was inserted
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 if (off_new < 0 || off_new >= dp->df_count[i_new])
631 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000632 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 dp->df_lnum[i_new] + off_new, FALSE)) != 0)
634 break;
635 }
636 vim_free(line_org);
637
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100638 // Stop when a line isn't equal in all diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000639 if (i_new != DB_COUNT)
640 break;
641
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100642 // Line matched in all buffers, remove it from the diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000643 for (i_new = i_org; i_new < DB_COUNT; ++i_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000644 if (tp->tp_diffbuf[i_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645 {
646 if (dir == FORWARD)
647 ++dp->df_lnum[i_new];
648 --dp->df_count[i_new];
649 }
650 }
651 if (dir == BACKWARD)
652 break;
653 dir = BACKWARD;
654 }
655}
656
657/*
658 * Check if a diff block doesn't contain invalid line numbers.
659 * This can happen when the diff program returns invalid results.
660 */
661 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100662diff_check_sanity(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000663{
664 int i;
665
666 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000667 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 if (dp->df_lnum[i] + dp->df_count[i] - 1
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000669 > tp->tp_diffbuf[i]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 return FAIL;
671 return OK;
672}
673
674/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000675 * Mark all diff buffers in the current tab page for redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000676 */
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200677 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100678diff_redraw(
Bram Moolenaare3521d92018-09-16 14:10:31 +0200679 int dofold) // also recompute the folds
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680{
681 win_T *wp;
Bram Moolenaar04626c22021-09-01 16:02:07 +0200682 win_T *wp_other = NULL;
Bram Moolenaar841c2252021-10-22 20:56:55 +0100683 int used_max_fill_other = FALSE;
684 int used_max_fill_curwin = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000685 int n;
686
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200687 need_diff_redraw = FALSE;
Bram Moolenaar29323592016-07-24 22:04:11 +0200688 FOR_ALL_WINDOWS(wp)
zeertzjq101d57b2022-07-31 18:34:32 +0100689 {
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100690 // when closing windows or wiping buffers skip invalid window
zeertzjq101d57b2022-07-31 18:34:32 +0100691 if (!wp->w_p_diff || !buf_valid(wp->w_buffer))
692 continue;
693
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100694 redraw_win_later(wp, UPD_SOME_VALID);
zeertzjq101d57b2022-07-31 18:34:32 +0100695 if (wp != curwin)
696 wp_other = wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000697#ifdef FEAT_FOLDING
zeertzjq101d57b2022-07-31 18:34:32 +0100698 if (dofold && foldmethodIsDiff(wp))
699 foldUpdateAll(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000700#endif
zeertzjq101d57b2022-07-31 18:34:32 +0100701 // A change may have made filler lines invalid, need to take care of
702 // that for other windows.
703 n = diff_check(wp, wp->w_topline);
704 if ((wp != curwin && wp->w_topfill > 0) || n > 0)
705 {
706 if (wp->w_topfill > n)
707 wp->w_topfill = (n < 0 ? 0 : n);
708 else if (n > 0 && n > wp->w_topfill)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000709 {
zeertzjq101d57b2022-07-31 18:34:32 +0100710 wp->w_topfill = n;
711 if (wp == curwin)
712 used_max_fill_curwin = TRUE;
713 else if (wp_other != NULL)
714 used_max_fill_other = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715 }
zeertzjq101d57b2022-07-31 18:34:32 +0100716 check_topfill(wp, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000717 }
zeertzjq101d57b2022-07-31 18:34:32 +0100718 }
Bram Moolenaar04626c22021-09-01 16:02:07 +0200719
Bram Moolenaar841c2252021-10-22 20:56:55 +0100720 if (wp_other != NULL && curwin->w_p_scb)
721 {
722 if (used_max_fill_curwin)
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000723 // The current window was set to use the maximum number of filler
Bram Moolenaar841c2252021-10-22 20:56:55 +0100724 // lines, may need to reduce them.
725 diff_set_topline(wp_other, curwin);
726 else if (used_max_fill_other)
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000727 // The other window was set to use the maximum number of filler
Bram Moolenaar841c2252021-10-22 20:56:55 +0100728 // lines, may need to reduce them.
729 diff_set_topline(curwin, wp_other);
730 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000731}
732
Bram Moolenaare828b762018-09-10 17:51:58 +0200733 static void
734clear_diffin(diffin_T *din)
735{
736 if (din->din_fname == NULL)
Yegappan Lakshmanan960dcbd2023-03-07 17:45:11 +0000737 VIM_CLEAR(din->din_mmfile.ptr);
Bram Moolenaare828b762018-09-10 17:51:58 +0200738 else
739 mch_remove(din->din_fname);
740}
741
742 static void
743clear_diffout(diffout_T *dout)
744{
745 if (dout->dout_fname == NULL)
746 ga_clear_strings(&dout->dout_ga);
747 else
748 mch_remove(dout->dout_fname);
749}
750
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751/*
Bram Moolenaare828b762018-09-10 17:51:58 +0200752 * Write buffer "buf" to a memory buffer.
753 * Return FAIL for failure.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754 */
755 static int
Bram Moolenaare828b762018-09-10 17:51:58 +0200756diff_write_buffer(buf_T *buf, diffin_T *din)
757{
758 linenr_T lnum;
759 char_u *s;
760 long len = 0;
761 char_u *ptr;
762
763 // xdiff requires one big block of memory with all the text.
764 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
Bram Moolenaar6e272ac2018-09-16 14:51:36 +0200765 len += (long)STRLEN(ml_get_buf(buf, lnum, FALSE)) + 1;
Bram Moolenaar18a4ba22019-05-24 19:39:03 +0200766 ptr = alloc(len);
Bram Moolenaare828b762018-09-10 17:51:58 +0200767 if (ptr == NULL)
768 {
769 // Allocating memory failed. This can happen, because we try to read
770 // the whole buffer text into memory. Set the failed flag, the diff
771 // will be retried with external diff. The flag is never reset.
772 buf->b_diff_failed = TRUE;
773 if (p_verbose > 0)
774 {
775 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100776 smsg(_("Not enough memory to use internal diff for buffer \"%s\""),
Bram Moolenaare828b762018-09-10 17:51:58 +0200777 buf->b_fname);
778 verbose_leave();
779 }
780 return FAIL;
781 }
782 din->din_mmfile.ptr = (char *)ptr;
783 din->din_mmfile.size = len;
784
785 len = 0;
786 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
787 {
788 for (s = ml_get_buf(buf, lnum, FALSE); *s != NUL; )
789 {
790 if (diff_flags & DIFF_ICASE)
791 {
792 int c;
Bram Moolenaare828b762018-09-10 17:51:58 +0200793 int orig_len;
794 char_u cbuf[MB_MAXBYTES + 1];
795
Bram Moolenaar06f60952021-12-28 18:30:05 +0000796 if (*s == NL)
797 c = NUL;
798 else
799 {
800 // xdiff doesn't support ignoring case, fold-case the text.
801 c = PTR2CHAR(s);
802 c = MB_CASEFOLD(c);
803 }
Bram Moolenaar1614a142019-10-06 22:00:13 +0200804 orig_len = mb_ptr2len(s);
Bram Moolenaare828b762018-09-10 17:51:58 +0200805 if (mb_char2bytes(c, cbuf) != orig_len)
806 // TODO: handle byte length difference
807 mch_memmove(ptr + len, s, orig_len);
808 else
809 mch_memmove(ptr + len, cbuf, orig_len);
810
811 s += orig_len;
812 len += orig_len;
Bram Moolenaare828b762018-09-10 17:51:58 +0200813 }
814 else
Bram Moolenaar06f60952021-12-28 18:30:05 +0000815 {
816 ptr[len++] = *s == NL ? NUL : *s;
817 s++;
818 }
Bram Moolenaare828b762018-09-10 17:51:58 +0200819 }
820 ptr[len++] = NL;
821 }
822 return OK;
823}
824
825/*
826 * Write buffer "buf" to file or memory buffer.
827 * Return FAIL for failure.
828 */
829 static int
830diff_write(buf_T *buf, diffin_T *din)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831{
832 int r;
833 char_u *save_ff;
Bram Moolenaare1004402020-10-24 20:49:43 +0200834 int save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835
Bram Moolenaare828b762018-09-10 17:51:58 +0200836 if (din->din_fname == NULL)
837 return diff_write_buffer(buf, din);
838
839 // Always use 'fileformat' set to "unix".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000840 save_ff = buf->b_p_ff;
841 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
Bram Moolenaare1004402020-10-24 20:49:43 +0200842 save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100843 // Writing the buffer is an implementation detail of performing the diff,
844 // so it shouldn't update the '[ and '] marks.
Bram Moolenaare1004402020-10-24 20:49:43 +0200845 cmdmod.cmod_flags |= CMOD_LOCKMARKS;
Bram Moolenaare828b762018-09-10 17:51:58 +0200846 r = buf_write(buf, din->din_fname, NULL,
847 (linenr_T)1, buf->b_ml.ml_line_count,
848 NULL, FALSE, FALSE, FALSE, TRUE);
Bram Moolenaare1004402020-10-24 20:49:43 +0200849 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850 free_string_option(buf->b_p_ff);
851 buf->b_p_ff = save_ff;
852 return r;
853}
854
855/*
Bram Moolenaare828b762018-09-10 17:51:58 +0200856 * Update the diffs for all buffers involved.
857 */
858 static void
859diff_try_update(
860 diffio_T *dio,
861 int idx_orig,
862 exarg_T *eap) // "eap" can be NULL
863{
864 buf_T *buf;
865 int idx_new;
866
867 if (dio->dio_internal)
868 {
869 ga_init2(&dio->dio_diff.dout_ga, sizeof(char *), 1000);
870 }
871 else
872 {
873 // We need three temp file names.
874 dio->dio_orig.din_fname = vim_tempname('o', TRUE);
875 dio->dio_new.din_fname = vim_tempname('n', TRUE);
876 dio->dio_diff.dout_fname = vim_tempname('d', TRUE);
877 if (dio->dio_orig.din_fname == NULL
878 || dio->dio_new.din_fname == NULL
879 || dio->dio_diff.dout_fname == NULL)
880 goto theend;
881 }
882
883 // Check external diff is actually working.
884 if (!dio->dio_internal && check_external_diff(dio) == FAIL)
885 goto theend;
886
887 // :diffupdate!
888 if (eap != NULL && eap->forceit)
889 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
890 {
891 buf = curtab->tp_diffbuf[idx_new];
892 if (buf_valid(buf))
893 buf_check_timestamp(buf, FALSE);
894 }
895
896 // Write the first buffer to a tempfile or mmfile_t.
897 buf = curtab->tp_diffbuf[idx_orig];
898 if (diff_write(buf, &dio->dio_orig) == FAIL)
899 goto theend;
900
901 // Make a difference between the first buffer and every other.
902 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
903 {
904 buf = curtab->tp_diffbuf[idx_new];
905 if (buf == NULL || buf->b_ml.ml_mfp == NULL)
906 continue; // skip buffer that isn't loaded
907
908 // Write the other buffer and diff with the first one.
909 if (diff_write(buf, &dio->dio_new) == FAIL)
910 continue;
911 if (diff_file(dio) == FAIL)
912 continue;
913
914 // Read the diff output and add each entry to the diff list.
Lewis Russelld9da86e2021-12-28 13:54:41 +0000915 diff_read(idx_orig, idx_new, dio);
Bram Moolenaare828b762018-09-10 17:51:58 +0200916
917 clear_diffin(&dio->dio_new);
918 clear_diffout(&dio->dio_diff);
919 }
920 clear_diffin(&dio->dio_orig);
921
922theend:
923 vim_free(dio->dio_orig.din_fname);
924 vim_free(dio->dio_new.din_fname);
925 vim_free(dio->dio_diff.dout_fname);
926}
927
928/*
929 * Return TRUE if the options are set to use the internal diff library.
930 * Note that if the internal diff failed for one of the buffers, the external
931 * diff will be used anyway.
932 */
Bram Moolenaare3521d92018-09-16 14:10:31 +0200933 int
Bram Moolenaare828b762018-09-10 17:51:58 +0200934diff_internal(void)
935{
Bram Moolenaar975880b2019-03-03 14:42:11 +0100936 return (diff_flags & DIFF_INTERNAL) != 0
937#ifdef FEAT_EVAL
938 && *p_dex == NUL
939#endif
940 ;
Bram Moolenaare828b762018-09-10 17:51:58 +0200941}
942
943/*
944 * Return TRUE if the internal diff failed for one of the diff buffers.
945 */
946 static int
947diff_internal_failed(void)
948{
949 int idx;
950
951 // Only need to do something when there is another buffer.
952 for (idx = 0; idx < DB_COUNT; ++idx)
953 if (curtab->tp_diffbuf[idx] != NULL
954 && curtab->tp_diffbuf[idx]->b_diff_failed)
955 return TRUE;
956 return FALSE;
957}
958
959/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960 * Completely update the diffs for the buffers involved.
Bram Moolenaare3521d92018-09-16 14:10:31 +0200961 * When using the external "diff" command the buffers are written to a file,
962 * also for unmodified buffers (the file could have been produced by
963 * autocommands, e.g. the netrw plugin).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000964 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000965 void
Bram Moolenaare828b762018-09-10 17:51:58 +0200966ex_diffupdate(exarg_T *eap) // "eap" can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000968 int idx_orig;
969 int idx_new;
Bram Moolenaare828b762018-09-10 17:51:58 +0200970 diffio_T diffio;
Bram Moolenaar198fa062018-09-18 21:20:26 +0200971 int had_diffs = curtab->tp_first_diff != NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000972
Bram Moolenaard2b58c02018-09-16 18:10:48 +0200973 if (diff_busy)
974 {
975 diff_need_update = TRUE;
976 return;
977 }
978
Bram Moolenaare828b762018-09-10 17:51:58 +0200979 // Delete all diffblocks.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000980 diff_clear(curtab);
981 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000982
Bram Moolenaare828b762018-09-10 17:51:58 +0200983 // Use the first buffer as the original text.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000984 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000985 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000986 break;
987 if (idx_orig == DB_COUNT)
Bram Moolenaar198fa062018-09-18 21:20:26 +0200988 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989
Bram Moolenaare828b762018-09-10 17:51:58 +0200990 // Only need to do something when there is another buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000991 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000992 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000993 break;
994 if (idx_new == DB_COUNT)
Bram Moolenaar198fa062018-09-18 21:20:26 +0200995 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000996
Bram Moolenaare828b762018-09-10 17:51:58 +0200997 // Only use the internal method if it did not fail for one of the buffers.
Bram Moolenaara80faa82020-04-12 19:37:17 +0200998 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +0200999 diffio.dio_internal = diff_internal() && !diff_internal_failed();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001000
Bram Moolenaare828b762018-09-10 17:51:58 +02001001 diff_try_update(&diffio, idx_orig, eap);
1002 if (diffio.dio_internal && diff_internal_failed())
1003 {
1004 // Internal diff failed, use external diff instead.
Bram Moolenaara80faa82020-04-12 19:37:17 +02001005 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +02001006 diff_try_update(&diffio, idx_orig, eap);
1007 }
1008
1009 // force updating cursor position on screen
1010 curwin->w_valid_cursor.lnum = 0;
1011
Bram Moolenaar198fa062018-09-18 21:20:26 +02001012theend:
1013 // A redraw is needed if there were diffs and they were cleared, or there
1014 // are diffs now, which means they got updated.
1015 if (had_diffs || curtab->tp_first_diff != NULL)
1016 {
1017 diff_redraw(TRUE);
1018 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
1019 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001020}
1021
1022/*
1023 * Do a quick test if "diff" really works. Otherwise it looks like there
1024 * are no differences. Can't use the return value, it's non-zero when
1025 * there are differences.
1026 */
1027 static int
1028check_external_diff(diffio_T *diffio)
1029{
1030 FILE *fd;
1031 int ok;
1032 int io_error = FALSE;
1033
1034 // May try twice, first with "-a" and then without.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035 for (;;)
1036 {
1037 ok = FALSE;
Bram Moolenaare828b762018-09-10 17:51:58 +02001038 fd = mch_fopen((char *)diffio->dio_orig.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001039 if (fd == NULL)
1040 io_error = TRUE;
1041 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001042 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001043 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
1044 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001046 fd = mch_fopen((char *)diffio->dio_new.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001047 if (fd == NULL)
1048 io_error = TRUE;
1049 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001050 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001051 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
1052 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001053 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001054 fd = NULL;
1055 if (diff_file(diffio) == OK)
1056 fd = mch_fopen((char *)diffio->dio_diff.dout_fname, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001057 if (fd == NULL)
1058 io_error = TRUE;
1059 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060 {
1061 char_u linebuf[LBUFLEN];
1062
1063 for (;;)
1064 {
glacambread5c1782021-05-24 14:20:53 +02001065 // For normal diff there must be a line that contains
1066 // "1c1". For unified diff "@@ -1 +1 @@".
Bram Moolenaar00590742019-02-15 21:06:09 +01001067 if (vim_fgets(linebuf, LBUFLEN, fd))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001068 break;
glacambread5c1782021-05-24 14:20:53 +02001069 if (STRNCMP(linebuf, "1c1", 3) == 0
1070 || STRNCMP(linebuf, "@@ -1 +1 @@", 11) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001071 ok = TRUE;
1072 }
1073 fclose(fd);
1074 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001075 mch_remove(diffio->dio_diff.dout_fname);
1076 mch_remove(diffio->dio_new.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001077 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001078 mch_remove(diffio->dio_orig.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001079 }
1080
1081#ifdef FEAT_EVAL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001082 // When using 'diffexpr' break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001083 if (*p_dex != NUL)
1084 break;
1085#endif
1086
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001087#if defined(MSWIN)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001088 // If the "-a" argument works, also check if "--binary" works.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001089 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
1090 {
1091 diff_a_works = TRUE;
1092 diff_bin_works = TRUE;
1093 continue;
1094 }
1095 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
1096 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001097 // Tried --binary, but it failed. "-a" works though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001098 diff_bin_works = FALSE;
1099 ok = TRUE;
1100 }
1101#endif
1102
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001103 // If we checked if "-a" works already, break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001104 if (diff_a_works != MAYBE)
1105 break;
1106 diff_a_works = ok;
1107
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001108 // If "-a" works break here, otherwise retry without "-a".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001109 if (ok)
1110 break;
1111 }
1112 if (!ok)
1113 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001114 if (io_error)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001115 emsg(_(e_cannot_read_or_write_temp_files));
Bram Moolenaare1242042021-12-16 20:56:57 +00001116 emsg(_(e_cannot_create_diffs));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001117 diff_a_works = MAYBE;
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001118#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001119 diff_bin_works = MAYBE;
1120#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001121 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001123 return OK;
1124}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125
Bram Moolenaare828b762018-09-10 17:51:58 +02001126/*
1127 * Invoke the xdiff function.
1128 */
1129 static int
1130diff_file_internal(diffio_T *diffio)
1131{
1132 xpparam_t param;
1133 xdemitconf_t emit_cfg;
1134 xdemitcb_t emit_cb;
Bram Moolenaarbd1d5602012-05-18 18:47:17 +02001135
Bram Moolenaara80faa82020-04-12 19:37:17 +02001136 CLEAR_FIELD(param);
1137 CLEAR_FIELD(emit_cfg);
1138 CLEAR_FIELD(emit_cb);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001139
Bram Moolenaare828b762018-09-10 17:51:58 +02001140 param.flags = diff_algorithm;
1141
1142 if (diff_flags & DIFF_IWHITE)
1143 param.flags |= XDF_IGNORE_WHITESPACE_CHANGE;
Bram Moolenaar785fc652018-09-15 19:17:38 +02001144 if (diff_flags & DIFF_IWHITEALL)
1145 param.flags |= XDF_IGNORE_WHITESPACE;
1146 if (diff_flags & DIFF_IWHITEEOL)
1147 param.flags |= XDF_IGNORE_WHITESPACE_AT_EOL;
1148 if (diff_flags & DIFF_IBLANK)
1149 param.flags |= XDF_IGNORE_BLANK_LINES;
Bram Moolenaare828b762018-09-10 17:51:58 +02001150
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001151 emit_cfg.ctxlen = diffio->dio_ctxlen;
Bram Moolenaare828b762018-09-10 17:51:58 +02001152 emit_cb.priv = &diffio->dio_diff;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001153 if (diffio->dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01001154 emit_cfg.hunk_func = xdiff_out_indices;
1155 else
1156 emit_cb.out_line = xdiff_out_unified;
Bram Moolenaare828b762018-09-10 17:51:58 +02001157 if (xdl_diff(&diffio->dio_orig.din_mmfile,
1158 &diffio->dio_new.din_mmfile,
1159 &param, &emit_cfg, &emit_cb) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001160 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001161 emsg(_(e_problem_creating_internal_diff));
Bram Moolenaare828b762018-09-10 17:51:58 +02001162 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001163 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001164 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001165}
1166
1167/*
1168 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
Bram Moolenaare828b762018-09-10 17:51:58 +02001169 * return OK or FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001170 */
Bram Moolenaare828b762018-09-10 17:51:58 +02001171 static int
1172diff_file(diffio_T *dio)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001173{
1174 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001175 size_t len;
Bram Moolenaare828b762018-09-10 17:51:58 +02001176 char_u *tmp_orig = dio->dio_orig.din_fname;
1177 char_u *tmp_new = dio->dio_new.din_fname;
1178 char_u *tmp_diff = dio->dio_diff.dout_fname;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001179
1180#ifdef FEAT_EVAL
1181 if (*p_dex != NUL)
Bram Moolenaare828b762018-09-10 17:51:58 +02001182 {
1183 // Use 'diffexpr' to generate the diff file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001184 eval_diff(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaare828b762018-09-10 17:51:58 +02001185 return OK;
1186 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 else
1188#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001189 // Use xdiff for generating the diff.
1190 if (dio->dio_internal)
Bram Moolenaare828b762018-09-10 17:51:58 +02001191 return diff_file_internal(dio);
Bram Moolenaar95fb60a2005-03-08 22:29:20 +00001192
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001193 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
1194 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
1195 cmd = alloc(len);
1196 if (cmd == NULL)
1197 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02001198
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001199 // We don't want $DIFF_OPTIONS to get in the way.
1200 if (getenv("DIFF_OPTIONS"))
1201 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
1202
1203 // Build the diff command and execute it. Always use -a, binary
1204 // differences are of no use. Ignore errors, diff returns
1205 // non-zero when differences have been found.
1206 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s%s%s%s %s",
1207 diff_a_works == FALSE ? "" : "-a ",
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001208#if defined(MSWIN)
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001209 diff_bin_works == TRUE ? "--binary " : "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210#else
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001211 "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001212#endif
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001213 (diff_flags & DIFF_IWHITE) ? "-b " : "",
1214 (diff_flags & DIFF_IWHITEALL) ? "-w " : "",
1215 (diff_flags & DIFF_IWHITEEOL) ? "-Z " : "",
1216 (diff_flags & DIFF_IBLANK) ? "-B " : "",
1217 (diff_flags & DIFF_ICASE) ? "-i " : "",
1218 tmp_orig, tmp_new);
1219 append_redir(cmd, (int)len, p_srr, tmp_diff);
1220 block_autocmds(); // avoid ShellCmdPost stuff
1221 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
1222 unblock_autocmds();
1223 vim_free(cmd);
1224 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001225}
1226
1227/*
1228 * Create a new version of a file from the current buffer and a diff file.
1229 * The buffer is written to a file, also for unmodified buffers (the file
1230 * could have been produced by autocommands, e.g. the netrw plugin).
1231 */
1232 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001233ex_diffpatch(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001234{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001235 char_u *tmp_orig; // name of original temp file
1236 char_u *tmp_new; // name of patched temp file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001237 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001238 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001239 win_T *old_curwin = curwin;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001240 char_u *newname = NULL; // name of patched file buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241#ifdef UNIX
1242 char_u dirbuf[MAXPATHL];
1243 char_u *fullname = NULL;
1244#endif
1245#ifdef FEAT_BROWSE
1246 char_u *browseFile = NULL;
Bram Moolenaare1004402020-10-24 20:49:43 +02001247 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02001249 stat_T st;
Bram Moolenaara95ab322017-03-11 19:21:53 +01001250 char_u *esc_name = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001251
1252#ifdef FEAT_BROWSE
Bram Moolenaare1004402020-10-24 20:49:43 +02001253 if (cmdmod.cmod_flags & CMOD_BROWSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001254 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +00001255 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaarc36651b2018-04-29 12:22:56 +02001256 eap->arg, NULL, NULL,
1257 (char_u *)_(BROWSE_FILTER_ALL_FILES), NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001258 if (browseFile == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001259 return; // operation cancelled
Bram Moolenaar071d4272004-06-13 20:20:40 +00001260 eap->arg = browseFile;
Bram Moolenaare1004402020-10-24 20:49:43 +02001261 cmdmod.cmod_flags &= ~CMOD_BROWSE; // don't let do_ecmd() browse again
Bram Moolenaar071d4272004-06-13 20:20:40 +00001262 }
1263#endif
1264
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001265 // We need two temp file names.
Bram Moolenaare5c421c2015-03-31 13:33:08 +02001266 tmp_orig = vim_tempname('o', FALSE);
1267 tmp_new = vim_tempname('n', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001268 if (tmp_orig == NULL || tmp_new == NULL)
1269 goto theend;
1270
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001271 // Write the current buffer to "tmp_orig".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001272 if (buf_write(curbuf, tmp_orig, NULL,
1273 (linenr_T)1, curbuf->b_ml.ml_line_count,
1274 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
1275 goto theend;
1276
1277#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001278 // Get the absolute path of the patchfile, changing directory below.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001279 fullname = FullName_save(eap->arg, FALSE);
1280#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001281 esc_name = vim_strsave_shellescape(
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282# ifdef UNIX
Bram Moolenaara95ab322017-03-11 19:21:53 +01001283 fullname != NULL ? fullname :
Bram Moolenaar071d4272004-06-13 20:20:40 +00001284# endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001285 eap->arg, TRUE, TRUE);
1286 if (esc_name == NULL)
1287 goto theend;
1288 buflen = STRLEN(tmp_orig) + STRLEN(esc_name) + STRLEN(tmp_new) + 16;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001289 buf = alloc(buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001290 if (buf == NULL)
1291 goto theend;
1292
1293#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001294 // Temporarily chdir to /tmp, to avoid patching files in the current
1295 // directory when the patch file contains more than one patch. When we
1296 // have our own temp dir use that instead, it will be cleaned up when we
1297 // exit (any .rej files created). Don't change directory if we can't
1298 // return to the current.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001299 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
1300 dirbuf[0] = NUL;
1301 else
1302 {
1303# ifdef TEMPDIRNAMES
1304 if (vim_tempdir != NULL)
Bram Moolenaar42335f52018-09-13 15:33:43 +02001305 vim_ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001306 else
1307# endif
Bram Moolenaar42335f52018-09-13 15:33:43 +02001308 vim_ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001309 shorten_fnames(TRUE);
1310 }
1311#endif
1312
1313#ifdef FEAT_EVAL
1314 if (*p_pex != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001315 // Use 'patchexpr' to generate the new file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001316 eval_patch(tmp_orig,
1317# ifdef UNIX
1318 fullname != NULL ? fullname :
1319# endif
1320 eap->arg, tmp_new);
1321 else
1322#endif
1323 {
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001324 if (check_restricted())
1325 goto theend;
1326
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001327 // Build the patch command and execute it. Ignore errors. Switch to
1328 // cooked mode to allow the user to respond to prompts.
Bram Moolenaara95ab322017-03-11 19:21:53 +01001329 vim_snprintf((char *)buf, buflen, "patch -o %s %s < %s",
1330 tmp_new, tmp_orig, esc_name);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001331 block_autocmds(); // Avoid ShellCmdPost stuff
Bram Moolenaar071d4272004-06-13 20:20:40 +00001332 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar78ab3312007-09-29 12:16:41 +00001333 unblock_autocmds();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001334 }
1335
1336#ifdef UNIX
1337 if (dirbuf[0] != NUL)
1338 {
1339 if (mch_chdir((char *)dirbuf) != 0)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001340 emsg(_(e_cannot_go_back_to_previous_directory));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001341 shorten_fnames(TRUE);
1342 }
1343#endif
1344
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001345 // patch probably has written over the screen
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001346 redraw_later(UPD_CLEAR);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001347
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001348 // Delete any .orig or .rej file created.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001349 STRCPY(buf, tmp_new);
1350 STRCAT(buf, ".orig");
1351 mch_remove(buf);
1352 STRCPY(buf, tmp_new);
1353 STRCAT(buf, ".rej");
1354 mch_remove(buf);
1355
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001356 // Only continue if the output file was created.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001357 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001358 emsg(_(e_cannot_read_patch_output));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001359 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001360 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001361 if (curbuf->b_fname != NULL)
1362 {
1363 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaardf44a272020-06-07 20:49:05 +02001364 STRLEN(curbuf->b_fname) + 4);
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001365 if (newname != NULL)
1366 STRCAT(newname, ".new");
1367 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001368
1369#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001370 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001371#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001372 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001373 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001374
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001375 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001376 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001377 // Pretend it was a ":split fname" command
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001378 eap->cmdidx = CMD_split;
1379 eap->arg = tmp_new;
1380 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001381
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001382 // check that split worked and editing tmp_new
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001383 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001384 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001385 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001386 diff_win_options(curwin, TRUE);
1387 diff_win_options(old_curwin, TRUE);
1388
1389 if (newname != NULL)
1390 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001391 // do a ":file filename.new" on the patched buffer
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001392 eap->arg = newname;
1393 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001394
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001395 // Do filetype detection with the new name.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001396 if (au_has_group((char_u *)"filetypedetect"))
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001397 do_cmdline_cmd(
1398 (char_u *)":doau filetypedetect BufRead");
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001399 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001400 }
1401 }
1402 }
1403
1404theend:
1405 if (tmp_orig != NULL)
1406 mch_remove(tmp_orig);
1407 vim_free(tmp_orig);
1408 if (tmp_new != NULL)
1409 mch_remove(tmp_new);
1410 vim_free(tmp_new);
1411 vim_free(newname);
1412 vim_free(buf);
1413#ifdef UNIX
1414 vim_free(fullname);
1415#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001416 vim_free(esc_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001417#ifdef FEAT_BROWSE
1418 vim_free(browseFile);
Bram Moolenaare1004402020-10-24 20:49:43 +02001419 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001420#endif
1421}
1422
1423/*
1424 * Split the window and edit another file, setting options to show the diffs.
1425 */
1426 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001427ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001428{
1429 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001430 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001431
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001432 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001433#ifdef FEAT_GUI
1434 need_mouse_correct = TRUE;
1435#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001436 // Need to compute w_fraction when no redraw happened yet.
Bram Moolenaar46328f92016-08-28 15:39:57 +02001437 validate_cursor();
1438 set_fraction(curwin);
1439
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001440 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001441 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001442
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001443 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) == FAIL)
1444 return;
1445
1446 // Pretend it was a ":split fname" command
1447 eap->cmdidx = CMD_split;
1448 curwin->w_p_diff = TRUE;
1449 do_exedit(eap, old_curwin);
1450
1451 if (curwin == old_curwin) // split didn't work
1452 return;
1453
1454 // Set 'diff', 'scrollbind' on and 'wrap' off.
1455 diff_win_options(curwin, TRUE);
1456 if (win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001457 {
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001458 diff_win_options(old_curwin, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001459
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001460 if (bufref_valid(&old_curbuf))
1461 // Move the cursor position to that of the old window.
1462 curwin->w_cursor.lnum = diff_get_corresponding_line(
1463 old_curbuf.br_buf, old_curwin->w_cursor.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001464 }
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001465 // Now that lines are folded scroll to show the cursor at the same
1466 // relative position.
1467 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001468}
1469
1470/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001471 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001472 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001473 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001474ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001476 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001477 diff_win_options(curwin, TRUE);
1478}
1479
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001480 static void
1481set_diff_option(win_T *wp, int value)
1482{
1483 win_T *old_curwin = curwin;
1484
1485 curwin = wp;
1486 curbuf = curwin->w_buffer;
1487 ++curbuf_lock;
Bram Moolenaar31e5c602022-04-15 13:53:33 +01001488 set_option_value_give_err((char_u *)"diff", (long)value, NULL, OPT_LOCAL);
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001489 --curbuf_lock;
1490 curwin = old_curwin;
1491 curbuf = curwin->w_buffer;
1492}
1493
Bram Moolenaar071d4272004-06-13 20:20:40 +00001494/*
1495 * Set options in window "wp" for diff mode.
1496 */
1497 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001498diff_win_options(
1499 win_T *wp,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001500 int addbuf) // Add buffer to diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001501{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001502# ifdef FEAT_FOLDING
1503 win_T *old_curwin = curwin;
1504
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001505 // close the manually opened folds
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001506 curwin = wp;
1507 newFoldLevel();
1508 curwin = old_curwin;
1509# endif
1510
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001511 // Use 'scrollbind' and 'cursorbind' when available
Bram Moolenaar43929962015-07-03 15:06:56 +02001512 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001513 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001514 wp->w_p_scb = TRUE;
Bram Moolenaar43929962015-07-03 15:06:56 +02001515 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001516 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001517 wp->w_p_crb = TRUE;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001518 if (!(diff_flags & DIFF_FOLLOWWRAP))
1519 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001520 if (!wp->w_p_diff)
Bram Moolenaar4223d432021-02-10 13:18:17 +01001521 wp->w_p_wrap_save = wp->w_p_wrap;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001522 wp->w_p_wrap = FALSE;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001523 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001524# ifdef FEAT_FOLDING
Bram Moolenaar43929962015-07-03 15:06:56 +02001525 if (!wp->w_p_diff)
1526 {
1527 if (wp->w_p_diff_saved)
1528 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001529 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001530 }
Bram Moolenaar20c023a2019-05-26 21:03:24 +02001531 set_string_option_direct_in_win(wp, (char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001532 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar43929962015-07-03 15:06:56 +02001533 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001534 {
1535 wp->w_p_fdc_save = wp->w_p_fdc;
1536 wp->w_p_fen_save = wp->w_p_fen;
1537 wp->w_p_fdl_save = wp->w_p_fdl;
1538 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001539 wp->w_p_fdc = diff_foldcolumn;
1540 wp->w_p_fen = TRUE;
1541 wp->w_p_fdl = 0;
1542 foldUpdateAll(wp);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001543 // make sure topline is not halfway a fold
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001544 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001545# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001546 if (vim_strchr(p_sbo, 'h') == NULL)
1547 do_cmdline_cmd((char_u *)"set sbo+=hor");
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001548 // Save the current values, to be restored in ex_diffoff().
Bram Moolenaara87aa802013-07-03 15:47:03 +02001549 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001550
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001551 set_diff_option(wp, TRUE);
Bram Moolenaar43929962015-07-03 15:06:56 +02001552
Bram Moolenaar071d4272004-06-13 20:20:40 +00001553 if (addbuf)
1554 diff_buf_add(wp->w_buffer);
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001555 redraw_win_later(wp, UPD_NOT_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001556}
1557
1558/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001559 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001560 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001561 */
1562 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001563ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001564{
1565 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001566 int diffwin = FALSE;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001567
Bram Moolenaar29323592016-07-24 22:04:11 +02001568 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001569 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001570 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001571 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001572 // Set 'diff' off. If option values were saved in
1573 // diff_win_options(), restore the ones whose settings seem to have
1574 // been left over from diff mode.
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001575 set_diff_option(wp, FALSE);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001576
Bram Moolenaara87aa802013-07-03 15:47:03 +02001577 if (wp->w_p_diff_saved)
1578 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001579
Bram Moolenaar43929962015-07-03 15:06:56 +02001580 if (wp->w_p_scb)
1581 wp->w_p_scb = wp->w_p_scb_save;
Bram Moolenaar43929962015-07-03 15:06:56 +02001582 if (wp->w_p_crb)
1583 wp->w_p_crb = wp->w_p_crb_save;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001584 if (!(diff_flags & DIFF_FOLLOWWRAP))
1585 {
1586 if (!wp->w_p_wrap)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001587 wp->w_p_wrap = wp->w_p_wrap_save;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001588 }
Bram Moolenaar43929962015-07-03 15:06:56 +02001589#ifdef FEAT_FOLDING
1590 free_string_option(wp->w_p_fdm);
Bram Moolenaar79a213d2017-05-16 13:15:18 +02001591 wp->w_p_fdm = vim_strsave(
1592 *wp->w_p_fdm_save ? wp->w_p_fdm_save : (char_u*)"manual");
Bram Moolenaar43929962015-07-03 15:06:56 +02001593
1594 if (wp->w_p_fdc == diff_foldcolumn)
1595 wp->w_p_fdc = wp->w_p_fdc_save;
1596 if (wp->w_p_fdl == 0)
1597 wp->w_p_fdl = wp->w_p_fdl_save;
1598
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001599 // Only restore 'foldenable' when 'foldmethod' is not
1600 // "manual", otherwise we continue to show the diff folds.
Bram Moolenaar43929962015-07-03 15:06:56 +02001601 if (wp->w_p_fen)
1602 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1603 : wp->w_p_fen_save;
1604
1605 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001606#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001607 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001608 // remove filler lines
Bram Moolenaare67d5462016-08-27 22:40:42 +02001609 wp->w_topfill = 0;
1610
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001611 // make sure topline is not halfway a fold and cursor is
1612 // invalidated
Bram Moolenaare67d5462016-08-27 22:40:42 +02001613 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001614
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001615 // Note: 'sbo' is not restored, it's a global option.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001616 diff_buf_adjust(wp);
1617 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001618 diffwin |= wp->w_p_diff;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001619 }
1620
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001621 // Also remove hidden buffers from the list.
Bram Moolenaar25ea0542017-02-03 23:16:28 +01001622 if (eap->forceit)
1623 diff_buf_clear();
1624
Bram Moolenaarc8234772019-11-10 21:00:27 +01001625 if (!diffwin)
1626 {
1627 diff_need_update = FALSE;
1628 curtab->tp_diff_invalid = FALSE;
1629 curtab->tp_diff_update = FALSE;
1630 diff_clear(curtab);
1631 }
1632
Dominique Pelleaf4a61a2021-12-27 17:21:41 +00001633 // Remove "hor" from 'scrollopt' if there are no diff windows left.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001634 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1635 do_cmdline_cmd((char_u *)"set sbo-=hor");
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001636}
1637
1638/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001639 * Read the diff output and add each entry to the diff list.
1640 */
1641 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001642diff_read(
Bram Moolenaare828b762018-09-10 17:51:58 +02001643 int idx_orig, // idx of original file
1644 int idx_new, // idx of new file
Lewis Russelld9da86e2021-12-28 13:54:41 +00001645 diffio_T *dio) // diff output
Bram Moolenaar071d4272004-06-13 20:20:40 +00001646{
Bram Moolenaare828b762018-09-10 17:51:58 +02001647 FILE *fd = NULL;
1648 int line_idx = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001649 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001650 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001651 diff_T *dn, *dpl;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001652 diffout_T *dout = &dio->dio_diff;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001653 char_u linebuf[LBUFLEN]; // only need to hold the diff line
Bram Moolenaare828b762018-09-10 17:51:58 +02001654 char_u *line;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001655 long off;
1656 int i;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001657 int notset = TRUE; // block "*dp" not set yet
Bram Moolenaar49166972021-12-30 10:51:45 +00001658 diffhunk_T *hunk = NULL; // init to avoid gcc warning
Lewis Russelld9da86e2021-12-28 13:54:41 +00001659
Bram Moolenaare828b762018-09-10 17:51:58 +02001660 enum {
1661 DIFF_ED,
1662 DIFF_UNIFIED,
1663 DIFF_NONE
1664 } diffstyle = DIFF_NONE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001665
Bram Moolenaare828b762018-09-10 17:51:58 +02001666 if (dout->dout_fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001667 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001668 diffstyle = DIFF_UNIFIED;
1669 }
1670 else
1671 {
1672 fd = mch_fopen((char *)dout->dout_fname, "r");
1673 if (fd == NULL)
1674 {
Bram Moolenaare1242042021-12-16 20:56:57 +00001675 emsg(_(e_cannot_read_diff_output));
Bram Moolenaare828b762018-09-10 17:51:58 +02001676 return;
1677 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678 }
1679
Lewis Russelld9da86e2021-12-28 13:54:41 +00001680 if (!dio->dio_internal)
1681 {
1682 hunk = ALLOC_ONE(diffhunk_T);
1683 if (hunk == NULL)
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001684 {
1685 if (fd != NULL)
1686 fclose(fd);
Lewis Russelld9da86e2021-12-28 13:54:41 +00001687 return;
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001688 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00001689 }
1690
Bram Moolenaar071d4272004-06-13 20:20:40 +00001691 for (;;)
1692 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001693 if (dio->dio_internal)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001694 {
Bram Moolenaarebfec1c2023-01-22 21:14:53 +00001695 if (line_idx >= dout->dout_ga.ga_len)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001696 break; // did last line
Lewis Russelld9da86e2021-12-28 13:54:41 +00001697 hunk = ((diffhunk_T **)dout->dout_ga.ga_data)[line_idx++];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001698 }
1699 else
1700 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001701 if (fd == NULL)
1702 {
1703 if (line_idx >= dout->dout_ga.ga_len)
1704 break; // did last line
1705 line = ((char_u **)dout->dout_ga.ga_data)[line_idx++];
1706 }
Bram Moolenaar3b8defd2018-09-13 13:03:11 +02001707 else
Lewis Russelld9da86e2021-12-28 13:54:41 +00001708 {
1709 if (vim_fgets(linebuf, LBUFLEN, fd))
1710 break; // end of file
1711 line = linebuf;
1712 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001713
Lewis Russelld9da86e2021-12-28 13:54:41 +00001714 if (diffstyle == DIFF_NONE)
1715 {
1716 // Determine diff style.
1717 // ed like diff looks like this:
1718 // {first}[,{last}]c{first}[,{last}]
1719 // {first}a{first}[,{last}]
1720 // {first}[,{last}]d{first}
1721 //
1722 // unified diff looks like this:
1723 // --- file1 2018-03-20 13:23:35.783153140 +0100
1724 // +++ file2 2018-03-20 13:23:41.183156066 +0100
1725 // @@ -1,3 +1,5 @@
Keith Thompson184f71c2024-01-04 21:19:04 +01001726 if (SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001727 diffstyle = DIFF_ED;
1728 else if ((STRNCMP(line, "@@ ", 3) == 0))
1729 diffstyle = DIFF_UNIFIED;
1730 else if ((STRNCMP(line, "--- ", 4) == 0)
1731 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1732 && (STRNCMP(line, "+++ ", 4) == 0)
1733 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1734 && (STRNCMP(line, "@@ ", 3) == 0))
1735 diffstyle = DIFF_UNIFIED;
1736 else
1737 // Format not recognized yet, skip over this line. Cygwin
1738 // diff may put a warning at the start of the file.
1739 continue;
1740 }
1741
1742 if (diffstyle == DIFF_ED)
1743 {
Keith Thompson184f71c2024-01-04 21:19:04 +01001744 if (!SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001745 continue; // not the start of a diff block
1746 if (parse_diff_ed(line, hunk) == FAIL)
1747 continue;
1748 }
1749 else if (diffstyle == DIFF_UNIFIED)
1750 {
1751 if (STRNCMP(line, "@@ ", 3) != 0)
1752 continue; // not the start of a diff block
1753 if (parse_diff_unified(line, hunk) == FAIL)
1754 continue;
1755 }
1756 else
1757 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001758 emsg(_(e_invalid_diff_format));
Lewis Russelld9da86e2021-12-28 13:54:41 +00001759 break;
1760 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001761 }
1762
1763 // Go over blocks before the change, for which orig and new are equal.
1764 // Copy blocks from orig to new.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001765 while (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001766 && hunk->lnum_orig > dp->df_lnum[idx_orig]
1767 + dp->df_count[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768 {
1769 if (notset)
1770 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1771 dprev = dp;
1772 dp = dp->df_next;
1773 notset = TRUE;
1774 }
1775
1776 if (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001777 && hunk->lnum_orig <= dp->df_lnum[idx_orig]
1778 + dp->df_count[idx_orig]
1779 && hunk->lnum_orig + hunk->count_orig >= dp->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001780 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001781 // New block overlaps with existing block(s).
1782 // First find last block that overlaps.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001783 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001784 if (hunk->lnum_orig + hunk->count_orig
1785 < dpl->df_next->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001786 break;
1787
Bram Moolenaare828b762018-09-10 17:51:58 +02001788 // If the newly found block starts before the old one, set the
1789 // start back a number of lines.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001790 off = dp->df_lnum[idx_orig] - hunk->lnum_orig;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001791 if (off > 0)
1792 {
1793 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001794 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 dp->df_lnum[i] -= off;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001796 dp->df_lnum[idx_new] = hunk->lnum_new;
1797 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001798 }
1799 else if (notset)
1800 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001801 // new block inside existing one, adjust new block
Lewis Russelld9da86e2021-12-28 13:54:41 +00001802 dp->df_lnum[idx_new] = hunk->lnum_new + off;
1803 dp->df_count[idx_new] = hunk->count_new - off;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001804 }
1805 else
Bram Moolenaare828b762018-09-10 17:51:58 +02001806 // second overlap of new block with existing block
Lewis Russelld9da86e2021-12-28 13:54:41 +00001807 dp->df_count[idx_new] += hunk->count_new - hunk->count_orig
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001808 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1809 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001810
Bram Moolenaare828b762018-09-10 17:51:58 +02001811 // Adjust the size of the block to include all the lines to the
1812 // end of the existing block or the new diff, whatever ends last.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001813 off = (hunk->lnum_orig + hunk->count_orig)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1815 if (off < 0)
1816 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001817 // new change ends in existing block, adjust the end if not
1818 // done already
Bram Moolenaar071d4272004-06-13 20:20:40 +00001819 if (notset)
1820 dp->df_count[idx_new] += -off;
1821 off = 0;
1822 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001823 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001824 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001825 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1826 - dp->df_lnum[i] + off;
1827
Bram Moolenaare828b762018-09-10 17:51:58 +02001828 // Delete the diff blocks that have been merged into one.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001829 dn = dp->df_next;
1830 dp->df_next = dpl->df_next;
1831 while (dn != dp->df_next)
1832 {
1833 dpl = dn->df_next;
1834 vim_free(dn);
1835 dn = dpl;
1836 }
1837 }
1838 else
1839 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001840 // Allocate a new diffblock.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001841 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001843 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844
Lewis Russelld9da86e2021-12-28 13:54:41 +00001845 dp->df_lnum[idx_orig] = hunk->lnum_orig;
1846 dp->df_count[idx_orig] = hunk->count_orig;
1847 dp->df_lnum[idx_new] = hunk->lnum_new;
1848 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001849
Bram Moolenaare828b762018-09-10 17:51:58 +02001850 // Set values for other buffers, these must be equal to the
1851 // original buffer, otherwise there would have been a change
1852 // already.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001853 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001854 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001855 diff_copy_entry(dprev, dp, idx_orig, i);
1856 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001857 notset = FALSE; // "*dp" has been set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001858 }
1859
Bram Moolenaare828b762018-09-10 17:51:58 +02001860 // for remaining diff blocks orig and new are equal
Bram Moolenaar071d4272004-06-13 20:20:40 +00001861 while (dp != NULL)
1862 {
1863 if (notset)
1864 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1865 dprev = dp;
1866 dp = dp->df_next;
1867 notset = TRUE;
1868 }
1869
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001870done:
Lewis Russelld9da86e2021-12-28 13:54:41 +00001871 if (!dio->dio_internal)
1872 vim_free(hunk);
1873
Bram Moolenaare828b762018-09-10 17:51:58 +02001874 if (fd != NULL)
1875 fclose(fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876}
1877
1878/*
1879 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1880 */
1881 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001882diff_copy_entry(
1883 diff_T *dprev,
1884 diff_T *dp,
1885 int idx_orig,
1886 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001887{
1888 long off;
1889
1890 if (dprev == NULL)
1891 off = 0;
1892 else
1893 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1894 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1895 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1896 dp->df_count[idx_new] = dp->df_count[idx_orig];
1897}
1898
1899/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001900 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001902 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001903diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904{
1905 diff_T *p, *next_p;
1906
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001907 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908 {
1909 next_p = p->df_next;
1910 vim_free(p);
1911 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001912 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913}
1914
1915/*
1916 * Check diff status for line "lnum" in buffer "buf":
1917 * Returns 0 for nothing special
1918 * Returns -1 for a line that should be highlighted as changed.
1919 * Returns -2 for a line that should be highlighted as added/deleted.
1920 * Returns > 0 for inserting that many filler lines above it (never happens
1921 * when 'diffopt' doesn't contain "filler").
1922 * This should only be used for windows where 'diff' is set.
1923 */
1924 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001925diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001926{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001927 int idx; // index in tp_diffbuf[] for this buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928 diff_T *dp;
1929 int maxcount;
1930 int i;
1931 buf_T *buf = wp->w_buffer;
1932 int cmp;
1933
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001934 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001935 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00001936
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001937 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) // no diffs at all
Bram Moolenaar071d4272004-06-13 20:20:40 +00001938 return 0;
1939
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001940 // safety check: "lnum" must be a buffer line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001941 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1942 return 0;
1943
1944 idx = diff_buf_idx(buf);
1945 if (idx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001946 return 0; // no diffs for buffer "buf"
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947
1948#ifdef FEAT_FOLDING
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001949 // A closed fold never has filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001950 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1951 return 0;
1952#endif
1953
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001954 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02001955 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001956 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1957 break;
1958 if (dp == NULL || lnum < dp->df_lnum[idx])
1959 return 0;
1960
1961 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1962 {
1963 int zero = FALSE;
1964
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001965 // Changed or inserted line. If the other buffers have a count of
1966 // zero, the lines were inserted. If the other buffers have the same
1967 // count, check if the lines are identical.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001968 cmp = FALSE;
1969 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001970 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001971 {
1972 if (dp->df_count[i] == 0)
1973 zero = TRUE;
1974 else
1975 {
1976 if (dp->df_count[i] != dp->df_count[idx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001977 return -1; // nr of lines changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978 cmp = TRUE;
1979 }
1980 }
1981 if (cmp)
1982 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001983 // Compare all lines. If they are equal the lines were inserted
1984 // in some buffers, deleted in others, but not changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001985 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001986 if (i != idx && curtab->tp_diffbuf[i] != NULL
1987 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001988 if (!diff_equal_entry(dp, idx, i))
1989 return -1;
1990 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001991 // If there is no buffer with zero lines then there is no difference
1992 // any longer. Happens when making a change (or undo) that removes
1993 // the difference. Can't remove the entry here, we might be halfway
1994 // updating the window. Just report the text as unchanged. Other
1995 // windows might still show the change though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996 if (zero == FALSE)
1997 return 0;
1998 return -2;
1999 }
2000
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002001 // If 'diffopt' doesn't contain "filler", return 0.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002 if (!(diff_flags & DIFF_FILLER))
2003 return 0;
2004
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002005 // Insert filler lines above the line just below the change. Will return
2006 // 0 when this buf had the max count.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002007 maxcount = 0;
2008 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002009 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010 maxcount = dp->df_count[i];
2011 return maxcount - dp->df_count[idx];
2012}
2013
2014/*
2015 * Compare two entries in diff "*dp" and return TRUE if they are equal.
2016 */
2017 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002018diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002019{
2020 int i;
2021 char_u *line;
2022 int cmp;
2023
2024 if (dp->df_count[idx1] != dp->df_count[idx2])
2025 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002026 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002027 return FALSE;
2028 for (i = 0; i < dp->df_count[idx1]; ++i)
2029 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002030 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002031 dp->df_lnum[idx1] + i, FALSE));
2032 if (line == NULL)
2033 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002034 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 dp->df_lnum[idx2] + i, FALSE));
2036 vim_free(line);
2037 if (cmp != 0)
2038 return FALSE;
2039 }
2040 return TRUE;
2041}
2042
2043/*
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002044 * Compare the characters at "p1" and "p2". If they are equal (possibly
2045 * ignoring case) return TRUE and set "len" to the number of bytes.
2046 */
2047 static int
2048diff_equal_char(char_u *p1, char_u *p2, int *len)
2049{
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002050 int l = (*mb_ptr2len)(p1);
2051
2052 if (l != (*mb_ptr2len)(p2))
2053 return FALSE;
2054 if (l > 1)
2055 {
2056 if (STRNCMP(p1, p2, l) != 0
2057 && (!enc_utf8
2058 || !(diff_flags & DIFF_ICASE)
2059 || utf_fold(utf_ptr2char(p1))
2060 != utf_fold(utf_ptr2char(p2))))
2061 return FALSE;
2062 *len = l;
2063 }
2064 else
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002065 {
2066 if ((*p1 != *p2)
2067 && (!(diff_flags & DIFF_ICASE)
2068 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
2069 return FALSE;
2070 *len = 1;
2071 }
2072 return TRUE;
2073}
2074
2075/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002076 * Compare strings "s1" and "s2" according to 'diffopt'.
2077 * Return non-zero when they are different.
2078 */
2079 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002080diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002081{
2082 char_u *p1, *p2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002083 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084
Bram Moolenaar785fc652018-09-15 19:17:38 +02002085 if ((diff_flags & DIFF_IBLANK)
2086 && (*skipwhite(s1) == NUL || *skipwhite(s2) == NUL))
2087 return 0;
2088
2089 if ((diff_flags & (DIFF_ICASE | ALL_WHITE_DIFF)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002090 return STRCMP(s1, s2);
Bram Moolenaar785fc652018-09-15 19:17:38 +02002091 if ((diff_flags & DIFF_ICASE) && !(diff_flags & ALL_WHITE_DIFF))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002092 return MB_STRICMP(s1, s2);
2093
Bram Moolenaar071d4272004-06-13 20:20:40 +00002094 p1 = s1;
2095 p2 = s2;
Bram Moolenaar785fc652018-09-15 19:17:38 +02002096
2097 // Ignore white space changes and possibly ignore case.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098 while (*p1 != NUL && *p2 != NUL)
2099 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002100 if (((diff_flags & DIFF_IWHITE)
2101 && VIM_ISWHITE(*p1) && VIM_ISWHITE(*p2))
2102 || ((diff_flags & DIFF_IWHITEALL)
2103 && (VIM_ISWHITE(*p1) || VIM_ISWHITE(*p2))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002104 {
2105 p1 = skipwhite(p1);
2106 p2 = skipwhite(p2);
2107 }
2108 else
2109 {
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002110 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002111 break;
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002112 p1 += l;
2113 p2 += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114 }
2115 }
2116
Bram Moolenaar785fc652018-09-15 19:17:38 +02002117 // Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002118 p1 = skipwhite(p1);
2119 p2 = skipwhite(p2);
2120 if (*p1 != NUL || *p2 != NUL)
2121 return 1;
2122 return 0;
2123}
2124
2125/*
2126 * Return the number of filler lines above "lnum".
2127 */
2128 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002129diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002130{
2131 int n;
2132
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002133 // be quick when there are no filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002134 if (!(diff_flags & DIFF_FILLER))
2135 return 0;
2136 n = diff_check(wp, lnum);
2137 if (n <= 0)
2138 return 0;
2139 return n;
2140}
2141
2142/*
2143 * Set the topline of "towin" to match the position in "fromwin", so that they
2144 * show the same diff'ed lines.
2145 */
2146 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002147diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002148{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002149 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002150 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002151 int fromidx;
2152 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002153 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002154 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002155 int i;
2156
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002157 fromidx = diff_buf_idx(frombuf);
2158 if (fromidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002159 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002161 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002162 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163
2164 towin->w_topfill = 0;
2165
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002166 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002167 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002168 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002169 break;
2170 if (dp == NULL)
2171 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002172 // After last change, compute topline relative to end of file; no
2173 // filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002174 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002175 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002176 }
2177 else
2178 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002179 // Find index for "towin".
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002180 toidx = diff_buf_idx(towin->w_buffer);
2181 if (toidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002182 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002183
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002184 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
2185 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002186 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002187 // Inside a change: compute filler lines. With three or more
2188 // buffers we need to know the largest count.
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002189 max_count = 0;
2190 for (i = 0; i < DB_COUNT; ++i)
2191 if (curtab->tp_diffbuf[i] != NULL
2192 && max_count < dp->df_count[i])
2193 max_count = dp->df_count[i];
2194
2195 if (dp->df_count[toidx] == dp->df_count[fromidx])
2196 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002197 // same number of lines: use same filler count
Bram Moolenaar071d4272004-06-13 20:20:40 +00002198 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002199 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002200 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002201 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002202 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002203 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002204 // more lines in towin and fromwin doesn't show diff
2205 // lines, only filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002206 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
2207 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002208 // towin also only shows filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002209 towin->w_topline = dp->df_lnum[toidx]
2210 + dp->df_count[toidx];
2211 towin->w_topfill = fromwin->w_topfill;
2212 }
2213 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002214 // towin still has some diff lines to show
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002215 towin->w_topline = dp->df_lnum[toidx]
2216 + max_count - fromwin->w_topfill;
2217 }
2218 }
2219 else if (towin->w_topline >= dp->df_lnum[toidx]
2220 + dp->df_count[toidx])
2221 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002222 // less lines in towin and no diff lines to show: compute
2223 // filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002224 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
2225 if (diff_flags & DIFF_FILLER)
2226 {
2227 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002228 // fromwin is also out of diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002229 towin->w_topfill = fromwin->w_topfill;
2230 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002231 // fromwin has some diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002232 towin->w_topfill = dp->df_lnum[fromidx]
2233 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234 }
2235 }
2236 }
2237 }
2238
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002239 // safety check (if diff info gets outdated strange things may happen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002240 towin->w_botfill = FALSE;
2241 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
2242 {
2243 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
2244 towin->w_botfill = TRUE;
2245 }
2246 if (towin->w_topline < 1)
2247 {
2248 towin->w_topline = 1;
2249 towin->w_topfill = 0;
2250 }
2251
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002252 // When w_topline changes need to recompute w_botline and cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00002253 invalidate_botline_win(towin);
2254 changed_line_abv_curs_win(towin);
2255
2256 check_topfill(towin, FALSE);
2257#ifdef FEAT_FOLDING
2258 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
2259 NULL, TRUE, NULL);
2260#endif
2261}
2262
2263/*
2264 * This is called when 'diffopt' is changed.
2265 */
2266 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002267diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268{
2269 char_u *p;
2270 int diff_context_new = 6;
2271 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002272 int diff_foldcolumn_new = 2;
Bram Moolenaare828b762018-09-10 17:51:58 +02002273 long diff_algorithm_new = 0;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002274 long diff_indent_heuristic = 0;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002275 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002276
2277 p = p_dip;
2278 while (*p != NUL)
2279 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002280 // Note: Keep this in sync with p_dip_values
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281 if (STRNCMP(p, "filler", 6) == 0)
2282 {
2283 p += 6;
2284 diff_flags_new |= DIFF_FILLER;
2285 }
2286 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
2287 {
2288 p += 8;
2289 diff_context_new = getdigits(&p);
2290 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002291 else if (STRNCMP(p, "iblank", 6) == 0)
2292 {
2293 p += 6;
2294 diff_flags_new |= DIFF_IBLANK;
2295 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002296 else if (STRNCMP(p, "icase", 5) == 0)
2297 {
2298 p += 5;
2299 diff_flags_new |= DIFF_ICASE;
2300 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002301 else if (STRNCMP(p, "iwhiteall", 9) == 0)
2302 {
2303 p += 9;
2304 diff_flags_new |= DIFF_IWHITEALL;
2305 }
2306 else if (STRNCMP(p, "iwhiteeol", 9) == 0)
2307 {
2308 p += 9;
2309 diff_flags_new |= DIFF_IWHITEEOL;
2310 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002311 else if (STRNCMP(p, "iwhite", 6) == 0)
2312 {
2313 p += 6;
2314 diff_flags_new |= DIFF_IWHITE;
2315 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002316 else if (STRNCMP(p, "horizontal", 10) == 0)
2317 {
2318 p += 10;
2319 diff_flags_new |= DIFF_HORIZONTAL;
2320 }
2321 else if (STRNCMP(p, "vertical", 8) == 0)
2322 {
2323 p += 8;
2324 diff_flags_new |= DIFF_VERTICAL;
2325 }
2326 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
2327 {
2328 p += 11;
2329 diff_foldcolumn_new = getdigits(&p);
2330 }
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002331 else if (STRNCMP(p, "hiddenoff", 9) == 0)
2332 {
2333 p += 9;
2334 diff_flags_new |= DIFF_HIDDEN_OFF;
2335 }
Bram Moolenaarc8234772019-11-10 21:00:27 +01002336 else if (STRNCMP(p, "closeoff", 8) == 0)
2337 {
2338 p += 8;
2339 diff_flags_new |= DIFF_CLOSE_OFF;
2340 }
Bram Moolenaar4223d432021-02-10 13:18:17 +01002341 else if (STRNCMP(p, "followwrap", 10) == 0)
2342 {
2343 p += 10;
2344 diff_flags_new |= DIFF_FOLLOWWRAP;
2345 }
Bram Moolenaare828b762018-09-10 17:51:58 +02002346 else if (STRNCMP(p, "indent-heuristic", 16) == 0)
2347 {
2348 p += 16;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002349 diff_indent_heuristic = XDF_INDENT_HEURISTIC;
Bram Moolenaare828b762018-09-10 17:51:58 +02002350 }
2351 else if (STRNCMP(p, "internal", 8) == 0)
2352 {
2353 p += 8;
2354 diff_flags_new |= DIFF_INTERNAL;
2355 }
2356 else if (STRNCMP(p, "algorithm:", 10) == 0)
2357 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002358 // Note: Keep this in sync with p_dip_algorithm_values.
Bram Moolenaare828b762018-09-10 17:51:58 +02002359 p += 10;
2360 if (STRNCMP(p, "myers", 5) == 0)
2361 {
2362 p += 5;
2363 diff_algorithm_new = 0;
2364 }
2365 else if (STRNCMP(p, "minimal", 7) == 0)
2366 {
2367 p += 7;
2368 diff_algorithm_new = XDF_NEED_MINIMAL;
2369 }
2370 else if (STRNCMP(p, "patience", 8) == 0)
2371 {
2372 p += 8;
2373 diff_algorithm_new = XDF_PATIENCE_DIFF;
2374 }
2375 else if (STRNCMP(p, "histogram", 9) == 0)
2376 {
2377 p += 9;
2378 diff_algorithm_new = XDF_HISTOGRAM_DIFF;
2379 }
Bram Moolenaard0721052018-11-05 21:21:33 +01002380 else
2381 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02002382 }
2383
Bram Moolenaar071d4272004-06-13 20:20:40 +00002384 if (*p != ',' && *p != NUL)
2385 return FAIL;
2386 if (*p == ',')
2387 ++p;
2388 }
2389
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002390 diff_algorithm_new |= diff_indent_heuristic;
2391
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002392 // Can't have both "horizontal" and "vertical".
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002393 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
2394 return FAIL;
2395
Bram Moolenaar198fa062018-09-18 21:20:26 +02002396 // If flags were added or removed, or the algorithm was changed, need to
2397 // update the diff.
Bram Moolenaare828b762018-09-10 17:51:58 +02002398 if (diff_flags != diff_flags_new || diff_algorithm != diff_algorithm_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02002399 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002400 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002401
2402 diff_flags = diff_flags_new;
Bram Moolenaarb9ddda62019-02-19 23:00:50 +01002403 diff_context = diff_context_new == 0 ? 1 : diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002404 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaare828b762018-09-10 17:51:58 +02002405 diff_algorithm = diff_algorithm_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002406
2407 diff_redraw(TRUE);
2408
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002409 // recompute the scroll binding with the new option value, may
2410 // remove or add filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411 check_scrollbind((linenr_T)0, 0L);
2412
2413 return OK;
2414}
2415
2416/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002417 * Return TRUE if 'diffopt' contains "horizontal".
2418 */
2419 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002420diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002421{
2422 return (diff_flags & DIFF_HORIZONTAL) != 0;
2423}
2424
2425/*
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002426 * Return TRUE if 'diffopt' contains "hiddenoff".
2427 */
2428 int
2429diffopt_hiddenoff(void)
2430{
2431 return (diff_flags & DIFF_HIDDEN_OFF) != 0;
2432}
2433
2434/*
Bram Moolenaarc8234772019-11-10 21:00:27 +01002435 * Return TRUE if 'diffopt' contains "closeoff".
2436 */
2437 int
2438diffopt_closeoff(void)
2439{
2440 return (diff_flags & DIFF_CLOSE_OFF) != 0;
2441}
2442
2443/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002444 * Find the difference within a changed line.
2445 * Returns TRUE if the line was added, no other buffer has it.
2446 */
2447 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002448diff_find_change(
2449 win_T *wp,
2450 linenr_T lnum,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002451 int *startp, // first char of the change
2452 int *endp) // last char of the change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453{
2454 char_u *line_org;
2455 char_u *line_new;
2456 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002457 int si_org, si_new;
2458 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002459 diff_T *dp;
2460 int idx;
2461 int off;
2462 int added = TRUE;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002463 char_u *p1, *p2;
2464 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002465
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002466 // Make a copy of the line, the next ml_get() will invalidate it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002467 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
2468 if (line_org == NULL)
2469 return FALSE;
2470
2471 idx = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002472 if (idx == DB_COUNT) // cannot happen
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002473 {
2474 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002475 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002476 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002477
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002478 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002479 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2481 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002482 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002483 {
2484 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002486 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487
2488 off = lnum - dp->df_lnum[idx];
2489
2490 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002491 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002492 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002493 // Skip lines that are not in the other change (filler lines).
Bram Moolenaar071d4272004-06-13 20:20:40 +00002494 if (off >= dp->df_count[i])
2495 continue;
2496 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002497 line_new = ml_get_buf(curtab->tp_diffbuf[i],
2498 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002499
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002500 // Search for start of difference
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002501 si_org = si_new = 0;
2502 while (line_org[si_org] != NUL)
2503 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002504 if (((diff_flags & DIFF_IWHITE)
2505 && VIM_ISWHITE(line_org[si_org])
2506 && VIM_ISWHITE(line_new[si_new]))
2507 || ((diff_flags & DIFF_IWHITEALL)
2508 && (VIM_ISWHITE(line_org[si_org])
2509 || VIM_ISWHITE(line_new[si_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002510 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002511 si_org = (int)(skipwhite(line_org + si_org) - line_org);
2512 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002513 }
2514 else
2515 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002516 if (!diff_equal_char(line_org + si_org, line_new + si_new,
2517 &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002518 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002519 si_org += l;
2520 si_new += l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002521 }
2522 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002523 if (has_mbyte)
2524 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002525 // Move back to first byte of character in both lines (may
2526 // have "nn^" in line_org and "n^ in line_new).
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002527 si_org -= (*mb_head_off)(line_org, line_org + si_org);
2528 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002529 }
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002530 if (*startp > si_org)
2531 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002532
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002533 // Search for end of difference, if any.
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002534 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002535 {
2536 ei_org = (int)STRLEN(line_org);
2537 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002538 while (ei_org >= *startp && ei_new >= si_new
2539 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002540 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002541 if (((diff_flags & DIFF_IWHITE)
2542 && VIM_ISWHITE(line_org[ei_org])
2543 && VIM_ISWHITE(line_new[ei_new]))
2544 || ((diff_flags & DIFF_IWHITEALL)
2545 && (VIM_ISWHITE(line_org[ei_org])
2546 || VIM_ISWHITE(line_new[ei_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002547 {
2548 while (ei_org >= *startp
Bram Moolenaar1c465442017-03-12 20:10:05 +01002549 && VIM_ISWHITE(line_org[ei_org]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002550 --ei_org;
2551 while (ei_new >= si_new
Bram Moolenaar1c465442017-03-12 20:10:05 +01002552 && VIM_ISWHITE(line_new[ei_new]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002553 --ei_new;
2554 }
2555 else
2556 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002557 p1 = line_org + ei_org;
2558 p2 = line_new + ei_new;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002559 p1 -= (*mb_head_off)(line_org, p1);
2560 p2 -= (*mb_head_off)(line_new, p2);
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002561 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002562 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002563 ei_org -= l;
2564 ei_new -= l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002565 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002566 }
2567 if (*endp < ei_org)
2568 *endp = ei_org;
2569 }
2570 }
2571
2572 vim_free(line_org);
2573 return added;
2574}
2575
2576#if defined(FEAT_FOLDING) || defined(PROTO)
2577/*
2578 * Return TRUE if line "lnum" is not close to a diff block, this line should
2579 * be in a fold.
2580 * Return FALSE if there are no diff blocks at all in this window.
2581 */
2582 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002583diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584{
2585 int i;
2586 int idx = -1;
2587 int other = FALSE;
2588 diff_T *dp;
2589
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002590 // Return if 'diff' isn't set.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591 if (!wp->w_p_diff)
2592 return FALSE;
2593
2594 for (i = 0; i < DB_COUNT; ++i)
2595 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002596 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002597 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002598 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002599 other = TRUE;
2600 }
2601
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002602 // return here if there are no diffs in the window
Bram Moolenaar071d4272004-06-13 20:20:40 +00002603 if (idx == -1 || !other)
2604 return FALSE;
2605
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002606 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002607 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002608
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002609 // Return if there are no diff blocks. All lines will be folded.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002610 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002611 return TRUE;
2612
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002613 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002615 // If this change is below the line there can't be any further match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002616 if (dp->df_lnum[idx] - diff_context > lnum)
2617 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002618 // If this change ends before the line we have a match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002619 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2620 return FALSE;
2621 }
2622 return TRUE;
2623}
2624#endif
2625
2626/*
2627 * "dp" and "do" commands.
2628 */
2629 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002630nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002631{
2632 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002633 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002634
Bram Moolenaarf2732452018-06-03 14:47:35 +02002635#ifdef FEAT_JOB_CHANNEL
2636 if (bt_prompt(curbuf))
2637 {
2638 vim_beep(BO_OPER);
2639 return;
2640 }
2641#endif
Bram Moolenaar6a643652014-10-31 13:54:25 +01002642 if (count == 0)
2643 ea.arg = (char_u *)"";
2644 else
2645 {
2646 vim_snprintf((char *)buf, 30, "%ld", count);
2647 ea.arg = buf;
2648 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002649 if (put)
2650 ea.cmdidx = CMD_diffput;
2651 else
2652 ea.cmdidx = CMD_diffget;
2653 ea.addr_count = 0;
2654 ea.line1 = curwin->w_cursor.lnum;
2655 ea.line2 = curwin->w_cursor.lnum;
2656 ex_diffgetput(&ea);
2657}
2658
2659/*
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002660 * Return TRUE if "diff" appears in the list of diff blocks of the current tab.
2661 */
2662 static int
2663valid_diff(diff_T *diff)
2664{
2665 diff_T *dp;
2666
Yegappan Lakshmanan14113fd2023-03-07 17:13:51 +00002667 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002668 if (dp == diff)
2669 return TRUE;
2670 return FALSE;
2671}
2672
2673/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002674 * ":diffget"
2675 * ":diffput"
2676 */
2677 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002678ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002679{
2680 linenr_T lnum;
2681 int count;
2682 linenr_T off = 0;
2683 diff_T *dp;
2684 diff_T *dprev;
2685 diff_T *dfree;
2686 int idx_cur;
2687 int idx_other;
2688 int idx_from;
2689 int idx_to;
2690 int i;
2691 int added;
2692 char_u *p;
2693 aco_save_T aco;
2694 buf_T *buf;
2695 int start_skip, end_skip;
2696 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002697 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002698 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002700 // Find the current buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002701 idx_cur = diff_buf_idx(curbuf);
2702 if (idx_cur == DB_COUNT)
2703 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002704 emsg(_(e_current_buffer_is_not_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705 return;
2706 }
2707
2708 if (*eap->arg == NUL)
2709 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002710 // No argument: Find the other buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002711 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002712 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002713 && curtab->tp_diffbuf[idx_other] != NULL)
2714 {
2715 if (eap->cmdidx != CMD_diffput
2716 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2717 break;
2718 found_not_ma = TRUE;
2719 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002720 if (idx_other == DB_COUNT)
2721 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002722 if (found_not_ma)
Bram Moolenaar677658a2022-01-05 16:09:06 +00002723 emsg(_(e_no_other_buffer_in_diff_mode_is_modifiable));
Bram Moolenaar602eb742007-02-20 03:43:38 +00002724 else
Bram Moolenaare1242042021-12-16 20:56:57 +00002725 emsg(_(e_no_other_buffer_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002726 return;
2727 }
2728
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002729 // Check that there isn't a third buffer in the list
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002731 if (curtab->tp_diffbuf[i] != curbuf
2732 && curtab->tp_diffbuf[i] != NULL
2733 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002734 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002735 emsg(_(e_more_than_two_buffers_in_diff_mode_dont_know_which_one_to_use));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736 return;
2737 }
2738 }
2739 else
2740 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002741 // Buffer number or pattern given. Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002742 p = eap->arg + STRLEN(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01002743 while (p > eap->arg && VIM_ISWHITE(p[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002744 --p;
2745 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2746 ;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002747 if (eap->arg + i == p) // digits only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002748 i = atol((char *)eap->arg);
2749 else
2750 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002751 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002752 if (i < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002753 return; // error message already given
Bram Moolenaar071d4272004-06-13 20:20:40 +00002754 }
2755 buf = buflist_findnr(i);
2756 if (buf == NULL)
2757 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002758 semsg(_(e_cant_find_buffer_str), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002759 return;
2760 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002761 if (buf == curbuf)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002762 return; // nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +00002763 idx_other = diff_buf_idx(buf);
2764 if (idx_other == DB_COUNT)
2765 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002766 semsg(_(e_buffer_str_is_not_in_diff_mode), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002767 return;
2768 }
2769 }
2770
2771 diff_busy = TRUE;
2772
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002773 // When no range given include the line above or below the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002774 if (eap->addr_count == 0)
2775 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002776 // Make it possible that ":diffget" on the last line gets line below
2777 // the cursor line when there is no difference above the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778 if (eap->cmdidx == CMD_diffget
2779 && eap->line1 == curbuf->b_ml.ml_line_count
2780 && diff_check(curwin, eap->line1) == 0
2781 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2782 ++eap->line2;
2783 else if (eap->line1 > 0)
2784 --eap->line1;
2785 }
2786
2787 if (eap->cmdidx == CMD_diffget)
2788 {
2789 idx_from = idx_other;
2790 idx_to = idx_cur;
2791 }
2792 else
2793 {
2794 idx_from = idx_cur;
2795 idx_to = idx_other;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002796 // Need to make the other buffer the current buffer to be able to make
2797 // changes in it.
Bram Moolenaare76062c2022-11-28 18:51:43 +00002798 // Set curwin/curbuf to buf and save a few things.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002799 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaare76062c2022-11-28 18:51:43 +00002800 if (curbuf != curtab->tp_diffbuf[idx_other])
2801 // Could not find a window for this buffer, the rest is likely to
2802 // fail.
2803 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002804 }
2805
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002806 // May give the warning for a changed buffer here, which can trigger the
2807 // FileChangedRO autocommand, which may do nasty things and mess
2808 // everything up.
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002809 if (!curbuf->b_changed)
2810 {
2811 change_warning(0);
2812 if (diff_buf_idx(curbuf) != idx_to)
2813 {
Bram Moolenaar677658a2022-01-05 16:09:06 +00002814 emsg(_(e_buffer_changed_unexpectedly));
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002815 goto theend;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002816 }
2817 }
2818
Bram Moolenaar071d4272004-06-13 20:20:40 +00002819 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002820 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002821 {
2822 if (dp->df_lnum[idx_cur] > eap->line2 + off)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002823 break; // past the range that was specified
Bram Moolenaar071d4272004-06-13 20:20:40 +00002824
2825 dfree = NULL;
2826 lnum = dp->df_lnum[idx_to];
2827 count = dp->df_count[idx_to];
2828 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2829 && u_save(lnum - 1, lnum + count) != FAIL)
2830 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002831 // Inside the specified range and saving for undo worked.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002832 start_skip = 0;
2833 end_skip = 0;
2834 if (eap->addr_count > 0)
2835 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002836 // A range was specified: check if lines need to be skipped.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002837 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2838 if (start_skip > 0)
2839 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002840 // range starts below start of current diff block
Bram Moolenaar071d4272004-06-13 20:20:40 +00002841 if (start_skip > count)
2842 {
2843 lnum += count;
2844 count = 0;
2845 }
2846 else
2847 {
2848 count -= start_skip;
2849 lnum += start_skip;
2850 }
2851 }
2852 else
2853 start_skip = 0;
2854
2855 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2856 - (eap->line2 + off);
2857 if (end_skip > 0)
2858 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002859 // range ends above end of current/from diff block
2860 if (idx_cur == idx_from) // :diffput
Bram Moolenaar071d4272004-06-13 20:20:40 +00002861 {
2862 i = dp->df_count[idx_cur] - start_skip - end_skip;
2863 if (count > i)
2864 count = i;
2865 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002866 else // :diffget
Bram Moolenaar071d4272004-06-13 20:20:40 +00002867 {
2868 count -= end_skip;
2869 end_skip = dp->df_count[idx_from] - start_skip - count;
2870 if (end_skip < 0)
2871 end_skip = 0;
2872 }
2873 }
2874 else
2875 end_skip = 0;
2876 }
2877
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002878 buf_empty = BUFEMPTY();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002879 added = 0;
2880 for (i = 0; i < count; ++i)
2881 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002882 // remember deleting the last line of the buffer
Bram Moolenaar280f1262006-01-30 00:14:18 +00002883 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar4e677b92022-07-28 18:44:27 +01002884 if (ml_delete(lnum) == OK)
2885 --added;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002886 }
2887 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2888 {
2889 linenr_T nr;
2890
2891 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002892 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002893 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002894 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2895 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002896 if (p != NULL)
2897 {
2898 ml_append(lnum + i - 1, p, 0, FALSE);
2899 vim_free(p);
2900 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002901 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2902 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002903 // Added the first line into an empty buffer, need to
2904 // delete the dummy empty line.
Bram Moolenaar280f1262006-01-30 00:14:18 +00002905 buf_empty = FALSE;
Bram Moolenaarca70c072020-05-30 20:30:46 +02002906 ml_delete((linenr_T)2);
Bram Moolenaar280f1262006-01-30 00:14:18 +00002907 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002908 }
2909 }
2910 new_count = dp->df_count[idx_to] + added;
2911 dp->df_count[idx_to] = new_count;
2912
2913 if (start_skip == 0 && end_skip == 0)
2914 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002915 // Check if there are any other buffers and if the diff is
2916 // equal in them.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002917 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002918 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2919 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002920 && !diff_equal_entry(dp, idx_from, i))
2921 break;
2922 if (i == DB_COUNT)
2923 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002924 // delete the diff entry, the buffers are now equal here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002925 dfree = dp;
2926 dp = dp->df_next;
2927 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002928 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002929 else
2930 dprev->df_next = dp;
2931 }
2932 }
2933
Bram Moolenaar071d4272004-06-13 20:20:40 +00002934 if (added != 0)
2935 {
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002936 // Adjust marks. This will change the following entries!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002937 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2938 if (curwin->w_cursor.lnum >= lnum)
2939 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002940 // Adjust the cursor position if it's in/after the changed
2941 // lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942 if (curwin->w_cursor.lnum >= lnum + count)
2943 curwin->w_cursor.lnum += added;
2944 else if (added < 0)
2945 curwin->w_cursor.lnum = lnum;
2946 }
2947 }
2948 changed_lines(lnum, 0, lnum + count, (long)added);
2949
2950 if (dfree != NULL)
2951 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002952 // Diff is deleted, update folds in other windows.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002953#ifdef FEAT_FOLDING
2954 diff_fold_update(dfree, idx_to);
2955#endif
2956 vim_free(dfree);
2957 }
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002958
2959 // mark_adjust() may have made "dp" invalid. We don't know where
2960 // to continue then, bail out.
2961 if (added != 0 && !valid_diff(dp))
2962 break;
2963
2964 if (dfree == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002965 // mark_adjust() may have changed the count in a wrong way
Bram Moolenaar071d4272004-06-13 20:20:40 +00002966 dp->df_count[idx_to] = new_count;
2967
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002968 // When changing the current buffer, keep track of line numbers
Bram Moolenaar071d4272004-06-13 20:20:40 +00002969 if (idx_cur == idx_to)
2970 off += added;
2971 }
2972
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002973 // If before the range or not deleted, go to next diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002974 if (dfree == NULL)
2975 {
2976 dprev = dp;
2977 dp = dp->df_next;
2978 }
2979 }
2980
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002981 // restore curwin/curbuf and a few other things
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002982 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002984 // Syncing undo only works for the current buffer, but we change
2985 // another buffer. Sync undo if the command was typed. This isn't
2986 // 100% right when ":diffput" is used in a function or mapping.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002988 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002989 aucmd_restbuf(&aco);
2990 }
2991
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002992theend:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002993 diff_busy = FALSE;
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002994 if (diff_need_update)
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002995 ex_diffupdate(NULL);
Bram Moolenaardf77cef2018-10-07 17:46:42 +02002996
Bram Moolenaar5f57bdc2018-10-25 17:52:23 +02002997 // Check that the cursor is on a valid character and update its
Bram Moolenaardf77cef2018-10-07 17:46:42 +02002998 // position. When there were filler lines the topline has become
2999 // invalid.
3000 check_cursor();
3001 changed_line_abv_curs();
3002
3003 if (diff_need_update)
3004 // redraw already done by ex_diffupdate()
3005 diff_need_update = FALSE;
Bram Moolenaar198fa062018-09-18 21:20:26 +02003006 else
3007 {
Bram Moolenaar198fa062018-09-18 21:20:26 +02003008 // Also need to redraw the other buffers.
3009 diff_redraw(FALSE);
3010 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
3011 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012}
3013
3014#ifdef FEAT_FOLDING
3015/*
3016 * Update folds for all diff buffers for entry "dp".
3017 * Skip buffer with index "skip_idx".
3018 * When there are no diffs, all folds are removed.
3019 */
3020 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01003021diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003022{
3023 int i;
3024 win_T *wp;
3025
Bram Moolenaar29323592016-07-24 22:04:11 +02003026 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003027 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003028 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003029 foldUpdate(wp, dp->df_lnum[i],
3030 dp->df_lnum[i] + dp->df_count[i]);
3031}
3032#endif
3033
3034/*
3035 * Return TRUE if buffer "buf" is in diff-mode.
3036 */
3037 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003038diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003039{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003040 tabpage_T *tp;
3041
Bram Moolenaar29323592016-07-24 22:04:11 +02003042 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003043 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
3044 return TRUE;
3045 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046}
3047
3048/*
3049 * Move "count" times in direction "dir" to the next diff block.
3050 * Return FAIL if there isn't such a diff block.
3051 */
3052 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003053diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003054{
3055 int idx;
3056 linenr_T lnum = curwin->w_cursor.lnum;
3057 diff_T *dp;
3058
3059 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003060 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003061 return FAIL;
3062
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003063 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003064 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003065
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003066 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar071d4272004-06-13 20:20:40 +00003067 return FAIL;
3068
3069 while (--count >= 0)
3070 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003071 // Check if already before first diff.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003072 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00003073 break;
3074
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003075 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003076 {
3077 if (dp == NULL)
3078 break;
3079 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
3080 || (dir == BACKWARD
3081 && (dp->df_next == NULL
3082 || lnum <= dp->df_next->df_lnum[idx])))
3083 {
3084 lnum = dp->df_lnum[idx];
3085 break;
3086 }
3087 }
3088 }
3089
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003090 // don't end up past the end of the file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091 if (lnum > curbuf->b_ml.ml_line_count)
3092 lnum = curbuf->b_ml.ml_line_count;
3093
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003094 // When the cursor didn't move at all we fail.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003095 if (lnum == curwin->w_cursor.lnum)
3096 return FAIL;
3097
3098 setpcmark();
3099 curwin->w_cursor.lnum = lnum;
3100 curwin->w_cursor.col = 0;
3101
3102 return OK;
3103}
3104
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003105/*
3106 * Return the line number in the current window that is closest to "lnum1" in
3107 * "buf1" in diff mode.
3108 */
3109 static linenr_T
3110diff_get_corresponding_line_int(
Bram Moolenaar7454a062016-01-30 15:14:10 +01003111 buf_T *buf1,
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003112 linenr_T lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003113{
3114 int idx1;
3115 int idx2;
3116 diff_T *dp;
3117 int baseline = 0;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003118
3119 idx1 = diff_buf_idx(buf1);
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003120 idx2 = diff_buf_idx(curbuf);
Bram Moolenaar860cae12010-06-05 23:22:07 +02003121 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
3122 return lnum1;
3123
3124 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003125 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar860cae12010-06-05 23:22:07 +02003126
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003127 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar860cae12010-06-05 23:22:07 +02003128 return lnum1;
3129
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003130 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003131 {
3132 if (dp->df_lnum[idx1] > lnum1)
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003133 return lnum1 - baseline;
3134 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003135 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003136 // Inside the diffblock
Bram Moolenaar860cae12010-06-05 23:22:07 +02003137 baseline = lnum1 - dp->df_lnum[idx1];
3138 if (baseline > dp->df_count[idx2])
3139 baseline = dp->df_count[idx2];
3140
3141 return dp->df_lnum[idx2] + baseline;
3142 }
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003143 if ( (dp->df_lnum[idx1] == lnum1)
3144 && (dp->df_count[idx1] == 0)
3145 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum)
3146 && ((dp->df_lnum[idx2] + dp->df_count[idx2])
3147 > curwin->w_cursor.lnum))
Bram Moolenaar860cae12010-06-05 23:22:07 +02003148 /*
3149 * Special case: if the cursor is just after a zero-count
3150 * block (i.e. all filler) and the target cursor is already
3151 * inside the corresponding block, leave the target cursor
3152 * unmoved. This makes repeated CTRL-W W operations work
3153 * as expected.
3154 */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003155 return curwin->w_cursor.lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003156 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
3157 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
3158 }
3159
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003160 // If we get here then the cursor is after the last diff
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003161 return lnum1 - baseline;
3162}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003163
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003164/*
3165 * Return the line number in the current window that is closest to "lnum1" in
3166 * "buf1" in diff mode. Checks the line number to be valid.
3167 */
3168 linenr_T
3169diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1)
3170{
3171 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1);
3172
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003173 // don't end up past the end of the file
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003174 if (lnum > curbuf->b_ml.ml_line_count)
3175 return curbuf->b_ml.ml_line_count;
3176 return lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003177}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003178
Bram Moolenaar071d4272004-06-13 20:20:40 +00003179/*
3180 * For line "lnum" in the current window find the equivalent lnum in window
3181 * "wp", compensating for inserted/deleted lines.
3182 */
3183 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01003184diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003185{
3186 diff_T *dp;
3187 int idx;
3188 int i;
3189 linenr_T n;
3190
3191 idx = diff_buf_idx(curbuf);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003192 if (idx == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003193 return (linenr_T)0;
3194
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003195 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003196 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003198 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003199 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003200 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
3201 break;
3202
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003203 // When after the last change, compute relative to the last line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204 if (dp == NULL)
3205 return wp->w_buffer->b_ml.ml_line_count
3206 - (curbuf->b_ml.ml_line_count - lnum);
3207
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003208 // Find index for "wp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003209 i = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003210 if (i == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003211 return (linenr_T)0;
3212
3213 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
3214 if (n > dp->df_lnum[i] + dp->df_count[i])
3215 n = dp->df_lnum[i] + dp->df_count[i];
3216 return n;
3217}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003218
Bram Moolenaare828b762018-09-10 17:51:58 +02003219/*
3220 * Handle an ED style diff line.
3221 * Return FAIL if the line does not contain diff info.
3222 */
3223 static int
3224parse_diff_ed(
3225 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003226 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003227{
3228 char_u *p;
3229 long f1, l1, f2, l2;
3230 int difftype;
3231
3232 // The line must be one of three formats:
3233 // change: {first}[,{last}]c{first}[,{last}]
3234 // append: {first}a{first}[,{last}]
3235 // delete: {first}[,{last}]d{first}
3236 p = line;
3237 f1 = getdigits(&p);
3238 if (*p == ',')
3239 {
3240 ++p;
3241 l1 = getdigits(&p);
3242 }
3243 else
3244 l1 = f1;
3245 if (*p != 'a' && *p != 'c' && *p != 'd')
3246 return FAIL; // invalid diff format
3247 difftype = *p++;
3248 f2 = getdigits(&p);
3249 if (*p == ',')
3250 {
3251 ++p;
3252 l2 = getdigits(&p);
3253 }
3254 else
3255 l2 = f2;
3256 if (l1 < f1 || l2 < f2)
3257 return FAIL;
3258
3259 if (difftype == 'a')
3260 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003261 hunk->lnum_orig = f1 + 1;
3262 hunk->count_orig = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003263 }
3264 else
3265 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003266 hunk->lnum_orig = f1;
3267 hunk->count_orig = l1 - f1 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003268 }
3269 if (difftype == 'd')
3270 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003271 hunk->lnum_new = f2 + 1;
3272 hunk->count_new = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003273 }
3274 else
3275 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003276 hunk->lnum_new = f2;
3277 hunk->count_new = l2 - f2 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003278 }
3279 return OK;
3280}
3281
3282/*
3283 * Parses unified diff with zero(!) context lines.
3284 * Return FAIL if there is no diff information in "line".
3285 */
3286 static int
3287parse_diff_unified(
3288 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003289 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003290{
3291 char_u *p;
3292 long oldline, oldcount, newline, newcount;
3293
3294 // Parse unified diff hunk header:
3295 // @@ -oldline,oldcount +newline,newcount @@
3296 p = line;
3297 if (*p++ == '@' && *p++ == '@' && *p++ == ' ' && *p++ == '-')
3298 {
3299 oldline = getdigits(&p);
3300 if (*p == ',')
3301 {
3302 ++p;
3303 oldcount = getdigits(&p);
3304 }
3305 else
3306 oldcount = 1;
3307 if (*p++ == ' ' && *p++ == '+')
3308 {
3309 newline = getdigits(&p);
3310 if (*p == ',')
3311 {
3312 ++p;
3313 newcount = getdigits(&p);
3314 }
3315 else
3316 newcount = 1;
3317 }
3318 else
3319 return FAIL; // invalid diff format
3320
3321 if (oldcount == 0)
3322 oldline += 1;
3323 if (newcount == 0)
3324 newline += 1;
3325 if (newline == 0)
3326 newline = 1;
3327
Lewis Russelld9da86e2021-12-28 13:54:41 +00003328 hunk->lnum_orig = oldline;
3329 hunk->count_orig = oldcount;
3330 hunk->lnum_new = newline;
3331 hunk->count_new = newcount;
Bram Moolenaare828b762018-09-10 17:51:58 +02003332
3333 return OK;
3334 }
3335
3336 return FAIL;
3337}
3338
3339/*
3340 * Callback function for the xdl_diff() function.
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003341 * Stores the diff output (indices) in a grow array.
Bram Moolenaare828b762018-09-10 17:51:58 +02003342 */
3343 static int
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003344xdiff_out_indices(
Lewis Russelld9da86e2021-12-28 13:54:41 +00003345 long start_a,
3346 long count_a,
3347 long start_b,
3348 long count_b,
3349 void *priv)
Bram Moolenaare828b762018-09-10 17:51:58 +02003350{
3351 diffout_T *dout = (diffout_T *)priv;
Lewis Russelld9da86e2021-12-28 13:54:41 +00003352 diffhunk_T *p = ALLOC_ONE(diffhunk_T);
Bram Moolenaare828b762018-09-10 17:51:58 +02003353
Lewis Russelld9da86e2021-12-28 13:54:41 +00003354 if (p == NULL)
3355 return -1;
Bram Moolenaarf080d702018-10-31 22:57:26 +01003356
3357 if (ga_grow(&dout->dout_ga, 1) == FAIL)
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003358 {
3359 vim_free(p);
Bram Moolenaarf080d702018-10-31 22:57:26 +01003360 return -1;
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003361 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00003362
3363 p->lnum_orig = start_a + 1;
3364 p->count_orig = count_a;
3365 p->lnum_new = start_b + 1;
3366 p->count_new = count_b;
3367 ((diffhunk_T **)dout->dout_ga.ga_data)[dout->dout_ga.ga_len++] = p;
Bram Moolenaare828b762018-09-10 17:51:58 +02003368 return 0;
3369}
3370
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003371/*
3372 * Callback function for the xdl_diff() function.
3373 * Stores the unified diff output in a grow array.
3374 */
3375 static int
3376xdiff_out_unified(
3377 void *priv,
3378 mmbuffer_t *mb,
3379 int nbuf)
3380{
3381 diffout_T *dout = (diffout_T *)priv;
3382 int i;
3383
3384 for (i = 0; i < nbuf; i++)
3385 ga_concat_len(&dout->dout_ga, (char_u *)mb[i].ptr, mb[i].size);
3386
3387 return 0;
3388}
3389
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003390#endif // FEAT_DIFF
3391
3392#if defined(FEAT_EVAL) || defined(PROTO)
3393
3394/*
3395 * "diff_filler()" function
3396 */
3397 void
3398f_diff_filler(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3399{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003400# ifdef FEAT_DIFF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02003401 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3402 return;
3403
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003404 rettv->vval.v_number = diff_check_fill(curwin, tv_get_lnum(argvars));
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003405# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003406}
3407
3408/*
3409 * "diff_hlID()" function
3410 */
3411 void
3412f_diff_hlID(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3413{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003414# ifdef FEAT_DIFF
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003415 linenr_T lnum;
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003416 static linenr_T prev_lnum = 0;
3417 static varnumber_T changedtick = 0;
3418 static int fnum = 0;
3419 static int change_start = 0;
3420 static int change_end = 0;
3421 static hlf_T hlID = (hlf_T)0;
3422 int filler_lines;
3423 int col;
3424
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003425 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +02003426 && (check_for_lnum_arg(argvars,0) == FAIL
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003427 || check_for_number_arg(argvars, 1) == FAIL))
3428 return;
3429
3430 lnum = tv_get_lnum(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003431 if (lnum < 0) // ignore type error in {lnum} arg
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003432 lnum = 0;
3433 if (lnum != prev_lnum
3434 || changedtick != CHANGEDTICK(curbuf)
3435 || fnum != curbuf->b_fnum)
3436 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003437 // New line, buffer, change: need to get the values.
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003438 filler_lines = diff_check(curwin, lnum);
3439 if (filler_lines < 0)
3440 {
3441 if (filler_lines == -1)
3442 {
3443 change_start = MAXCOL;
3444 change_end = -1;
3445 if (diff_find_change(curwin, lnum, &change_start, &change_end))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003446 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003447 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003448 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003449 }
3450 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003451 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003452 }
3453 else
3454 hlID = (hlf_T)0;
3455 prev_lnum = lnum;
3456 changedtick = CHANGEDTICK(curbuf);
3457 fnum = curbuf->b_fnum;
3458 }
3459
3460 if (hlID == HLF_CHD || hlID == HLF_TXD)
3461 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003462 col = tv_get_number(&argvars[1]) - 1; // ignore type error in {col}
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003463 if (col >= change_start && col <= change_end)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003464 hlID = HLF_TXD; // changed text
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003465 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003466 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003467 }
3468 rettv->vval.v_number = hlID == (hlf_T)0 ? 0 : (int)hlID;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003469# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003470}
3471
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003472# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003473/*
3474 * Parse the diff options passed in "optarg" to the diff() function and return
3475 * the options in "diffopts" and the diff algorithm in "diffalgo".
3476 */
3477 static int
3478parse_diff_optarg(
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003479 typval_T *opts,
3480 int *diffopts,
3481 long *diffalgo,
3482 dio_outfmt_T *diff_output_fmt,
3483 int *diff_ctxlen)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003484{
3485 dict_T *d = opts->vval.v_dict;
3486
3487 char_u *algo = dict_get_string(d, "algorithm", FALSE);
3488 if (algo != NULL)
3489 {
3490 if (STRNCMP(algo, "myers", 5) == 0)
3491 *diffalgo = 0;
3492 else if (STRNCMP(algo, "minimal", 7) == 0)
3493 *diffalgo = XDF_NEED_MINIMAL;
3494 else if (STRNCMP(algo, "patience", 8) == 0)
3495 *diffalgo = XDF_PATIENCE_DIFF;
3496 else if (STRNCMP(algo, "histogram", 9) == 0)
3497 *diffalgo = XDF_HISTOGRAM_DIFF;
3498 }
3499
3500 char_u *output_fmt = dict_get_string(d, "output", FALSE);
3501 if (output_fmt != NULL)
3502 {
3503 if (STRNCMP(output_fmt, "unified", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003504 *diff_output_fmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003505 else if (STRNCMP(output_fmt, "indices", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003506 *diff_output_fmt = DIO_OUTPUT_INDICES;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003507 else
3508 {
3509 semsg(_(e_unsupported_diff_output_format_str), output_fmt);
3510 return FAIL;
3511 }
3512 }
3513
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003514 *diff_ctxlen = dict_get_number_def(d, "context", 1);
3515 if (*diff_ctxlen < 0)
3516 *diff_ctxlen = 1;
3517
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003518 if (dict_get_bool(d, "iblank", FALSE))
3519 *diffopts |= DIFF_IBLANK;
3520 if (dict_get_bool(d, "icase", FALSE))
3521 *diffopts |= DIFF_ICASE;
3522 if (dict_get_bool(d, "iwhite", FALSE))
3523 *diffopts |= DIFF_IWHITE;
3524 if (dict_get_bool(d, "iwhiteall", FALSE))
3525 *diffopts |= DIFF_IWHITEALL;
3526 if (dict_get_bool(d, "iwhiteeol", FALSE))
3527 *diffopts |= DIFF_IWHITEEOL;
3528 if (dict_get_bool(d, "indent-heuristic", FALSE))
3529 *diffalgo |= XDF_INDENT_HEURISTIC;
3530
3531 return OK;
3532}
3533
3534/*
3535 * Concatenate the List of strings in "l" and store the result in
3536 * "din->din_mmfile.ptr" and the length in "din->din_mmfile.size".
3537 */
3538 static void
3539list_to_diffin(list_T *l, diffin_T *din, int icase)
3540{
3541 garray_T ga;
3542 listitem_T *li;
3543 char_u *str;
3544
3545 ga_init2(&ga, 512, 4);
3546
3547 FOR_ALL_LIST_ITEMS(l, li)
3548 {
3549 str = tv_get_string(&li->li_tv);
3550 if (icase)
3551 {
3552 str = strlow_save(str);
3553 if (str == NULL)
3554 continue;
3555 }
3556 ga_concat(&ga, str);
3557 ga_concat(&ga, (char_u *)NL_STR);
3558 if (icase)
3559 vim_free(str);
3560 }
3561 if (ga.ga_len > 0)
3562 ((char *)ga.ga_data)[ga.ga_len] = NUL;
3563
3564 din->din_mmfile.ptr = (char *)ga.ga_data;
3565 din->din_mmfile.size = ga.ga_len;
3566}
3567
3568/*
3569 * Get the start and end indices from the diff "hunk".
3570 */
3571 static dict_T *
3572get_diff_hunk_indices(diffhunk_T *hunk)
3573{
3574 dict_T *hunk_dict;
3575
3576 hunk_dict = dict_alloc();
3577 if (hunk_dict == NULL)
3578 return NULL;
3579
3580 dict_add_number(hunk_dict, "from_idx", hunk->lnum_orig - 1);
3581 dict_add_number(hunk_dict, "from_count", hunk->count_orig);
3582 dict_add_number(hunk_dict, "to_idx", hunk->lnum_new - 1);
3583 dict_add_number(hunk_dict, "to_count", hunk->count_new);
3584
3585 return hunk_dict;
3586}
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003587# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003588
3589/*
3590 * "diff()" function
3591 */
3592 void
3593f_diff(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3594{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003595# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003596 diffio_T dio;
3597
3598 if (check_for_nonnull_list_arg(argvars, 0) == FAIL
3599 || check_for_nonnull_list_arg(argvars, 1) == FAIL
3600 || check_for_opt_nonnull_dict_arg(argvars, 2) == FAIL)
3601 return;
3602
3603 CLEAR_FIELD(dio);
3604 dio.dio_internal = TRUE;
3605 ga_init2(&dio.dio_diff.dout_ga, sizeof(char *), 1000);
3606
3607 list_T *orig_list = argvars[0].vval.v_list;
3608 list_T *new_list = argvars[1].vval.v_list;
3609
3610 // Save the 'diffopt' option value and restore it after getting the diff.
3611 int save_diff_flags = diff_flags;
3612 long save_diff_algorithm = diff_algorithm;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003613 diff_flags = DIFF_INTERNAL;
3614 diff_algorithm = 0;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003615 dio.dio_outfmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003616 if (argvars[2].v_type != VAR_UNKNOWN)
3617 if (parse_diff_optarg(&argvars[2], &diff_flags, &diff_algorithm,
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003618 &dio.dio_outfmt, &dio.dio_ctxlen) == FAIL)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003619 return;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003620
3621 // Concatenate the List of strings into a single string using newline
3622 // separator. Internal diff library expects a single string.
3623 list_to_diffin(orig_list, &dio.dio_orig, diff_flags & DIFF_ICASE);
3624 list_to_diffin(new_list, &dio.dio_new, diff_flags & DIFF_ICASE);
3625
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003626 // If 'diffexpr' is set, then the internal diff is not used. Set
3627 // 'diffexpr' to an empty string temporarily.
3628 int restore_diffexpr = FALSE;
3629 char_u cc = *p_dex;
3630 if (*p_dex != NUL)
3631 {
3632 restore_diffexpr = TRUE;
3633 *p_dex = NUL;
3634 }
3635
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003636 // Compute the diff
3637 int diff_status = diff_file(&dio);
3638
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003639 // restore 'diffexpr'
3640 if (restore_diffexpr)
3641 *p_dex = cc;
3642
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003643 if (diff_status == FAIL)
3644 goto done;
3645
3646 int hunk_idx = 0;
3647 dict_T *hunk_dict;
3648
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003649 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003650 {
3651 if (rettv_list_alloc(rettv) != OK)
3652 goto done;
3653 list_T *l = rettv->vval.v_list;
3654
3655 // Process each diff hunk
3656 diffhunk_T *hunk = NULL;
3657 while (hunk_idx < dio.dio_diff.dout_ga.ga_len)
3658 {
3659 hunk = ((diffhunk_T **)dio.dio_diff.dout_ga.ga_data)[hunk_idx++];
3660
3661 hunk_dict = get_diff_hunk_indices(hunk);
3662 if (hunk_dict == NULL)
3663 goto done;
3664
3665 list_append_dict(l, hunk_dict);
3666 }
3667 }
3668 else
3669 {
3670 ga_append(&dio.dio_diff.dout_ga, NUL);
3671 rettv->v_type = VAR_STRING;
3672 rettv->vval.v_string =
3673 vim_strsave((char_u *)dio.dio_diff.dout_ga.ga_data);
3674 }
3675
3676done:
3677 clear_diffin(&dio.dio_new);
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003678 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003679 clear_diffout(&dio.dio_diff);
3680 else
3681 ga_clear(&dio.dio_diff.dout_ga);
3682 clear_diffin(&dio.dio_orig);
3683 // Restore the 'diffopt' option value.
3684 diff_flags = save_diff_flags;
3685 diff_algorithm = save_diff_algorithm;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003686# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003687}
3688
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003689#endif