]> Sergey Matveev's repositories - public-inbox.git/blob - lib/PublicInbox/LeiExternal.pm
lei query + pagination sorta working
[public-inbox.git] / lib / PublicInbox / LeiExternal.pm
1 # Copyright (C) 2020-2021 all contributors <meta@public-inbox.org>
2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
3
4 # *-external commands of lei
5 package PublicInbox::LeiExternal;
6 use strict;
7 use v5.10.1;
8 use parent qw(Exporter);
9 our @EXPORT = qw(lei_ls_external lei_add_external lei_forget_external);
10
11 sub _externals_each {
12         my ($self, $cb, @arg) = @_;
13         my $cfg = $self->_lei_cfg(0);
14         my %boost;
15         for my $sec (grep(/\Aexternal\./, @{$cfg->{-section_order}})) {
16                 my $loc = substr($sec, length('external.'));
17                 $boost{$loc} = $cfg->{"$sec.boost"};
18         }
19         return \%boost if !wantarray && !$cb;
20
21         # highest boost first, but stable for alphabetic tie break
22         use sort 'stable';
23         my @order = sort { $boost{$b} <=> $boost{$a} } sort keys %boost;
24         return @order if !$cb;
25         for my $loc (@order) {
26                 $cb->(@arg, $loc, $boost{$loc});
27         }
28         @order; # scalar or array
29 }
30
31 sub lei_ls_external {
32         my ($self, @argv) = @_;
33         my $stor = $self->_lei_store(0);
34         my $out = $self->{1};
35         my ($OFS, $ORS) = $self->{opt}->{z} ? ("\0", "\0\0") : (" ", "\n");
36         $self->_externals_each(sub {
37                 my ($loc, $boost_val) = @_;
38                 print $out $loc, $OFS, 'boost=', $boost_val, $ORS;
39         });
40 }
41
42 sub lei_add_external {
43         my ($self, $url_or_dir) = @_;
44         my $cfg = $self->_lei_cfg(1);
45         if ($url_or_dir !~ m!\Ahttps?://!) {
46                 $url_or_dir = File::Spec->canonpath($url_or_dir);
47         }
48         my $new_boost = $self->{opt}->{boost} // 0;
49         my $key = "external.$url_or_dir.boost";
50         my $cur_boost = $cfg->{$key};
51         return if defined($cur_boost) && $cur_boost == $new_boost; # idempotent
52         $self->lei_config($key, $new_boost);
53         my $stor = $self->_lei_store(1);
54         # TODO: add to MiscIdx
55         $stor->done;
56 }
57
58 sub lei_forget_external {
59         # TODO
60 }
61
62 1;