1 files changed, 6 insertions, 4 deletions
diff --git a/lib/PublicInbox/Import.pm b/lib/PublicInbox/Import.pm
index 7258e848..60cff9c2 100644
--- a/lib/PublicInbox/Import.pm
+++ b/lib/PublicInbox/Import.pm
@@ -463,16 +463,18 @@ EOD
  EOC
  
  sub init_bare {
-        my ($dir) = @_; # or self
+        my ($dir, $head) = @_; # or self
          $dir = $dir->{git}->{git_dir} if ref($dir);
          require File::Path;
          File::Path::mkpath([ map { "$dir/$_" } qw(objects/info refs/heads) ]);
          $INIT_FILES[1] //= 'ref: '.default_branch."\n";
-        for (my $i = 0; $i < @INIT_FILES; $i++) {
-                my $f = $dir.'/'.$INIT_FILES[$i++];
+        my @fn_contents = @INIT_FILES;
+        $fn_contents[1] = "ref: refs/heads/$head\n" if defined $head;
+        while (my ($fn, $contents) = splice(@fn_contents, 0, 2)) {
+                my $f = $dir.'/'.$fn;
                  next if -f $f;
                  open my $fh, '>', $f or die "open $f: $!";
-                print $fh $INIT_FILES[$i] or die "print $f: $!";
+                print $fh $contents or die "print $f: $!";
                  close $fh or die "close $f: $!";
          }
  }