]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/Config.pm
view: implement optional address obfuscation
[public-inbox.git] / lib / PublicInbox / Config.pm
index 876e5d0239d2797c8e600f783a6cdadcc6bba690..0597a527baf7b0aa0a58d61d1187d09da9d69252 100644 (file)
@@ -1,45 +1,39 @@
-# Copyright (C) 2014, Eric Wong <normalperson@yhbt.net> and all contributors
+# Copyright (C) 2014-2015 all contributors <meta@public-inbox.org>
 # License: AGPLv3 or later (https://www.gnu.org/licenses/agpl-3.0.txt)
+#
+# Used throughout the project for reading configuration
 package PublicInbox::Config;
 use strict;
 use warnings;
-use File::Path::Expand qw/expand_filename/;
-use IPC::Run;
+require PublicInbox::Inbox;
+use PublicInbox::Spawn qw(popen_rd);
 
 # returns key-value pairs of config directives in a hash
 # if keys may be multi-value, the value is an array ref containing all values
 sub new {
        my ($class, $file) = @_;
-       my ($in, $out);
-
        $file = default_file() unless defined($file);
-       IPC::Run::run([qw/git config --file/, $file, '-l'], \$in, \$out);
-       $? == 0 or die "git config --file $file -l failed: $?\n";
-       my %rv;
-       foreach my $line (split(/\n/, $out)) {
-               my ($k, $v) = split(/=/, $line, 2);
-               my $cur = $rv{$k};
+       $file = ref $file ? $file : git_config_dump($file);
+       my $self = bless $file, $class;
 
-               if (defined $cur) {
-                       if (ref($cur) eq "ARRAY") {
-                               push @$cur, $v;
-                       } else {
-                               $rv{$k} = [ $cur, $v ];
-                       }
-               } else {
-                       $rv{$k} = $v;
-               }
-       }
-       bless \%rv, $class;
+       # caches
+       $self->{-by_addr} ||= {};
+       $self->{-by_name} ||= {};
+       $self->{-by_newsgroup} ||= {};
+       $self->{-limiters} ||= {};
+       $self;
 }
 
 sub lookup {
        my ($self, $recipient) = @_;
        my $addr = lc($recipient);
+       my $inbox = $self->{-by_addr}->{$addr};
+       return $inbox if $inbox;
+
        my $pfx;
 
        foreach my $k (keys %$self) {
-               $k =~ /\A(publicinbox\.[A-Z0-9a-z-]+)\.address\z/ or next;
+               $k =~ /\A(publicinbox\.[\w-]+)\.address\z/ or next;
                my $v = $self->{$k};
                if (ref($v) eq "ARRAY") {
                        foreach my $alias (@$v) {
@@ -53,33 +47,135 @@ sub lookup {
                        last;
                }
        }
-
        defined $pfx or return;
+       _fill($self, $pfx);
+}
 
-       my %rv;
-       foreach my $k (qw(mainrepo address)) {
-               my $v = $self->{"$pfx.$k"};
-               $rv{$k} = $v if defined $v;
+sub lookup_name ($$) {
+       my ($self, $name) = @_;
+       $self->{-by_name}->{$name} || _fill($self, "publicinbox.$name");
+}
+
+sub each_inbox {
+       my ($self, $cb) = @_;
+       my %seen;
+       foreach my $k (keys %$self) {
+               $k =~ /\Apublicinbox\.([A-Z0-9a-z-]+)\.mainrepo\z/ or next;
+               next if $seen{$1};
+               $seen{$1} = 1;
+               my $ibx = lookup_name($self, $1) or next;
+               $cb->($ibx);
        }
-       my $listname = $pfx;
-       $listname =~ s/\Apublicinbox\.//;
-       $rv{listname} = $listname;
-       my $v = $rv{address};
-       $rv{-primary_address} = ref($v) eq 'ARRAY' ? $v->[0] : $v;
-       \%rv;
 }
 
-sub get {
-       my ($self, $listname, $key) = @_;
+sub lookup_newsgroup {
+       my ($self, $ng) = @_;
+       $ng = lc($ng);
+       my $rv = $self->{-by_newsgroup}->{$ng};
+       return $rv if $rv;
 
-       $self->{"publicinbox.$listname.$key"};
+       foreach my $k (keys %$self) {
+               $k =~ /\A(publicinbox\.[\w-]+)\.newsgroup\z/ or next;
+               my $v = $self->{$k};
+               my $pfx = $1;
+               if ($v eq $ng) {
+                       $rv = _fill($self, $pfx);
+                       return $rv;
+               }
+       }
+       undef;
 }
 
+sub limiter {
+       my ($self, $name) = @_;
+       $self->{-limiters}->{$name} ||= do {
+               require PublicInbox::Qspawn;
+               my $max = $self->{"publicinboxlimiter.$name.max"};
+               PublicInbox::Qspawn::Limiter->new($max);
+       };
+}
+
+sub config_dir { $ENV{PI_DIR} || "$ENV{HOME}/.public-inbox" }
+
 sub default_file {
        my $f = $ENV{PI_CONFIG};
        return $f if defined $f;
-       my $pi_dir = $ENV{PI_DIR} || expand_filename('~/.public-inbox/');
-       "$pi_dir/config";
+       config_dir() . '/config';
+}
+
+sub git_config_dump {
+       my ($file) = @_;
+       my ($in, $out);
+       my @cmd = (qw/git config/, "--file=$file", '-l');
+       my $cmd = join(' ', @cmd);
+       my $fh = popen_rd(\@cmd) or die "popen_rd failed for $file: $!\n";
+       my %rv;
+       local $/ = "\n";
+       while (defined(my $line = <$fh>)) {
+               chomp $line;
+               my ($k, $v) = split(/=/, $line, 2);
+               my $cur = $rv{$k};
+
+               if (defined $cur) {
+                       if (ref($cur) eq "ARRAY") {
+                               push @$cur, $v;
+                       } else {
+                               $rv{$k} = [ $cur, $v ];
+                       }
+               } else {
+                       $rv{$k} = $v;
+               }
+       }
+       close $fh or die "failed to close ($cmd) pipe: $?";
+       \%rv;
+}
+
+sub _fill {
+       my ($self, $pfx) = @_;
+       my $rv = {};
+
+       foreach my $k (qw(mainrepo address filter url newsgroup
+                       infourl watch watchheader httpbackendmax
+                       replyto feedmax nntpserver)) {
+               my $v = $self->{"$pfx.$k"};
+               $rv->{$k} = $v if defined $v;
+       }
+       foreach my $k (qw(obfuscate)) {
+               my $v = $self->{"$pfx.$k"};
+               defined $v or next;
+               if ($v =~ /\A(?:false|no|off|0)\z/) {
+                       $rv->{$k} = 0;
+               } elsif ($v =~ /\A(?:true|yes|on|1)\z/) {
+                       $rv->{$k} = 1;
+               } else {
+                       warn "Ignoring $pfx.$k=$v in config, not boolean\n";
+               }
+       }
+
+       # TODO: more arrays, we should support multi-value for
+       # more things to encourage decentralization
+       foreach my $k (qw(altid nntpmirror)) {
+               if (defined(my $v = $self->{"$pfx.$k"})) {
+                       $rv->{$k} = ref($v) eq 'ARRAY' ? $v : [ $v ];
+               }
+       }
+
+       return unless $rv->{mainrepo};
+       my $name = $pfx;
+       $name =~ s/\Apublicinbox\.//;
+       $rv->{name} = $name;
+       $rv->{-pi_config} = $self;
+       $rv = PublicInbox::Inbox->new($rv);
+       my $v = $rv->{address};
+       if (ref($v) eq 'ARRAY') {
+               $self->{-by_addr}->{lc($_)} = $rv foreach @$v;
+       } else {
+               $self->{-by_addr}->{lc($v)} = $rv;
+       }
+       if (my $ng = $rv->{newsgroup}) {
+               $self->{-by_newsgroup}->{$ng} = $rv;
+       }
+       $self->{-by_name}->{$name} = $rv;
 }
 
 1;