Teach git-commit-graph to write graph files. Create new test script to verify
this command succeeds without failure.

Signed-off-by: Derrick Stolee <dsto...@microsoft.com>
---
 Documentation/git-commit-graph.txt |  39 +++++++++++++
 builtin/commit-graph.c             |  43 +++++++++++++++
 t/t5318-commit-graph.sh            | 109 +++++++++++++++++++++++++++++++++++++
 3 files changed, 191 insertions(+)
 create mode 100755 t/t5318-commit-graph.sh

diff --git a/Documentation/git-commit-graph.txt 
b/Documentation/git-commit-graph.txt
index e1c3078ca1..55dfe5c3d8 100644
--- a/Documentation/git-commit-graph.txt
+++ b/Documentation/git-commit-graph.txt
@@ -5,6 +5,45 @@ NAME
 ----
 git-commit-graph - Write and verify Git commit graphs (.graph files)
 
+
+SYNOPSIS
+--------
+[verse]
+'git commit-graph write' <options> [--pack-dir <pack_dir>]
+
+
+DESCRIPTION
+-----------
+
+Manage the serialized commit graph file.
+
+
+OPTIONS
+-------
+--pack-dir::
+       Use given directory for the location of packfiles, graph-head,
+       and graph files.
+
+
+COMMANDS
+--------
+'write'::
+
+Write a commit graph file based on the commits found in packfiles.
+Includes all commits from the existing commit graph file. Outputs the
+checksum hash of the written file.
+
+
+EXAMPLES
+--------
+
+* Write a commit graph file for the packed commits in your local .git folder.
++
+------------------------------------------------
+$ git commit-graph write
+------------------------------------------------
+
+
 GIT
 ---
 Part of the linkgit:git[1] suite
diff --git a/builtin/commit-graph.c b/builtin/commit-graph.c
index a01c5d9981..5dac033bfe 100644
--- a/builtin/commit-graph.c
+++ b/builtin/commit-graph.c
@@ -1,9 +1,16 @@
 #include "builtin.h"
 #include "config.h"
 #include "parse-options.h"
+#include "commit-graph.h"
 
 static char const * const builtin_commit_graph_usage[] = {
        N_("git commit-graph [--pack-dir <packdir>]"),
+       N_("git commit-graph write [--pack-dir <packdir>]"),
+       NULL
+};
+
+static const char * const builtin_commit_graph_write_usage[] = {
+       N_("git commit-graph write [--pack-dir <packdir>]"),
        NULL
 };
 
@@ -11,6 +18,37 @@ static struct opts_commit_graph {
        const char *pack_dir;
 } opts;
 
+static int graph_write(int argc, const char **argv)
+{
+       struct object_id *graph_hash;
+
+       static struct option builtin_commit_graph_write_options[] = {
+               { OPTION_STRING, 'p', "pack-dir", &opts.pack_dir,
+                       N_("dir"),
+                       N_("The pack directory to store the graph") },
+               OPT_END(),
+       };
+
+       argc = parse_options(argc, argv, NULL,
+                            builtin_commit_graph_write_options,
+                            builtin_commit_graph_write_usage, 0);
+
+       if (!opts.pack_dir) {
+               struct strbuf path = STRBUF_INIT;
+               strbuf_addstr(&path, get_object_directory());
+               strbuf_addstr(&path, "/pack");
+               opts.pack_dir = strbuf_detach(&path, NULL);
+       }
+
+       graph_hash = write_commit_graph(opts.pack_dir);
+
+       if (graph_hash) {
+               printf("%s\n", oid_to_hex(graph_hash));
+               FREE_AND_NULL(graph_hash);
+       }
+
+       return 0;
+}
 
 int cmd_commit_graph(int argc, const char **argv, const char *prefix)
 {
@@ -31,6 +69,11 @@ int cmd_commit_graph(int argc, const char **argv, const char 
*prefix)
                             builtin_commit_graph_usage,
                             PARSE_OPT_STOP_AT_NON_OPTION);
 
+       if (argc > 0) {
+               if (!strcmp(argv[0], "write"))
+                       return graph_write(argc, argv);
+       }
+
        usage_with_options(builtin_commit_graph_usage,
                           builtin_commit_graph_options);
 }
diff --git a/t/t5318-commit-graph.sh b/t/t5318-commit-graph.sh
new file mode 100755
index 0000000000..b762587595
--- /dev/null
+++ b/t/t5318-commit-graph.sh
@@ -0,0 +1,109 @@
+#!/bin/sh
+
+test_description='commit graph'
+. ./test-lib.sh
+
+test_expect_success 'setup full repo' '
+       rm -rf .git &&
+       mkdir full &&
+       cd full &&
+       git init &&
+       packdir=".git/objects/pack"'
+
+test_expect_success 'write graph with no packs' '
+       git commit-graph write --pack-dir .'
+
+test_expect_success 'create commits and repack' '
+       for i in $(test_seq 3)
+       do
+               test_commit $i &&
+               git branch commits/$i
+       done &&
+       git repack'
+
+test_expect_success 'write graph' '
+       graph1=$(git commit-graph write) &&
+       test_path_is_file $packdir/graph-$graph1.graph'
+
+test_expect_success 'Add more commits' '
+       git reset --hard commits/1 &&
+       for i in $(test_seq 4 5)
+       do
+               test_commit $i &&
+               git branch commits/$i
+       done &&
+       git reset --hard commits/2 &&
+       for i in $(test_seq 6 7)
+       do
+               test_commit $i &&
+               git branch commits/$i
+       done &&
+       git reset --hard commits/2 &&
+       git merge commits/4 &&
+       git branch merge/1 &&
+       git reset --hard commits/4 &&
+       git merge commits/6 &&
+       git branch merge/2 &&
+       git reset --hard commits/3 &&
+       git merge commits/5 commits/7 &&
+       git branch merge/3 &&
+       git repack'
+
+# Current graph structure:
+#
+#   __M3___
+#  /   |   \
+# 3 M1 5 M2 7
+# |/  \|/  \|
+# 2    4    6
+# |___/____/
+# 1
+
+
+test_expect_success 'write graph with merges' '
+       graph2=$(git commit-graph write)&&
+       test_path_is_file $packdir/graph-$graph2.graph'
+
+test_expect_success 'Add one more commit' '
+       test_commit 8 &&
+       git branch commits/8 &&
+       ls $packdir | grep idx >existing-idx &&
+       git repack &&
+       ls $packdir | grep idx | grep -v --file=existing-idx >new-idx'
+
+# Current graph structure:
+#
+#      8
+#      |
+#   __M3___
+#  /   |   \
+# 3 M1 5 M2 7
+# |/  \|/  \|
+# 2    4    6
+# |___/____/
+# 1
+
+test_expect_success 'write graph with new commit' '
+       graph3=$(git commit-graph write) &&
+       test_path_is_file $packdir/graph-$graph3.graph'
+
+
+test_expect_success 'write graph with nothing new' '
+       graph4=$(git commit-graph write) &&
+       test_path_is_file $packdir/graph-$graph4.graph &&
+       printf $graph3 >expect &&
+       printf $graph4 >output &&
+       test_cmp expect output'
+
+test_expect_success 'setup bare repo' '
+       cd .. &&
+       git clone --bare --no-local full bare &&
+       cd bare &&
+       baredir="./objects/pack"'
+
+test_expect_success 'write graph in bare repo' '
+       graphbare=$(git commit-graph write) &&
+       test_path_is_file $baredir/graph-$graphbare.graph'
+
+test_done
+
-- 
2.15.1.45.g9b7079f

Reply via email to