4 #include "levenshtein.h"
7 /* most GUI terminals set COLUMNS (although some don't export it) */
8 static int term_columns(void)
10 char *col_string
= getenv("COLUMNS");
13 if (col_string
&& (n_cols
= atoi(col_string
)) > 0)
19 if (!ioctl(1, TIOCGWINSZ
, &ws
)) {
29 void add_cmdname(struct cmdnames
*cmds
, const char *name
, int len
)
31 struct cmdname
*ent
= xmalloc(sizeof(*ent
) + len
+ 1);
34 memcpy(ent
->name
, name
, len
);
37 ALLOC_GROW(cmds
->names
, cmds
->cnt
+ 1, cmds
->alloc
);
38 cmds
->names
[cmds
->cnt
++] = ent
;
41 static void clean_cmdnames(struct cmdnames
*cmds
)
44 for (i
= 0; i
< cmds
->cnt
; ++i
)
51 static int cmdname_compare(const void *a_
, const void *b_
)
53 struct cmdname
*a
= *(struct cmdname
**)a_
;
54 struct cmdname
*b
= *(struct cmdname
**)b_
;
55 return strcmp(a
->name
, b
->name
);
58 static void uniq(struct cmdnames
*cmds
)
65 for (i
= j
= 1; i
< cmds
->cnt
; i
++)
66 if (strcmp(cmds
->names
[i
]->name
, cmds
->names
[i
-1]->name
))
67 cmds
->names
[j
++] = cmds
->names
[i
];
72 void exclude_cmds(struct cmdnames
*cmds
, struct cmdnames
*excludes
)
78 while (ci
< cmds
->cnt
&& ei
< excludes
->cnt
) {
79 cmp
= strcmp(cmds
->names
[ci
]->name
, excludes
->names
[ei
]->name
);
81 cmds
->names
[cj
++] = cmds
->names
[ci
++];
88 while (ci
< cmds
->cnt
)
89 cmds
->names
[cj
++] = cmds
->names
[ci
++];
94 static void pretty_print_string_list(struct cmdnames
*cmds
, int longest
)
97 int space
= longest
+ 1; /* min 1 SP between words */
98 int max_cols
= term_columns() - 1; /* don't print *on* the edge */
101 if (space
< max_cols
)
102 cols
= max_cols
/ space
;
103 rows
= DIV_ROUND_UP(cmds
->cnt
, cols
);
105 for (i
= 0; i
< rows
; i
++) {
108 for (j
= 0; j
< cols
; j
++) {
109 int n
= j
* rows
+ i
;
113 if (j
== cols
-1 || n
+ rows
>= cmds
->cnt
)
115 printf("%-*s", size
, cmds
->names
[n
]->name
);
121 static int is_executable(const char *name
)
125 if (stat(name
, &st
) || /* stat, not lstat */
126 !S_ISREG(st
.st_mode
))
130 /* cannot trust the executable bit, peek into the file instead */
133 int fd
= open(name
, O_RDONLY
);
134 st
.st_mode
&= ~S_IXUSR
;
136 n
= read(fd
, buf
, 2);
138 /* DOS executables start with "MZ" */
139 if (!strcmp(buf
, "#!") || !strcmp(buf
, "MZ"))
140 st
.st_mode
|= S_IXUSR
;
144 return st
.st_mode
& S_IXUSR
;
147 static void list_commands_in_dir(struct cmdnames
*cmds
,
152 DIR *dir
= opendir(path
);
154 struct strbuf buf
= STRBUF_INIT
;
161 prefix_len
= strlen(prefix
);
163 strbuf_addf(&buf
, "%s/", path
);
166 while ((de
= readdir(dir
)) != NULL
) {
169 if (prefixcmp(de
->d_name
, prefix
))
172 strbuf_setlen(&buf
, len
);
173 strbuf_addstr(&buf
, de
->d_name
);
174 if (!is_executable(buf
.buf
))
177 entlen
= strlen(de
->d_name
) - prefix_len
;
178 if (has_extension(de
->d_name
, ".exe"))
181 add_cmdname(cmds
, de
->d_name
+ prefix_len
, entlen
);
184 strbuf_release(&buf
);
187 void load_command_list(const char *prefix
,
188 struct cmdnames
*main_cmds
,
189 struct cmdnames
*other_cmds
)
191 const char *env_path
= getenv("PATH");
192 const char *exec_path
= git_exec_path();
195 list_commands_in_dir(main_cmds
, exec_path
, prefix
);
196 qsort(main_cmds
->names
, main_cmds
->cnt
,
197 sizeof(*main_cmds
->names
), cmdname_compare
);
202 char *paths
, *path
, *colon
;
203 path
= paths
= xstrdup(env_path
);
205 if ((colon
= strchr(path
, PATH_SEP
)))
207 if (!exec_path
|| strcmp(path
, exec_path
))
208 list_commands_in_dir(other_cmds
, path
, prefix
);
216 qsort(other_cmds
->names
, other_cmds
->cnt
,
217 sizeof(*other_cmds
->names
), cmdname_compare
);
220 exclude_cmds(other_cmds
, main_cmds
);
223 void list_commands(const char *title
, struct cmdnames
*main_cmds
,
224 struct cmdnames
*other_cmds
)
228 for (i
= 0; i
< main_cmds
->cnt
; i
++)
229 if (longest
< main_cmds
->names
[i
]->len
)
230 longest
= main_cmds
->names
[i
]->len
;
231 for (i
= 0; i
< other_cmds
->cnt
; i
++)
232 if (longest
< other_cmds
->names
[i
]->len
)
233 longest
= other_cmds
->names
[i
]->len
;
235 if (main_cmds
->cnt
) {
236 const char *exec_path
= git_exec_path();
237 printf("available %s in '%s'\n", title
, exec_path
);
238 printf("----------------");
239 mput_char('-', strlen(title
) + strlen(exec_path
));
241 pretty_print_string_list(main_cmds
, longest
);
245 if (other_cmds
->cnt
) {
246 printf("%s available from elsewhere on your $PATH\n", title
);
247 printf("---------------------------------------");
248 mput_char('-', strlen(title
));
250 pretty_print_string_list(other_cmds
, longest
);
255 int is_in_cmdlist(struct cmdnames
*c
, const char *s
)
258 for (i
= 0; i
< c
->cnt
; i
++)
259 if (!strcmp(s
, c
->names
[i
]->name
))
264 static int autocorrect
;
265 static struct cmdnames aliases
;
267 static int git_unknown_cmd_config(const char *var
, const char *value
, void *cb
)
269 if (!strcmp(var
, "help.autocorrect"))
270 autocorrect
= git_config_int(var
,value
);
271 /* Also use aliases for command lookup */
272 if (!prefixcmp(var
, "alias."))
273 add_cmdname(&aliases
, var
+ 6, strlen(var
+ 6));
275 return git_default_config(var
, value
, cb
);
278 static int levenshtein_compare(const void *p1
, const void *p2
)
280 const struct cmdname
*const *c1
= p1
, *const *c2
= p2
;
281 const char *s1
= (*c1
)->name
, *s2
= (*c2
)->name
;
284 return l1
!= l2
? l1
- l2
: strcmp(s1
, s2
);
287 static void add_cmd_list(struct cmdnames
*cmds
, struct cmdnames
*old
)
290 ALLOC_GROW(cmds
->names
, cmds
->cnt
+ old
->cnt
, cmds
->alloc
);
292 for (i
= 0; i
< old
->cnt
; i
++)
293 cmds
->names
[cmds
->cnt
++] = old
->names
[i
];
299 const char *help_unknown_cmd(const char *cmd
)
301 int i
, n
, best_similarity
= 0;
302 struct cmdnames main_cmds
, other_cmds
;
304 memset(&main_cmds
, 0, sizeof(main_cmds
));
305 memset(&other_cmds
, 0, sizeof(main_cmds
));
306 memset(&aliases
, 0, sizeof(aliases
));
308 git_config(git_unknown_cmd_config
, NULL
);
310 load_command_list("git-", &main_cmds
, &other_cmds
);
312 add_cmd_list(&main_cmds
, &aliases
);
313 add_cmd_list(&main_cmds
, &other_cmds
);
314 qsort(main_cmds
.names
, main_cmds
.cnt
,
315 sizeof(main_cmds
.names
), cmdname_compare
);
318 /* This reuses cmdname->len for similarity index */
319 for (i
= 0; i
< main_cmds
.cnt
; ++i
)
320 main_cmds
.names
[i
]->len
=
321 levenshtein(cmd
, main_cmds
.names
[i
]->name
, 0, 2, 1, 4);
323 qsort(main_cmds
.names
, main_cmds
.cnt
,
324 sizeof(*main_cmds
.names
), levenshtein_compare
);
327 die ("Uh oh. Your system reports no Git commands at all.");
329 best_similarity
= main_cmds
.names
[0]->len
;
331 while (n
< main_cmds
.cnt
&& best_similarity
== main_cmds
.names
[n
]->len
)
333 if (autocorrect
&& n
== 1) {
334 const char *assumed
= main_cmds
.names
[0]->name
;
335 main_cmds
.names
[0] = NULL
;
336 clean_cmdnames(&main_cmds
);
337 fprintf(stderr
, "WARNING: You called a Git program named '%s', "
338 "which does not exist.\n"
339 "Continuing under the assumption that you meant '%s'\n",
341 if (autocorrect
> 0) {
342 fprintf(stderr
, "in %0.1f seconds automatically...\n",
343 (float)autocorrect
/10.0);
344 poll(NULL
, 0, autocorrect
* 100);
349 fprintf(stderr
, "git: '%s' is not a git-command. See 'git --help'.\n", cmd
);
351 if (best_similarity
< 6) {
352 fprintf(stderr
, "\nDid you mean %s?\n",
353 n
< 2 ? "this": "one of these");
355 for (i
= 0; i
< n
; i
++)
356 fprintf(stderr
, "\t%s\n", main_cmds
.names
[i
]->name
);
362 int cmd_version(int argc
, const char **argv
, const char *prefix
)
364 printf("git version %s\n", git_version_string
);