9002f3f51825ea051fbfd1dbec786e384b5bc95a
[apt.git] / apt-pkg / rpm / rpmindexfile.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description                                                          /*{{{*/
3 // $Id: rpmindexfile.cc,v 1.4 2002/11/27 16:22:40 niemeyer Exp $
4 /* ######################################################################
5
6    RPM Specific sources.list types and the three sorts of RPM
7    index files.
8    
9    ##################################################################### */
10                                                                         /*}}}*/
11 // Include Files                                                        /*{{{*/
12 #ifdef __GNUG__
13 #pragma implementation "apt-pkg/rpmindexfile.h"
14 #endif
15
16 #include <config.h>
17
18 #ifdef HAVE_RPM
19
20 #include <apt-pkg/rpmindexfile.h>
21 #include <apt-pkg/rpmsrcrecords.h>
22 #include <apt-pkg/rpmlistparser.h>
23 #include <apt-pkg/rpmrecords.h>
24 #include <apt-pkg/rpmsystem.h>
25 #include <apt-pkg/rpmhandler.h>
26 #include <apt-pkg/rpmpackagedata.h>
27 #include <apt-pkg/sourcelist.h>
28 #include <apt-pkg/configuration.h>
29 #include <apt-pkg/progress.h>
30 #include <apt-pkg/error.h>
31 #include <apt-pkg/strutl.h>
32 #include <apt-pkg/acquire-item.h>
33 #include <apt-pkg/repomd.h>
34
35 #include <apti18n.h>
36
37 #include <sys/stat.h>
38                                                                         /*}}}*/
39 vector<pkgRepository *> RepList;
40
41 // rpmListIndex::Release* - Return the URI to the release file          /*{{{*/
42 // ---------------------------------------------------------------------
43 /* */
44 inline string rpmListIndex::ReleaseFile(string Type) const
45 {
46    return URItoFileName(ReleaseURI(Type));
47 }
48
49 string rpmListIndex::ReleaseURI(string Type) const
50 {
51    RPMPackageData *rpmdata = RPMPackageData::Singleton();
52    string Res;
53    if (Dist[Dist.size() - 1] == '/')
54    {
55       if (Dist != "/")
56          Res = URI + Dist;
57       else
58          Res = URI;
59    }
60    else
61       Res = URI + Dist + "/base/";
62
63    Res += Type;
64    
65    if (rpmdata->HasIndexTranslation() == true)
66    {
67       map<string,string> Dict;
68       Dict["uri"] = URI;
69       Dict["dist"] = Dist;
70       Dict["sect"] = "";
71       Dict["type"] = Type;
72       rpmdata->TranslateIndex(Res, Dict);
73    }
74
75    return Res;
76 }
77                                                                         /*}}}*/
78 // rpmListIndex::ReleaseInfo - One liner describing the index URI       /*{{{*/
79 // ---------------------------------------------------------------------
80 /* */
81 string rpmListIndex::ReleaseInfo(string Type) const 
82 {
83    string Info = ::URI::SiteOnly(URI) + ' ';
84    if (Dist[Dist.size() - 1] == '/')
85    {
86       if (Dist != "/")
87          Info += Dist;
88    }
89    else
90       Info += Dist;   
91    Info += " ";
92    Info += Type;
93    return Info;
94 }
95                                                                         /*}}}*/
96 // rpmListIndex::GetReleases - Fetch the index files                    /*{{{*/
97 // ---------------------------------------------------------------------
98 /* */
99 bool rpmListIndex::GetReleases(pkgAcquire *Owner) const
100 {
101    if (!Repository->Acquire)
102       return true;
103    Repository->Acquire = false;
104    new pkgAcqIndexRel(Owner,Repository,ReleaseURI("release"),
105                       ReleaseInfo("release"), "release", true);
106    return true;
107 }
108                                                                         /*}}}*/
109 // rpmListIndex::Info - One liner describing the index URI              /*{{{*/
110 // ---------------------------------------------------------------------
111 /* */
112 string rpmListIndex::Info(string Type) const 
113 {
114    string Info = ::URI::SiteOnly(URI) + ' ';
115    if (Dist[Dist.size() - 1] == '/')
116    {
117       if (Dist != "/")
118          Info += Dist;
119    }
120    else
121       Info += Dist + '/' + Section;   
122    Info += " ";
123    Info += Type;
124    return Info;
125 }
126                                                                         /*}}}*/
127 // rpmListIndex::Index* - Return the URI to the index files             /*{{{*/
128 // ---------------------------------------------------------------------
129 /* */
130 inline string rpmListIndex::IndexFile(string Type) const
131 {
132    return _config->FindDir("Dir::State::lists")+URItoFileName(IndexURI(Type));
133 }
134
135
136 string rpmListIndex::IndexURI(string Type) const
137 {
138    RPMPackageData *rpmdata = RPMPackageData::Singleton();
139    string Res;
140    if (Dist[Dist.size() - 1] == '/')
141    {
142       if (Dist != "/")
143          Res = URI + Dist;
144       else 
145          Res = URI;
146    }
147    else
148       Res = URI + Dist + "/base/";
149    
150    Res += Type + '.' + Section;
151
152    if (rpmdata->HasIndexTranslation() == true)
153    {
154       map<string,string> Dict;
155       Dict["uri"] = URI;
156       Dict["dist"] = Dist; 
157       Dict["sect"] = Section;
158       Dict["type"] = Type;
159       rpmdata->TranslateIndex(Res, Dict);
160    }
161
162    return Res;
163 }
164                                                                         /*}}}*/
165 // rpmListIndex::Exists - Check if the index is available               /*{{{*/
166 // ---------------------------------------------------------------------
167 /* */
168 bool rpmListIndex::Exists() const
169 {
170    return FileExists(IndexPath());
171 }
172                                                                         /*}}}*/
173 // rpmListIndex::Size - Return the size of the index                    /*{{{*/
174 // ---------------------------------------------------------------------
175 /* */
176 unsigned long rpmListIndex::Size() const
177 {
178    struct stat S;
179    if (stat(IndexPath().c_str(),&S) != 0)
180       return 0;
181    return S.st_size;
182 }
183                                                                         /*}}}*/
184 // rpmListIndex::Describe - Give a descriptive path to the index        /*{{{*/
185 // ---------------------------------------------------------------------
186 /* */
187 string rpmListIndex::Describe(bool Short) const
188 {
189    char S[300];
190    if (Short == true)
191       snprintf(S,sizeof(S),"%s",Info(MainType()).c_str());
192    else
193       snprintf(S,sizeof(S),"%s (%s)",Info(MainType()).c_str(),
194          IndexFile(MainType()).c_str());
195    return S;
196 }
197                                                                         /*}}}*/
198
199 // SrcListIndex::SourceInfo - Short 1 liner describing a source         /*{{{*/
200 // ---------------------------------------------------------------------
201 string rpmSrcListIndex::SourceInfo(pkgSrcRecords::Parser const &Record,
202                                    pkgSrcRecords::File const &File) const
203 {
204    string Res;
205    Res = ::URI::SiteOnly(URI) + ' ';
206    if (Dist[Dist.size() - 1] == '/')
207    {
208       if (Dist != "/")
209          Res += Dist;
210    }      
211    else
212       Res += Dist + '/' + Section;
213    
214    Res += " ";
215    Res += Record.Package();
216    Res += " ";
217    Res += Record.Version();
218    if (File.Type.empty() == false)
219       Res += " (" + File.Type + ")";
220    return Res;
221 }
222                                                                         /*}}}*/
223 // SrcListIndex::ArchiveURI - URI for the archive                       /*{{{*/
224 // ---------------------------------------------------------------------
225 string rpmSrcListIndex::ArchiveURI(string File) const
226 {
227    RPMPackageData *rpmdata = RPMPackageData::Singleton();
228    string Res;
229    
230    if (Dist[Dist.size() - 1] == '/')
231    {
232       if (Dist != "/")
233          Res = URI + Dist;
234       else
235          Res = URI;
236    }
237    else
238       Res = URI + Dist;
239    
240    if (File.find("/") != string::npos)
241       Res += '/' + File;
242    else
243       Res += "/SRPMS."+Section + '/' + File;
244
245    if (rpmdata->HasSourceTranslation() == true)
246    {
247       map<string,string> Dict;
248       Dict["uri"] = URI;
249       Dict["dist"] = Dist; 
250       Dict["sect"] = Section;
251       string::size_type pos = File.rfind("/");
252       if (pos != string::npos)
253          Dict["file"] = string(File, pos+1);
254       else
255          Dict["file"] = File;
256       
257       rpmdata->TranslateSource(Res, Dict);
258    }
259          
260    return Res;
261 }
262                                                                         /*}}}*/
263 // SrcListIndex::CreateSrcParser - Get a parser for the source files    /*{{{*/
264 // ---------------------------------------------------------------------
265 /* */
266 pkgSrcRecords::Parser *rpmSrcListIndex::CreateSrcParser() const
267 {
268    return new rpmSrcRecordParser(IndexPath(), this);
269 }
270                                                                         /*}}}*/
271 // SrcListIndex::GetIndexes - Fetch the index files                     /*{{{*/
272 // ---------------------------------------------------------------------
273 /* */
274 bool rpmSrcListIndex::GetIndexes(pkgAcquire *Owner) const
275 {
276    // Ignore indexes for repositories that could not be authenticated
277    if (Repository->IsAuthenticated() == true && 
278        Repository->HasRelease() == false)
279       return true;
280    new pkgAcqIndex(Owner,Repository,IndexURI("srclist"),Info("srclist"),
281                    "srclist");
282    return true;
283 }
284                                                                         /*}}}*/
285
286 // PkgListIndex::ArchiveInfo - Short version of the archive url         /*{{{*/
287 // ---------------------------------------------------------------------
288 /* This is a shorter version that is designed to be < 60 chars or so */
289 string rpmPkgListIndex::ArchiveInfo(pkgCache::VerIterator Ver) const
290 {
291    string Res = ::URI::SiteOnly(URI) + ' ';
292    if (Dist[Dist.size() - 1] == '/')
293    {
294       if (Dist != "/")
295          Res += Dist;
296    }
297    else
298       Res += Dist + '/' + Section;
299    
300    Res += " ";
301    Res += Ver.ParentPkg().Name();
302    Res += " ";
303    Res += Ver.VerStr();
304    return Res;
305 }
306                                                                         /*}}}*/
307 // PkgListIndex::ArchiveURI - URI for the archive                       /*{{{*/
308 // ---------------------------------------------------------------------
309 string rpmPkgListIndex::ArchiveURI(string File) const
310 {
311    RPMPackageData *rpmdata = RPMPackageData::Singleton();
312    string Res;
313    if (Dist[Dist.size() - 1] == '/')
314    {
315       if (Dist != "/")
316          Res = URI + Dist;
317       else
318          Res = URI;
319    }
320    else
321       Res = URI + Dist;
322
323    if (File.find("/") != string::npos)
324       Res += '/' + File;
325    else
326       Res += "/RPMS." + Section + '/' + File;
327
328    if (rpmdata->HasBinaryTranslation() == true)
329    {
330       map<string,string> Dict;
331       Dict["uri"] = URI;
332       Dict["dist"] = Dist; 
333       Dict["sect"] = Section;
334       string::size_type pos = File.rfind("/");
335       if (pos != string::npos)
336          Dict["file"] = string(File, pos+1);
337       else
338          Dict["file"] = File;
339       rpmdata->TranslateBinary(Res, Dict);
340    }
341          
342    return Res;
343 }
344                                                                         /*}}}*/
345 // PkgListIndex::GetIndexes - Fetch the index files                     /*{{{*/
346 // ---------------------------------------------------------------------
347 /* */
348 bool rpmPkgListIndex::GetIndexes(pkgAcquire *Owner) const
349 {
350    // Ignore indexes for repositories that could not be authenticated
351    if (Repository->IsAuthenticated() == true && 
352        Repository->HasRelease() == false)
353       return true;
354    new pkgAcqIndex(Owner,Repository,IndexURI("pkglist"),Info("pkglist"),
355                    "pkglist");
356    new pkgAcqIndexRel(Owner,Repository,IndexURI("release"),Info("release"),
357                       "release");
358    return true;
359 }
360                                                                         /*}}}*/
361 // PkgListIndex::Merge - Load the index file into a cache               /*{{{*/
362 // ---------------------------------------------------------------------
363 /* */
364 bool rpmPkgListIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
365 {
366    string PackageFile = IndexPath();
367    RPMHandler *Handler = CreateHandler();
368
369    Prog.SubProgress(0,Info(MainType()));
370    ::URI Tmp(URI);
371    if (Gen.SelectFile(PackageFile,Tmp.Host,*this) == false)
372    {
373       delete Handler;
374       return _error->Error(_("Problem with SelectFile %s"),PackageFile.c_str());
375    }
376
377    // Store the IMS information
378    pkgCache::PkgFileIterator File = Gen.GetCurFile();
379    struct stat St;
380    if (stat(PackageFile.c_str(),&St) != 0) 
381    {
382       delete Handler;
383       return _error->Errno("stat",_("Failed to stat %s"), PackageFile.c_str());
384    }
385    File->Size = St.st_size;
386    File->mtime = St.st_mtime;
387    
388    rpmListParser Parser(Handler);
389    if (_error->PendingError() == true) 
390    {
391       delete Handler;
392       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
393    }
394    
395    if (Gen.MergeList(Parser) == false)
396    {
397       delete Handler;
398       return _error->Error(_("Problem with MergeList %s"),PackageFile.c_str());
399    }
400    
401    delete Handler;
402
403    // Check the release file
404    string RelFile = ReleasePath();
405    if (FileExists(RelFile) == true)
406    {
407       FileFd Rel(RelFile,FileFd::ReadOnly);
408       if (_error->PendingError() == true)
409          return false;
410       Parser.LoadReleaseInfo(File,Rel);
411       Rel.Seek(0);
412    }
413
414    return true;
415 }
416                                                                         /*}}}*/
417 // PkgListIndex::MergeFileProvides - Process file dependencies if any   /*{{{*/
418 // ---------------------------------------------------------------------
419 /* */
420 bool rpmPkgListIndex::MergeFileProvides(pkgCacheGenerator &Gen,
421                                         OpProgress &Prog) const
422 {
423    string PackageFile = IndexPath();
424    RPMHandler *Handler = CreateHandler();
425    rpmListParser Parser(Handler);
426    if (_error->PendingError() == true) {
427       delete Handler;
428       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
429    }
430    // We call SubProgress with Size(), since we won't call SelectFile() here.
431    Prog.SubProgress(Size(),Info("pkglist"));
432    if (Gen.MergeFileProvides(Parser) == false)
433       return _error->Error(_("Problem with MergeFileProvides %s"),
434                            PackageFile.c_str());
435    delete Handler;
436    return true;
437 }
438                                                                         /*}}}*/
439 // PkgListIndex::FindInCache - Find this index                          /*{{{*/
440 // ---------------------------------------------------------------------
441 /* */
442 pkgCache::PkgFileIterator rpmPkgListIndex::FindInCache(pkgCache &Cache) const
443 {
444    string FileName = IndexPath();
445    pkgCache::PkgFileIterator File = Cache.FileBegin();
446    for (; File.end() == false; File++)
447    {
448       if (FileName != File.FileName())
449          continue;
450       
451       struct stat St;
452       if (stat(File.FileName(),&St) != 0)
453          return pkgCache::PkgFileIterator(Cache);
454
455       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
456          return pkgCache::PkgFileIterator(Cache);
457       return File;
458    }
459    
460    return File;
461 }
462                                                                         /*}}}*/
463
464 // PkgDirIndex::Index* - Return the URI to the index files              /*{{{*/
465 // ---------------------------------------------------------------------
466 /* */
467 string rpmPkgDirIndex::IndexPath() const
468 {
469    return ::URI(ArchiveURI("")).Path;
470 }
471                                                                         /*}}}*/
472 // PkgDirIndex::Release* - Return the URI to the index files            /*{{{*/
473 // ---------------------------------------------------------------------
474 /* */
475 string rpmPkgDirIndex::ReleasePath() const
476 {
477    return ::URI(IndexURI("release")).Path;
478 }
479                                                                         /*}}}*/
480 // PkgDirIndex::Size - Return the size of the index                     /*{{{*/
481 // ---------------------------------------------------------------------
482 /* This is really only used for progress reporting. */
483 unsigned long rpmPkgDirIndex::Size() const
484 {
485    // XXX: Must optimize this somehow.
486    RPMHandler *Handler = CreateHandler();
487    unsigned long Res = Handler->Size();
488    delete Handler;
489    return Res;
490 }
491                                                                         /*}}}*/
492
493 // SrcDirIndex::Index* - Return the URI to the index files              /*{{{*/
494 // ---------------------------------------------------------------------
495 /* */
496 string rpmSrcDirIndex::IndexPath() const
497 {
498    return ::URI(ArchiveURI("")).Path;
499 }
500                                                                         /*}}}*/
501 // SrcDirIndex::Size - Return the size of the index                     /*{{{*/
502 // ---------------------------------------------------------------------
503 /* This is really only used for progress reporting. */
504 unsigned long rpmSrcDirIndex::Size() const
505 {
506    // XXX: Must optimize this somehow.
507    RPMHandler *Handler = CreateHandler();
508    unsigned long Res = Handler->Size();
509    delete Handler;
510    return Res;
511 }
512
513 // SinglePkgIndex::ArchiveURI - URI for the archive                     /*{{{*/
514 // ---------------------------------------------------------------------
515 string rpmSinglePkgIndex::ArchiveURI(string File) const
516 {
517    char *cwd = getcwd(NULL,0);
518    if (File[0] == '.' && File[1] == '/')
519       File = string(File, 2);
520    string URI = "file://"+flCombine(cwd, File);
521    free(cwd);
522    return URI;
523 }
524                                                                         /*}}}*/
525 // SinglePkgIndex::ArchiveURI - URI for the archive                     /*{{{*/
526 // ---------------------------------------------------------------------
527 string rpmSingleSrcIndex::ArchiveURI(string File) const
528 {
529    char *cwd = getcwd(NULL,0);
530    if (File[0] == '.' && File[1] == '/')
531       File = string(File, 2);
532    string URI = "file://"+flCombine(cwd, File);
533    free(cwd);
534    return URI;
535 }
536
537 string rpmRepomdIndex::ArchiveURI(string File) const
538 {
539    string Res;
540
541    //cout << Dist << File << endl;
542
543    Res += URI + '/' + Dist + '/' + File;
544    //cout << "repomd archiveuri " << " " << Res << endl;
545
546    return Res;
547 }
548
549 string rpmRepomdIndex::ArchiveInfo(pkgCache::VerIterator Ver) const
550 {
551    string Res = ::URI::SiteOnly(URI) + ' ';
552    if (Dist[Dist.size() - 1] == '/')
553    {
554       if (Dist != "/")
555          Res += Dist;
556    }
557    else
558       Res += Dist + '/';
559    
560    Res += " ";
561    Res += Ver.ParentPkg().Name();
562    Res += " ";
563    Res += Ver.VerStr();
564    return Res;
565 }
566 pkgCache::PkgFileIterator rpmRepomdIndex::FindInCache(pkgCache &Cache) const
567 {
568    string FileName = IndexPath();
569    pkgCache::PkgFileIterator File = Cache.FileBegin();
570    for (; File.end() == false; File++)
571    {
572       if (FileName != File.FileName())
573          continue;
574       
575       struct stat St;
576       if (stat(File.FileName(),&St) != 0)
577          return pkgCache::PkgFileIterator(Cache);
578
579       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
580          return pkgCache::PkgFileIterator(Cache);
581       return File;
582    }
583    
584    return File;
585 }
586 string rpmRepomdIndex::ReleaseURI(string Type) const
587 {
588    string Res;
589    Res = URI + Dist + "/repodata/" + "repomd.xml";
590
591    //cout << "XXXXX repomd releaseuri " << Res << endl;
592    
593    return Res;
594 }
595
596 string rpmRepomdIndex::ReleaseInfo(string Type) const
597 {
598    string Info = ::URI::SiteOnly(URI) + ' ';
599    if (Dist[Dist.size() - 1] == '/')
600    {
601       if (Dist != "/")
602          Info += Dist;
603    }
604    else
605       Info += Dist;   
606    Info += " ";
607    Info += Type;
608    //cout << "repomd releaseinfo " << Info << endl;
609    return Info;
610 };
611
612 string rpmRepomdIndex::Info(string Type) const 
613 {
614    string Info = ::URI::SiteOnly(URI) + ' ';
615    if (Dist[Dist.size() - 1] == '/')
616    {
617       if (Dist != "/")
618          Info += Dist;
619    }
620    else
621       Info += Dist + '/' ;   
622    Info += " ";
623    Info += Type;
624    //cout << "repomd info " << Info << endl;
625    return Info;
626 }
627
628 string rpmRepomdIndex::IndexURI(string Type) const
629 {
630    string Res = URI + Dist;
631    if (Dist[Dist.size() - 1] != '/') {
632          Res += "/";
633    }
634    // XXXX eh, this doesn't look sane .. 
635    if (Type == "repomd")
636       Res += "repodata/primary.xml";
637    else
638       Res += "repodata/" + Type;
639    return Res;
640 }
641                                                                         /*}}}*/
642 bool rpmRepomdIndex::GetReleases(pkgAcquire *Owner) const
643 {
644    if (!Repository->Acquire)
645       return true;
646    Repository->Acquire = false;
647    // ignore for now - we need to parse the file for checksum and 
648    // optimally for location of other xml files as well
649    new pkgAcqIndexRel(Owner,Repository,ReleaseURI("repomd.xml"),
650                       ReleaseInfo("repomd.xml"), "repomd.xml", true);
651    return true;
652 }
653
654 bool rpmRepomdIndex::GetIndexes(pkgAcquire *Owner) const
655 {
656    new pkgAcqIndex(Owner,Repository,IndexURI("primary.xml"),
657                    Info("primary.xml"), "primary.xml");
658    new pkgAcqIndex(Owner,Repository,IndexURI("filelists.xml"),
659                    Info("filelists.xml"), "filelists.xml");
660 #if 0
661    new pkgAcqIndex(Owner,Repository,IndexURI("other.xml"),
662                    Info("other.xml"), "other.xml");
663 #endif
664    return true;
665 }
666
667 string rpmRepomdIndex::Describe(bool Short) const
668 {
669    char S[300];
670    if (Short == true)
671       snprintf(S,sizeof(S),"%s",Info(MainType()).c_str());
672    else
673       snprintf(S,sizeof(S),"%s (%s)",Info(MainType()).c_str(),
674          IndexFile(MainType()).c_str());
675    return S;
676 }
677
678 string rpmRepomdIndex::IndexFile(string Type) const
679 {
680    return _config->FindDir("Dir::State::lists")+URItoFileName(IndexURI(Type));
681 };
682
683
684 bool rpmRepomdIndex::Exists() const
685 {
686    return FileExists(IndexPath());
687 }
688
689 string rpmRepomdIndex::ReleasePath() const
690 {
691    return _config->FindDir("Dir::State::lists")+URItoFileName(ReleaseURI("release.xml"));
692 }
693
694 unsigned long rpmRepomdIndex::Size() const
695 {
696    // ugh.. all this just to get the package count ;(
697    RPMHandler *Handler = CreateHandler();
698    unsigned long Res = Handler->Size();
699    delete Handler;
700    return Res;
701 }
702
703 bool rpmRepomdIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
704 {
705    //cout << "repomd MERGE" << endl;
706    string PackageFile = IndexPath();
707    RPMHandler *Handler = CreateHandler();
708
709    Prog.SubProgress(0,Info(MainType()));
710    ::URI Tmp(URI);
711    if (Gen.SelectFile(PackageFile,Tmp.Host,*this) == false)
712    {
713       delete Handler;
714       return _error->Error(_("Problem with SelectFile %s"),PackageFile.c_str());
715    }
716
717    // Store the IMS information
718    pkgCache::PkgFileIterator File = Gen.GetCurFile();
719    //cout << "merge pkgfile " << PackageFile << endl;
720    struct stat St;
721    if (stat(PackageFile.c_str(),&St) != 0) 
722    {
723       delete Handler;
724       return _error->Errno("stat",_("Failed to stat %s"), PackageFile.c_str());
725    }
726    File->Size = St.st_size;
727    File->mtime = St.st_mtime;
728    
729    rpmRepomdParser Parser(Handler);
730    if (_error->PendingError() == true) 
731    {
732       delete Handler;
733       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
734    }
735    
736    if (Gen.MergeList(Parser) == false)
737    {
738       delete Handler;
739       return _error->Error(_("Problem with MergeList %s"),PackageFile.c_str());
740    }
741    
742    delete Handler;
743
744    // Check the release file
745    string RelFile = ReleasePath();
746    if (FileExists(RelFile) == true)
747    {
748       Parser.LoadReleaseInfo(File,RelFile);
749    }
750
751    return true;
752 }
753
754 bool rpmRepomdIndex::MergeFileProvides(pkgCacheGenerator &Gen,
755                                         OpProgress &Prog) const
756 {
757    string PackageFile = IndexPath();
758    RPMHandler *Handler = CreateHandler();
759    rpmListParser Parser(Handler);
760    if (_error->PendingError() == true) {
761       delete Handler;
762       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
763    }
764    // We call SubProgress with Size(), since we won't call SelectFile() here.
765    Prog.SubProgress(Size(),Info("pkglist"));
766    if (Gen.MergeFileProvides(Parser) == false)
767       return _error->Error(_("Problem with MergeFileProvides %s"),
768                            PackageFile.c_str());
769    delete Handler;
770    return true;
771 }
772
773 pkgSrcRecords::Parser *rpmRepomdSrcIndex::CreateSrcParser() const
774 {
775    return new rpmSrcRecordParser(IndexPath(), this);
776 }
777
778 string rpmRepomdSrcIndex::SourceInfo(pkgSrcRecords::Parser const &Record,
779                                      pkgSrcRecords::File const &File) const
780 {
781    string Res;
782    Res = ::URI::SiteOnly(URI) + ' ';
783    if (Dist[Dist.size() - 1] == '/')
784    {
785       if (Dist != "/")
786          Res += Dist;
787    }      
788    else
789       Res += Dist + '/' + Section;
790    
791    Res += " ";
792    Res += Record.Package();
793    Res += " ";
794    Res += Record.Version();
795    if (File.Type.empty() == false)
796       Res += " (" + File.Type + ")";
797    return Res;
798 }
799
800 // DatabaseIndex::rpmDatabaseIndex - Constructor                        /*{{{*/
801 // ---------------------------------------------------------------------
802 /* */
803 rpmDatabaseIndex::rpmDatabaseIndex()
804 {
805 }
806                                                                         /*}}}*/
807 // DatabaseIndex::Size - Return the size of the index                   /*{{{*/
808 // ---------------------------------------------------------------------
809 /* */
810 unsigned long rpmDatabaseIndex::Size() const
811 {
812    return rpmSys.GetDBHandler()->Size();
813 }
814                                                                         /*}}}*/
815 // DatabaseIndex::CreateHandler - Create a RPMHandler for this file     /*{{{*/
816 // ---------------------------------------------------------------------
817 RPMHandler *rpmDatabaseIndex::CreateHandler() const
818 {
819    return rpmSys.GetDBHandler();
820 }
821                                                                         /*}}}*/
822 // DatabaseIndex::Merge - Load the index file into a cache              /*{{{*/
823 // ---------------------------------------------------------------------
824 /* */
825 bool rpmDatabaseIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
826 {
827    RPMDBHandler *Handler = rpmSys.GetDBHandler();
828    rpmListParser Parser(Handler);
829    if (_error->PendingError() == true)
830       return _error->Error(_("Problem opening RPM database"));
831    
832    Prog.SubProgress(0,"RPM Database");
833    if (Gen.SelectFile(Handler->DataPath(false),string(),*this,pkgCache::Flag::NotSource) == false)
834       return _error->Error(_("Problem with SelectFile RPM Database"));
835
836    // Store the IMS information
837    pkgCache::PkgFileIterator CFile = Gen.GetCurFile();
838    struct stat St;
839    if (stat(Handler->DataPath(false).c_str(),&St) != 0)
840       return _error->Errno("fstat",_("Failed to stat %s"), Handler->DataPath(false).c_str());
841    CFile->Size = St.st_size;
842    CFile->mtime = Handler->Mtime();
843    
844    if (Gen.MergeList(Parser) == false)
845       return _error->Error(_("Problem with MergeList %s"),
846                            Handler->DataPath(false).c_str());
847    return true;
848 }
849                                                                         /*}}}*/
850 // DatabaseIndex::MergeFileProvides - Process file dependencies if any  /*{{{*/
851 // ---------------------------------------------------------------------
852 /* */
853 bool rpmDatabaseIndex::MergeFileProvides(pkgCacheGenerator &Gen,
854                                          OpProgress &Prog) const
855 {
856    RPMDBHandler *Handler = rpmSys.GetDBHandler();
857    rpmListParser Parser(Handler);
858    if (_error->PendingError() == true)
859       return _error->Error(_("Problem opening RPM database"));
860    // We call SubProgress with Size(), since we won't call SelectFile() here.
861    Prog.SubProgress(Size(),"RPM Database");
862    if (Gen.MergeFileProvides(Parser) == false)
863       return _error->Error(_("Problem with MergeFileProvides %s"),
864                            Handler->DataPath(false).c_str());
865    return true;
866 }
867                                                                         /*}}}*/
868 // DatabaseIndex::FindInCache - Find this index                         /*{{{*/
869 // ---------------------------------------------------------------------
870 /* */
871 pkgCache::PkgFileIterator rpmDatabaseIndex::FindInCache(pkgCache &Cache) const
872 {
873    pkgCache::PkgFileIterator File = Cache.FileBegin();
874    for (; File.end() == false; File++)
875    {
876       if (rpmSys.GetDBHandler()->DataPath(false) != File.FileName())
877          continue;
878       struct stat St;
879       if (stat(File.FileName(),&St) != 0)
880          return pkgCache::PkgFileIterator(Cache);
881       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
882          return pkgCache::PkgFileIterator(Cache);
883       return File;
884    }   
885    return File;
886 }
887                                                                         /*}}}*/
888
889 // Source List types for rpm                                            /*{{{*/
890
891 class rpmSLTypeGen : public pkgSourceList::Type
892 {
893    public:
894
895    rpmSLTypeGen()
896    {
897       Name = "rpm";
898       Label = "Standard RPM source tree";
899    }   
900
901    pkgRepository *FindRepository(string URI,string Dist,
902                                  const pkgSourceList::Vendor *Vendor) const
903    {
904       for (vector<pkgRepository *>::const_iterator iter = RepList.begin();
905            iter != RepList.end(); iter++) 
906       {
907          if ((*iter)->URI == URI && (*iter)->Dist == Dist) 
908          {       
909             if (Vendor != NULL)
910                (*iter)->FingerPrint = Vendor->FingerPrint;
911             return *iter;
912          }
913       }
914       return NULL;
915    }
916
917    pkgRepository *GetRepository(string URI,string Dist,
918                                 const pkgSourceList::Vendor *Vendor) const
919    {
920       pkgRepository *Rep = FindRepository(URI,Dist,Vendor);
921       if (Rep != NULL)
922          return Rep;
923
924       string BaseURI;
925       if (Dist[Dist.size() - 1] == '/')
926       {
927          if (Dist != "/")
928             BaseURI = URI + Dist;
929          else 
930             BaseURI = URI + '/';
931       }
932       else
933          BaseURI = URI + Dist + '/';
934
935       Rep = new pkgRepository(URI,Dist,Vendor,BaseURI);
936       RepList.push_back(Rep);
937       return Rep;
938    }
939 };
940
941
942 class rpmSLTypeRpm : public rpmSLTypeGen
943 {
944    public:
945
946    bool CreateItem(vector<pkgIndexFile *> &List,
947                    string URI, string Dist, string Section,
948                    pkgSourceList::Vendor const *Vendor) const 
949    {
950       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
951       List.push_back(new rpmPkgListIndex(URI,Dist,Section,Rep));
952       return true;
953    };
954
955    rpmSLTypeRpm()
956    {
957       Name = "rpm";
958       Label = "Standard RPM binary tree";
959    }   
960 };
961
962 class rpmSLTypeSrpm : public rpmSLTypeGen
963 {
964    public:
965
966    bool CreateItem(vector<pkgIndexFile *> &List,
967                    string URI, string Dist, string Section,
968                    pkgSourceList::Vendor const *Vendor) const 
969    {
970       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
971       List.push_back(new rpmSrcListIndex(URI,Dist,Section,Rep));
972       return true;
973    };  
974    
975    rpmSLTypeSrpm()
976    {
977       Name = "rpm-src";
978       Label = "Standard RPM source tree";
979    }   
980 };
981
982 class rpmSLTypeRpmDir : public rpmSLTypeGen
983 {
984    public:
985
986    bool CreateItem(vector<pkgIndexFile *> &List,
987                    string URI, string Dist, string Section,
988                    pkgSourceList::Vendor const *Vendor) const 
989    {
990       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
991       List.push_back(new rpmPkgDirIndex(URI,Dist,Section,Rep));
992       return true;
993    };
994
995    rpmSLTypeRpmDir()
996    {
997       Name = "rpm-dir";
998       Label = "Local RPM directory tree";
999    }   
1000 };
1001
1002 class rpmSLTypeSrpmDir : public rpmSLTypeGen
1003 {
1004    public:
1005
1006    bool CreateItem(vector<pkgIndexFile *> &List,
1007                    string URI, string Dist, string Section,
1008                    pkgSourceList::Vendor const *Vendor) const 
1009    {
1010       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1011       List.push_back(new rpmSrcDirIndex(URI,Dist,Section,Rep));
1012       return true;
1013    };
1014
1015    rpmSLTypeSrpmDir()
1016    {
1017       Name = "rpm-src-dir";
1018       Label = "Local SRPM directory tree";
1019    }   
1020 };
1021
1022 class rpmSLTypeRepomd : public rpmSLTypeGen
1023 {
1024    public:
1025
1026    pkgRepository *GetRepository(string URI,string Dist,
1027                                 const pkgSourceList::Vendor *Vendor) const
1028    {
1029       pkgRepository *Rep = FindRepository(URI,Dist,Vendor);
1030       if (Rep != NULL)
1031          return Rep;
1032
1033       string BaseURI;
1034       if (Dist[Dist.size() - 1] == '/')
1035       {
1036          if (Dist != "/")
1037             BaseURI = URI + Dist;
1038          else 
1039             BaseURI = URI + '/';
1040       }
1041       else
1042          BaseURI = URI + Dist + '/';
1043
1044       Rep = new repomdRepository(URI,Dist,Vendor,BaseURI);
1045       RepList.push_back(Rep);
1046       return Rep;
1047    }
1048
1049    bool CreateItem(vector<pkgIndexFile *> &List,
1050                    string URI, string Dist, string Section,
1051                    pkgSourceList::Vendor const *Vendor) const 
1052    {
1053       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1054       List.push_back(new rpmRepomdPkgIndex(URI,Dist,Section,Rep));
1055       return true;
1056    };
1057
1058    bool ParseLine(vector<pkgIndexFile *> &List,
1059                   pkgSourceList::Vendor const *Vendor,
1060                   const char *Buffer,
1061                   unsigned long CurLine,string File) const
1062    {
1063       string URI;
1064       string Dist;
1065       if (ParseQuoteWord(Buffer,URI) == false)
1066          return _error->Error(_("Malformed line %lu in source list %s (URI)"),CurLine,File.c_str());
1067       if (ParseQuoteWord(Buffer,Dist) == false)
1068          return _error->Error(_("Malformed line %lu in source list %s (dist)"),CurLine,File.c_str());
1069
1070       if (FixupURI(URI) == false)
1071          return _error->Error(_("Malformed line %lu in source list %s (URI parse)"),CurLine,File.c_str());
1072       
1073       Dist = SubstVar(Dist,"$(ARCH)",_config->Find("APT::Architecture"));
1074       Dist = SubstVar(Dist,"$(VERSION)",_config->Find("APT::DistroVersion"));
1075
1076       if (CreateItem(List,URI,Dist,"",Vendor) == false)
1077          return false;
1078
1079       return true;
1080    };
1081
1082
1083    rpmSLTypeRepomd()
1084    {
1085       Name = "repomd";
1086       Label = "RepoMD tree";
1087    }   
1088 };
1089 class rpmSLTypeRepomdSrc : public rpmSLTypeRepomd
1090 {
1091    public:
1092
1093    bool CreateItem(vector<pkgIndexFile *> &List,
1094                    string URI, string Dist, string Section,
1095                    pkgSourceList::Vendor const *Vendor) const 
1096    {
1097       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1098       List.push_back(new rpmRepomdSrcIndex(URI,Dist,Section,Rep));
1099       return true;
1100    };
1101
1102    rpmSLTypeRepomdSrc()
1103    {
1104       Name = "repomd-src";
1105       Label = "RepoMD src tree";
1106    }   
1107 };
1108
1109 rpmSLTypeRpm _apt_rpmType;
1110 rpmSLTypeSrpm _apt_rpmSrcType;
1111 rpmSLTypeRpmDir _apt_rpmDirType;
1112 rpmSLTypeSrpmDir _apt_rpmSrcDirType;
1113 rpmSLTypeRepomd _apt_repomdType;
1114 rpmSLTypeRepomdSrc _apt_repomdSrcType;
1115                                                                         /*}}}*/
1116 // Index File types for rpm                                             /*{{{*/
1117 class rpmIFTypeSrc : public pkgIndexFile::Type
1118 {
1119    public:
1120    
1121    rpmIFTypeSrc() {Label = "RPM Source Index";};
1122 };
1123 class rpmIFTypePkg : public pkgIndexFile::Type
1124 {
1125    public:
1126    
1127    virtual pkgRecords::Parser *CreatePkgParser(pkgCache::PkgFileIterator File) const
1128    {
1129       return new rpmRecordParser(File.FileName(),*File.Cache());
1130    };
1131    rpmIFTypePkg() {Label = "RPM Package Index";};
1132 };
1133 class rpmIFTypeDatabase : public pkgIndexFile::Type
1134 {
1135    public:
1136    
1137    virtual pkgRecords::Parser *CreatePkgParser(pkgCache::PkgFileIterator File) const
1138    {
1139       return new rpmRecordParser(File.FileName(),*File.Cache());
1140    };
1141    rpmIFTypeDatabase() {Label = "RPM Database";};
1142 };
1143 static rpmIFTypeSrc _apt_Src;
1144 static rpmIFTypePkg _apt_Pkg;
1145 static rpmIFTypeDatabase _apt_DB;
1146
1147 const pkgIndexFile::Type *rpmSrcListIndex::GetType() const
1148 {
1149    return &_apt_Src;
1150 }
1151 const pkgIndexFile::Type *rpmPkgListIndex::GetType() const
1152 {
1153    return &_apt_Pkg;
1154 }
1155 const pkgIndexFile::Type *rpmSrcDirIndex::GetType() const
1156 {
1157    return &_apt_Src;
1158 }
1159 const pkgIndexFile::Type *rpmPkgDirIndex::GetType() const
1160 {
1161    return &_apt_Pkg;
1162 }
1163 const pkgIndexFile::Type *rpmSinglePkgIndex::GetType() const
1164 {
1165    return &_apt_Pkg;
1166 }
1167 const pkgIndexFile::Type *rpmSingleSrcIndex::GetType() const
1168 {
1169    return &_apt_Src;
1170 }
1171 const pkgIndexFile::Type *rpmDatabaseIndex::GetType() const
1172 {
1173    return &_apt_DB;
1174 }
1175 const pkgIndexFile::Type *rpmRepomdPkgIndex::GetType() const
1176 {
1177    return &_apt_Pkg;
1178 }
1179 const pkgIndexFile::Type *rpmRepomdSrcIndex::GetType() const
1180 {
1181    return &_apt_Src;
1182 }
1183
1184                                                                         /*}}}*/
1185 #endif /* HAVE_RPM */
1186
1187 // vim:sts=3:sw=3