builtin-cat-file.con commit git-svn: Speed up fetch (ffe256f)
   1/*
   2 * GIT - The information manager from hell
   3 *
   4 * Copyright (C) Linus Torvalds, 2005
   5 */
   6#include "cache.h"
   7#include "exec_cmd.h"
   8#include "tag.h"
   9#include "tree.h"
  10#include "builtin.h"
  11
  12static void pprint_tag(const unsigned char *sha1, const char *buf, unsigned long size)
  13{
  14        /* the parser in tag.c is useless here. */
  15        const char *endp = buf + size;
  16        const char *cp = buf;
  17
  18        while (cp < endp) {
  19                char c = *cp++;
  20                if (c != '\n')
  21                        continue;
  22                if (7 <= endp - cp && !memcmp("tagger ", cp, 7)) {
  23                        const char *tagger = cp;
  24
  25                        /* Found the tagger line.  Copy out the contents
  26                         * of the buffer so far.
  27                         */
  28                        write_or_die(1, buf, cp - buf);
  29
  30                        /*
  31                         * Do something intelligent, like pretty-printing
  32                         * the date.
  33                         */
  34                        while (cp < endp) {
  35                                if (*cp++ == '\n') {
  36                                        /* tagger to cp is a line
  37                                         * that has ident and time.
  38                                         */
  39                                        const char *sp = tagger;
  40                                        char *ep;
  41                                        unsigned long date;
  42                                        long tz;
  43                                        while (sp < cp && *sp != '>')
  44                                                sp++;
  45                                        if (sp == cp) {
  46                                                /* give up */
  47                                                write_or_die(1, tagger,
  48                                                             cp - tagger);
  49                                                break;
  50                                        }
  51                                        while (sp < cp &&
  52                                               !('0' <= *sp && *sp <= '9'))
  53                                                sp++;
  54                                        write_or_die(1, tagger, sp - tagger);
  55                                        date = strtoul(sp, &ep, 10);
  56                                        tz = strtol(ep, NULL, 10);
  57                                        sp = show_date(date, tz, 0);
  58                                        write_or_die(1, sp, strlen(sp));
  59                                        xwrite(1, "\n", 1);
  60                                        break;
  61                                }
  62                        }
  63                        break;
  64                }
  65                if (cp < endp && *cp == '\n')
  66                        /* end of header */
  67                        break;
  68        }
  69        /* At this point, we have copied out the header up to the end of
  70         * the tagger line and cp points at one past \n.  It could be the
  71         * next header line after the tagger line, or it could be another
  72         * \n that marks the end of the headers.  We need to copy out the
  73         * remainder as is.
  74         */
  75        if (cp < endp)
  76                write_or_die(1, cp, endp - cp);
  77}
  78
  79static int cat_one_file(int opt, const char *exp_type, const char *obj_name)
  80{
  81        unsigned char sha1[20];
  82        enum object_type type;
  83        void *buf;
  84        unsigned long size;
  85
  86        if (get_sha1(obj_name, sha1))
  87                die("Not a valid object name %s", obj_name);
  88
  89        buf = NULL;
  90        switch (opt) {
  91        case 't':
  92                type = sha1_object_info(sha1, NULL);
  93                if (type > 0) {
  94                        printf("%s\n", typename(type));
  95                        return 0;
  96                }
  97                break;
  98
  99        case 's':
 100                type = sha1_object_info(sha1, &size);
 101                if (type > 0) {
 102                        printf("%lu\n", size);
 103                        return 0;
 104                }
 105                break;
 106
 107        case 'e':
 108                return !has_sha1_file(sha1);
 109
 110        case 'p':
 111                type = sha1_object_info(sha1, NULL);
 112                if (type < 0)
 113                        die("Not a valid object name %s", obj_name);
 114
 115                /* custom pretty-print here */
 116                if (type == OBJ_TREE) {
 117                        const char *ls_args[3] = {"ls-tree", obj_name, NULL};
 118                        return cmd_ls_tree(2, ls_args, NULL);
 119                }
 120
 121                buf = read_sha1_file(sha1, &type, &size);
 122                if (!buf)
 123                        die("Cannot read object %s", obj_name);
 124                if (type == OBJ_TAG) {
 125                        pprint_tag(sha1, buf, size);
 126                        return 0;
 127                }
 128
 129                /* otherwise just spit out the data */
 130                break;
 131        case 0:
 132                buf = read_object_with_reference(sha1, exp_type, &size, NULL);
 133                break;
 134
 135        default:
 136                die("git-cat-file: unknown option: %s\n", exp_type);
 137        }
 138
 139        if (!buf)
 140                die("git-cat-file %s: bad file", obj_name);
 141
 142        write_or_die(1, buf, size);
 143        return 0;
 144}
 145
 146static int batch_one_object(const char *obj_name, int print_contents)
 147{
 148        unsigned char sha1[20];
 149        enum object_type type;
 150        unsigned long size;
 151        void *contents = contents;
 152
 153        if (!obj_name)
 154           return 1;
 155
 156        if (get_sha1(obj_name, sha1)) {
 157                printf("%s missing\n", obj_name);
 158                return 0;
 159        }
 160
 161        if (print_contents)
 162                contents = read_sha1_file(sha1, &type, &size);
 163        else
 164                type = sha1_object_info(sha1, &size);
 165
 166        if (type <= 0)
 167                return 1;
 168
 169        printf("%s %s %lu\n", sha1_to_hex(sha1), typename(type), size);
 170        fflush(stdout);
 171
 172        if (print_contents) {
 173                write_or_die(1, contents, size);
 174                printf("\n");
 175                fflush(stdout);
 176        }
 177
 178        return 0;
 179}
 180
 181static int batch_objects(int print_contents)
 182{
 183        struct strbuf buf;
 184
 185        strbuf_init(&buf, 0);
 186        while (strbuf_getline(&buf, stdin, '\n') != EOF) {
 187                int error = batch_one_object(buf.buf, print_contents);
 188                if (error)
 189                        return error;
 190        }
 191
 192        return 0;
 193}
 194
 195static const char cat_file_usage[] = "git-cat-file [ [-t|-s|-e|-p|<type>] <sha1> | [--batch|--batch-check] < <list_of_sha1s> ]";
 196
 197int cmd_cat_file(int argc, const char **argv, const char *prefix)
 198{
 199        int i, opt = 0, batch = 0, batch_check = 0;
 200        const char *exp_type = NULL, *obj_name = NULL;
 201
 202        git_config(git_default_config);
 203
 204        for (i = 1; i < argc; ++i) {
 205                const char *arg = argv[i];
 206                int is_batch = 0, is_batch_check = 0;
 207
 208                is_batch = !strcmp(arg, "--batch");
 209                if (!is_batch)
 210                        is_batch_check = !strcmp(arg, "--batch-check");
 211
 212                if (is_batch || is_batch_check) {
 213                        if (opt) {
 214                                error("git-cat-file: Can't use %s with -%c", arg, opt);
 215                                usage(cat_file_usage);
 216                        } else if (exp_type) {
 217                                error("git-cat-file: Can't use %s when a type (\"%s\") is specified", arg, exp_type);
 218                                usage(cat_file_usage);
 219                        } else if (obj_name) {
 220                                error("git-cat-file: Can't use %s when an object (\"%s\") is specified", arg, obj_name);
 221                                usage(cat_file_usage);
 222                        }
 223
 224                        if ((is_batch && batch_check) || (is_batch_check && batch)) {
 225                                error("git-cat-file: Can't use %s with %s", arg, is_batch ? "--batch-check" : "--batch");
 226                                usage(cat_file_usage);
 227                        }
 228
 229                        if (is_batch)
 230                                batch = 1;
 231                        else
 232                                batch_check = 1;
 233
 234                        continue;
 235                }
 236
 237                if (!strcmp(arg, "-t") || !strcmp(arg, "-s") || !strcmp(arg, "-e") || !strcmp(arg, "-p")) {
 238                        if (batch || batch_check) {
 239                                error("git-cat-file: Can't use %s with %s", arg, batch ? "--batch" : "--batch-check");
 240                                usage(cat_file_usage);
 241                        }
 242
 243                        exp_type = arg;
 244                        opt = exp_type[1];
 245                        continue;
 246                }
 247
 248                if (arg[0] == '-')
 249                        usage(cat_file_usage);
 250
 251                if (!exp_type) {
 252                        if (batch || batch_check) {
 253                                error("git-cat-file: Can't specify a type (\"%s\") with %s", arg, batch ? "--batch" : "--batch-check");
 254                                usage(cat_file_usage);
 255                        }
 256
 257                        exp_type = arg;
 258                        continue;
 259                }
 260
 261                if (obj_name)
 262                        usage(cat_file_usage);
 263
 264                // We should have hit one of the earlier if (batch || batch_check) cases before
 265                // getting here.
 266                assert(!batch);
 267                assert(!batch_check);
 268
 269                obj_name = arg;
 270                break;
 271        }
 272
 273        if (batch || batch_check)
 274                return batch_objects(batch);
 275
 276        if (!exp_type || !obj_name)
 277                usage(cat_file_usage);
 278
 279        return cat_one_file(opt, exp_type, obj_name);
 280}