thodg/libgit2/src/checkout.c

• Show log

  Commit

• Hash : 095ccc01
  Author :
  Date : 2012-07-26T16:31:49
  

  Checkout: implementation of most options

Download

• src/checkout.c

• /*
   * Copyright (C) 2009-2012 the libgit2 contributors
   *
   * This file is part of libgit2, distributed under the GNU GPL v2 with
   * a Linking Exception. For full terms see the included COPYING file.
   */
  
  #include <assert.h>
  
  #include "git2/checkout.h"
  #include "git2/repository.h"
  #include "git2/refs.h"
  #include "git2/tree.h"
  #include "git2/commit.h"
  #include "git2/blob.h"
  #include "git2/config.h"
  
  #include "common.h"
  #include "refs.h"
  #include "buffer.h"
  #include "repository.h"
  #include "filter.h"
  #include "blob.h"
  
  GIT_BEGIN_DECL
  
  
  typedef struct tree_walk_data
  {
  	git_indexer_stats *stats;
  	git_checkout_opts *opts;
  	git_repository *repo;
  	git_odb *odb;
  	bool do_symlinks;
  } tree_walk_data;
  
  
  static int blob_contents_to_link(tree_walk_data *data, git_buf *fnbuf,
  											const git_oid *id)
  {
  	int retcode = GIT_ERROR;
  	git_blob *blob;
  
  	/* Get the link target */
  	if (!(retcode = git_blob_lookup(&blob, data->repo, id))) {
  		git_buf linktarget = GIT_BUF_INIT;
  		if (!(retcode = git_blob__getbuf(&linktarget, blob))) {
  			/* Create the link */
  			const char *new = git_buf_cstr(&linktarget),
  						  *old = git_buf_cstr(fnbuf);
  			retcode = data->do_symlinks
  				? p_symlink(new, old)
  				: git_futils_fake_symlink(new, old);
  		}
  		git_buf_free(&linktarget);
  		git_blob_free(blob);
  	}
  
  	return retcode;
  }
  
  
  static int blob_contents_to_file(git_repository *repo, git_buf *fnbuf,
  											const git_oid *id, tree_walk_data *data)
  {
  	int retcode = GIT_ERROR;
  	git_buf contents = GIT_BUF_INIT;
  
  	/* Allow disabling of filters */
  	if (data->opts->disable_filters) {
  		git_blob *blob;
  		if (!(retcode = git_blob_lookup(&blob, repo, id))) {
  			retcode = git_blob__getbuf(&contents, blob);
  			git_blob_free(blob);
  		}
  	} else {
  		retcode = git_filter_blob_contents(&contents, repo, id, git_buf_cstr(fnbuf));
  	}
  	if (retcode < 0) goto bctf_cleanup;
  
  	/* Deal with pre-existing files */
  	if (git_path_exists(git_buf_cstr(fnbuf)) &&
  		 data->opts->existing_file_action == GIT_CHECKOUT_SKIP_EXISTING)
  		goto bctf_cleanup;
  
  	/* TODO: use p_open with flags */
  	int fd = git_futils_creat_withpath(git_buf_cstr(fnbuf),
  												  data->opts->dir_mode,
  												  data->opts->file_mode);
  	if (fd >= 0) {
  		if (!p_write(fd, git_buf_cstr(&contents),
  						 git_buf_len(&contents)))
  			retcode = 0;
  		else
  			retcode = GIT_ERROR;
  		p_close(fd);
  	}
  
  bctf_cleanup:
  	git_buf_free(&contents);
  	return retcode;
  }
  
  static int checkout_walker(const char *path, const git_tree_entry *entry, void *payload)
  {
  	int retcode = 0;
  	tree_walk_data *data = (tree_walk_data*)payload;
  	int attr = git_tree_entry_attributes(entry);
  	git_buf fnbuf = GIT_BUF_INIT;
  	git_buf_join_n(&fnbuf, '/', 3,
  						git_repository_workdir(data->repo),
  						path,
  						git_tree_entry_name(entry));
  
  	switch(git_tree_entry_type(entry))
  	{
  	case GIT_OBJ_TREE:
  		/* Nothing to do; the blob handling creates necessary directories. */
  		break;
  
  	case GIT_OBJ_COMMIT:
  		/* Submodule */
  		retcode = p_mkdir(git_buf_cstr(&fnbuf), 0644);
  		break;
  
  	case GIT_OBJ_BLOB:
  		if (S_ISLNK(attr)) {
  			retcode = blob_contents_to_link(data, &fnbuf,
  													  git_tree_entry_id(entry));
  		} else {
  			retcode = blob_contents_to_file(data->repo, &fnbuf,
  													  git_tree_entry_id(entry), data);
  		}
  		break;
  
  	default:
  		retcode = -1;
  		break;
  	}
  
  	git_buf_free(&fnbuf);
  	data->stats->processed++;
  	return retcode;
  }
  
  
  int git_checkout_index(git_repository *repo, git_checkout_opts *opts, git_indexer_stats *stats)
  {
  	int retcode = GIT_ERROR;
  	git_indexer_stats dummy_stats;
  	git_checkout_opts default_opts = {0};
  	git_tree *tree;
  	tree_walk_data payload;
  	git_config *cfg;
  
  	assert(repo);
  	if (!opts) opts = &default_opts;
  	if (!stats) stats = &dummy_stats;
  
  	/* Default options */
  	if (!opts->existing_file_action)
  		opts->existing_file_action = GIT_CHECKOUT_OVERWRITE_EXISTING;
  	/* opts->disable_filters is false by default */
  	if (!opts->dir_mode) opts->dir_mode = GIT_DIR_MODE;
  	if (!opts->file_mode)
  		opts->file_mode = 0644; 
  	if (!opts->file_open_flags)
  		opts->file_open_flags = O_CREAT | O_TRUNC | O_WRONLY;
  
  	if (git_repository_is_bare(repo)) {
  		giterr_set(GITERR_INVALID, "Checkout is not allowed for bare repositories");
  		return GIT_ERROR;
  	}
  
  	/* Determine if symlinks should be handled */
  	if (!git_repository_config(&cfg, repo)) {
  		int temp = true;
  		if (!git_config_get_bool(&temp, cfg, "core.symlinks")) {
  			payload.do_symlinks = !!temp;
  		}
  		git_config_free(cfg);
  	}
  
  	stats->total = stats->processed = 0;
  	payload.stats = stats;
  	payload.opts = opts;
  	payload.repo = repo;
  	if (git_repository_odb(&payload.odb, repo) < 0) return GIT_ERROR;
  
  	/* TODO: stats->total is never calculated. */
  
  	if (!git_repository_head_tree(&tree, repo)) {
  		/* Checkout the files */
  		if (!git_tree_walk(tree, checkout_walker, GIT_TREEWALK_POST, &payload)) {
  			retcode = 0;
  		}
  		git_tree_free(tree);
  	}
  
  	git_odb_free(payload.odb);
  	return retcode;
  }
  
  
  int git_checkout_head(git_repository *repo, git_checkout_opts *opts, git_indexer_stats *stats)
  {
  	/* TODO: read HEAD into index */
  
  	return git_checkout_index(repo, opts, stats);
  }
  
  
  GIT_END_DECL
  

Download