]> git.saurik.com Git - apt.git/blob - ftparchive/writer.h
* Applied patch from Guillem Jover <guillem.jover@nokia.com> to add
[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 cout << 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
90 // General options
91 string PathPrefix;
92 string DirStrip;
93 FILE *Output;
94 struct CacheDB::Stats &Stats;
95 string Arch;
96
97 inline bool ReadOverride(string File) {return Over.ReadOverride(File);};
98 inline bool ReadExtraOverride(string File)
99 {return Over.ReadExtraOverride(File);};
100 virtual bool DoPackage(string FileName);
101
102 PackagesWriter(string DB,string Overrides,string ExtOverrides=string(),
103 string Arch=string());
104 virtual ~PackagesWriter() {};
105 };
106
107 class ContentsWriter : public FTWScanner
108 {
109 CacheDB Db;
110
111 GenContents Gen;
112
113 public:
114
115 // General options
116 FILE *Output;
117 struct CacheDB::Stats &Stats;
118 string Prefix;
119
120 bool DoPackage(string FileName,string Package);
121 virtual bool DoPackage(string FileName)
122 {return DoPackage(FileName,string());};
123 bool ReadFromPkgs(string PkgFile,string PkgCompress);
124
125 void Finish() {Gen.Print(Output);};
126 inline bool ReadyDB(string DB) {return Db.ReadyDB(DB);};
127
128 ContentsWriter(string DB);
129 virtual ~ContentsWriter() {};
130 };
131
132 class SourcesWriter : public FTWScanner
133 {
134 Override BOver;
135 Override SOver;
136 char *Buffer;
137 unsigned long BufSize;
138
139 public:
140
141 bool NoOverride;
142
143 // General options
144 string PathPrefix;
145 string DirStrip;
146 FILE *Output;
147 struct CacheDB::Stats Stats;
148
149 virtual bool DoPackage(string FileName);
150
151 SourcesWriter(string BOverrides,string SOverrides,
152 string ExtOverrides=string());
153 virtual ~SourcesWriter() {free(Buffer);};
154 };
155
156 class ReleaseWriter : public FTWScanner
157 {
158 public:
159 ReleaseWriter(string DB);
160 virtual bool DoPackage(string FileName);
161 void Finish();
162
163 FILE *Output;
164 // General options
165 string PathPrefix;
166 string DirStrip;
167
168 protected:
169 struct CheckSum
170 {
171 string MD5;
172 string SHA1;
173 string SHA256;
174 // Limited by FileFd::Size()
175 unsigned long size;
176 ~CheckSum() {};
177 };
178 map<string,struct CheckSum> CheckSums;
179 };
180
181 #endif