2 # Copyright (C) 2018-2020 all contributors <meta@public-inbox.org>
3 # License: AGPL-3.0+ <http://www.gnu.org/licenses/agpl-3.0.txt>
6 use Getopt::Long qw(:config gnu_getopt no_ignore_case auto_abbrev);
7 my $usage = 'Usage: public-inbox-convert [options] OLD NEW';
8 my $help = <<EOF; # the following should fit w/o scrolling in 80x24 term:
11 convert v1 format inboxes to v2
15 --no-index do not index after conversion
16 --jobs=NUM set shards (NUM=0)
17 --verbose | -v increase verbosity (may be repeated)
18 --help | -? show this help
20 index options (see public-inbox-index(1) manpage for full description):
22 --no-fsync speed up indexing, risk corruption on power outage
23 -L LEVEL `basic', `medium', or `full' (default: full)
24 --compact | -c run public-inbox-compact(1) after indexing
25 --sequential-shard index Xapian shards sequentially for slow storage
26 --batch-size=BYTES flush changes to OS after a given number of bytes
27 --max-size=BYTES do not index messages larger than the given size
29 See public-inbox-convert(1) man page for full documentation.
35 quiet => -1, compact => 0, maxsize => undef, fsync => 1,
36 reindex => 1, # we always reindex
38 GetOptions($opt, qw(jobs|j=i index! help|?),
40 qw(verbose|v+ rethread compact|c+ fsync|sync!
41 indexlevel|index-level|L=s max_size|max-size=s
42 batch_size|batch-size=s
43 sequential_shard|sequential-shard|seq-shard
45 bad command-line args\n$usage
47 if ($opt->{help}) { print $help; exit 0 };
48 my $old_dir = shift(@ARGV) // '';
49 my $new_dir = shift(@ARGV) // '';
50 die $usage if (scalar(@ARGV) || $new_dir eq '' || $old_dir eq '');
51 die "$new_dir exists\n" if -d $new_dir;
52 die "$old_dir not a directory\n" unless -d $old_dir;
55 Cwd->import('abs_path');
56 require PublicInbox::Config;
57 require PublicInbox::InboxWritable;
59 $old_dir = abs_path($old_dir);
60 my $cfg = PublicInbox::Config->new;
62 $cfg->each_inbox(sub {
63 $old = $_[0] if abs_path($_[0]->{inboxdir}) eq $old_dir;
66 $old = PublicInbox::InboxWritable->new($old);
68 warn "W: $old_dir not configured in " .
69 PublicInbox::Config::default_file() . "\n";
70 $old = PublicInbox::InboxWritable->new({
73 -primary_address => 'old@example.com',
74 address => [ 'old@example.com' ],
77 die "Only conversion from v1 inboxes is supported\n" if $old->version >= 2;
79 require PublicInbox::Admin;
80 my $detected = PublicInbox::Admin::detect_indexlevel($old);
81 $old->{indexlevel} //= $detected;
83 if ($opt->{'index'}) {
85 PublicInbox::Admin::scan_ibx_modules($mods, $old);
86 PublicInbox::Admin::require_or_die(keys %$mods);
87 PublicInbox::Admin::progress_prepare($opt);
88 $env = PublicInbox::Admin::index_prepare($opt, $cfg);
90 local %ENV = (%$env, %ENV) if $env;
92 $new->{inboxdir} = abs_path($new_dir);
94 $new = PublicInbox::InboxWritable->new($new, { nproc => $opt->{jobs} });
95 $new->{-no_fsync} = 1 if !$opt->{fsync};
99 sub link_or_copy ($$) {
100 my ($src, $dst) = @_;
101 link($src, $dst) and return;
102 $!{EXDEV} or warn "link $src, $dst failed: $!, trying cp\n";
103 require File::Copy; # preserves permissions:
104 File::Copy::cp($src, $dst) or die "cp $src, $dst failed: $!\n";
107 $old->with_umask(sub {
108 my $old_cfg = "$old->{inboxdir}/config";
109 local $ENV{GIT_CONFIG} = $old_cfg;
110 my $new_cfg = "$new->{inboxdir}/all.git/config";
111 $v2w = $new->importer(1);
112 $v2w->init_inbox(delete $opt->{jobs});
114 link_or_copy($old_cfg, $new_cfg);
115 if (my $alt = $new->{altid}) {
116 require PublicInbox::AltId;
117 foreach my $i (0..$#$alt) {
118 my $src = PublicInbox::AltId->new($old, $alt->[$i], 0);
119 $src = $src->mm_alt or next;
120 $src = $src->{dbh}->sqlite_db_filename;
121 my $dst = PublicInbox::AltId->new($new, $alt->[$i], 1);
122 $dst->mm_alt->{dbh}->sqlite_backup_from_file($src);
125 my $desc = "$old->{inboxdir}/description";
126 link_or_copy($desc, "$new->{inboxdir}/description") if -e $desc;
127 my $clone = "$old->{inboxdir}/cloneurl";
130 $clone may not be valid after migrating to v2, not copying
135 my $head = $old->{ref_head} || 'HEAD';
136 my ($rd, $pid) = $old->git->popen(qw(fast-export --use-done-feature), $head);
137 $v2w->idx_init($opt);
138 my $im = $v2w->importer;
139 my ($r, $w) = $im->gfi_start;
144 if ($_ eq "blob\n") {
146 } elsif (/^commit /) {
148 } elsif (/^data ([0-9]+)/) {
150 print $w $_ or $im->wfail;
152 my $n = read($rd, my $tmp, $len) or die "read: $!";
153 warn "$n != $len\n" if $n != $len;
155 print $w $tmp or $im->wfail;
158 } elsif ($state eq 'commit') {
159 if (m{^M 100644 :([0-9]+) (${h}{2}/${h}{38})}o) {
160 my ($mark, $path) = ($1, $2);
162 if ($last && $last ne 'm') {
163 print $w "D $last\n" or $im->wfail;
165 print $w "M 100644 :$mark m\n" or $im->wfail;
169 if (m{^D (${h}{2}/${h}{38})}o) {
170 my $mark = delete $D{$1};
171 defined $mark or die "undeleted path: $1\n";
172 if ($last && $last ne 'd') {
173 print $w "D $last\n" or $im->wfail;
175 print $w "M 100644 :$mark d\n" or $im->wfail;
180 last if $_ eq "done\n";
181 print $w $_ or $im->wfail;
183 close $rd or die "close fast-export: $!\n";
184 waitpid($pid, 0) or die "waitpid failed: $!\n";
185 $? == 0 or die "fast-export failed: $?\n";
186 $r = $w = undef; # v2w->done does the actual close and error checking
188 if (my $old_mm = $old->mm) {
190 $old_mm = $old_mm->{dbh}->sqlite_db_filename;
192 # we want to trigger a reindex, not a from scratch index if
193 # we're reusing the msgmap from an existing v1 installation.
194 $v2w->idx_init($opt);
195 $v2w->{mm}->{dbh}->sqlite_backup_from_file($old_mm);
197 my $epoch0 = PublicInbox::Git->new($v2w->git_init(0));
198 chop(my $cmt = $epoch0->qx(qw(rev-parse --verify), $head));
199 $v2w->last_epoch_commit(0, $cmt);
201 $v2w->index_sync($opt) if delete $opt->{'index'};