thodg/libgit2/src/attr.c

• Show log

  Commit

• Hash : 2e9d813b
  Author :
  Date : 2014-04-11T12:12:47
  

  Fix tests with new attr cache code

Download

• src/attr.c

• #include "common.h"
  #include "repository.h"
  #include "sysdir.h"
  #include "config.h"
  #include "attr_file.h"
  #include "ignore.h"
  #include "git2/oid.h"
  #include <ctype.h>
  
  GIT__USE_STRMAP;
  
  const char *git_attr__true  = "[internal]__TRUE__";
  const char *git_attr__false = "[internal]__FALSE__";
  const char *git_attr__unset = "[internal]__UNSET__";
  
  git_attr_t git_attr_value(const char *attr)
  {
  	if (attr == NULL || attr == git_attr__unset)
  		return GIT_ATTR_UNSPECIFIED_T;
  
  	if (attr == git_attr__true)
  		return GIT_ATTR_TRUE_T;
  
  	if (attr == git_attr__false)
  		return GIT_ATTR_FALSE_T;
  
  	return GIT_ATTR_VALUE_T;
  }
  
  static int collect_attr_files(
  	git_repository *repo,
  	uint32_t flags,
  	const char *path,
  	git_vector *files);
  
  static void release_attr_files(git_vector *files);
  
  int git_attr_get(
  	const char **value,
  	git_repository *repo,
  	uint32_t flags,
  	const char *pathname,
  	const char *name)
  {
  	int error;
  	git_attr_path path;
  	git_vector files = GIT_VECTOR_INIT;
  	size_t i, j;
  	git_attr_file *file;
  	git_attr_name attr;
  	git_attr_rule *rule;
  
  	assert(value && repo && name);
  
  	*value = NULL;
  
  	if (git_attr_path__init(&path, pathname, git_repository_workdir(repo)) < 0)
  		return -1;
  
  	if ((error = collect_attr_files(repo, flags, pathname, &files)) < 0)
  		goto cleanup;
  
  	memset(&attr, 0, sizeof(attr));
  	attr.name = name;
  	attr.name_hash = git_attr_file__name_hash(name);
  
  	git_vector_foreach(&files, i, file) {
  
  		git_attr_file__foreach_matching_rule(file, &path, j, rule) {
  			size_t pos;
  
  			if (!git_vector_bsearch(&pos, &rule->assigns, &attr)) {
  				*value = ((git_attr_assignment *)git_vector_get(
  							  &rule->assigns, pos))->value;
  				goto cleanup;
  			}
  		}
  	}
  
  cleanup:
  	release_attr_files(&files);
  	git_attr_path__free(&path);
  
  	return error;
  }
  
  
  typedef struct {
  	git_attr_name name;
  	git_attr_assignment *found;
  } attr_get_many_info;
  
  int git_attr_get_many(
  	const char **values,
  	git_repository *repo,
  	uint32_t flags,
  	const char *pathname,
  	size_t num_attr,
  	const char **names)
  {
  	int error;
  	git_attr_path path;
  	git_vector files = GIT_VECTOR_INIT;
  	size_t i, j, k;
  	git_attr_file *file;
  	git_attr_rule *rule;
  	attr_get_many_info *info = NULL;
  	size_t num_found = 0;
  
  	if (!num_attr)
  		return 0;
  
  	assert(values && repo && names);
  
  	if (git_attr_path__init(&path, pathname, git_repository_workdir(repo)) < 0)
  		return -1;
  
  	if ((error = collect_attr_files(repo, flags, pathname, &files)) < 0)
  		goto cleanup;
  
  	info = git__calloc(num_attr, sizeof(attr_get_many_info));
  	GITERR_CHECK_ALLOC(info);
  
  	git_vector_foreach(&files, i, file) {
  
  		git_attr_file__foreach_matching_rule(file, &path, j, rule) {
  
  			for (k = 0; k < num_attr; k++) {
  				size_t pos;
  
  				if (info[k].found != NULL) /* already found assignment */
  					continue;
  
  				if (!info[k].name.name) {
  					info[k].name.name = names[k];
  					info[k].name.name_hash = git_attr_file__name_hash(names[k]);
  				}
  
  				if (!git_vector_bsearch(&pos, &rule->assigns, &info[k].name)) {
  					info[k].found = (git_attr_assignment *)
  						git_vector_get(&rule->assigns, pos);
  					values[k] = info[k].found->value;
  
  					if (++num_found == num_attr)
  						goto cleanup;
  				}
  			}
  		}
  	}
  
  	for (k = 0; k < num_attr; k++) {
  		if (!info[k].found)
  			values[k] = NULL;
  	}
  
  cleanup:
  	release_attr_files(&files);
  	git_attr_path__free(&path);
  	git__free(info);
  
  	return error;
  }
  
  
  int git_attr_foreach(
  	git_repository *repo,
  	uint32_t flags,
  	const char *pathname,
  	int (*callback)(const char *name, const char *value, void *payload),
  	void *payload)
  {
  	int error;
  	git_attr_path path;
  	git_vector files = GIT_VECTOR_INIT;
  	size_t i, j, k;
  	git_attr_file *file;
  	git_attr_rule *rule;
  	git_attr_assignment *assign;
  	git_strmap *seen = NULL;
  
  	assert(repo && callback);
  
  	if (git_attr_path__init(&path, pathname, git_repository_workdir(repo)) < 0)
  		return -1;
  
  	if ((error = collect_attr_files(repo, flags, pathname, &files)) < 0 ||
  		(error = git_strmap_alloc(&seen)) < 0)
  		goto cleanup;
  
  	git_vector_foreach(&files, i, file) {
  
  		git_attr_file__foreach_matching_rule(file, &path, j, rule) {
  
  			git_vector_foreach(&rule->assigns, k, assign) {
  				/* skip if higher priority assignment was already seen */
  				if (git_strmap_exists(seen, assign->name))
  					continue;
  
  				git_strmap_insert(seen, assign->name, assign, error);
  				if (error < 0)
  					goto cleanup;
  
  				error = callback(assign->name, assign->value, payload);
  				if (error) {
  					giterr_set_after_callback(error);
  					goto cleanup;
  				}
  			}
  		}
  	}
  
  cleanup:
  	git_strmap_free(seen);
  	release_attr_files(&files);
  	git_attr_path__free(&path);
  
  	return error;
  }
  
  int git_attr_add_macro(
  	git_repository *repo,
  	const char *name,
  	const char *values)
  {
  	int error;
  	git_attr_rule *macro = NULL;
  	git_pool *pool;
  
  	if (git_attr_cache__init(repo) < 0)
  		return -1;
  
  	macro = git__calloc(1, sizeof(git_attr_rule));
  	GITERR_CHECK_ALLOC(macro);
  
  	pool = &git_repository_attr_cache(repo)->pool;
  
  	macro->match.pattern = git_pool_strdup(pool, name);
  	GITERR_CHECK_ALLOC(macro->match.pattern);
  
  	macro->match.length = strlen(macro->match.pattern);
  	macro->match.flags = GIT_ATTR_FNMATCH_MACRO;
  
  	error = git_attr_assignment__parse(repo, pool, &macro->assigns, &values);
  
  	if (!error)
  		error = git_attr_cache__insert_macro(repo, macro);
  
  	if (error < 0)
  		git_attr_rule__free(macro);
  
  	return error;
  }
  
  typedef struct {
  	git_repository *repo;
  	uint32_t flags;
  	const char *workdir;
  	git_index *index;
  	git_vector *files;
  } attr_walk_up_info;
  
  static int attr_decide_sources(
  	uint32_t flags, bool has_wd, bool has_index, git_attr_cache_source *srcs)
  {
  	int count = 0;
  
  	switch (flags & 0x03) {
  	case GIT_ATTR_CHECK_FILE_THEN_INDEX:
  		if (has_wd)
  			srcs[count++] = GIT_ATTR_CACHE__FROM_FILE;
  		if (has_index)
  			srcs[count++] = GIT_ATTR_CACHE__FROM_INDEX;
  		break;
  	case GIT_ATTR_CHECK_INDEX_THEN_FILE:
  		if (has_index)
  			srcs[count++] = GIT_ATTR_CACHE__FROM_INDEX;
  		if (has_wd)
  			srcs[count++] = GIT_ATTR_CACHE__FROM_FILE;
  		break;
  	case GIT_ATTR_CHECK_INDEX_ONLY:
  		if (has_index)
  			srcs[count++] = GIT_ATTR_CACHE__FROM_INDEX;
  		break;
  	}
  
  	return count;
  }
  
  static int push_attr_file(
  	git_repository *repo,
  	git_vector *list,
  	git_attr_cache_source source,
  	const char *base,
  	const char *filename)
  {
  	int error = 0;
  	git_attr_file *file = NULL;
  
  	error = git_attr_cache__get(
  		&file, repo, source, base, filename, git_attr_file__parse_buffer, NULL);
  	if (error < 0)
  		return error;
  
  	if (file != NULL) {
  		if ((error = git_vector_insert(list, file)) < 0)
  			git_attr_file__free(file);
  	}
  
  	return error;
  }
  
  static int push_one_attr(void *ref, git_buf *path)
  {
  	int error = 0, n_src, i;
  	attr_walk_up_info *info = (attr_walk_up_info *)ref;
  	git_attr_cache_source src[2];
  
  	n_src = attr_decide_sources(
  		info->flags, info->workdir != NULL, info->index != NULL, src);
  
  	for (i = 0; !error && i < n_src; ++i)
  		error = push_attr_file(
  			info->repo, info->files, src[i], path->ptr, GIT_ATTR_FILE);
  
  	return error;
  }
  
  static void release_attr_files(git_vector *files)
  {
  	size_t i;
  	git_attr_file *file;
  
  	git_vector_foreach(files, i, file) {
  		git_attr_file__free(file);
  		files->contents[i] = NULL;
  	}
  	git_vector_free(files);
  }
  
  static int collect_attr_files(
  	git_repository *repo,
  	uint32_t flags,
  	const char *path,
  	git_vector *files)
  {
  	int error;
  	git_buf dir = GIT_BUF_INIT;
  	const char *workdir = git_repository_workdir(repo);
  	attr_walk_up_info info = { NULL };
  
  	if ((error = git_attr_cache__init(repo)) < 0)
  		return error;
  
  	/* Resolve path in a non-bare repo */
  	if (workdir != NULL)
  		error = git_path_find_dir(&dir, path, workdir);
  	else
  		error = git_path_dirname_r(&dir, path);
  	if (error < 0)
  		goto cleanup;
  
  	/* in precendence order highest to lowest:
  	 * - $GIT_DIR/info/attributes
  	 * - path components with .gitattributes
  	 * - config core.attributesfile
  	 * - $GIT_PREFIX/etc/gitattributes
  	 */
  
  	error = push_attr_file(
  		repo, files, GIT_ATTR_CACHE__FROM_FILE,
  		git_repository_path(repo), GIT_ATTR_FILE_INREPO);
  	if (error < 0)
  		goto cleanup;
  
  	info.repo  = repo;
  	info.flags = flags;
  	info.workdir = workdir;
  	if (git_repository_index__weakptr(&info.index, repo) < 0)
  		giterr_clear(); /* no error even if there is no index */
  	info.files = files;
  
  	error = git_path_walk_up(&dir, workdir, push_one_attr, &info);
  	if (error < 0)
  		goto cleanup;
  
  	if (git_repository_attr_cache(repo)->cfg_attr_file != NULL) {
  		error = push_attr_file(
  			repo, files, GIT_ATTR_CACHE__FROM_FILE,
  			NULL, git_repository_attr_cache(repo)->cfg_attr_file);
  		if (error < 0)
  			goto cleanup;
  	}
  
  	if ((flags & GIT_ATTR_CHECK_NO_SYSTEM) == 0) {
  		error = git_sysdir_find_system_file(&dir, GIT_ATTR_FILE_SYSTEM);
  		if (!error)
  			error = push_attr_file(
  				repo, files, GIT_ATTR_CACHE__FROM_FILE, NULL, dir.ptr);
  		else if (error == GIT_ENOTFOUND) {
  			giterr_clear();
  			error = 0;
  		}
  	}
  
   cleanup:
  	if (error < 0)
  		release_attr_files(files);
  	git_buf_free(&dir);
  
  	return error;
  }
  

Download