/[Search-Estraier]/trunk/Estraier.pm

This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!

Diff of /trunk/Estraier.pm

Parent Directory | Revision Log | View Patch Patch

-revision 19 by dpavlin,
Wed Jan  4 23:10:48 2006 UTC
+revision 41 by dpavlin,
Thu Jan  5 23:32:31 2006 UTC
 Line 272 
 sub dump_draft {
          $draft .= "\n";
-         $draft .= join("\n", @{ $self->{dtexts} }) . "\n";
+         $draft .= join("\n", @{ $self->{dtexts} }) . "\n" if ($self->{dtexts});
-         $draft .= "\t" . join("\n\t", @{ $self->{htexts} }) . "\n";
+         $draft .= "\t" . join("\n\t", @{ $self->{htexts} }) . "\n" if ($self->{htexts});
          return $draft;
  }
 Line 476 
 sub options {
  }
- package Search::Estraier::Master;
+ package Search::Estraier::ResultDocument;
- use Carp;
+ use Carp qw/croak/;
- =head1 Search::Estraier::Master
+ #use Search::Estraier;
+ #our @ISA = qw/Search::Estraier/;
- Controll node master. This requires user with administration priviledges.
+ =head1 Search::Estraier::ResultDocument
+ =head2 new
+   my $rdoc = new Search::HyperEstraier::ResultDocument(
+         uri => 'http://localhost/document/uri/42',
+         attrs => {
+                 foo => 1,
+                 bar => 2,
+         },
+         snippet => 'this is a text of snippet'
+         keywords => 'this\tare\tkeywords'
+   );
  =cut
