more flexible git-mr

1 files changed, 15 insertions, 5 deletions

diff --git a/bin/git-mr b/bin/git-mr
index a71f50eb..bb53d7b2 100755
--- a/bin/git-mr
+++ b/bin/git-mr
@@ -1,10 +1,20 @@
 #!/usr/bin/perl
 
-use 5.028;
 use strict;
 use warnings;
 
-die "usage: git mr REMOTE_NAME MERGE_REQUEST_ID BRANCH_NAME\n"
-  unless @ARGV == 3;
-system "git fetch $ARGV[0] merge-requests/$ARGV[1]/head";
-system "git checkout -b mr/$ARGV[2] FETCH_HEAD";
+my ($remote, $id, $branch);
+if (@ARGV == 1 and $ARGV[0] =~ /^[0-9]+$/) {
+    $remote = "origin";
+    $id = $branch = shift @ARGV;
+} elsif (@ARGV == 2) {
+    $remote = shift @ARGV;
+    $id = $branch = shift @ARGV;
+} elsif (@ARGV == 3) {
+    ($remote, $id, $branch) = @ARGV;
+} else {
+    die "usage: git mr [REMOTE_NAME] MERGE_REQUEST_ID [BRANCH_NAME]\n";
+}
+
+system "git fetch $remote merge-requests/$id/head";
+system "git checkout -b mr/$branch FETCH_HEAD";