Allow getting/setting a repo's odb.

Author: arthurschreiber

ext/rugged/rugged.h

@@ -98,6 +98,7 @@ VALUE rugged_signature_new(const git_signature *sig, const char *encoding_name);
 VALUE rugged_repo_new(VALUE klass, git_repository *repo);
 VALUE rugged_index_new(VALUE klass, VALUE owner, git_index *index);
 VALUE rugged_config_new(VALUE klass, VALUE owner, git_config *cfg);
+VALUE rugged_odb_new(VALUE klass, VALUE owner, git_odb *odb);
 VALUE rugged_object_new(VALUE owner, git_object *object);
 VALUE rugged_object_rev_parse(VALUE rb_repo, VALUE rb_spec, int as_obj);
 VALUE rugged_ref_new(VALUE klass, VALUE owner, git_reference *ref);

ext/rugged/rugged_odb.c

@@ -27,6 +27,15 @@
 extern VALUE rb_mRugged;
 VALUE rb_cRuggedOdb;
 
+VALUE rugged_odb_new(VALUE klass, VALUE owner, git_odb *odb)
+{
+	VALUE rb_odb = Data_Wrap_Struct(klass, NULL, &git_odb_free, odb);
+
+	rugged_set_owner(rb_odb, owner);
+
+	return rb_odb;
+}
+
 /*
  *  call-seq:
  *    Odb.new() -> odb
@@ -38,8 +47,10 @@ VALUE rb_cRuggedOdb;
  */
 static VALUE rb_git_odb_new(VALUE klass) {
 	git_odb *odb;
+
 	rugged_exception_check(git_odb_new(&odb));
-	return Data_Wrap_Struct(klass, NULL, git_odb_free, odb);
+
+	return rugged_odb_new(klass, Qnil, odb);
 }
 
 /*
@@ -57,7 +68,7 @@ static VALUE rb_git_odb_open(VALUE klass, VALUE rb_path) {
 
 	rugged_exception_check(git_odb_open(&odb, StringValueCStr(rb_path)));
 
-	return Data_Wrap_Struct(klass, NULL, git_odb_free, odb);
+	return rugged_odb_new(klass, Qnil, odb);
 }
 
 /*

ext/rugged/rugged_repo.c

@@ -39,6 +39,7 @@ extern VALUE rb_cRuggedTag;
 extern VALUE rb_cRuggedTree;
 extern VALUE rb_cRuggedReference;
 extern VALUE rb_cRuggedBackend;
+extern VALUE rb_cRuggedOdb;
 
 extern VALUE rb_cRuggedCredPlaintext;
 extern VALUE rb_cRuggedCredSshKey;
@@ -590,6 +591,24 @@ static VALUE rb_git_repo_get_config(VALUE self)
 	RB_GIT_REPO_OWNED_GET(rb_cRuggedConfig, config);
 }
 
+/*
+ *  call-seq:
+ *    repo.odb = odb
+ */
+static VALUE rb_git_repo_set_odb(VALUE self, VALUE rb_data)
+{
+	RB_GIT_REPO_OWNED_SET(rb_cRuggedOdb, odb);
+}
+
+/*
+ *  call-seq:
+ *    repo.odb -> odb
+ */
+static VALUE rb_git_repo_get_odb(VALUE self)
+{
+	RB_GIT_REPO_OWNED_GET(rb_cRuggedOdb, odb);
+}
+
 /*
  *  call-seq:
  *    repo.ident = ident
@@ -2512,6 +2531,8 @@ void Init_rugged_repo(void)
 	rb_define_method(rb_cRuggedRepo, "index=",  rb_git_repo_set_index,  1);
 	rb_define_method(rb_cRuggedRepo, "config",  rb_git_repo_get_config,  0);
 	rb_define_method(rb_cRuggedRepo, "config=",  rb_git_repo_set_config,  1);
+	rb_define_method(rb_cRuggedRepo, "odb",  rb_git_repo_get_odb,  0);
+	rb_define_method(rb_cRuggedRepo, "odb=",  rb_git_repo_set_odb,  1);
 	rb_define_method(rb_cRuggedRepo, "refdb=", rb_git_repository_set_refdb, 1);
 
 	rb_define_method(rb_cRuggedRepo, "ident", rb_git_repo_get_ident, 0);