ref-filter: use string_list_split over strbuf_split

diff --git a/ref-filter.c b/ref-filter.c
index 7bef7f8dac644b49a2cb13d8d5154fd563d6e8a0..19367ce705cb0ddd4dc54c82cef0fde4d1ac868b 100644 (file)
--- a/ref-filter.c
+++ b/ref-filter.c
@@ -763,6 +763,29 @@ static inline char *copy_advance(char *dst, const char *src)
        return dst;
 }
 
+static const char *strip_ref_components(const char *refname, const char *nr_arg)
+{
+       char *end;
+       long nr = strtol(nr_arg, &end, 10);
+       long remaining = nr;
+       const char *start = refname;
+
+       if (nr < 1 || *end != '\0')
+               die(":strip= requires a positive integer argument");
+
+       while (remaining) {
+               switch (*start++) {
+               case '\0':
+                       die("ref '%s' does not have %ld components to :strip",
+                           refname, nr);
+               case '/':
+                       remaining--;
+                       break;
+               }
+       }
+       return start;
+}
+
 /*
  * Parse the object referred by ref, and grab needed value.
  */
@@ -863,41 +886,34 @@ static void populate_value(struct ref_array_item *ref)
                        continue;
                } else if (match_atom_name(name, "align", &valp)) {
                        struct align *align = &v->u.align;
-                       struct strbuf **s, **to_free;
+                       struct string_list params = STRING_LIST_INIT_DUP;
+                       int i;
                        int width = -1;
 
                        if (!valp)
                                die(_("expected format: %%(align:<width>,<position>)"));
 
-                       /*
-                        * TODO: Implement a function similar to strbuf_split_str()
-                        * which would omit the separator from the end of each value.
-                        */
-                       s = to_free = strbuf_split_str(valp, ',', 0);
-
                        align->position = ALIGN_LEFT;
 
-                       while (*s) {
-                               /*  Strip trailing comma */
-                               if (s[1])
-                                       strbuf_setlen(s[0], s[0]->len - 1);
-                               if (!strtoul_ui(s[0]->buf, 10, (unsigned int *)&width))
+                       string_list_split(&params, valp, ',', -1);
+                       for (i = 0; i < params.nr; i++) {
+                               const char *s = params.items[i].string;
+                               if (!strtoul_ui(s, 10, (unsigned int *)&width))
                                        ;
-                               else if (!strcmp(s[0]->buf, "left"))
+                               else if (!strcmp(s, "left"))
                                        align->position = ALIGN_LEFT;
-                               else if (!strcmp(s[0]->buf, "right"))
+                               else if (!strcmp(s, "right"))
                                        align->position = ALIGN_RIGHT;
-                               else if (!strcmp(s[0]->buf, "middle"))
+                               else if (!strcmp(s, "middle"))
                                        align->position = ALIGN_MIDDLE;
                                else
-                                       die(_("improper format entered align:%s"), s[0]->buf);
-                               s++;
+                                       die(_("improper format entered align:%s"), s);
                        }
 
                        if (width < 0)
                                die(_("positive width expected with the %%(align) atom"));
                        align->width = width;
-                       strbuf_list_free(to_free);
+                       string_list_clear(&params, 0);
                        v->handler = align_atom_handler;
                        continue;
                } else if (!strcmp(name, "end")) {
@@ -909,11 +925,14 @@ static void populate_value(struct ref_array_item *ref)
                formatp = strchr(name, ':');
                if (formatp) {
                        int num_ours, num_theirs;
+                       const char *arg;
 
                        formatp++;
                        if (!strcmp(formatp, "short"))
                                refname = shorten_unambiguous_ref(refname,
                                                      warn_ambiguous_refs);
+                       else if (skip_prefix(formatp, "strip=", &arg))
+                               refname = strip_ref_components(refname, arg);
                        else if (!strcmp(formatp, "track") &&
                                 (starts_with(name, "upstream") ||
                                  starts_with(name, "push"))) {