- {
+ sub new {
-         package RequestAgent;
+         my $class = shift;
-         our @ISA = qw(LWP::UserAgent);
+         my $self = {@_};
+         bless($self, $class);
-         sub new {
+         foreach my $f (qw/uri attrs snippet keywords/) {
-                 my $self = LWP::UserAgent::new(@_);
+                 croak "missing $f for ResultDocument" unless defined($self->{$f});
-                 $self->agent("Search-Estraier/$Search::Estraer::VERSION");
-                 $self;
          }
-         sub get_basic_credentials {
+         $self ? return $self : return undef;
-                 my($self, $realm, $uri) = @_;
- #               return ($user, $password);
-         }
  }
+ =head2 uri
+ Return URI of result document
+   print $rdoc->uri;
+ =cut
+ sub uri {
+         my $self = shift;
+         return $self->{uri};
+ }
+ =head2 attr_names
+ Returns array with attribute names from result document object.
+   my @attrs = $rdoc->attr_names;
+ =cut
+ sub attr_names {
+         my $self = shift;
+         croak "attr_names return array, not scalar" if (! wantarray);
+         return sort keys %{ $self->{attrs} };
+ }
+ =head2 attr
+ Returns value of an attribute.
+   my $value = $rdoc->attr( 'attribute' );
+ =cut
+ sub attr {
+         my $self = shift;
+         my $name = shift || return;
+         return $self->{attrs}->{ $name };
+ }
+ =head2 snippet
+ Return snippet from result document
+   print $rdoc->snippet;
+ =cut
+ sub snippet {
+         my $self = shift;
+         return $self->{snippet};
+ }
+ =head2 keywords
+ Return keywords from result document
+   print $rdoc->keywords;
+ =cut
+ sub keywords {
+         my $self = shift;
+         return $self->{keywords};
+ }
- =head2 new
- Create new connection to node master.
+ package Search::Estraier::NodeResult;
-   my $master = new Search::Estraier::Master(
+ use Carp qw/croak/;
-         url => 'http://localhost:1978',
-         user => 'admin',
+ #use Search::Estraier;
-         passwd => 'admin',
+ #our @ISA = qw/Search::Estraier/;
+ =head1 Search::Estraier::NodeResult
+ =head2 new
+   my $res = new Search::HyperEstraier::NodeResult(
+         docs => @array_of_rdocs,
+         hits => %hash_with_hints,
    );
  =cut
-Line 521 
 sub new {
+Line 603 
 sub new {
          my $self = {@_};
          bless($self, $class);
-         foreach my $p (qw/url user passwd/) {
+         foreach my $f (qw/docs hints/) {
-                 croak "need $p" unless ($self->{$p});
+                 croak "missing $f for ResultDocument" unless defined($self->{$f});
+         }
+         $self ? return $self : return undef;
+ }
+ =head2 doc_num
+ Return number of documents
+   print $res->doc_num;
+ =cut
+ sub doc_num {
+         my $self = shift;
+         return $#{$self->{docs}};
+ }
+ =head2 get_doc
+ Return single document
+   my $doc = $res->get_doc( 42 );
+ Returns undef if document doesn't exist.
+ =cut
+ sub get_doc {
+         my $self = shift;
+         my $num = shift;
+         croak "expect number as argument" unless ($num =~ m/^\d+$/);
+         return undef if ($num < 0 || $num > $self->{docs});
+         return $self->{docs}->[$num];
+ }
+ =head2 hint
+ Return specific hint from results.
+   print $rec->hint( 'VERSION' );
+ Possible hints are: C<VERSION>, C<NODE>, C<HIT>, C<HINT#n>, C<DOCNUM>, C<WORDNUM>,
+ C<TIME>, C<LINK#n>, C<VIEW>.
+ =cut
+ sub hint {
+         my $self = shift;
+         my $key = shift || return;
+         return $self->{hints}->{$key};
+ }
+ package Search::Estraier::Node;
+ use Carp qw/carp croak/;
+ use URI;
+ use MIME::Base64;
+ use IO::Socket::INET;
+ =head1 Search::Estraier::Node
+ =head2 new
+   my $node = new Search::HyperEstraier::Node;
+ =cut
+ sub new {
+         my $class = shift;
+         my $self = {
+                 pxport => -1,
+                 timeout => 0,   # this used to be -1
+                 dnum => -1,
+                 wnum => -1,
+                 size => -1.0,
+                 wwidth => 480,
+                 hwidth => 96,
+                 awidth => 96,
+                 status => -1,
+         };
+         bless($self, $class);
+         if (@_) {
+                 $self->{debug} = shift;
+                 warn "## Node debug on\n";
          }
          $self ? return $self : return undef;
  }
+ =head2 set_url
+ Specify URL to node server
+   $node->set_url('http://localhost:1978');
+ =cut
+ sub set_url {
+         my $self = shift;
+         $self->{url} = shift;
+ }
+ =head2 set_proxy
+ Specify proxy server to connect to node server
+   $node->set_proxy('proxy.example.com', 8080);
+ =cut
+ sub set_proxy {
+         my $self = shift;
+         my ($host,$port) = @_;
+         croak "proxy port must be number" unless ($port =~ m/^\d+$/);
+         $self->{pxhost} = $host;
+         $self->{pxport} = $port;
+ }
+ =head2 set_timeout
+ Specify timeout of connection in seconds
+   $node->set_timeout( 15 );
+ =cut
+ sub set_timeout {
+         my $self = shift;
+         my $sec = shift;
+         croak "timeout must be number" unless ($sec =~ m/^\d+$/);
+         $self->{timeout} = $sec;
+ }
+ =head2 set_auth
+ Specify name and password for authentication to node server.
+   $node->set_auth('clint','eastwood');
+ =cut
+ sub set_auth {
+         my $self = shift;
+         my ($login,$passwd) = @_;
+         my $basic_auth = encode_base64( "$login:$passwd" );
+         chomp($basic_auth);
+         $self->{auth} = $basic_auth;
+ }
+ =head2 status
+ Return status code of last request.
+   print $node->status;
+ C<-1> means connection failure.
+ =cut
+ sub status {
+         my $self = shift;
+         return $self->{status};
+ }
+ =head2 put_doc
+ Add a document
+   $node->put_doc( $document_draft ) or die "can't add document";
+ Return true on success or false on failture.
+ =cut
+ sub put_doc {
+         my $self = shift;
+         my $doc = shift || return;
+         return unless ($self->{url});
+         $self->shuttle_url( $self->{url} . '/put_doc',
+                 'text/x-estraier-draft',
+                 $doc->dump_draft,
+                 undef
+         ) == 200;
+ }
+ =head2 out_doc
+ Remove a document
+   $node->out_doc( document_id ) or "can't remove document";
+ Return true on success or false on failture.
+ =cut
+ sub out_doc {
+         my $self = shift;
+         my $id = shift || return;
+         return unless ($self->{url});
+         croak "id must be number" unless ($id =~ m/^\d+$/);
+         $self->shuttle_url( $self->{url} . '/out_doc',
+                 'application/x-www-form-urlencoded',
+                 "id=$id",
+                 undef
+         ) == 200;
+ }
+ =head2 out_doc_by_uri
+ Remove a registrated document using it's uri
+   $node->out_doc_by_uri( 'file:///document_url' ) or "can't remove document";
+ Return true on success or false on failture.
+ =cut
+ sub out_doc_by_uri {
+         my $self = shift;
+         my $uri = shift || return;
+         return unless ($self->{url});
+         $self->shuttle_url( $self->{url} . '/out_doc',
+                 'application/x-www-form-urlencoded',
+                 "uri=$uri",
+                 undef
+         ) == 200;
+ }
+ =head2 shuttle_url
+ This is method which uses C<IO::Socket::INET> to communicate with Hyper Estraier node
+ master.
+   my $rv = shuttle_url( $url, $content_type, \$req_body, \$resbody );
+ C<$resheads> and C<$resbody> booleans controll if response headers and/or response
+ body will be saved within object.
+ =cut
+ sub shuttle_url {
+         my $self = shift;
+         my ($url, $content_type, $reqbody, $resbody) = @_;
+         $self->{status} = -1;
+         warn "## $url\n" if ($self->{debug});
+         $url = new URI($url);
+         if (
+                         !$url || !$url->scheme || !$url->scheme eq 'http' ||
+                         !$url->host || !$url->port || $url->port < 1
+                 ) {
+                 carp "can't parse $url\n";
+                 return -1;
+         }
+         my ($host,$port,$query) = ($url->host, $url->port, $url->path);
+         if ($self->{pxhost}) {
+                 ($host,$port) = ($self->{pxhost}, $self->{pxport});
+                 $query = "http://$host:$port/$query";
+         }
+         $query .= '?' . $url->query if ($url->query && ! $reqbody);
+         my $headers;
+         if ($reqbody) {
+                 $headers .= "POST $query HTTP/1.0\r\n";
+         } else {
+                 $headers .= "GET $query HTTP/1.0\r\n";
+         }
+         $headers .= "Host: " . $url->host . ":" . $url->port . "\r\n";
+         $headers .= "Connection: close\r\n";
+         $headers .= "User-Agent: Search-Estraier/$Search::Estraier::VERSION\r\n";
+         $headers .= "Content-Type: $content_type\r\n";
+         $headers .= "Authorization: Basic $self->{auth}\r\n";
+         my $len = 0;
+         {
+                 use bytes;
+                 $len = length($reqbody) if ($reqbody);
+         }
+         $headers .= "Content-Length: $len\r\n";
+         $headers .= "\r\n";
+         my $sock = IO::Socket::INET->new(
+                 PeerAddr        => $host,
+                 PeerPort        => $port,
+                 Proto           => 'tcp',
+                 Timeout         => $self->{timeout} || 90,
+         );
+         if (! $sock) {
+                 carp "can't open socket to $host:$port";
+                 return -1;
+         }
+         warn $headers if ($self->{debug});
+         print $sock $headers or
+                 carp "can't send headers to network:\n$headers\n" and return -1;
+         if ($reqbody) {
+                 warn "$reqbody\n" if ($self->{debug});
+                 print $sock $reqbody or
+                         carp "can't send request body to network:\n$$reqbody\n" and return -1;
+         }
+         my $line = <$sock>;
+         chomp($line);
+         my ($schema, $res_status, undef) = split(/  */, $line, 3);
+         return if ($schema !~ /^HTTP/ || ! $res_status);
+         $self->{status} = $res_status;
+         warn "## response status: $res_status\n" if ($self->{debug});
+         # skip rest of headers
+         $line = <$sock>;
+         while ($line) {
+                 $line = <$sock>;
+                 $line =~ s/[\r\n]+$//;
+                 warn "## ", $line || 'NULL', " ##\n" if ($self->{debug});
+         };
+         # read body
+         $len = 0;
+         do {
+                 $len = read($sock, my $buf, 8192);
+                 $$resbody .= $buf if ($resbody);
+         } while ($len);
+         warn "## response body:\n$$resbody\n" if ($resbody && $self->{debug});
+         return $self->{status};
+ }
  ###

 Legend:



Removed from v.19
 


changed lines


 
Added in v.41
 Legend:



Removed from v.19
 


changed lines


 
Added in v.41
-Removed from v.19
+Added in v.41

	ViewVC Help
Powered by ViewVC 1.1.26