2 # Convert Packages.gz files into Sleepycat db files for efficient usage of
7 # Copyright (C) 2006 Jeroen van Wolffelaar <jeroen@wolffelaar.nl>
8 # This program is free software; you can redistribute it and/or modify
9 # it under the terms of the GNU General Public License as published by
10 # the Free Software Foundation; either version 2 of the License, or
11 # (at your option) any later version.
13 # This program is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU General Public License for more details.
18 # You should have received a copy of the GNU General Public License
19 # along with this program; if not, write to the Free Software
20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
28 my $what = $ARGV[0] ? "non-free" : "*";
29 # max. distinct results for a given package postfix
30 my $MAX_PACKAGE_POSTFIXES = 100;
36 use Lingua::Stem v0.82;
38 use Packages::Config qw( $TOPDIR $DBDIR @ARCHIVES @SUITES );
39 &Packages::Config::init( './' );
40 my %packages_small = ();
41 my %virtual_packages = ();
42 my %package_names = ();
43 my %package_postfixes = ();
44 my %sources_packages = ();
45 my %descriptions = ();
46 my @descriptions = ("we count lines one-based\000");
47 my %packages_descriptions = ();
48 my %descriptions_packages = ();
56 -d $DBDIR || mkpath( $DBDIR );
57 -d "$DBDIR/xapian.new" && rmtree("$DBDIR/xapian.new");
58 -d "$DBDIR/xapian.old" && rmtree("$DBDIR/xapian.old");
59 mkpath( "$DBDIR/xapian.new" );
61 for my $suite (@SUITES) {
62 my %package_names_suite = ();
64 tie %packages_all_db, "DB_File", "$DBDIR/packages_all_$suite.db.new",
65 O_RDWR|O_CREAT, 0666, $DB_BTREE
66 or die "Error creating DB: $!";
68 for my $archive (@ARCHIVES) {
69 print "Reading $archive/$suite...\n";
70 if (!-d "$TOPDIR/archive/$archive/$suite/") {
71 print "\tseems not to exist, skipping...\n";
74 open PKG, "zcat $TOPDIR/archive/$archive/$suite/$what/{,debian-installer/}binary-*/Packages.gz|";
81 while (/^(\S+):\s*(.*)\s*$/mg) {
82 my ($key, $value) = ($1, $2);
83 $value =~ s/\377/\n /g;
84 $key =~ tr [A-Z] [a-z];
88 next if exists($packages_all_db{"$data{'package'} $data{'architecture'} $data{'version'}"});
89 # Skip arch:all for amd64 & gnuab, any non-redundancy is
90 # usually a bug anyway
91 next if ($archive eq 'amd64' or $archive eq 'gnuab')
92 and $data{architecture} eq 'all';
94 if ($data{'provides'}) {
95 foreach (split /\s*,\s*/, $data{'provides'}) {
96 $virtual_packages{$_}{$suite}{$data{'package'}}++;
97 $packages_small{$_} ||= {};
100 $package_names{$data{'package'}} = 1;
101 $package_names_suite{$data{'package'}} = 1;
102 my $src = $data{'package'};
103 my $src_version = '';
104 if ($data{'source'}) {
105 $src = $data{'source'};
106 $src =~ s/\s+.*//o; # strip version info
108 $data{'source'} = $src;
109 # we add some additional data here
110 my $descr = "$data{'description'}\000$data{'package'}\000"
112 my $sdescr = $data{'description'};
113 $sdescr =~ s/\n.*//os;
115 if (exists($descriptions{$descr})) {
116 $did = $descriptions{$descr};
118 $did = 1 + $#descriptions;
119 $descriptions[$did] = $descr;
120 $descriptions{$descr} = $did;
122 $data{'description'} = $did;
123 $packages_descriptions{"$data{'package'} $data{'version'} $data{'architecture'}"} = $did;
124 $descriptions_packages{$did} .=
125 "$data{'package'} $data{'version'} $data{'architecture'}\000";
127 my $section = 'main';
128 my $subsection = $data{section} || '-';
129 if ($data{section} && ($data{section} =~ m=/=o)) {
130 ($section, $subsection) = split m=/=o, $data{section}, 2;
131 ($subsection, $section) = split m=/=o, $data{section}, 2
132 if $section eq 'non-US';
134 $data{'section'} = $section;
135 $data{'subsection'} = $subsection;
136 $data{'priority'} ||= '-';
137 $sections{$suite}{$section}++;
138 $subsections{$suite}{$subsection}++;
139 $priorities{$suite}{$data{priority}}++;
140 my $pkgitem = "$archive $suite $data{'architecture'} ".
141 "$section $subsection $data{'priority'} $data{'version'} $sdescr\0";
142 my $previtem = ($packages_small{$data{'package'}}{$suite}{$data{'architecture'}}
144 $packages_small{$data{'package'}}{$suite}{$data{'architecture'}} = $pkgitem
145 if version_cmp($data{'version'}, (split /\s/o, $previtem)[6]) > 0;
146 $previtem = ($packages_small{$data{'package'}}{$suite}{'any'}
148 $packages_small{$data{'package'}}{$suite}{'any'} = $pkgitem
149 if version_cmp($data{'version'}, (split /\s/o, $previtem)[6]) > 0;
150 $sources_packages{$src} .=
151 "$archive $suite $data{'package'} $data{'version'} $data{'architecture'}\000";
152 $data{archive} = $archive;
153 while (my ($key, $value) = each (%data)) {
154 next if $key eq 'package' or $key eq 'version' or $key eq 'architecture'
155 or $key eq 'maintainer';
156 print STDERR "WARN: $key ($suite/$archive/$data{package}/$data{architecture}\n" unless defined $value;
157 $data .= "$key\00$value\00";
160 $packages_all_db{"$data{'package'} $data{'architecture'} $data{'version'}"}
165 open NAMES, '>', "$DBDIR/package_names_$suite.txt.new"
166 or die "Error creating package names list: $!";
167 foreach (sort keys %package_names_suite) {
172 untie %packages_all_db;
175 print "Writing databases...\n";
176 my %packages_small_db;
177 tie %packages_small_db, "DB_File", "$DBDIR/packages_small.db.new",
178 O_RDWR|O_CREAT, 0666, $DB_BTREE
179 or die "Error creating DB: $!";
180 while (my ($pkg, $v) = each(%packages_small)) {
181 my ($res1, $res2, $res3) = ("", "", "");
182 while (my ($suite, $v2) = each %$v) {
183 $res2 .= $v2->{'any'};
184 while (my ($arch, $v3) = each %$v2) {
185 next if $arch eq 'any' or $v3 eq $v2->{'any'};
190 if (exists $virtual_packages{$pkg}) {
191 while (my ($suite, $v2) = each %{$virtual_packages{$pkg}}) {
192 $res1 .= "$suite\01".(join ' ', keys %$v2)."\01";
199 my $res = "$res1$res2$res3";
201 $packages_small_db{$pkg} = $res;
203 untie %packages_small_db;
205 my %sources_packages_db;
206 tie %sources_packages_db, "DB_File", "$DBDIR/sources_packages.db.new",
207 O_RDWR|O_CREAT, 0666, $DB_BTREE
208 or die "Error creating DB: $!";
209 while (my ($k, $v) = each(%sources_packages)) {
211 $sources_packages_db{$k} = $v;
213 untie %sources_packages_db;
215 my %packages_descriptions_db;
216 tie %packages_descriptions_db, "DB_File", "$DBDIR/packages_descriptions.db.new",
217 O_RDWR|O_CREAT, 0666, $DB_BTREE
218 or die "Error creating DB: $!";
219 while (my ($k, $v) = each(%packages_descriptions)) {
220 $packages_descriptions_db{$k} = $v;
222 untie %packages_descriptions_db;
224 my %descriptions_packages_db;
225 tie %descriptions_packages_db, "DB_File", "$DBDIR/descriptions_packages.db.new",
226 O_RDWR|O_CREAT, 0666, $DB_BTREE
227 or die "Error creating DB: $!";
228 while (my ($k, $v) = each(%descriptions_packages)) {
230 $descriptions_packages_db{$k} = $v;
232 untie %descriptions_packages_db;
234 my $stemmer = Lingua::Stem->new();
235 $stemmer->stem_caching({ -level => 2 });
238 $xapian_db = Search::Xapian::WritableDatabase->new("$DBDIR/xapian.new",
239 Search::Xapian::DB_CREATE_OR_OPEN)
240 or die "can't create write-able db object: $!\n";
244 tie %descriptions_db, "DB_File", "$DBDIR/descriptions.db.new",
245 O_RDWR|O_CREAT, 0666, $DB_BTREE
246 or die "Error creating DB: $!";
247 open DESCR, ">", "$DBDIR/descriptions.txt" or die "Error creating descriptions textfile";
248 print "Index $#descriptions descriptions\n";
249 for (my $i=1; $i<= $#descriptions; $i++) {
250 my $plain_description = $descriptions[$i];
251 # strip away additional data
252 my ($only_desc) = split /\000/o, $plain_description, 2;
253 # WARNING: This needs to correspond with what happens in
254 # Packages/Search.pm:do_fulltext_search
255 $plain_description =~ tr [A-Z] [a-z];
256 # ensure one space on both ends
257 $plain_description = " $plain_description ";
258 $plain_description =~ s/[(),.-]+//og;
259 $plain_description =~ s#[^a-z0-9_/+]+# #og;
260 print DESCR "$plain_description\n";
264 my @words = split /\s+/, $plain_description;
265 my $stem_words = $stemmer->stem( \@words );
266 my $doc = Search::Xapian::Document->new()
267 or die "can't create doc object for $i: $!\n";
268 if ($doc->set_data($i)){
269 warn "can't set_data in doc object for $i: $!\n";
271 for my $j (0 .. (@$stem_words-1)) {
272 next if $stem_words->[$j] =~ /^\s*$/o;
273 if ($doc->add_posting($stem_words->[$j], $j)) {
274 warn "can't add word $stem_words->[$j] $j: $!\n";
277 $xapian_db->add_document($doc)
278 or warn "failed to add document: $i\n";
282 $descriptions_db{$i} = $only_desc;
285 untie %descriptions_db;
289 # package names stuff:
290 for my $pkg (keys %package_names) {
291 for (my $i=0;$i<length($pkg)-1;$i++) {
292 my $before = substr($pkg, 0, $i);
293 my $after = substr($pkg, $i);
294 $before = "^" if $before eq ""; # otherwise split doesn't work properly
295 $package_postfixes{$after} .= "$before\0";
298 my %package_postfixes_db;
299 tie %package_postfixes_db, "DB_File", "$DBDIR/package_postfixes.db.new",
300 O_RDWR|O_CREAT, 0666, $DB_BTREE
301 or die "Error creating DB: $!";
302 while (my ($k, $v) = each(%package_postfixes)) {
306 $nr = length($nr) + 1; # < number of hits
307 if ($nr > $MAX_PACKAGE_POSTFIXES) {
310 $package_postfixes_db{$k} = $v;
312 untie %package_postfixes_db;
314 store \%sections, "$DBDIR/sections.info";
315 store \%subsections, "$DBDIR/subsections.info";
316 store \%priorities, "$DBDIR/priorities.info";
318 rename("$DBDIR/packages_small.db.new", "$DBDIR/packages_small.db");
319 rename("$DBDIR/sources_packages.db.new", "$DBDIR/sources_packages.db");
320 for my $suite (@SUITES) {
321 rename("$DBDIR/packages_all_$suite.db.new",
322 "$DBDIR/packages_all_$suite.db");
323 rename("$DBDIR/package_names_$suite.txt.new",
324 "$DBDIR/package_names_$suite.txt");
326 rename("$DBDIR/xapian", "$DBDIR/xapian.old");
327 rename("$DBDIR/xapian.new","$DBDIR/xapian");
328 rmtree("$DBDIR/xapian.old");
329 rename("$DBDIR/packages_descriptions.db.new",
330 "$DBDIR/packages_descriptions.db");
331 rename("$DBDIR/descriptions_packages.db.new",
332 "$DBDIR/descriptions_packages.db");
333 rename("$DBDIR/descriptions.txt.new", "$DBDIR/descriptions.txt");
334 rename("$DBDIR/descriptions.db.new", "$DBDIR/descriptions.db");
335 rename("$DBDIR/package_postfixes.db.new", "$DBDIR/package_postfixes.db");