notice-sender/trunk/sender.pl

#!/usr/bin/perl -w

use strict;
use Class::DBI::Loader::Pg;
use Getopt::Long;
use Mail::CheckUser qw(check_email);
use Email::Valid;

=head1 NAME

sender.pl - command line notify sender utility

=head1 SYNOPSYS

 sender.pl --add=mylist members.txt
 sender.pl --list[=mylist]
 sender.pl --queue=mylist message.txt
 sender.pl --send=mylist

=head2 All options

=over 20

=item --debug

Turn on debugging output from C<Class::DBI>

=cut

my $debug = 0;
my $list_opt;
my $add_opt;
my $queue_opt;

my $result = GetOptions(
        "list:s" => \$list_opt,
        "add=s" => \$add_opt,
        "queue=s" => \$queue_opt,
        "debug" => \$debug,
);


my $loader = Class::DBI::Loader::Pg->new(
        debug           => $debug,
        dsn             => "dbi:Pg:dbname=notices",
        user            => "dpavlin",
        password        => "",
        namespace       => "Noticer",
#       additional_classes      => qw/Class::DBI::AbstractSearch/,
#       additional_base_classes => qw/My::Stuff/,
        relationships   => 1,
);

my $lists = $loader->find_class('lists');
my $users = $loader->find_class('users');
my $user_list = $loader->find_class('user_list');
my $messages = $loader->find_class('messages');
my $queue = $loader->find_class('queue');

=item --list[=list_name]

List all available lists and users on them. Optional value is name of list
and it will produce users just on that list.

=cut

