Minimal command for branches

CMakeLists.txt

@@ -41,6 +41,8 @@ find_package(libgit2)
 set(GIT2CPP_SRC
     ${GIT2CPP_SOURCE_DIR}/subcommand/add_subcommand.cpp
     ${GIT2CPP_SOURCE_DIR}/subcommand/add_subcommand.hpp
+    ${GIT2CPP_SOURCE_DIR}/subcommand/branch_subcommand.cpp
+    ${GIT2CPP_SOURCE_DIR}/subcommand/branch_subcommand.hpp
     ${GIT2CPP_SOURCE_DIR}/subcommand/init_subcommand.cpp
     ${GIT2CPP_SOURCE_DIR}/subcommand/init_subcommand.hpp
     ${GIT2CPP_SOURCE_DIR}/subcommand/status_subcommand.cpp
@@ -49,6 +51,8 @@ set(GIT2CPP_SRC
     ${GIT2CPP_SOURCE_DIR}/utils/common.hpp
     ${GIT2CPP_SOURCE_DIR}/utils/git_exception.cpp
     ${GIT2CPP_SOURCE_DIR}/utils/git_exception.hpp
+    ${GIT2CPP_SOURCE_DIR}/wrapper/branch_wrapper.cpp
+    ${GIT2CPP_SOURCE_DIR}/wrapper/branch_wrapper.hpp
     ${GIT2CPP_SOURCE_DIR}/wrapper/commit_wrapper.cpp
     ${GIT2CPP_SOURCE_DIR}/wrapper/commit_wrapper.hpp
     ${GIT2CPP_SOURCE_DIR}/wrapper/index_wrapper.cpp

src/main.cpp

@@ -5,6 +5,7 @@
 #include "utils/git_exception.hpp"
 #include "version.hpp"
 #include "subcommand/add_subcommand.hpp"
+#include "subcommand/branch_subcommand.hpp"
 #include "subcommand/init_subcommand.hpp"
 #include "subcommand/status_subcommand.hpp"
 
@@ -23,6 +24,7 @@ int main(int argc, char** argv)
         init_subcommand init(lg2_obj, app);
         status_subcommand status(lg2_obj, app);
         add_subcommand add(lg2_obj, app);
+        branch_subcommand(lg2_obj, app);
 
         app.parse(argc, argv);
 

src/subcommand/branch_subcommand.cpp

@@ -0,0 +1,60 @@
+#include <iostream>
+
+#include "../subcommand/branch_subcommand.hpp"
+#include "../wrapper/repository_wrapper.hpp"
+
+branch_subcommand::branch_subcommand(const libgit2_object&, CLI::App& app)
+{
+    auto* sub = app.add_subcommand("branch", "List, create or delete branches");
+
+    sub->add_option("<branchname>", m_branch_name, "The name of the branch to create or delete");
+
+    sub->add_flag("-d,--delete", m_deletion_flag, "Delete a branch");
+    sub->add_flag("-a,--all", m_all_flag, "List both remote-tracking branches and local branches");
+    sub->add_flag("-r,--remotes", m_remote_flag, "List or delete (if used with -d) the remote-tracking branches");
+    sub->add_flag("-l,--list", m_list_flag, "List branches");
+    sub->add_flag("-f,--force", m_force_flag, "Skips confirmation");
+
+    sub->callback([this]() { this->run(); });
+}
+
+void branch_subcommand::run()
+{
+    auto directory = get_current_git_path();
+    auto bare = false;
+    auto repo = repository_wrapper::open(directory);
+
+    if (m_list_flag || m_branch_name.empty())
+    {
+        git_branch_t type = m_all_flag ? GIT_BRANCH_ALL : (m_remote_flag ? GIT_BRANCH_REMOTE : GIT_BRANCH_LOCAL); 
+        auto iter = repo.iterate_branches(type);
+        auto br = iter.next();
+        while (br)
+        {
+            std::cout << br->name() << std::endl;
+            br = iter.next();
+        }
+    }
+    else if (m_deletion_flag)
+    {
+        run_deletion(repo);
+    }
+    else
+    {
+        run_creation(repo);
+    }
+}
+
+void branch_subcommand::run_deletion(repository_wrapper& repo)
+{
+    auto branch = repo.find_branch(m_branch_name);
+    // TODO: handle unmerged stated once we handle upstream repos
+    delete_branch(std::move(branch));
+}
+
+
+void branch_subcommand::run_creation(repository_wrapper& repo)
+{
+    // TODO: handle specification of starting commit
+    repo.create_branch(m_branch_name, m_force_flag);
+}

src/subcommand/branch_subcommand.hpp

@@ -0,0 +1,28 @@
+#pragma once
+
+#include <string>
+
+#include <CLI/CLI.hpp>
+
+#include "../utils/common.hpp"
+#include "../wrapper/repository_wrapper.hpp"
+
+class branch_subcommand
+{
+public:
+
+    explicit branch_subcommand(const libgit2_object&, CLI::App& app);
+    void run();
+
+private:
+
+    void run_deletion(repository_wrapper& repo);
+    void run_creation(repository_wrapper& repo);
+
+    std::string m_branch_name = {};
+    bool m_deletion_flag = false;
+    bool m_all_flag = false;
+    bool m_remote_flag = false;
+    bool m_list_flag = false;
+    bool m_force_flag = false;
+};

src/wrapper/branch_wrapper.cpp

@@ -0,0 +1,56 @@
+#include "../utils/git_exception.hpp"
+#include "../wrapper/branch_wrapper.hpp"
+#include "../wrapper/commit_wrapper.hpp"
+#include "../wrapper/repository_wrapper.hpp"
+
+#include <iostream>
+
+branch_wrapper::branch_wrapper(git_reference* polna_ref)
+    : base_type(polna_ref)
+{    
+}
+
+branch_wrapper::~branch_wrapper()
+{
+    git_reference_free(p_resource);
+    p_resource=nullptr;
+}
+
+std::string_view branch_wrapper::name() const
+{
+    const char* out = nullptr;
+    throwIfError(git_branch_name(&out, p_resource));
+    return std::string_view(out);
+}
+
+void delete_branch(branch_wrapper&& branch)
+{
+    throwIfError(git_branch_delete(branch));
+}
+
+branch_iterator::branch_iterator(git_branch_iterator* iter)
+    : base_type(iter)
+{
+}
+
+branch_iterator::~branch_iterator()
+{
+    git_branch_iterator_free(p_resource);
+    p_resource = nullptr;
+}
+
+
+std::optional<branch_wrapper> branch_iterator::next()
+{
+    git_reference* ref = nullptr;
+    git_branch_t type;
+    int res = git_branch_next(&ref, &type, p_resource);
+    if (res == 0)
+    {
+        return branch_wrapper(ref);
+    }
+    else
+    {
+        return std::nullopt;
+    }
+}

src/wrapper/branch_wrapper.hpp

@@ -0,0 +1,56 @@
+#pragma once
+
+#include <optional>
+#include <string_view>
+
+#include <git2.h>
+
+#include "../wrapper/wrapper_base.hpp"
+
+class commit_wrapper;
+
+class branch_wrapper : public wrapper_base<git_reference>
+{
+public:
+
+    using base_type = wrapper_base<git_reference>;
+
+    ~branch_wrapper();
+
+    branch_wrapper(branch_wrapper&&) = default;
+    branch_wrapper& operator=(branch_wrapper&&) = default;
+
+    std::string_view name() const;
+
+private:
+
+    explicit branch_wrapper(git_reference* ref);
+
+    friend class repository_wrapper;
+    friend class branch_iterator;
+};
+
+void delete_branch(branch_wrapper&& br);
+
+// Rust / Python-like iterator instead of regular C++ iterator,
+// because of the libgit2 API. Implementing the postfix increment
+// operator of an input iterator would be overcomplicated.
+class branch_iterator : public wrapper_base<git_branch_iterator>
+{
+public:
+
+    using base_type = wrapper_base<git_branch_iterator>;
+
+    ~branch_iterator();
+
+    branch_iterator(branch_iterator&&) = default;
+    branch_iterator& operator=(branch_iterator&&) = default;
+
+    std::optional<branch_wrapper> next();
+
+private:
+
+    explicit branch_iterator(git_branch_iterator* iter);
+
+    friend class repository_wrapper;
+};

src/wrapper/commit_wrapper.cpp

@@ -1,5 +1,6 @@
-#include "commit_wrapper.hpp"
 #include "../utils/git_exception.hpp"
+#include "../wrapper/commit_wrapper.hpp"
+#include "../wrapper/repository_wrapper.hpp"
 
 commit_wrapper::~commit_wrapper()
 {
@@ -8,7 +9,7 @@ commit_wrapper::~commit_wrapper()
 }
 
 
-commit_wrapper commit_wrapper::last_commit(const repository_wrapper& repo, const std::string& ref_name)
+commit_wrapper commit_wrapper::from_reference_name(const repository_wrapper& repo, const std::string& ref_name)
 {
     git_oid oid_parent_commit;
     throwIfError(git_reference_name_to_id(&oid_parent_commit, repo, ref_name.c_str()));

src/wrapper/commit_wrapper.hpp

@@ -4,9 +4,10 @@
 
 #include <git2.h>
 
-#include "../wrapper/repository_wrapper.hpp"
 #include "../wrapper/wrapper_base.hpp"
 
+class repository_wrapper;
+
 class commit_wrapper : public wrapper_base<git_commit>
 {
 public:
@@ -17,7 +18,7 @@ class commit_wrapper : public wrapper_base<git_commit>
     commit_wrapper& operator=(commit_wrapper&&) noexcept = default;
 
     static commit_wrapper
-    last_commit(const repository_wrapper& repo, const std::string& ref_name = "HEAD");
+    from_reference_name(const repository_wrapper& repo, const std::string& ref_name = "HEAD");
 
 private:
 

src/wrapper/repository_wrapper.cpp

@@ -1,5 +1,5 @@
-#include "repository_wrapper.hpp"
 #include "../utils/git_exception.hpp"
+#include "../wrapper/repository_wrapper.hpp"
 
 repository_wrapper::~repository_wrapper()
 {
@@ -26,3 +26,29 @@ index_wrapper repository_wrapper::make_index()
     index_wrapper index = index_wrapper::init(*this);
     return index;
 }
+
+branch_wrapper repository_wrapper::create_branch(const std::string& name, bool force)
+{
+    return create_branch(name, commit_wrapper::from_reference_name(*this), force);
+}
+
+branch_wrapper repository_wrapper::create_branch(const std::string& name, const commit_wrapper& commit, bool force)
+{
+    git_reference* branch = nullptr;
+    throwIfError(git_branch_create(&branch, *this, name.c_str(), commit, force));
+    return branch_wrapper(branch);
+}
+
+branch_wrapper repository_wrapper::find_branch(const std::string& name)
+{
+    git_reference* branch = nullptr;
+    throwIfError(git_branch_lookup(&branch, *this, name.c_str(), GIT_BRANCH_LOCAL));
+    return branch_wrapper(branch);
+}
+
+branch_iterator repository_wrapper::iterate_branches(git_branch_t type) const
+{
+    git_branch_iterator* iter = nullptr;
+    git_branch_iterator_new(&iter, *this, type);
+    return branch_iterator(iter);
+}

src/wrapper/repository_wrapper.hpp

@@ -4,6 +4,8 @@
 
 #include <git2.h>
 
+#include "../wrapper/branch_wrapper.hpp"
+#include "../wrapper/commit_wrapper.hpp"
 #include "../wrapper/index_wrapper.hpp"
 #include "../wrapper/wrapper_base.hpp"
 
@@ -18,8 +20,16 @@ class repository_wrapper : public wrapper_base<git_repository>
 
     static repository_wrapper init(const std::string& directory, bool bare);
     static repository_wrapper open(const std::string& directory);
+
     index_wrapper make_index();
 
+    branch_wrapper create_branch(const std::string& name, bool force);
+    branch_wrapper create_branch(const std::string& name, const commit_wrapper& commit, bool force);
+
+    branch_wrapper find_branch(const std::string& name);
+
+    branch_iterator iterate_branches(git_branch_t type) const;
+
 private:
 
     repository_wrapper() = default;

src/wrapper/wrapper_base.hpp

@@ -28,7 +28,11 @@ class wrapper_base
 
 protected:
     // Allocation and deletion of p_resource must be handled by inheriting class.
-    wrapper_base() = default;
+    explicit wrapper_base(resource_type* resource = nullptr)
+        : p_resource(resource)
+    {
+    }
+
     ~wrapper_base() = default;
 
     resource_type* p_resource = nullptr;

test/test_branch.py

@@ -0,0 +1,28 @@
+import os
+import subprocess
+
+import pytest
+
+@pytest.fixture
+def rename_git():
+    os.rename("test/data/status_data/embeded_git/", "test/data/status_data/.git/")
+    yield
+    os.rename("test/data/status_data/.git/", "test/data/status_data/embeded_git/")
+
+def test_branch_list(rename_git, git2cpp_path):
+    cmd = [git2cpp_path, 'branch']
+    p = subprocess.run(cmd, capture_output=True, cwd="test/data/status_data", text=True)
+    assert('main' in p.stdout == 'main\n')
+
+def test_branch_create_delete(rename_git, git2cpp_path):
+    create_cmd = [git2cpp_path, 'branch', 'foregone']
+    subprocess.run(create_cmd, capture_output=True, cwd="test/data/status_data", text=True)
+    list_cmd = [git2cpp_path, 'branch']
+    p = subprocess.run(list_cmd, capture_output=True, cwd="test/data/status_data", text=True)
+    assert(p.stdout == 'foregone\nmain\n')
+    del_cmd = [git2cpp_path, 'branch', '-d', 'foregone']
+    subprocess.run(del_cmd, capture_output=True, cwd="test/data/status_data", text=True)
+    p2 = subprocess.run(list_cmd, capture_output=True, cwd="test/data/status_data", text=True)
+    assert(p2.stdout == 'main\n')
+
+