ftparchive/writer.h

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: writer.h,v 1.4.2.2 2003/12/26 22:55:43 mdz Exp $
   4 /* ######################################################################
   5
   6    Writer
   7
   8    The file writer classes. These write various types of output, sources,
   9    packages and contents.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13 #ifndef WRITER_H
  14 #define WRITER_H
  15
  16
  17 #include <string>
  18 #include <stdio.h>
  19 #include <iostream>
  20 #include <vector>
  21 #include <map>
  22
  23 #include "cachedb.h"
  24 #include "override.h"
  25 #include "apt-ftparchive.h"
  26
  27 using std::string;
  28 using std::cout;
  29 using std::endl;
  30 using std::vector;
  31 using std::map;
  32
  33 class FTWScanner
  34 {
  35    protected:
  36    vector<string> Patterns;
  37    const char *OriginalPath;
  38    char *RealPath;
  39    bool ErrorPrinted;
  40
  41    // Stuff for the delinker
  42    bool NoLinkAct;
  43
  44    static FTWScanner *Owner;
  45    static int ScannerFTW(const char *File,const struct stat *sb,int Flag);
  46    static int ScannerFile(const char *File, bool ReadLink);
  47
  48    bool Delink(string &FileName,const char *OriginalPath,
  49                unsigned long &Bytes,off_t FileSize);
  50
  51    inline void NewLine(unsigned Priority)
  52    {
  53       if (ErrorPrinted == false && Quiet <= Priority)
  54       {
  55          c1out << endl;
  56          ErrorPrinted = true;
  57       }
  58    }
  59
  60    public:
  61
  62    unsigned long DeLinkLimit;
  63    string InternalPrefix;
  64
  65    virtual bool DoPackage(string FileName) = 0;
  66    bool RecursiveScan(string Dir);
  67    bool LoadFileList(string BaseDir,string File);
  68    void ClearPatterns() { Patterns.clear(); };
  69    void AddPattern(string Pattern) { Patterns.push_back(Pattern); };
  70    bool SetExts(string Vals);
  71
  72    FTWScanner();
  73    virtual ~FTWScanner() {delete [] RealPath;};
  74 };
  75
  76 class PackagesWriter : public FTWScanner
  77 {
  78    Override Over;
  79    CacheDB Db;
  80
  81    public:
  82
  83    // Some flags
  84    bool DoMD5;
  85    bool DoSHA1;
  86    bool DoSHA256;
  87    bool NoOverride;
  88    bool DoContents;
  89    bool LongDescription;
  90
  91    // General options
  92    string PathPrefix;
  93    string DirStrip;
  94    FILE *Output;
  95    struct CacheDB::Stats &Stats;
  96    string Arch;
  97
  98    inline bool ReadOverride(string File) {return Over.ReadOverride(File);};
  99    inline bool ReadExtraOverride(string File)
 100       {return Over.ReadExtraOverride(File);};
 101    virtual bool DoPackage(string FileName);
 102
 103    PackagesWriter(string DB,string Overrides,string ExtOverrides=string(),
 104                   string Arch=string());
 105    virtual ~PackagesWriter() {};
 106 };
 107
 108 class ContentsWriter : public FTWScanner
 109 {
 110    CacheDB Db;
 111
 112    GenContents Gen;
 113
 114    public:
 115
 116    // General options
 117    FILE *Output;
 118    struct CacheDB::Stats &Stats;
 119    string Prefix;
 120
 121    bool DoPackage(string FileName,string Package);
 122    virtual bool DoPackage(string FileName)
 123              {return DoPackage(FileName,string());};
 124    bool ReadFromPkgs(string PkgFile,string PkgCompress);
 125
 126    void Finish() {Gen.Print(Output);};
 127    inline bool ReadyDB(string DB) {return Db.ReadyDB(DB);};
 128
 129    ContentsWriter(string DB);
 130    virtual ~ContentsWriter() {};
 131 };
 132
 133 class SourcesWriter : public FTWScanner
 134 {
 135    Override BOver;
 136    Override SOver;
 137    char *Buffer;
 138    unsigned long BufSize;
 139
 140    public:
 141
 142    bool NoOverride;
 143
 144    // General options
 145    string PathPrefix;
 146    string DirStrip;
 147    FILE *Output;
 148    struct CacheDB::Stats Stats;
 149
 150    virtual bool DoPackage(string FileName);
 151
 152    SourcesWriter(string BOverrides,string SOverrides,
 153                  string ExtOverrides=string());
 154    virtual ~SourcesWriter() {free(Buffer);};
 155 };
 156
 157 class ReleaseWriter : public FTWScanner
 158 {
 159 public:
 160    ReleaseWriter(string DB);
 161    virtual bool DoPackage(string FileName);
 162    void Finish();
 163
 164    FILE *Output;
 165    // General options
 166    string PathPrefix;
 167    string DirStrip;
 168
 169 protected:
 170    struct CheckSum
 171    {
 172       string MD5;
 173       string SHA1;
 174       string SHA256;
 175       // Limited by FileFd::Size()
 176       unsigned long size;
 177       ~CheckSum() {};
 178    };
 179    map<string,struct CheckSum> CheckSums;
 180 };
 181
 182 #endif