]> 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 9ba4ad32901f501d011c6a88f531914471d98e09..0597a527baf7b0aa0a58d61d1187d09da9d69252 100644 (file)
-# 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/;
+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) = @_;
+       $file = default_file() unless defined($file);
+       $file = ref $file ? $file : git_config_dump($file);
+       my $self = bless $file, $class;
 
-       local $ENV{GIT_CONFIG} = defined $file ? $file : default_file();
-
-       my @cfg = `git config -l`;
-       $? == 0 or die "git config -l failed: $?\n";
-       chomp @cfg;
-       my %rv = map { split(/=/, $_, 2) } @cfg;
-       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;
-               (lc($self->{$k}) eq $addr) or next;
-               $pfx = $1;
-               last;
+               $k =~ /\A(publicinbox\.[\w-]+)\.address\z/ or next;
+               my $v = $self->{$k};
+               if (ref($v) eq "ARRAY") {
+                       foreach my $alias (@$v) {
+                               (lc($alias) eq $addr) or next;
+                               $pfx = $1;
+                               last;
+                       }
+               } else {
+                       (lc($v) eq $addr) or next;
+                       $pfx = $1;
+                       last;
+               }
        }
-
        defined $pfx or return;
+       _fill($self, $pfx);
+}
 
-       my %rv = map {
-               $_ => $self->{"$pfx.$_"}
-       } (qw(mainrepo failrepo description address));
-       \%rv;
+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);
+       }
+}
+
+sub lookup_newsgroup {
+       my ($self, $ng) = @_;
+       $ng = lc($ng);
+       my $rv = $self->{-by_newsgroup}->{$ng};
+       return $rv if $rv;
+
+       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;