Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 1 | #include "cache.h" |
| 2 | #include "refs.h" |
Junio C Hamano | cf0adba | 2006-11-19 13:22:44 -0800 | [diff] [blame] | 3 | #include "object.h" |
| 4 | #include "tag.h" |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 5 | |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 6 | static const char builtin_pack_refs_usage[] = |
Junio C Hamano | 0f018ba | 2007-01-08 14:40:33 -0800 | [diff] [blame] | 7 | "git-pack-refs [--all] [--prune | --no-prune]"; |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 8 | |
| 9 | struct ref_to_prune { |
| 10 | struct ref_to_prune *next; |
| 11 | unsigned char sha1[20]; |
| 12 | char name[FLEX_ARRAY]; |
| 13 | }; |
| 14 | |
| 15 | struct pack_refs_cb_data { |
| 16 | int prune; |
Alexandre Julliard | ca8e2d8 | 2006-11-02 12:13:32 +0100 | [diff] [blame] | 17 | int all; |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 18 | struct ref_to_prune *ref_to_prune; |
| 19 | FILE *refs_file; |
| 20 | }; |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 21 | |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 22 | static int do_not_prune(int flags) |
| 23 | { |
| 24 | /* If it is already packed or if it is a symref, |
| 25 | * do not prune it. |
| 26 | */ |
| 27 | return (flags & (REF_ISSYMREF|REF_ISPACKED)); |
| 28 | } |
| 29 | |
Junio C Hamano | 8da1977 | 2006-09-20 22:02:01 -0700 | [diff] [blame] | 30 | static int handle_one_ref(const char *path, const unsigned char *sha1, |
| 31 | int flags, void *cb_data) |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 32 | { |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 33 | struct pack_refs_cb_data *cb = cb_data; |
Junio C Hamano | cf0adba | 2006-11-19 13:22:44 -0800 | [diff] [blame] | 34 | int is_tag_ref; |
Junio C Hamano | cb5d709 | 2006-09-20 21:47:42 -0700 | [diff] [blame] | 35 | |
Junio C Hamano | 13e4aa9 | 2006-09-21 00:06:05 -0700 | [diff] [blame] | 36 | /* Do not pack the symbolic refs */ |
Junio C Hamano | cf0adba | 2006-11-19 13:22:44 -0800 | [diff] [blame] | 37 | if ((flags & REF_ISSYMREF)) |
| 38 | return 0; |
Junio C Hamano | cc44c76 | 2007-02-20 01:53:29 -0800 | [diff] [blame^] | 39 | is_tag_ref = !prefixcmp(path, "refs/tags/"); |
Linus Torvalds | 1b55593 | 2007-01-25 16:51:21 -0800 | [diff] [blame] | 40 | |
| 41 | /* ALWAYS pack refs that were already packed or are tags */ |
| 42 | if (!cb->all && !is_tag_ref && !(flags & REF_ISPACKED)) |
Junio C Hamano | cf0adba | 2006-11-19 13:22:44 -0800 | [diff] [blame] | 43 | return 0; |
| 44 | |
| 45 | fprintf(cb->refs_file, "%s %s\n", sha1_to_hex(sha1), path); |
| 46 | if (is_tag_ref) { |
| 47 | struct object *o = parse_object(sha1); |
| 48 | if (o->type == OBJ_TAG) { |
| 49 | o = deref_tag(o, path, 0); |
| 50 | if (o) |
Junio C Hamano | f4204ab | 2006-11-21 23:36:35 -0800 | [diff] [blame] | 51 | fprintf(cb->refs_file, "^%s\n", |
| 52 | sha1_to_hex(o->sha1)); |
Junio C Hamano | cf0adba | 2006-11-19 13:22:44 -0800 | [diff] [blame] | 53 | } |
| 54 | } |
| 55 | |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 56 | if (cb->prune && !do_not_prune(flags)) { |
| 57 | int namelen = strlen(path) + 1; |
| 58 | struct ref_to_prune *n = xcalloc(1, sizeof(*n) + namelen); |
| 59 | hashcpy(n->sha1, sha1); |
| 60 | strcpy(n->name, path); |
| 61 | n->next = cb->ref_to_prune; |
| 62 | cb->ref_to_prune = n; |
| 63 | } |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 64 | return 0; |
| 65 | } |
| 66 | |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 67 | /* make sure nobody touched the ref, and unlink */ |
| 68 | static void prune_ref(struct ref_to_prune *r) |
| 69 | { |
Junio C Hamano | 4431fcc | 2006-09-27 01:09:18 -0700 | [diff] [blame] | 70 | struct ref_lock *lock = lock_ref_sha1(r->name + 5, r->sha1); |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 71 | |
| 72 | if (lock) { |
Junio C Hamano | 6d15987 | 2006-09-22 21:31:40 -0700 | [diff] [blame] | 73 | unlink(git_path("%s", r->name)); |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 74 | unlock_ref(lock); |
| 75 | } |
| 76 | } |
| 77 | |
| 78 | static void prune_refs(struct ref_to_prune *r) |
| 79 | { |
| 80 | while (r) { |
| 81 | prune_ref(r); |
| 82 | r = r->next; |
| 83 | } |
| 84 | } |
| 85 | |
Junio C Hamano | 03a1821 | 2006-10-03 02:15:18 -0700 | [diff] [blame] | 86 | static struct lock_file packed; |
| 87 | |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 88 | int cmd_pack_refs(int argc, const char **argv, const char *prefix) |
| 89 | { |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 90 | int fd, i; |
| 91 | struct pack_refs_cb_data cbdata; |
| 92 | |
| 93 | memset(&cbdata, 0, sizeof(cbdata)); |
| 94 | |
Junio C Hamano | 0f018ba | 2007-01-08 14:40:33 -0800 | [diff] [blame] | 95 | cbdata.prune = 1; |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 96 | for (i = 1; i < argc; i++) { |
| 97 | const char *arg = argv[i]; |
| 98 | if (!strcmp(arg, "--prune")) { |
Junio C Hamano | 0f018ba | 2007-01-08 14:40:33 -0800 | [diff] [blame] | 99 | cbdata.prune = 1; /* now the default */ |
| 100 | continue; |
| 101 | } |
| 102 | if (!strcmp(arg, "--no-prune")) { |
| 103 | cbdata.prune = 0; |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 104 | continue; |
| 105 | } |
Junio C Hamano | b3d4204 | 2006-10-08 01:36:08 -0700 | [diff] [blame] | 106 | if (!strcmp(arg, "--all")) { |
Alexandre Julliard | ca8e2d8 | 2006-11-02 12:13:32 +0100 | [diff] [blame] | 107 | cbdata.all = 1; |
Junio C Hamano | b3d4204 | 2006-10-08 01:36:08 -0700 | [diff] [blame] | 108 | continue; |
| 109 | } |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 110 | /* perhaps other parameters later... */ |
| 111 | break; |
| 112 | } |
| 113 | if (i != argc) |
| 114 | usage(builtin_pack_refs_usage); |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 115 | |
Junio C Hamano | 03a1821 | 2006-10-03 02:15:18 -0700 | [diff] [blame] | 116 | fd = hold_lock_file_for_update(&packed, git_path("packed-refs"), 1); |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 117 | cbdata.refs_file = fdopen(fd, "w"); |
| 118 | if (!cbdata.refs_file) |
| 119 | die("unable to create ref-pack file structure (%s)", |
| 120 | strerror(errno)); |
Junio C Hamano | f4204ab | 2006-11-21 23:36:35 -0800 | [diff] [blame] | 121 | |
| 122 | /* perhaps other traits later as well */ |
| 123 | fprintf(cbdata.refs_file, "# pack-refs with: peeled \n"); |
| 124 | |
Alexandre Julliard | ca8e2d8 | 2006-11-02 12:13:32 +0100 | [diff] [blame] | 125 | for_each_ref(handle_one_ref, &cbdata); |
Junio C Hamano | 422b4a0 | 2006-10-04 21:37:15 -0700 | [diff] [blame] | 126 | fflush(cbdata.refs_file); |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 127 | fsync(fd); |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 128 | fclose(cbdata.refs_file); |
Junio C Hamano | 03a1821 | 2006-10-03 02:15:18 -0700 | [diff] [blame] | 129 | if (commit_lock_file(&packed) < 0) |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 130 | die("unable to overwrite old ref-pack file (%s)", strerror(errno)); |
Junio C Hamano | 9688460 | 2006-09-21 00:06:06 -0700 | [diff] [blame] | 131 | if (cbdata.prune) |
| 132 | prune_refs(cbdata.ref_to_prune); |
Linus Torvalds | e1e22e3 | 2006-09-11 16:37:32 -0700 | [diff] [blame] | 133 | return 0; |
| 134 | } |