[PATCH] Factor out useful test case infrastructure from t/t6001... into t/t6000-lib.sh

diff --git a/cat-file.c b/cat-file.c
index 3071672c6ce2ffb7c1c7d43c1a728f8e020e296b..fa0bb722a2455fd4274f61e800cd9f5a764dda1e 100644 (file)
--- a/cat-file.c
+++ b/cat-file.c
@@ -11,25 +11,31 @@ int main(int argc, char **argv)
        char type[20];
        void *buf;
        unsigned long size;
-       int fd;
 
-       if (argc != 3 || get_sha1_hex(argv[2], sha1))
-               usage("cat-file: cat-file [-t | tagname] <sha1>");
-       buf = read_sha1_file(sha1, type, &size);
-       if (!buf) {
-               fprintf(stderr, "cat-file %s: bad file\n", argv[2]);
-               exit(1);
-       }
-       if (!strcmp("-t", argv[1])) {
-               buf = type;
-               size = strlen(type);
-               type[size] = '\n';
-               size++;
-       } else if (strcmp(type, argv[1])) {
-               fprintf(stderr, "cat-file %s: bad tag\n", argv[2]);
-               exit(1);        /* bad tag */
+       if (argc != 3 || get_sha1(argv[2], sha1))
+               usage("git-cat-file [-t | -s | tagname] <sha1>");
+
+       if (!strcmp("-t", argv[1]) || !strcmp("-s", argv[1])) {
+               if (!sha1_object_info(sha1, type,
+                                     argv[1][1] == 's' ? &size : NULL)) {
+                       switch (argv[1][1]) {
+                       case 't':
+                               printf("%s\n", type);
+                               break;
+                       case 's':
+                               printf("%lu\n", size);
+                               break;
+                       }
+                       return 0;
+               }
+               buf = NULL;
+       } else {
+               buf = read_object_with_reference(sha1, argv[1], &size, NULL);
        }
 
+       if (!buf)
+               die("git-cat-file %s: bad file", argv[2]);
+
        while (size > 0) {
                long ret = write(1, buf, size);
                if (ret < 0) {
@@ -38,12 +44,9 @@ int main(int argc, char **argv)
                        /* Ignore epipe */
                        if (errno == EPIPE)
                                break;
-                       fprintf(stderr, "cat-file: %s\n", strerror(errno));
-                       exit(1);
-               }
-               if (!ret) {
-                       fprintf(stderr, "cat-file: disk full?");
-                       exit(1);
+                       die("git-cat-file: %s", strerror(errno));
+               } else if (!ret) {
+                       die("git-cat-file: disk full?");
                }
                size -= ret;
                buf += ret;