less output
[BackupPC.git] / lib / BackupPC / Search / Estraier.pm
1 package BackupPC::Search::Estraier;
2 use warnings;
3 use strict;
4
5 use Search::Estraier 0.04;
6 use Data::Dump qw(dump);
7
8 my $debug = $ENV{DEBUG} || 0;
9
10 sub new {
11         my $class = shift @_;
12         my %Conf = @_;
13
14         my $index_node_url = $Conf{HyperEstraierIndex} || die "no HyperEstraierIndex in config ",dump(%Conf);
15
16         warn "# using $index_node_url";
17
18         my $self = bless {
19                 node => Search::Estraier::Node->new(
20                         url => $index_node_url,
21                         user => 'admin',
22                         passwd => 'admin',
23                         croak_on_error => 1,
24                 ),
25         }, $class;
26         return $self;
27 }
28
29 sub node { $_[0]->{node} };
30
31 sub exists {
32         my ( $self, $row ) = @_;
33
34         my $uri = $row->{hname} . ':' . $row->{sname} . '#' . $row->{backupnum} . ' ' . $row->{filepath};
35         my $id = $self->node->uri_to_id($uri);
36         return $id && $id != -1;
37 }
38
39 sub add_doc {
40         my ( $self, $row ) = @_;
41
42         # create a document object 
43         my $doc = Search::Estraier::Document->new;
44
45         my $uri = $row->{hname} . ':' . $row->{sname} . '#' . $row->{backupnum} . ' ' . $row->{filepath};
46         # add attributes to the document object 
47         $doc->add_attr('@uri', $uri);
48
49         foreach my $c (keys %$row) {
50                 print STDERR "attr $c = $row->{$c}\n" if ($debug > 2);
51                 $doc->add_attr($c, $row->{$c}) if defined($row->{$c});
52         }
53
54         #$doc->add_attr('@cdate', fmt_date($row->{'date'}));
55
56         # add the body text to the document object 
57         my $path = $row->{'filepath'};
58         $doc->add_text($path);
59         $path =~ s/(.)/$1 /g;
60         $doc->add_hidden_text($path);
61
62         print STDERR $doc->dump_draft,"\n" if ($debug > 1);
63
64         # register the document object to the database
65         $self->node->put_doc($doc);
66
67 }
68
69 sub commit {
70         my $self = shift;
71         warn "# commit not needed";
72 }
73
74 my $sort_mapping = {
75         share_d => 'sname STRD',
76         share_a => 'sname STRA',
77         path_d => 'filepath STRD',
78         path_a => 'filepath STRA',
79         num_d => 'backupnum NUMD',
80         num_a => 'backupnum NUMA',
81         size_d => 'size NUMD',
82         size_a => 'size NUMA',
83         date_d => 'date NUMD',
84         date_a => 'date NUMA',
85 };
86
87 sub search {
88         my ( $self, $offset, $on_page, $sort, $q, $shareid, $backup_from, $backup_to, $files_from, $files_to ) = @_;
89
90         warn "# search $offset/$on_page [$q] shareid: $shareid backup: $backup_from - $backup_to files: $files_from - $files_to";
91
92         # create a search condition object
93         my $cond = Search::Estraier::Condition->new();
94
95         if (length($q) > 0) {
96                 # exact match
97                 $cond->add_attr("filepath ISTRINC $q");
98
99                 $q =~ s/(.)/$1 /g;
100                 # set the search phrase to the search condition object
101                 $cond->set_phrase($q);
102         }
103
104         $cond->add_attr("backup_date NUMGE $backup_from") if ($backup_from);
105         $cond->add_attr("backup_date NUMLE $backup_to") if ($backup_to);
106
107         $cond->add_attr("date NUMGE $files_from") if ($files_from);
108         $cond->add_attr("date NUMLE $files_to") if ($files_to);
109
110         $cond->add_attr("shareid NUMEQ $shareid") if ($shareid);
111
112         $cond->set_max( $offset + $on_page );
113         $cond->set_options( 'SURE' );
114         $cond->set_order( $sort_mapping->{$sort} );
115
116         # get the result of search
117         my @res;
118         my ($result, $hits);
119
120         $result = $self->node->search($cond, 0);
121         if ($result) {
122                 $hits = $result->hits;
123         } else {
124                 return (0,[]);
125         }
126
127         # for each document in result
128         for my $i ($offset .. ($offset + $on_page - 1)) {
129                 last if ($i >= $result->doc_num);
130
131                 my $doc = $result->get_doc($i);
132
133                 my $row;
134                 foreach my $c (qw/fid hname sname backupnum filepath date type size/) {
135                         $row->{$c} = $doc->attr($c);
136                 }
137                 push @res, $row;
138         }
139
140         warn "# $hits hits";
141
142         return ($hits, \@res);
143
144 }
145
146 1;