Merge branch 'wp/add-patch-find'
[gitweb.git] / builtin-cat-file.c
index b4d0c2545f2db6dbbb360b7ce99cb771490ae258..8fad19daedef8a38674ee35cd543983bad610857 100644 (file)
@@ -8,6 +8,10 @@
 #include "tag.h"
 #include "tree.h"
 #include "builtin.h"
+#include "parse-options.h"
+
+#define BATCH 1
+#define BATCH_CHECK 2
 
 static void pprint_tag(const unsigned char *sha1, const char *buf, unsigned long size)
 {
@@ -133,11 +137,11 @@ static int cat_one_file(int opt, const char *exp_type, const char *obj_name)
                break;
 
        default:
-               die("git-cat-file: unknown option: %s\n", exp_type);
+               die("git cat-file: unknown option: %s", exp_type);
        }
 
        if (!buf)
-               die("git-cat-file %s: bad file", obj_name);
+               die("git cat-file %s: bad file", obj_name);
 
        write_or_die(1, buf, size);
        return 0;
@@ -146,7 +150,7 @@ static int cat_one_file(int opt, const char *exp_type, const char *obj_name)
 static int batch_one_object(const char *obj_name, int print_contents)
 {
        unsigned char sha1[20];
-       enum object_type type;
+       enum object_type type = 0;
        unsigned long size;
        void *contents = contents;
 
@@ -155,24 +159,29 @@ static int batch_one_object(const char *obj_name, int print_contents)
 
        if (get_sha1(obj_name, sha1)) {
                printf("%s missing\n", obj_name);
+               fflush(stdout);
                return 0;
        }
 
-       if (print_contents)
+       if (print_contents == BATCH)
                contents = read_sha1_file(sha1, &type, &size);
        else
                type = sha1_object_info(sha1, &size);
 
-       if (type <= 0)
-               return 1;
+       if (type <= 0) {
+               printf("%s missing\n", obj_name);
+               fflush(stdout);
+               return 0;
+       }
 
        printf("%s %s %lu\n", sha1_to_hex(sha1), typename(type), size);
        fflush(stdout);
 
-       if (print_contents) {
+       if (print_contents == BATCH) {
                write_or_die(1, contents, size);
                printf("\n");
                fflush(stdout);
+               free(contents);
        }
 
        return 0;
@@ -180,9 +189,8 @@ static int batch_one_object(const char *obj_name, int print_contents)
 
 static int batch_objects(int print_contents)
 {
-       struct strbuf buf;
+       struct strbuf buf = STRBUF_INIT;
 
-       strbuf_init(&buf, 0);
        while (strbuf_getline(&buf, stdin, '\n') != EOF) {
                int error = batch_one_object(buf.buf, print_contents);
                if (error)
@@ -192,89 +200,58 @@ static int batch_objects(int print_contents)
        return 0;
 }
 
-static const char cat_file_usage[] = "git-cat-file [ [-t|-s|-e|-p|<type>] <sha1> | [--batch|--batch-check] < <list_of_sha1s> ]";
+static const char * const cat_file_usage[] = {
+       "git cat-file [-t|-s|-e|-p|<type>] <sha1>",
+       "git cat-file [--batch|--batch-check] < <list_of_sha1s>",
+       NULL
+};
 
 int cmd_cat_file(int argc, const char **argv, const char *prefix)
 {
-       int i, opt = 0, batch = 0, batch_check = 0;
+       int opt = 0, batch = 0;
        const char *exp_type = NULL, *obj_name = NULL;
 
-       git_config(git_default_config);
-
-       for (i = 1; i < argc; ++i) {
-               const char *arg = argv[i];
-               int is_batch = 0, is_batch_check = 0;
-
-               is_batch = !strcmp(arg, "--batch");
-               if (!is_batch)
-                       is_batch_check = !strcmp(arg, "--batch-check");
-
-               if (is_batch || is_batch_check) {
-                       if (opt) {
-                               error("git-cat-file: Can't use %s with -%c", arg, opt);
-                               usage(cat_file_usage);
-                       } else if (exp_type) {
-                               error("git-cat-file: Can't use %s when a type (\"%s\") is specified", arg, exp_type);
-                               usage(cat_file_usage);
-                       } else if (obj_name) {
-                               error("git-cat-file: Can't use %s when an object (\"%s\") is specified", arg, obj_name);
-                               usage(cat_file_usage);
-                       }
-
-                       if ((is_batch && batch_check) || (is_batch_check && batch)) {
-                               error("git-cat-file: Can't use %s with %s", arg, is_batch ? "--batch-check" : "--batch");
-                               usage(cat_file_usage);
-                       }
-
-                       if (is_batch)
-                               batch = 1;
-                       else
-                               batch_check = 1;
-
-                       continue;
-               }
-
-               if (!strcmp(arg, "-t") || !strcmp(arg, "-s") || !strcmp(arg, "-e") || !strcmp(arg, "-p")) {
-                       if (batch || batch_check) {
-                               error("git-cat-file: Can't use %s with %s", arg, batch ? "--batch" : "--batch-check");
-                               usage(cat_file_usage);
-                       }
-
-                       exp_type = arg;
-                       opt = exp_type[1];
-                       continue;
-               }
-
-               if (arg[0] == '-')
-                       usage(cat_file_usage);
-
-               if (!exp_type) {
-                       if (batch || batch_check) {
-                               error("git-cat-file: Can't specify a type (\"%s\") with %s", arg, batch ? "--batch" : "--batch-check");
-                               usage(cat_file_usage);
-                       }
-
-                       exp_type = arg;
-                       continue;
-               }
-
-               if (obj_name)
-                       usage(cat_file_usage);
-
-               // We should have hit one of the earlier if (batch || batch_check) cases before
-               // getting here.
-               assert(!batch);
-               assert(!batch_check);
-
-               obj_name = arg;
-               break;
+       const struct option options[] = {
+               OPT_GROUP("<type> can be one of: blob, tree, commit, tag"),
+               OPT_SET_INT('t', NULL, &opt, "show object type", 't'),
+               OPT_SET_INT('s', NULL, &opt, "show object size", 's'),
+               OPT_SET_INT('e', NULL, &opt,
+                           "exit with zero when there's no error", 'e'),
+               OPT_SET_INT('p', NULL, &opt, "pretty-print object's content", 'p'),
+               OPT_SET_INT(0, "batch", &batch,
+                           "show info and content of objects feeded on stdin", BATCH),
+               OPT_SET_INT(0, "batch-check", &batch,
+                           "show info about objects feeded on stdin",
+                           BATCH_CHECK),
+               OPT_END()
+       };
+
+       git_config(git_default_config, NULL);
+
+       if (argc != 3 && argc != 2)
+               usage_with_options(cat_file_usage, options);
+
+       argc = parse_options(argc, argv, options, cat_file_usage, 0);
+
+       if (opt) {
+               if (argc == 1)
+                       obj_name = argv[0];
+               else
+                       usage_with_options(cat_file_usage, options);
+       }
+       if (!opt && !batch) {
+               if (argc == 2) {
+                       exp_type = argv[0];
+                       obj_name = argv[1];
+               } else
+                       usage_with_options(cat_file_usage, options);
+       }
+       if (batch && (opt || argc)) {
+               usage_with_options(cat_file_usage, options);
        }
 
-       if (batch || batch_check)
+       if (batch)
                return batch_objects(batch);
 
-       if (!exp_type || !obj_name)
-               usage(cat_file_usage);
-
        return cat_one_file(opt, exp_type, obj_name);
 }