]>
Commit | Line | Data |
---|---|---|
c932a5ff | 1 | #include "test-tool.h" |
69a63fe6 | 2 | #include "strbuf.h" |
ff919f96 MH |
3 | #include "string-list.h" |
4 | ||
eb5f0c7a MH |
5 | /* |
6 | * Parse an argument into a string list. arg should either be a | |
7 | * ':'-separated list of strings, or "-" to indicate an empty string | |
8 | * list (as opposed to "", which indicates a string list containing a | |
9 | * single empty string). list->strdup_strings must be set. | |
10 | */ | |
65119878 | 11 | static void parse_string_list(struct string_list *list, const char *arg) |
eb5f0c7a MH |
12 | { |
13 | if (!strcmp(arg, "-")) | |
14 | return; | |
15 | ||
16 | (void)string_list_split(list, arg, ':', -1); | |
17 | } | |
18 | ||
65119878 | 19 | static void write_list(const struct string_list *list) |
ff919f96 MH |
20 | { |
21 | int i; | |
22 | for (i = 0; i < list->nr; i++) | |
23 | printf("[%d]: \"%s\"\n", i, list->items[i].string); | |
24 | } | |
25 | ||
65119878 | 26 | static void write_list_compact(const struct string_list *list) |
eb5f0c7a MH |
27 | { |
28 | int i; | |
29 | if (!list->nr) | |
30 | printf("-\n"); | |
31 | else { | |
32 | printf("%s", list->items[0].string); | |
33 | for (i = 1; i < list->nr; i++) | |
34 | printf(":%s", list->items[i].string); | |
35 | printf("\n"); | |
36 | } | |
37 | } | |
38 | ||
65119878 | 39 | static int prefix_cb(struct string_list_item *item, void *cb_data) |
eb5f0c7a MH |
40 | { |
41 | const char *prefix = (const char *)cb_data; | |
59556548 | 42 | return starts_with(item->string, prefix); |
eb5f0c7a MH |
43 | } |
44 | ||
c932a5ff | 45 | int cmd__string_list(int argc, const char **argv) |
ff919f96 MH |
46 | { |
47 | if (argc == 5 && !strcmp(argv[1], "split")) { | |
48 | struct string_list list = STRING_LIST_INIT_DUP; | |
49 | int i; | |
50 | const char *s = argv[2]; | |
51 | int delim = *argv[3]; | |
52 | int maxsplit = atoi(argv[4]); | |
53 | ||
54 | i = string_list_split(&list, s, delim, maxsplit); | |
55 | printf("%d\n", i); | |
56 | write_list(&list); | |
57 | string_list_clear(&list, 0); | |
58 | return 0; | |
59 | } | |
60 | ||
61 | if (argc == 5 && !strcmp(argv[1], "split_in_place")) { | |
62 | struct string_list list = STRING_LIST_INIT_NODUP; | |
63 | int i; | |
64 | char *s = xstrdup(argv[2]); | |
52acddf3 | 65 | const char *delim = argv[3]; |
ff919f96 MH |
66 | int maxsplit = atoi(argv[4]); |
67 | ||
68 | i = string_list_split_in_place(&list, s, delim, maxsplit); | |
69 | printf("%d\n", i); | |
70 | write_list(&list); | |
71 | string_list_clear(&list, 0); | |
72 | free(s); | |
73 | return 0; | |
74 | } | |
75 | ||
eb5f0c7a MH |
76 | if (argc == 4 && !strcmp(argv[1], "filter")) { |
77 | /* | |
78 | * Retain only the items that have the specified prefix. | |
79 | * Arguments: list|- prefix | |
80 | */ | |
81 | struct string_list list = STRING_LIST_INIT_DUP; | |
82 | const char *prefix = argv[3]; | |
83 | ||
84 | parse_string_list(&list, argv[2]); | |
85 | filter_string_list(&list, 0, prefix_cb, (void *)prefix); | |
86 | write_list_compact(&list); | |
87 | string_list_clear(&list, 0); | |
88 | return 0; | |
89 | } | |
90 | ||
31d5451e MH |
91 | if (argc == 3 && !strcmp(argv[1], "remove_duplicates")) { |
92 | struct string_list list = STRING_LIST_INIT_DUP; | |
93 | ||
94 | parse_string_list(&list, argv[2]); | |
95 | string_list_remove_duplicates(&list, 0); | |
96 | write_list_compact(&list); | |
97 | string_list_clear(&list, 0); | |
98 | return 0; | |
99 | } | |
100 | ||
564e94e6 RS |
101 | if (argc == 2 && !strcmp(argv[1], "sort")) { |
102 | struct string_list list = STRING_LIST_INIT_NODUP; | |
103 | struct strbuf sb = STRBUF_INIT; | |
104 | struct string_list_item *item; | |
105 | ||
106 | strbuf_read(&sb, 0, 0); | |
107 | ||
108 | /* | |
109 | * Split by newline, but don't create a string_list item | |
110 | * for the empty string after the last separator. | |
111 | */ | |
97487ea1 | 112 | if (sb.len && sb.buf[sb.len - 1] == '\n') |
564e94e6 | 113 | strbuf_setlen(&sb, sb.len - 1); |
52acddf3 | 114 | string_list_split_in_place(&list, sb.buf, "\n", -1); |
564e94e6 RS |
115 | |
116 | string_list_sort(&list); | |
117 | ||
118 | for_each_string_list_item(item, &list) | |
119 | puts(item->string); | |
120 | ||
121 | string_list_clear(&list, 0); | |
122 | strbuf_release(&sb); | |
123 | return 0; | |
124 | } | |
125 | ||
ff919f96 MH |
126 | fprintf(stderr, "%s: unknown function name: %s\n", argv[0], |
127 | argv[1] ? argv[1] : "(there was none)"); | |
128 | return 1; | |
129 | } |