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 #include <set>
  23
  24 #include "cachedb.h"
  25 #include "override.h"
  26 #include "apt-ftparchive.h"
  27
  28 using std::string;
  29 using std::cout;
  30 using std::endl;
  31 using std::vector;
  32 using std::map;
  33
  34 class FTWScanner
  35 {
  36    protected:
  37    vector<string> Patterns;
  38    string Arch;
  39    const char *OriginalPath;
  40    bool ErrorPrinted;
  41
  42    // Stuff for the delinker
  43    bool NoLinkAct;
  44
  45    static FTWScanner *Owner;
  46    static int ScannerFTW(const char *File,const struct stat *sb,int Flag);
  47    static int ScannerFile(const char *File, bool const &ReadLink);
  48
  49    bool Delink(string &FileName,const char *OriginalPath,
  50                unsigned long &Bytes,off_t const &FileSize);
  51
  52    inline void NewLine(unsigned const &Priority)
  53    {
  54       if (ErrorPrinted == false && Quiet <= Priority)
  55       {
  56          c1out << endl;
  57          ErrorPrinted = true;
  58       }
  59    }
  60
  61    public:
  62
  63    unsigned long DeLinkLimit;
  64    string InternalPrefix;
  65
  66    virtual bool DoPackage(string FileName) = 0;
  67    bool RecursiveScan(string const &Dir);
  68    bool LoadFileList(string const &BaseDir,string const &File);
  69    void ClearPatterns() { Patterns.clear(); };
  70    void AddPattern(string const &Pattern) { Patterns.push_back(Pattern); };
  71    bool SetExts(string const &Vals);
  72
  73    FTWScanner(string const &Arch = string());
  74 };
  75
  76 class TranslationWriter
  77 {
  78    FILE *Output;
  79    std::set<string> Included;
  80    unsigned short RefCounter;
  81
  82    public:
  83    void IncreaseRefCounter() { ++RefCounter; };
  84    unsigned short DecreaseRefCounter() { return (RefCounter == 0) ? 0 : --RefCounter; };
  85    unsigned short GetRefCounter() const { return RefCounter; };
  86    bool DoPackage(string const &Pkg, string const &Desc, string const &MD5);
  87
  88    TranslationWriter(string const &File);
  89    TranslationWriter() : Output(NULL), RefCounter(0) {};
  90    ~TranslationWriter();
  91 };
  92
  93 class PackagesWriter : public FTWScanner
  94 {
  95    Override Over;
  96    CacheDB Db;
  97
  98    public:
  99
 100    // Some flags
 101    bool DoMD5;
 102    bool DoSHA1;
 103    bool DoSHA256;
 104    bool DoAlwaysStat;
 105    bool NoOverride;
 106    bool DoContents;
 107    bool LongDescription;
 108
 109    // General options
 110    string PathPrefix;
 111    string DirStrip;
 112    FILE *Output;
 113    struct CacheDB::Stats &Stats;
 114    TranslationWriter *TransWriter;
 115
 116    inline bool ReadOverride(string const &File) {return Over.ReadOverride(File);};
 117    inline bool ReadExtraOverride(string const &File)
 118       {return Over.ReadExtraOverride(File);};
 119    virtual bool DoPackage(string FileName);
 120
 121    PackagesWriter(string const &DB,string const &Overrides,string const &ExtOverrides=string(),
 122                   string const &Arch=string());
 123    virtual ~PackagesWriter() {};
 124 };
 125
 126 class ContentsWriter : public FTWScanner
 127 {
 128    CacheDB Db;
 129
 130    GenContents Gen;
 131
 132    public:
 133
 134    // General options
 135    FILE *Output;
 136    struct CacheDB::Stats &Stats;
 137    string Prefix;
 138
 139    bool DoPackage(string FileName,string Package);
 140    virtual bool DoPackage(string FileName)
 141              {return DoPackage(FileName,string());};
 142    bool ReadFromPkgs(string const &PkgFile,string const &PkgCompress);
 143
 144    void Finish() {Gen.Print(Output);};
 145    inline bool ReadyDB(string const &DB) {return Db.ReadyDB(DB);};
 146
 147    ContentsWriter(string const &DB, string const &Arch = string());
 148    virtual ~ContentsWriter() {};
 149 };
 150
 151 class SourcesWriter : public FTWScanner
 152 {
 153    Override BOver;
 154    Override SOver;
 155    char *Buffer;
 156    unsigned long BufSize;
 157
 158    public:
 159
 160    bool NoOverride;
 161
 162    // General options
 163    string PathPrefix;
 164    string DirStrip;
 165    FILE *Output;
 166    struct CacheDB::Stats Stats;
 167
 168    virtual bool DoPackage(string FileName);
 169
 170    SourcesWriter(string const &BOverrides,string const &SOverrides,
 171                  string const &ExtOverrides=string());
 172    virtual ~SourcesWriter() {free(Buffer);};
 173 };
 174
 175 class ReleaseWriter : public FTWScanner
 176 {
 177 public:
 178    ReleaseWriter(string const &DB);
 179    virtual bool DoPackage(string FileName);
 180    void Finish();
 181
 182    FILE *Output;
 183    // General options
 184    string PathPrefix;
 185    string DirStrip;
 186
 187 protected:
 188    struct CheckSum
 189    {
 190       string MD5;
 191       string SHA1;
 192       string SHA256;
 193       // Limited by FileFd::Size()
 194       unsigned long size;
 195       ~CheckSum() {};
 196    };
 197    map<string,struct CheckSum> CheckSums;
 198 };
 199
 200 #endif