patch 8.1.0353: an "after" directory of a package is appended to 'rtp'

Problem:    An "after" directory of a package is appended to 'rtp', which
            will be after the user's "after" directory. ()
Solution:   Insert the package "after" directory before any other "after"
            directory in 'rtp'. (closes #3409)

diff --git a/src/ex_cmds2.c b/src/ex_cmds2.c
index a58b955f3c7b2eb594ebf5e22d7f3a9e88fdbe5d..08eb181787ac329a8e32a9af960ffc90a87c4676 100644 (file)
--- a/src/ex_cmds2.c
+++ b/src/ex_cmds2.c
@@ -3691,14 +3691,17 @@ source_all_matches(char_u *pat)
 add_pack_dir_to_rtp(char_u *fname)
 {
     char_u  *p4, *p3, *p2, *p1, *p;
-    char_u  *insp;
+    char_u  *entry;
+    char_u  *insp = NULL;
     int            c;
     char_u  *new_rtp;
     int            keep;
     size_t  oldlen;
     size_t  addlen;
+    size_t  new_rtp_len;
     char_u  *afterdir = NULL;
     size_t  afterlen = 0;
+    char_u  *after_insp = NULL;
     char_u  *ffname = NULL;
     size_t  fname_len;
     char_u  *buf = NULL;
@@ -3725,54 +3728,99 @@ add_pack_dir_to_rtp(char_u *fname)
     if (ffname == NULL)
        return FAIL;
 
-    /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
+    // Find "ffname" in "p_rtp", ignoring '/' vs '\' differences.
+    // Also stop at the first "after" directory.
     fname_len = STRLEN(ffname);
-    insp = p_rtp;
     buf = alloc(MAXPATHL);
     if (buf == NULL)
        goto theend;
-    while (*insp != NUL)
-    {
-       copy_option_part(&insp, buf, MAXPATHL, ",");
-       add_pathsep(buf);
-       rtp_ffname = fix_fname(buf);
-       if (rtp_ffname == NULL)
-           goto theend;
-       match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
-       vim_free(rtp_ffname);
-       if (match)
+    for (entry = p_rtp; *entry != NUL; )
+    {
+       char_u *cur_entry = entry;
+
+       copy_option_part(&entry, buf, MAXPATHL, ",");
+       if (insp == NULL)
+       {
+           add_pathsep(buf);
+           rtp_ffname = fix_fname(buf);
+           if (rtp_ffname == NULL)
+               goto theend;
+           match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
+           vim_free(rtp_ffname);
+           if (match)
+               // Insert "ffname" after this entry (and comma).
+               insp = entry;
+       }
+
+       if ((p = (char_u *)strstr((char *)buf, "after")) != NULL
+               && p > buf
+               && vim_ispathsep(p[-1])
+               && (vim_ispathsep(p[5]) || p[5] == NUL || p[5] == ','))
+       {
+           if (insp == NULL)
+               // Did not find "ffname" before the first "after" directory,
+               // insert it before this entry.
+               insp = cur_entry;
+           after_insp = cur_entry;
            break;
+       }
     }
 
-    if (*insp == NUL)
-       /* not found, append at the end */
+    if (insp == NULL)
+       // Both "fname" and "after" not found, append at the end.
        insp = p_rtp + STRLEN(p_rtp);
-    else
-       /* append after the matching directory. */
-       --insp;
 
-    /* check if rtp/pack/name/start/name/after exists */
+    // check if rtp/pack/name/start/name/after exists
     afterdir = concat_fnames(fname, (char_u *)"after", TRUE);
     if (afterdir != NULL && mch_isdir(afterdir))
-       afterlen = STRLEN(afterdir) + 1; /* add one for comma */
+       afterlen = STRLEN(afterdir) + 1; // add one for comma
 
     oldlen = STRLEN(p_rtp);
-    addlen = STRLEN(fname) + 1; /* add one for comma */
-    new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
-    /* add one for NUL */
+    addlen = STRLEN(fname) + 1; // add one for comma
+    new_rtp = alloc((int)(oldlen + addlen + afterlen + 1)); // add one for NUL
     if (new_rtp == NULL)
        goto theend;
+
+    // We now have 'rtp' parts: {keep}{keep_after}{rest}.
+    // Create new_rtp, first: {keep},{fname}
     keep = (int)(insp - p_rtp);
     mch_memmove(new_rtp, p_rtp, keep);
-    new_rtp[keep] = ',';
-    mch_memmove(new_rtp + keep + 1, fname, addlen);
+    new_rtp_len = keep;
+    if (*insp == NUL)
+       new_rtp[new_rtp_len++] = ',';  // add comma before
+    mch_memmove(new_rtp + new_rtp_len, fname, addlen - 1);
+    new_rtp_len += addlen - 1;
+    if (*insp != NUL)
+       new_rtp[new_rtp_len++] = ',';  // add comma after
+
+    if (afterlen > 0 && after_insp != NULL)
+    {
+       int keep_after = (int)(after_insp - p_rtp);
+
+       // Add to new_rtp: {keep},{fname}{keep_after},{afterdir}
+       mch_memmove(new_rtp + new_rtp_len, p_rtp + keep,
+                                                       keep_after - keep);
+       new_rtp_len += keep_after - keep;
+       mch_memmove(new_rtp + new_rtp_len, afterdir, afterlen - 1);
+       new_rtp_len += afterlen - 1;
+       new_rtp[new_rtp_len++] = ',';
+       keep = keep_after;
+    }
+
     if (p_rtp[keep] != NUL)
-       mch_memmove(new_rtp + keep + addlen, p_rtp + keep, oldlen - keep + 1);
-    if (afterlen > 0)
+       // Append rest: {keep},{fname}{keep_after},{afterdir}{rest}
+       mch_memmove(new_rtp + new_rtp_len, p_rtp + keep, oldlen - keep + 1);
+    else
+       new_rtp[new_rtp_len] = NUL;
+
+    if (afterlen > 0 && after_insp == NULL)
     {
+       // Append afterdir when "after" was not found:
+       // {keep},{fname}{rest},{afterdir}
        STRCAT(new_rtp, ",");
        STRCAT(new_rtp, afterdir);
     }
+
     set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
     vim_free(new_rtp);
     retval = OK;

diff --git a/src/testdir/test_packadd.vim b/src/testdir/test_packadd.vim
index 64bd8d2b23e608898dd64aaee4c69e96dda4b481..f511a4d5872c4abaffc30e307f8610a729c65f4e 100644 (file)
--- a/src/testdir/test_packadd.vim
+++ b/src/testdir/test_packadd.vim
@@ -12,6 +12,11 @@ func TearDown()
 endfunc
 
 func Test_packadd()
+  if !exists('s:plugdir')
+    echomsg 'when running this test manually, call SetUp() first'
+    return
+  endif
+
   call mkdir(s:plugdir . '/plugin/also', 'p')
   call mkdir(s:plugdir . '/ftdetect', 'p')
   call mkdir(s:plugdir . '/after', 'p')
@@ -19,6 +24,14 @@ func Test_packadd()
   let rtp = &rtp
   filetype on
 
+  let rtp_entries = split(rtp, ',')
+  for entry in rtp_entries
+    if entry =~? '\<after\>' 
+      let first_after_entry = entry
+      break
+    endif
+  endfor
+
   exe 'split ' . s:plugdir . '/plugin/test.vim'
   call setline(1, 'let g:plugin_works = 42')
   wq
@@ -38,7 +51,12 @@ func Test_packadd()
   call assert_equal(17, g:ftdetect_works)
   call assert_true(len(&rtp) > len(rtp))
   call assert_match('/testdir/Xdir/pack/mine/opt/mytest\($\|,\)', &rtp)
-  call assert_match('/testdir/Xdir/pack/mine/opt/mytest/after$', &rtp)
+
+  let new_after = match(&rtp, '/testdir/Xdir/pack/mine/opt/mytest/after,')
+  let old_after = match(&rtp, ',' . first_after_entry . '\>')
+  call assert_true(new_after > 0, 'rtp is ' . &rtp)
+  call assert_true(old_after > 0, 'rtp is ' . &rtp)
+  call assert_true(new_after < old_after, 'rtp is ' . &rtp)
 
   " NOTE: '/.../opt/myte' forwardly matches with '/.../opt/mytest'
   call mkdir(fnamemodify(s:plugdir, ':h') . '/myte', 'p')

diff --git a/src/version.c b/src/version.c
index 772acc925dd55e1746b0871ffd11060a8a96cc59..840f288773e724b00dbeece0ba0466bb4d8dd8bf 100644 (file)
--- a/src/version.c
+++ b/src/version.c
@@ -794,6 +794,8 @@ static char *(features[]) =
 
 static int included_patches[] =
 {   /* Add new patch number below this line */
+/**/
+    353,
 /**/
     352,
 /**/