reftable/dump.c

   1 /*
   2 Copyright 2020 Google LLC
   3
   4 Use of this source code is governed by a BSD-style
   5 license that can be found in the LICENSE file or at
   6 https://developers.google.com/open-source/licenses/bsd
   7 */
   8
   9 #include "git-compat-util.h"
  10 #include "hash-ll.h"
  11
  12 #include "reftable-blocksource.h"
  13 #include "reftable-error.h"
  14 #include "reftable-record.h"
  15 #include "reftable-tests.h"
  16 #include "reftable-writer.h"
  17 #include "reftable-iterator.h"
  18 #include "reftable-reader.h"
  19 #include "reftable-stack.h"
  20
  21 #include <stddef.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <string.h>
  26
  27 static int compact_stack(const char *stackdir)
  28 {
  29         struct reftable_stack *stack = NULL;
  30         struct reftable_write_options cfg = { 0 };
  31
  32         int err = reftable_new_stack(&stack, stackdir, cfg);
  33         if (err < 0)
  34                 goto done;
  35
  36         err = reftable_stack_compact_all(stack, NULL);
  37         if (err < 0)
  38                 goto done;
  39 done:
  40         if (stack) {
  41                 reftable_stack_destroy(stack);
  42         }
  43         return err;
  44 }
  45
  46 static void print_help(void)
  47 {
  48         printf("usage: dump [-cst] arg\n\n"
  49                "options: \n"
  50                "  -c compact\n"
  51                "  -t dump table\n"
  52                "  -s dump stack\n"
  53                "  -6 sha256 hash format\n"
  54                "  -h this help\n"
  55                "\n");
  56 }
  57
  58 int reftable_dump_main(int argc, char *const *argv)
  59 {
  60         int err = 0;
  61         int opt_dump_table = 0;
  62         int opt_dump_stack = 0;
  63         int opt_compact = 0;
  64         uint32_t opt_hash_id = GIT_SHA1_FORMAT_ID;
  65         const char *arg = NULL, *argv0 = argv[0];
  66
  67         for (; argc > 1; argv++, argc--)
  68                 if (*argv[1] != '-')
  69                         break;
  70                 else if (!strcmp("-t", argv[1]))
  71                         opt_dump_table = 1;
  72                 else if (!strcmp("-6", argv[1]))
  73                         opt_hash_id = GIT_SHA256_FORMAT_ID;
  74                 else if (!strcmp("-s", argv[1]))
  75                         opt_dump_stack = 1;
  76                 else if (!strcmp("-c", argv[1]))
  77                         opt_compact = 1;
  78                 else if (!strcmp("-?", argv[1]) || !strcmp("-h", argv[1])) {
  79                         print_help();
  80                         return 2;
  81                 }
  82
  83         if (argc != 2) {
  84                 fprintf(stderr, "need argument\n");
  85                 print_help();
  86                 return 2;
  87         }
  88
  89         arg = argv[1];
  90
  91         if (opt_dump_table) {
  92                 err = reftable_reader_print_file(arg);
  93         } else if (opt_dump_stack) {
  94                 err = reftable_stack_print_directory(arg, opt_hash_id);
  95         } else if (opt_compact) {
  96                 err = compact_stack(arg);
  97         }
  98
  99         if (err < 0) {
 100                 fprintf(stderr, "%s: %s: %s\n", argv0, arg,
 101                         reftable_error_str(err));
 102                 return 1;
 103         }
 104         return 0;
 105 }