diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/status.c | 193 |
1 files changed, 193 insertions, 0 deletions
diff --git a/src/status.c b/src/status.c index 694ed921f..6e18b4edc 100644 --- a/src/status.c +++ b/src/status.c @@ -27,6 +27,9 @@ #include "git2.h" #include "fileops.h" #include "hash.h" +#include "vector.h" +#include "tree.h" +#include "git2/status.h" int git_status_hashfile(git_oid *out, const char *path) { @@ -75,3 +78,193 @@ int git_status_hashfile(git_oid *out, const char *path) return GIT_SUCCESS; } + +struct status_entry { + char path[GIT_PATH_MAX]; + + git_index_time mtime; + + git_oid head_oid; + git_oid index_oid; + git_oid wt_oid; + + unsigned int status_flags:6; +}; + +static int status_cmp(const void *a, const void *b) +{ + const struct status_entry *entry_a = (const struct status_entry *)(a); + const struct status_entry *entry_b = (const struct status_entry *)(b); + + return strcmp(entry_a->path, entry_b->path); +} + +static int status_srch(const void *key, const void *array_member) +{ + const char *path = (const char *)key; + const struct status_entry *entry = (const struct status_entry *)(array_member); + + return strcmp(path, entry->path); +} + +static int find_status_entry(git_vector *entries, const char *path) +{ + git_vector_sort(entries); + return git_vector_bsearch2(entries, status_srch, path); +} + +static struct status_entry *new_status_entry(git_vector *entries, const char *path) +{ + struct status_entry *e = git__malloc(sizeof(struct status_entry)); + memset(e, 0x0, sizeof(struct status_entry)); + git_vector_insert(entries, e); + strcpy(e->path, path); + return e; +} + +static void recurse_tree_entries(git_tree *tree, git_vector *entries, char *path) +{ + int i, cnt, idx; + struct status_entry *e; + char file_path[GIT_PATH_MAX]; + git_tree *subtree; + + cnt = git_tree_entrycount(tree); + for (i = 0; i < cnt; ++i) { + const git_tree_entry *tree_entry = git_tree_entry_byindex(tree, i); + + git_path_join(file_path, path, tree_entry->filename); + + if (git_tree_lookup(&subtree, tree->object.repo, &tree_entry->oid) == GIT_SUCCESS) { + recurse_tree_entries(subtree, entries, file_path); + return; + } + + if ((idx = find_status_entry(entries, file_path)) != GIT_ENOTFOUND) + e = (struct status_entry *)git_vector_get(entries, idx); + else + e = new_status_entry(entries, file_path); + + git_oid_cpy(&e->head_oid, &tree_entry->oid); + } + + git_tree_close(tree); +} + +static int workdir_path_len; +static int dirent_cb(void *state, char *full_path) +{ + int idx; + struct status_entry *e; + git_vector *entries = (git_vector *)state; + char *file_path = full_path + workdir_path_len; + struct stat filest; + git_oid oid; + + if ((git_futils_isdir(full_path) == GIT_SUCCESS) && (!strcmp(".git", file_path))) + return 0; + + if (git_futils_isdir(full_path) == GIT_SUCCESS) + return git_futils_direach(full_path, GIT_PATH_MAX, dirent_cb, state); + + if ((idx = find_status_entry(entries, file_path)) != GIT_ENOTFOUND) { + e = (struct status_entry *)git_vector_get(entries, idx); + + if (p_stat(full_path, &filest) < 0) + return git__throw(GIT_EOSERR, "Failed to read file %s", full_path); + + if (e->mtime.seconds == (git_time_t)filest.st_mtime) { + git_oid_cpy(&e->wt_oid, &e->index_oid); + return 0; + } + } else { + e = new_status_entry(entries, file_path); + } + + git_status_hashfile(&oid, full_path); + git_oid_cpy(&e->wt_oid, &oid); + + return 0; +} + +int git_status_foreach(git_repository *repo, int (*callback)(const char *, unsigned int, void *), void *payload) +{ + git_vector entries; + struct status_entry *e; + git_index *index; + unsigned int i, cnt; + git_index_entry *index_entry; + char temp_path[GIT_PATH_MAX]; + git_oid zero; + int error; + git_tree *tree; + + git_reference *head_ref, *resolved_head_ref; + git_commit *head_commit; + + git_repository_index(&index, repo); + + cnt = git_index_entrycount(index); + git_vector_init(&entries, cnt, status_cmp); + for (i = 0; i < cnt; ++i) { + index_entry = git_index_get(index, i); + + e = new_status_entry(&entries, index_entry->path); + git_oid_cpy(&e->index_oid, &index_entry->oid); + e->mtime = index_entry->mtime; + } + + git_reference_lookup(&head_ref, repo, GIT_HEAD_FILE); + git_reference_resolve(&resolved_head_ref, head_ref); + + git_commit_lookup(&head_commit, repo, git_reference_oid(resolved_head_ref)); + + // recurse through tree entries + git_commit_tree(&tree, head_commit); + recurse_tree_entries(tree, &entries, ""); + + workdir_path_len = strlen(repo->path_workdir); + strcpy(temp_path, repo->path_workdir); + git_futils_direach(temp_path, GIT_PATH_MAX, dirent_cb, &entries); + + memset(&zero, 0x0, sizeof(git_oid)); + for (i = 0; i < entries.length; ++i) { + int head_zero, index_zero, wt_zero; + e = (struct status_entry *)git_vector_get(&entries, i); + + head_zero = git_oid_cmp(&zero, &e->head_oid); + index_zero = git_oid_cmp(&zero, &e->index_oid); + wt_zero = git_oid_cmp(&zero, &e->wt_oid); + + if (head_zero == 0 && index_zero != 0) + e->status_flags |= GIT_STATUS_INDEX_NEW; + else if (index_zero == 0 && head_zero != 0) + e->status_flags |= GIT_STATUS_INDEX_DELETED; + else if (git_oid_cmp(&e->head_oid, &e->index_oid) != 0) + e->status_flags |= GIT_STATUS_INDEX_MODIFIED; + + if (index_zero == 0 && wt_zero != 0) + e->status_flags |= GIT_STATUS_WT_NEW; + else if (wt_zero == 0 && index_zero != 0) + e->status_flags |= GIT_STATUS_WT_DELETED; + else if (git_oid_cmp(&e->index_oid, &e->wt_oid) != 0) + e->status_flags |= GIT_STATUS_WT_MODIFIED; + } + + + for (i = 0; i < entries.length; ++i) { + e = (struct status_entry *)git_vector_get(&entries, i); + + if ((error = callback(e->path, e->status_flags, payload)) < GIT_SUCCESS) + return git__throw(error, "Failed to list statuses. User callback failed"); + } + + for (i = 0; i < entries.length; ++i) { + e = (struct status_entry *)git_vector_get(&entries, i); + free(e); + } + git_vector_free(&entries); + + return GIT_SUCCESS; +} + |