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
 106         if (set_config(MODE_ALL_PATTERNS))
 107                 return 1;
 108
 109         memset(&pl, 0, sizeof(pl));
 110
 111         sparse_filename = get_sparse_checkout_filename();
 112         res = add_patterns_from_file_to_list(sparse_filename, "", 0, &pl, NULL);
 113
 114         /* If we already have a sparse-checkout file, use it. */
 115         if (res >= 0) {
 116                 free(sparse_filename);
 117                 goto reset_dir;
 118         }
 119
 120         /* initial mode: all blobs at root */
 121         fp = xfopen(sparse_filename, "w");
 122         if (!fp)
 123                 die(_("failed to open '%s'"), sparse_filename);
 124
 125         free(sparse_filename);
 126         fprintf(fp, "/*\n!/*/\n");
 127         fclose(fp);
 128
 129 reset_dir:
 130         return update_working_directory();
 131 }
 132
 133 int cmd_sparse_checkout(int argc, const char **argv, const char *prefix)
 134 {
 135         static struct option builtin_sparse_checkout_options[] = {
 136                 OPT_END(),
 137         };
 138
 139         if (argc == 2 && !strcmp(argv[1], "-h"))
 140                 usage_with_options(builtin_sparse_checkout_usage,
 141                                    builtin_sparse_checkout_options);
 142
 143         argc = parse_options(argc, argv, prefix,
 144                              builtin_sparse_checkout_options,
 145                              builtin_sparse_checkout_usage,
 146                              PARSE_OPT_STOP_AT_NON_OPTION);
 147
 148         git_config(git_default_config, NULL);
 149
 150         if (argc > 0) {
 151                 if (!strcmp(argv[0], "list"))
 152                         return sparse_checkout_list(argc, argv);
 153                 if (!strcmp(argv[0], "init"))
 154                         return sparse_checkout_init(argc, argv);
 155         }
 156
 157         usage_with_options(builtin_sparse_checkout_usage,
 158                            builtin_sparse_checkout_options);
 159 }