if (defined($list_opt)) {
        my @lists;
        if ($list_opt ne '') {
                @lists = $lists->search( name=> $list_opt )->first || die "can't find list $list_opt";
        } else {
                @lists = $lists->retrieve_all;
        }

        foreach my $list (@lists) {
                print $list->name,"\n";
                foreach my $user_on_list ($user_list->search(list_id => $list->id)) {
                        my $user = $users->retrieve( id => $user_on_list->user_id );
                        print "\t",$user->full_name," <", $user->email, ">\n";
                }
        }

=item --add=list_name

Add users to list. Users are stored in file (which can be supplied as
argument) or read from C<STDIN>. List should be in following format:

 email@example.com      Optional full name of person
 dpavlin@rot13.org      Dobrica Pavlinusic

=cut

} elsif ($add_opt) {
        #my $noticer = $loader->find_class('Noticer') || die "can't find my class!";
        my $list = $lists->find_or_create({
                name => $add_opt,
        }) || die "can't add list $add_opt\n";

        my $added = 0;

        while(<>) {
                chomp;
                next if (/^#/ || /^\s*$/);
                my ($email, $name) = split(/\s+/,$_, 2);
                if (! Email::Valid->address($email)) {
                        print "SKIPPING $name <$email>\n";
                        next;
                }
                print "# $name <$email>\n";
                my $this_user = $users->find_or_create({
                        email => $email,
                        full_name => $name,
                }) || die "can't find or create member\n";
                my $user_on_list = $user_list->find_or_create({
                        user_id => $this_user->id,
                        list_id => $list->id,
                }) || die "can't add user to list";
                $added++;
        }

        foreach my $c_name ($loader->tables) {
                my $c = $loader->find_class($c_name)|| die "can't find $c_name";
                $c->dbi_commit();
        }

        print "list ",$list->name," has $added users\n";

=item --queue=list_name

Queue message for later delivery. Message can be read from file (specified as
argument) or read from C<STDIN>.

=cut

} elsif ($queue_opt) {
        my $this_list = $lists->search(
                name => $queue_opt,
        )->first || die "can't find list $queue_opt";

        my $message_text;
        while(<>) {
                $message_text .= $_;
        }

        die "no message" unless ($message_text);

        my $this_message = $messages->find_or_create({
                message => $message_text
        }) || die "can't insert message";

        $this_message->dbi_commit();

        $queue->find_or_create({
                message_id => $this_message->id,
                list_id => $this_list->id,
        }) || die "can't add message ",$this_message->id," to list ",$this_list->id, ": ",$this_list->name;

        print "added message ",$this_message->id, " to list ",$this_list->name,"\n";

} else  {
        die "see perldoc $0 for help";
}

=back

=head1 AUTHOR

Dobrica Pavlinusic <dpavlin@rot13.org>

=cut

1	dpavlin	1	#!/usr/bin/perl -w
2
3			use strict;
4			use Class::DBI::Loader::Pg;
5			use Getopt::Long;
6	dpavlin	13	use Mail::CheckUser qw(check_email);
7			use Email::Valid;
8	dpavlin	1
9	dpavlin	6	=head1 NAME
10
11			sender.pl - command line notify sender utility
12
13	dpavlin	8	=head1 SYNOPSYS
14
15			sender.pl --add=mylist members.txt
16	dpavlin	9	sender.pl --list[=mylist]
17	dpavlin	8	sender.pl --queue=mylist message.txt
18			sender.pl --send=mylist
19
20			=head2 All options
21
22			=over 20
23
24			=item --debug
25
26			Turn on debugging output from C<Class::DBI>
27
28	dpavlin	6	=cut
29
30	dpavlin	12	my $debug = 0;
31			my $list_opt;
32	dpavlin	2	my $add_opt;
33	dpavlin	6	my $queue_opt;
34	dpavlin	1
35			my $result = GetOptions(
36	dpavlin	9	"list:s" => \$list_opt,
37	dpavlin	2	"add=s" => \$add_opt,
38	dpavlin	6	"queue=s" => \$queue_opt,
39	dpavlin	1	"debug" => \$debug,
40			);
41
42
43			my $loader = Class::DBI::Loader::Pg->new(
44			debug => $debug,
45			dsn => "dbi:Pg:dbname=notices",
46			user => "dpavlin",
47			password => "",
48			namespace => "Noticer",
49			# additional_classes => qw/Class::DBI::AbstractSearch/,
50			# additional_base_classes => qw/My::Stuff/,
51	dpavlin	3	relationships => 1,
52	dpavlin	1	);
53
54	dpavlin	2	my $lists = $loader->find_class('lists');
55			my $users = $loader->find_class('users');
56			my $user_list = $loader->find_class('user_list');
57	dpavlin	6	my $messages = $loader->find_class('messages');
58	dpavlin	11	my $queue = $loader->find_class('queue');
59	dpavlin	2
60	dpavlin	9	=item --list[=list_name]
61	dpavlin	8
62	dpavlin	10	List all available lists and users on them. Optional value is name of list
63			and it will produce users just on that list.
64	dpavlin	8
65			=cut
66
67	dpavlin	9	if (defined($list_opt)) {
68			my @lists;
69			if ($list_opt ne '') {
70			@lists = $lists->search( name=> $list_opt )->first \|\| die "can't find list $list_opt";
71			} else {
72			@lists = $lists->retrieve_all;
73			}
74
75			foreach my $list (@lists) {
76	dpavlin	1	print $list->name,"\n";
77			foreach my $user_on_list ($user_list->search(list_id => $list->id)) {
78			my $user = $users->retrieve( id => $user_on_list->user_id );
79			print "\t",$user->full_name," <", $user->email, ">\n";
80			}
81			}
82	dpavlin	8
83			=item --add=list_name
84
85			Add users to list. Users are stored in file (which can be supplied as
86			argument) or read from C<STDIN>. List should be in following format:
87
88			email@example.com Optional full name of person
89			dpavlin@rot13.org Dobrica Pavlinusic
90
91			=cut
92
93	dpavlin	2	} elsif ($add_opt) {
94			#my $noticer = $loader->find_class('Noticer') \|\| die "can't find my class!";
95			my $list = $lists->find_or_create({
96			name => $add_opt,
97			}) \|\| die "can't add list $add_opt\n";
98	dpavlin	6
99			my $added = 0;
100
101	dpavlin	2	while(<>) {
102			chomp;
103			next if (/^#/ \|\| /^\s*$/);
104			my ($email, $name) = split(/\s+/,$_, 2);
105	dpavlin	13	if (! Email::Valid->address($email)) {
106			print "SKIPPING $name <$email>\n";
107			next;
108			}
109	dpavlin	2	print "# $name <$email>\n";
110			my $this_user = $users->find_or_create({
111			email => $email,
112			full_name => $name,
113			}) \|\| die "can't find or create member\n";
114			my $user_on_list = $user_list->find_or_create({
115			user_id => $this_user->id,
116			list_id => $list->id,
117			}) \|\| die "can't add user to list";
118	dpavlin	6	$added++;
119	dpavlin	2	}
120	dpavlin	3
121			foreach my $c_name ($loader->tables) {
122			my $c = $loader->find_class($c_name)\|\| die "can't find $c_name";
123			$c->dbi_commit();
124			}
125
126	dpavlin	6	print "list ",$list->name," has $added users\n";
127
128	dpavlin	8	=item --queue=list_name
129
130			Queue message for later delivery. Message can be read from file (specified as
131			argument) or read from C<STDIN>.
132
133			=cut
134
135	dpavlin	6	} elsif ($queue_opt) {
136			my $this_list = $lists->search(
137			name => $queue_opt,
138			)->first \|\| die "can't find list $queue_opt";
139
140			my $message_text;
141			while(<>) {
142			$message_text .= $_;
143			}
144
145			die "no message" unless ($message_text);
146
147			my $this_message = $messages->find_or_create({
148			message => $message_text
149			}) \|\| die "can't insert message";
150
151			$this_message->dbi_commit();
152
153	dpavlin	11	$queue->find_or_create({
154	dpavlin	6	message_id => $this_message->id,
155			list_id => $this_list->id,
156			}) \|\| die "can't add message ",$this_message->id," to list ",$this_list->id, ": ",$this_list->name;
157
158			print "added message ",$this_message->id, " to list ",$this_list->name,"\n";
159
160	dpavlin	1	} else {
161	dpavlin	8	die "see perldoc $0 for help";
162	dpavlin	1	}
163
164	dpavlin	8	=back
165
166			=head1 AUTHOR
167
168			Dobrica Pavlinusic <dpavlin@rot13.org>
169
170			=cut
171