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