#!/usr/bin/perl
-package Mubot4FB;
-
+#
+# Copyright (c) 2012 ISHIKAWA Mutsumi <ishikawa@hanzubon.jp>
+# This program is covered by the GNU General Public License 2
+#
+package Mubot4FB::FB;
use strict;
use utf8;
use Facebook::Graph;
-use base 'Bot::BasicBot';
-use Encode;
-use DBI qw/:sql_types/;
-use POSIX 'strftime';
+use LWP::UserAgent;
+use HTTP::Request::Common;
use Data::Dumper;
-my $mu_re = qr/^([^ ]+) (https?:\/\/[^ ]+) (.+)$/i;
-my $irc_type = 1;
+sub new {
+ my $proto = shift;
+ my $class = ref $proto || $proto;
+ my $self = {cfg => shift};
+ bless $self, $class;
+
+ $self->init();
+ return $self;
+}
-sub fb_init {
+sub init {
my ($me) = @_;
my $fb = Facebook::Graph->new(app_id => $me->{cfg}->{fb_app_id},
secret => $me->{cfg}->{fb_app_secret},
return $me->{fbo} = Facebook::Graph->new(access_token => $page_access_token);
}
-sub db_init {
- my ($me) = @_;
- $me->{dbh} = DBI->connect('DBI:mysql:'.$me->{cfg}->{database}, $me->{cfg}->{db_user}, $me->{cfg}->{db_pass}) || die $DBI::errstr;
-}
-
-sub misc_init {
- my ($me) = @_;
-
- $me->{last_search} = {};
-}
-
sub publish {
my ($me, $text, $uri) = @_;
->as_hashref();
}
-sub init {
- my ($me) = @_;
- $me->fb_init();
- $me->db_init();
+sub post_uri {
+ my ($me, $post_id) = @_;
+
+ return $me->{cfg}->{fb_page_url} . 'posts/' . $post_id;
}
-sub _check_dup {
- my ($me, $args, $uri) = @_;
+sub remove {
+ my ($me, $post_id) = @_;
- my $found = 0;
+ my $uri = $me->{fbo}->query->find($me->{cfg}->{fb_page_id}.'_'.$post_id)->uri_as_string;
+ my $req = HTTP::Request::Common::DELETE($uri);
+ warn Dumper($req) if ($me->{cfg}->{debug});
+ my $resp;
+ $resp = LWP::UserAgent->new->request($req);
+ warn Dumper($resp) if ($me->{cfg}->{debug});
+ if ($resp->is_success && $resp->code == 200 && $resp->content eq 'true') {
+ return 1;
+ } else {
+ warn 'DELETE ERROR: http code: ' . $resp->code() . ' , http content: ' . $resp->content;
+ return 0;
+ }
+}
+1;
+
+package Mubot4FB::FB;
+use strict;
+use utf8;
+
+use DBI qw/:sql_types/;
+
+use Data::Dumper;
+
+sub new {
+ my $proto = shift;
+ my $class = ref $proto || $proto;
+ my $self = {cfg => shift};
+ bless $self, $class;
+
+ $self->init();
+ return $self;
+}
+
+sub init {
+ my ($me) = @_;
+ return $me->{dbh} = DBI->connect('DBI:mysql:'.$me->{cfg}->{database}, $me->{cfg}->{db_user}, $me->{cfg}->{db_pass},{mysql_enable_utf8 => 1}) || die $DBI::errstr;
+}
+
+sub check_dup {
+ my ($me, $db_args) = @_;
my $sth = $me->{dbh}->prepare('select * from posts where uri = ? order by post_time desc limit 1');
- my $rv = $sth->execute($uri);
- my $res = $sth->fetchrow_hashref;
- if ($res) {
- if ($res->{post_time} < time() - 7 * 24 * 60 * 60) {
- $me->_response($args, 'だいぶ前 '.$me->_format_submit($res).'にいってたにゃー '.$me->_fb_post_uri($res->{fb_post_id}));
- } else {
- $me->_response($args, '既に '.$me->_format_submit($res).'に言ってますよ? '.$me->{cfg}->{fb_page_url}.'posts/'.$res->{fb_post_id});
- $found = 1;
- }
- }
- $sth->finish;
+ my $rv = $sth->execute($db_args->{uri});
+ my $ret = $sth->fetchrow_hashref;
- return $found;
-};
+ $sth->finish;
+ return $ret;
+}
-sub _db_insert {
+sub add {
my ($me, $db_args) = @_;
- my $sth = $me->{dbh}->prepare("insert into posts (submitter, fb_post_id, uri, prefix, suffix, post_time) values (?, ?, ?, ?, ?, ?)");
+ my ($scheme, $path) = split(/:\/\//, $db_args->{uri});
+ my $sth = $me->{dbh}->prepare("insert into posts (submitter, fb_post_id, uri, prefix, comment, scheme, path, post_time) values (?, ?, ?, ?, ?, ?, ?, ?)");
$sth->bind_param(1, $db_args->{submitter}, SQL_VARCHAR);
- $sth->bind_param(2, $db_args->{fb_post_id}, SQL_INTEGER);
+ $sth->bind_param(2, $db_args->{fb_post_id}, SQL_BIGINT);
$sth->bind_param(3, $db_args->{uri}, SQL_VARCHAR);
$sth->bind_param(4, $db_args->{prefix}, SQL_VARCHAR);
- $sth->bind_param(5, $db_args->{suffix}, SQL_VARCHAR);
- $sth->bind_param(6, time, SQL_INTEGER);
+ $sth->bind_param(5, $db_args->{comment}, SQL_VARCHAR);
+ $sth->bind_param(6, $scheme, SQL_VARCHAR);
+ $sth->bind_param(7, $path, SQL_VARCHAR);
+ $sth->bind_param(8, time, SQL_BIGINT);
my $rv = $sth->execute();
$sth->finish;
return $rv;
}
-sub _db_delete {
+sub remove {
my ($me, $db_args) = @_;
- $db_args->{submitter_type} = 1 unless defined $db_args->{submitter_type};
+ $db_args->{submitter_type} ||= 1;
my $sth = $me->{dbh}->prepare("delete from posts where fb_post_id = ? and submitter = ? and submitter_type = ?");
- $sth->bind_param(1, $db_args->{fb_post_id}, SQL_INTEGER);
+ $sth->bind_param(1, $db_args->{fb_post_id}, SQL_BIGINT);
$sth->bind_param(2, $db_args->{submitter}, SQL_VARCHAR);
$sth->bind_param(3, $db_args->{submitter_type}, SQL_INTEGER);
my $rv = $sth->execute();
-
my $ret = $rv ? $sth->rows : 0;
$sth->finish;
return $ret;
}
-sub _db_search {
- my ($me, $word) = @_;
+sub search_by_word {
+ my ($me, $db_args) = @_;
- my $sth = $me->{dbh}->prepare('select * from posts where match(suffix) against(?) or match(prefix) against(?) or match(uri) against(?) order by post_time desc limit 1000');
- $sth->bind_param(1, $word, SQL_VARCHAR);
- $sth->bind_param(2, $word, SQL_VARCHAR);
- $sth->bind_param(3, $word, SQL_VARCHAR);
+ my $column = $db_args->{word} =~ /:\/\// ? 'uri' : 'path';
+ my $w = '%' . $db_args->{word} . '%';
+ my $sth = $me->{dbh}->prepare('select * from posts where prefix like ? or '.$column.' like ? or comment like ? order by post_time desc limit 1000');
+ $sth->bind_param(1, $w, SQL_VARCHAR);
+ $sth->bind_param(2, $w, SQL_VARCHAR);
+ $sth->bind_param(3, $w, SQL_VARCHAR);
$sth->execute();
my $ret = $sth->fetchall_arrayref({});
return $ret;
}
-sub _fb_post_uri {
- my ($me, $post_id) = @_;
+sub search_lastpost_by_submitter {
+ my ($me, $db_args) = @_;
- return $me->{cfg}->{fb_page_url} . 'posts/' . $post_id;
+ my $sth = $me->{dbh}->prepare('select * from posts where submitter = ? order by post_time desc limit 1');
+ $sth->bind_param(1, $db_args->{who}, SQL_VARCHAR);
+ $sth->execute();
+
+ my $ret = $sth->fetchrow_hashref();
+ $sth->finish;
+
+ return $ret;
+}
+
+1;
+package Mubot4FB;
+
+use strict;
+use utf8;
+
+use base 'Bot::BasicBot';
+use DBI qw/:sql_types/;
+use POSIX 'strftime';
+
+use Data::Dumper;
+
+my $mu_re = qr/^([^\s]+)\s+((?:https?|ftps?):\/\/[^\s]+)\s+(.+)$/i;
+my $irc_type = 1;
+
+sub misc_init {
+ my ($me) = @_;
+
+ $me->{last_search} = {};
+}
+
+sub init {
+ my ($me) = @_;
+ $me->{fb} = Mubot4FB::FB->new($me->{cfg});
+ $me->{db} = Mubot4FB::DB->new($me->{cfg});
}
sub _format_submit {
my ($me, $e) = @_;
- return decode('utf8', $e->{submitter}).'が「'.decode('utf8', $e->{prefix}).' '.decode('utf8', $e->{uri}).' '.decode('utf8', $e->{suffix}).'」と'.strftime('%Y-%m-%d %H:%M:%S', localtime($e->{post_time}));
+ return $e->{submitter}.'が『'.$e->{prefix}.' '.$e->{uri}.' '.$e->{comment}.'』と'.strftime('%Y-%m-%d %H:%M:%S', localtime($e->{post_time}));
}
sub _response {
sub _add {
my ($me, $args) =@_;
- my $post_ok = 1;
my ($resp, $resp_msg);
if ($args->{body} =~ /$mu_re/) {
my $prefix = $1;
my $uri = $2;
- my $suffix = $3;
- my $text = $args->{who} . '曰く、'.$prefix.' '.$suffix;
-
- return 0 if ($me->_check_dup($args, $uri));
-
- eval{$resp = $me->publish($text, $uri)};
- if ($@) {
- $me->fb_init();
- eval{$resp = $me->publish($text, $uri)};
- $post_ok = 0 if ($@);
- }
-
- if ($post_ok) {
- my (undef, $post_id) = split(/_/, $resp->{id});
- $me->_db_insert({submitter => $args->{who},
- fb_post_id => $post_id,
- uri => $uri,
- prefix => $prefix,
- suffix => $suffix});
- $resp_msg = $args->{who} . ': うい '.$me->_fb_post_uri($post_id).' で登録';
+ my $comment = $3;
+ my $text = $args->{who} . '曰く、'.$prefix.' '.$comment;
+
+ if (my $res = $me->{db}->check_dup({uri =>$uri})) {
+ if ($res->{post_time} < time() - 7 * 24 * 60 * 60) {
+ $resp_msg = 'だいぶ前 '.$me->_format_submit($res).'にいってたにゃー '.$me->{fb}->post_uri($res->{fb_post_id});
+ } else {
+ $resp_msg = '既に '.$me->_format_submit($res).'に言ってますよ? '.$me->{fb}->post_uri($res->{fb_post_id});
+ }
} else {
- $resp_msg = 'can not post to facebook';
- }
+ my $post_ok = 1;
+ eval{$resp = $me->{fb}->publish($text, $uri)};
+ if ($@) {
+ $me->{fb}->init();
+ eval{$resp = $me->{fb}->publish($text, $uri)};
+ $post_ok = 0 if ($@);
+ }
+ if ($post_ok) {
+ my (undef, $post_id) = split(/_/, $resp->{id});
+ $me->{db}->add({submitter => $args->{who},
+ fb_post_id => $post_id,
+ uri => $uri,
+ prefix => $prefix,
+ comment => $comment});
+ $resp_msg = $args->{who} . ': うい '.$me->{fb}->post_uri($post_id).' で登録';
+ } else {
+ $resp_msg = 'can not post to facebook';
+ }
+ }
return $resp_msg;
}
return 0;
}
-sub _delete_prev {
+sub _remove_prev {
my ($me, $args) = @_;
- return _not_yet();
-}
-
-sub _delete_post_id {
- my ($me, $args) = @_;
+ my $last_post = $me->{db}->search_lastpost_by_submitter({who => $args->{who}});
- return _not_yet();
+ if (!defined $last_post) {
+ return $args->{who}.': いまのっていつの? というか ないし';
+ } elsif ($last_post->{post_time} < time() - 3600) {
+ return $args->{who}.': いまのっていつの? 最後のはこれだけど古いんだにゃ ' . $me->{fb}->post_uri($last_post->{fb_post_id});
+ } else {
+ return $me->_remove($args, $last_post->{'fb_post_id'});
+ }
}
-sub _delete {
+sub _remove {
my ($me, $args, $post_id) =@_;
+ my ($resp_msg, $resp);
$me->{dbh}->begin_work;
- if ($me->_db_delete(fb_post_id => $post_id, submitter => $args->{who})) {
+ if ($resp = $me->{db}->remove({fb_post_id => $post_id, submitter => $args->{who}})) {
# fb 側のエントリを削除しないといけない
- $me->{dbh}->commit;
+ if ($me->{fb}->remove($post_id)) {
+ $me->{dbh}->commit;
+ $resp_msg = $args->{who} . ': 削除しました ' . $me->{fb}->post_uri($post_id);
+ } else {
+ $resp_msg = $args->{who} . ': 削除に失敗しましたよ? ' . $me->{fb}->post_uri($post_id);
+ $me->{dbh}->rollback;
+ }
} else {
+ $resp_msg = $args->{who} . ': そんな投稿ないよ? ' . $me->{fb}->post_uri($post_id);
$me->{dbh}->rollback;
}
- return _not_yet();
+ return $resp_msg;
}
sub _search_start {
my ($me, $args) = @_;
- if ($args->{body} =~ /^ふみくん (.+)\?$/) {
+ if ($args->{body} =~ /^ふみくん\s+(.+)\?\s*$/) {
$me->{last_search}->{$args->{who}} = undef;
- $me->{last_search}->{$args->{who}} = $me->_db_search($1);
+ $me->{last_search}->{$args->{who}} = $me->{db}->search_by_word({word => $1});
return $me->_search_next($args);
}
}
my $count = @{$me->{last_search}->{$args->{who}}};
if ($count) {
}
- $resp_msg = $args->{who} . ': ' . $me->_format_submit($ent).'に言ってた'.($count ? '[ほか'.$count.'件] ' : '[ほかにはもうないよ] ').$me->_fb_post_uri($ent->{fb_post_id});
+ $resp_msg = $args->{who} . ': ' . $me->_format_submit($ent).'に言ってた '.($count ? '[ほか'.$count.'件] ' : '[ほかにはもうないよ] ').$me->{fb}->post_uri($ent->{fb_post_id});
}
}
return $resp_msg;
if ($args->{body} =~ /$mu_re/) {
$resp_msg = $me->_add($args) unless ($1 eq 'deb');
- } elsif ($args->{body} =~ /^ふみくん (.+)/) {
+ } elsif ($args->{body} =~ /^ふみくん\s+(.+)\s*$/) {
my $cmd = $1;
if ($cmd eq 'いまのなし') {
- $resp_msg = $me->_delete_prev($args);
- } elsif ($cmd =~ /削除 ([0-9]+)$/) {
- $resp_msg = $me->_delete_post_id($args, $1);
+ $resp_msg = $me->_remove_prev($args);
+ } elsif ($cmd =~ /削除\s+(?:$me->{cfg}->{fb_page_url}posts\/)?([0-9]+)$/) {
+ $resp_msg = $me->_remove($args, $1);
} elsif ($cmd =~ /\?$/) {
$resp_msg = $me->_search_start($args);
} elsif ($cmd =~ /つぎ/) {
use strict;
use utf8;
-use Config::Simple;
-use File::Path;
+use Config::Any;
+use Hash::Merge::Simple;
+use Data::Recursive::Encode;
+
+use Data::Dumper;
my $config_name = $ARGV[0] || 'not_found';
-my %cfg;
-my $config_path = ('/etc/mubot4fb/', $ENV{HOME} . '/.mubot4fb/', $ENV{PWD} . '/mubot4fb_');
-foreach my $c ($config_path) {
- my $config = $c . $config_name . '.conf';
- Config::Simple->import_from($config, \%cfg) if (-e $config);
+my $cfg = {};
+my $config_path = ['/etc/mubot4fb/'.$config_name,
+ $ENV{HOME} . '/.mubot4fb/'. $config_name,
+ $ENV{PWD} . '/mubot4fb_' . $config_name];
+
+my $c = Config::Any->load_stems({stems => $config_path, use_ext => 1, flatten_to_hash => 1});
+foreach my $i (keys %$c) {
+ $cfg = Hash::Merge::Simple->merge($cfg, $c->{$i});
}
-die 'missing config file' unless (keys %cfg);
+die 'missing config file' unless (keys %$cfg);
die 'missing some config parameters should be defined (irc_server, fb_app_id, fb_app_secret, fb_access_code, fb_page_id fb_postback_url)'
- if (!defined $cfg{'irc_server'}
- || !defined $cfg{'fb_app_id'}
- || !defined $cfg{'fb_app_secret'}
- || !defined $cfg{'fb_access_code'}
- || !defined $cfg{'fb_page_id'}
- || !defined $cfg{'fb_postback_url'}
- || !defined $cfg{'db_user'}
- || !defined $cfg{'db_pass'}
+ if (!defined $cfg->{'irc_server'}
+ || !defined $cfg->{'fb_app_id'}
+ || !defined $cfg->{'fb_app_secret'}
+ || !defined $cfg->{'fb_access_code'}
+ || !defined $cfg->{'fb_page_id'}
+ || !defined $cfg->{'fb_postback_url'}
+ || !defined $cfg->{'db_user'}
+ || !defined $cfg->{'db_pass'}
);
-
-$cfg{irc_port} ||= 6667;
-$cfg{irc_channels} ||= ['#mubot4fb'];
-$cfg{irc_nick} ||= 'mubot4fb';
-$cfg{irc_name}||= $cfg{irc_nick};
-$cfg{irc_charset} ||= 'utf8';
-$cfg{database} ||= 'mubot4fb';
-
-my $bot = Mubot4FB->new(server => $cfg{'irc_server'},
- port => $cfg{'irc_port'},
- channels => $cfg{'irc_channels'},
- nick => $cfg{'irc_nick'},
- username => $cfg{'irc_name'},
- name => $cfg{'irc_name'},
- charset => $cfg{'irc_charset'},
- cfg => \%cfg)->run();
+$cfg = Data::Recursive::Encode->decode('utf8', $cfg);
+
+$cfg->{irc_port} ||= 6667;
+$cfg->{irc_channels} ||= ['#mubot4fb'];
+$cfg->{irc_nick} ||= 'mubot4fb';
+$cfg->{irc_name}||= $cfg->{irc_nick};
+$cfg->{irc_charset} ||= 'utf8';
+$cfg->{database} ||= 'mubot4fb';
+$cfg->{debug} ||= 0;
+
+my $bot = Mubot4FB->new(server => $cfg->{'irc_server'},
+ port => $cfg->{'irc_port'},
+ channels => $cfg->{'irc_channels'},
+ nick => $cfg->{'irc_nick'},
+ username => $cfg->{'irc_name'},
+ name => $cfg->{'irc_name'},
+ charset => $cfg->{'irc_charset'},
+ cfg => $cfg)->run();
1;
OSDN Git Service
