Code

Merge branch 'jc/bare'
[git.git] / shallow.c
1 #include "cache.h"
2 #include "commit.h"
3 #include "tag.h"
5 static int is_shallow = -1;
7 int register_shallow(const unsigned char *sha1)
8 {
9         struct commit_graft *graft =
10                 xmalloc(sizeof(struct commit_graft));
11         struct commit *commit = lookup_commit(sha1);
13         hashcpy(graft->sha1, sha1);
14         graft->nr_parent = -1;
15         if (commit && commit->object.parsed)
16                 commit->parents = NULL;
17         return register_commit_graft(graft, 0);
18 }
20 int is_repository_shallow()
21 {
22         FILE *fp;
23         char buf[1024];
25         if (is_shallow >= 0)
26                 return is_shallow;
28         fp = fopen(git_path("shallow"), "r");
29         if (!fp) {
30                 is_shallow = 0;
31                 return is_shallow;
32         }
33         is_shallow = 1;
35         while (fgets(buf, sizeof(buf), fp)) {
36                 unsigned char sha1[20];
37                 if (get_sha1_hex(buf, sha1))
38                         die("bad shallow line: %s", buf);
39                 register_shallow(sha1);
40         }
41         fclose(fp);
42         return is_shallow;
43 }
45 struct commit_list *get_shallow_commits(struct object_array *heads, int depth,
46                 int shallow_flag, int not_shallow_flag)
47 {
48         int i = 0, cur_depth = 0;
49         struct commit_list *result = NULL;
50         struct object_array stack = {0, 0, NULL};
51         struct commit *commit = NULL;
53         while (commit || i < heads->nr || stack.nr) {
54                 struct commit_list *p;
55                 if (!commit) {
56                         if (i < heads->nr) {
57                                 commit = (struct commit *)
58                                         deref_tag(heads->objects[i++].item, NULL, 0);
59                                 if (commit->object.type != OBJ_COMMIT) {
60                                         commit = NULL;
61                                         continue;
62                                 }
63                                 if (!commit->util)
64                                         commit->util = xmalloc(sizeof(int));
65                                 *(int *)commit->util = 0;
66                                 cur_depth = 0;
67                         } else {
68                                 commit = (struct commit *)
69                                         stack.objects[--stack.nr].item;
70                                 cur_depth = *(int *)commit->util;
71                         }
72                 }
73                 parse_commit(commit);
74                 commit->object.flags |= not_shallow_flag;
75                 cur_depth++;
76                 for (p = commit->parents, commit = NULL; p; p = p->next) {
77                         if (!p->item->util) {
78                                 int *pointer = xmalloc(sizeof(int));
79                                 p->item->util = pointer;
80                                 *pointer =  cur_depth;
81                         } else {
82                                 int *pointer = p->item->util;
83                                 if (cur_depth >= *pointer)
84                                         continue;
85                                 *pointer = cur_depth;
86                         }
87                         if (cur_depth < depth) {
88                                 if (p->next)
89                                         add_object_array(&p->item->object,
90                                                         NULL, &stack);
91                                 else {
92                                         commit = p->item;
93                                         cur_depth = *(int *)commit->util;
94                                 }
95                         } else {
96                                 commit_list_insert(p->item, &result);
97                                 p->item->object.flags |= shallow_flag;
98                         }
99                 }
100         }
102         return result;