git.oblomov.eu Git - git/blob - builtin/sparse-checkout.c

   1 #include "builtin.h"
   2 #include "config.h"
   3 #include "dir.h"
   4 #include "parse-options.h"
   5 #include "pathspec.h"
   6 #include "repository.h"
   7 #include "run-command.h"
   8 #include "strbuf.h"
   9
  10 static char const * const builtin_sparse_checkout_usage[] = {
  11         N_("git sparse-checkout (init|list)"),
  12         NULL
  13 };
  14
  15 static char *get_sparse_checkout_filename(void)
  16 {
  17         return git_pathdup("info/sparse-checkout");
  18 }
  19
  20 static void write_patterns_to_file(FILE *fp, struct pattern_list *pl)
  21 {
  22         int i;
  23
  24         for (i = 0; i < pl->nr; i++) {
  25                 struct path_pattern *p = pl->patterns[i];
  26
  27                 if (p->flags & PATTERN_FLAG_NEGATIVE)
  28                         fprintf(fp, "!");
  29
  30                 fprintf(fp, "%s", p->pattern);
  31
  32                 if (p->flags & PATTERN_FLAG_MUSTBEDIR)
  33                         fprintf(fp, "/");
  34
  35                 fprintf(fp, "\n");
  36         }
  37 }
  38
  39 static int sparse_checkout_list(int argc, const char **argv)
  40 {
  41         struct pattern_list pl;
  42         char *sparse_filename;
  43         int res;
  44
  45         memset(&pl, 0, sizeof(pl));
  46
  47         sparse_filename = get_sparse_checkout_filename();
  48         res = add_patterns_from_file_to_list(sparse_filename, "", 0, &pl, NULL);
  49         free(sparse_filename);
  50
  51         if (res < 0) {
  52                 warning(_("this worktree is not sparse (sparse-checkout file may not exist)"));
  53                 return 0;
  54         }
  55
  56         write_patterns_to_file(stdout, &pl);
  57         clear_pattern_list(&pl);
  58
  59         return 0;
  60 }
  61
  62 static int update_working_directory(void)
  63 {
  64         struct argv_array argv = ARGV_ARRAY_INIT;
  65         int result = 0;
  66         argv_array_pushl(&argv, "read-tree", "-m", "-u", "HEAD", NULL);
  67
  68         if (run_command_v_opt(argv.argv, RUN_GIT_CMD)) {
  69                 error(_("failed to update index with new sparse-checkout paths"));
  70                 result = 1;
  71         }
  72
  73         argv_array_clear(&argv);
  74         return result;
  75 }
  76
  77 enum sparse_checkout_mode {
  78         MODE_NO_PATTERNS = 0,
  79         MODE_ALL_PATTERNS = 1,
  80 };
  81
  82 static int set_config(enum sparse_checkout_mode mode)
  83 {
  84         const char *config_path;
  85
  86         if (git_config_set_gently("extensions.worktreeConfig", "true")) {
  87                 error(_("failed to set extensions.worktreeConfig setting"));
  88                 return 1;
  89         }
  90
  91         config_path = git_path("config.worktree");
  92         git_config_set_in_file_gently(config_path,
  93                                       "core.sparseCheckout",
  94                                       mode ? "true" : NULL);
  95
  96         return 0;
  97 }
  98
  99 static int sparse_checkout_init(int argc, const char **argv)
 100 {
 101         struct pattern_list pl;
 102         char *sparse_filename;
 103         FILE *fp;
 104         int res;
 105         struct object_id oid;
 106
 107         if (set_config(MODE_ALL_PATTERNS))
 108                 return 1;
 109
 110         memset(&pl, 0, sizeof(pl));
 111
 112         sparse_filename = get_sparse_checkout_filename();
 113         res = add_patterns_from_file_to_list(sparse_filename, "", 0, &pl, NULL);
 114
 115         /* If we already have a sparse-checkout file, use it. */
 116         if (res >= 0) {
 117                 free(sparse_filename);
 118                 goto reset_dir;
 119         }
 120
 121         /* initial mode: all blobs at root */
 122         fp = xfopen(sparse_filename, "w");
 123         if (!fp)
 124                 die(_("failed to open '%s'"), sparse_filename);
 125
 126         free(sparse_filename);
 127         fprintf(fp, "/*\n!/*/\n");
 128         fclose(fp);
 129
 130         if (get_oid("HEAD", &oid)) {
 131                 /* assume we are in a fresh repo */
 132                 return 0;
 133         }
 134
 135 reset_dir:
 136         return update_working_directory();
 137 }
 138
 139 int cmd_sparse_checkout(int argc, const char **argv, const char *prefix)
 140 {
 141         static struct option builtin_sparse_checkout_options[] = {
 142                 OPT_END(),
 143         };
 144
 145         if (argc == 2 && !strcmp(argv[1], "-h"))
 146                 usage_with_options(builtin_sparse_checkout_usage,
 147                                    builtin_sparse_checkout_options);
 148
 149         argc = parse_options(argc, argv, prefix,
 150                              builtin_sparse_checkout_options,
 151                              builtin_sparse_checkout_usage,
 152                              PARSE_OPT_STOP_AT_NON_OPTION);
 153
 154         git_config(git_default_config, NULL);
 155
 156         if (argc > 0) {
 157                 if (!strcmp(argv[0], "list"))
 158                         return sparse_checkout_list(argc, argv);
 159                 if (!strcmp(argv[0], "init"))
 160                         return sparse_checkout_init(argc, argv);
 161         }
 162
 163         usage_with_options(builtin_sparse_checkout_usage,
 164                            builtin_sparse_checkout_options);
 165 }