Change check_ref_format() to take a flags argument
[git/git.git] / builtin / check-ref-format.c
1 /*
2 * GIT - The information manager from hell
3 */
4
5 #include "cache.h"
6 #include "refs.h"
7 #include "builtin.h"
8 #include "strbuf.h"
9
10 static const char builtin_check_ref_format_usage[] =
11 "git check-ref-format [--print] [options] <refname>\n"
12 " or: git check-ref-format --branch <branchname-shorthand>";
13
14 /*
15 * Remove leading slashes and replace each run of adjacent slashes in
16 * src with a single slash, and write the result to dst.
17 *
18 * This function is similar to normalize_path_copy(), but stripped down
19 * to meet check_ref_format's simpler needs.
20 */
21 static void collapse_slashes(char *dst, const char *src)
22 {
23 char ch;
24 char prev = '/';
25
26 while ((ch = *src++) != '\0') {
27 if (prev == '/' && ch == prev)
28 continue;
29
30 *dst++ = ch;
31 prev = ch;
32 }
33 *dst = '\0';
34 }
35
36 static int check_ref_format_branch(const char *arg)
37 {
38 struct strbuf sb = STRBUF_INIT;
39 int nongit;
40
41 setup_git_directory_gently(&nongit);
42 if (strbuf_check_branch_ref(&sb, arg))
43 die("'%s' is not a valid branch name", arg);
44 printf("%s\n", sb.buf + 11);
45 return 0;
46 }
47
48 static void refname_format_print(const char *arg)
49 {
50 char *refname = xmalloc(strlen(arg) + 1);
51
52 collapse_slashes(refname, arg);
53 printf("%s\n", refname);
54 }
55
56 int cmd_check_ref_format(int argc, const char **argv, const char *prefix)
57 {
58 int i;
59 int print = 0;
60 int flags = 0;
61
62 if (argc == 2 && !strcmp(argv[1], "-h"))
63 usage(builtin_check_ref_format_usage);
64
65 if (argc == 3 && !strcmp(argv[1], "--branch"))
66 return check_ref_format_branch(argv[2]);
67
68 for (i = 1; i < argc && argv[i][0] == '-'; i++) {
69 if (!strcmp(argv[i], "--print"))
70 print = 1;
71 else if (!strcmp(argv[i], "--allow-onelevel"))
72 flags |= REFNAME_ALLOW_ONELEVEL;
73 else if (!strcmp(argv[i], "--no-allow-onelevel"))
74 flags &= ~REFNAME_ALLOW_ONELEVEL;
75 else if (!strcmp(argv[i], "--refspec-pattern"))
76 flags |= REFNAME_REFSPEC_PATTERN;
77 else
78 usage(builtin_check_ref_format_usage);
79 }
80 if (! (i == argc - 1))
81 usage(builtin_check_ref_format_usage);
82
83 if (check_refname_format(argv[i], flags))
84 return 1;
85
86 if (print)
87 refname_format_print(argv[i]);
88
89 return 0;
90 }