]> git.saurik.com Git - apt.git/blob - ftparchive/writer.h
merged patches from david (many thanks)
[apt.git] / 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