prefer git_pathdup to git_path in some possibly-dangerous cases
[gitweb.git] / builtin / worktree.c
index 323d444b35cd465f5c7c61d4d935475c562be0d3..6a264ee749221cd6c6bf98ce68790bdec3dd82c5 100644 (file)
@@ -5,6 +5,7 @@
 #include "argv-array.h"
 #include "run-command.h"
 #include "sigchain.h"
+#include "refs.h"
 
 static const char * const worktree_usage[] = {
        N_("git worktree add [<options>] <path> <branch>"),
@@ -291,12 +292,16 @@ static int add(int ac, const char **av, const char *prefix)
                die(_("-b and -B are mutually exclusive"));
        if (ac < 1 || ac > 2)
                usage_with_options(worktree_usage, options);
-       if (ac < 2 && !new_branch && !new_branch_force)
-               usage_with_options(worktree_usage, options);
 
        path = prefix ? prefix_filename(prefix, strlen(prefix), av[0]) : av[0];
        branch = ac < 2 ? "HEAD" : av[1];
 
+       if (ac < 2 && !new_branch && !new_branch_force) {
+               int n;
+               const char *s = worktree_basename(path, &n);
+               new_branch = xstrndup(s, n);
+       }
+
        argv_array_push(&cmd, "checkout");
        if (force)
                argv_array_push(&cmd, "--ignore-other-worktrees");