]> git.decadent.org.uk Git - dak.git/blob - dak/generate_index_diffs.py
remove UserExtensions
[dak.git] / dak / generate_index_diffs.py
1 #!/usr/bin/env python
2
3 """ generates partial package updates list"""
4
5 ###########################################################
6
7 # idea and basic implementation by Anthony, some changes by Andreas
8 # parts are stolen from 'dak generate-releases'
9 #
10 # Copyright (C) 2004, 2005, 2006  Anthony Towns <aj@azure.humbug.org.au>
11 # Copyright (C) 2004, 2005  Andreas Barth <aba@not.so.argh.org>
12
13 # This program is free software; you can redistribute it and/or modify
14 # it under the terms of the GNU General Public License as published by
15 # the Free Software Foundation; either version 2 of the License, or
16 # (at your option) any later version.
17
18 # This program is distributed in the hope that it will be useful,
19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21 # GNU General Public License for more details.
22
23 # You should have received a copy of the GNU General Public License
24 # along with this program; if not, write to the Free Software
25 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
26
27
28 # < elmo> bah, don't bother me with annoying facts
29 # < elmo> I was on a roll
30
31
32 ################################################################################
33
34 import sys
35 import os
36 import tempfile
37 import subprocess
38 import time
39 import apt_pkg
40
41 from daklib import utils
42 from daklib.dbconn import get_suite
43
44 ################################################################################
45
46 Cnf = None
47 Logger = None
48 Options = None
49
50 ################################################################################
51
52 def usage (exit_code=0):
53     print """Usage: dak generate-index-diffs [OPTIONS] [suites]
54 Write out ed-style diffs to Packages/Source lists
55
56   -h, --help            show this help and exit
57   -c                    give the canonical path of the file
58   -p                    name for the patch (defaults to current time)
59   -n                    take no action
60     """
61     sys.exit(exit_code)
62
63
64 def tryunlink(file):
65     try:
66         os.unlink(file)
67     except OSError:
68         print "warning: removing of %s denied" % (file)
69
70 def smartstat(file):
71     for ext in ["", ".gz", ".bz2"]:
72         if os.path.isfile(file + ext):
73             return (ext, os.stat(file + ext))
74     return (None, None)
75
76 def smartlink(f, t):
77     if os.path.isfile(f):
78         os.link(f,t)
79     elif os.path.isfile("%s.gz" % (f)):
80         os.system("gzip -d < %s.gz > %s" % (f, t))
81     elif os.path.isfile("%s.bz2" % (f)):
82         os.system("bzip2 -d < %s.bz2 > %s" % (f, t))
83     else:
84         print "missing: %s" % (f)
85         raise IOError, f
86
87 def smartopen(file):
88     if os.path.isfile(file):
89         f = open(file, "r")
90     elif os.path.isfile("%s.gz" % file):
91         f = create_temp_file(os.popen("zcat %s.gz" % file, "r"))
92     elif os.path.isfile("%s.bz2" % file):
93         f = create_temp_file(os.popen("bzcat %s.bz2" % file, "r"))
94     else:
95         f = None
96     return f
97
98 def pipe_file(f, t):
99     f.seek(0)
100     while 1:
101         l = f.read()
102         if not l: break
103         t.write(l)
104     t.close()
105
106 class Updates:
107     def __init__(self, readpath = None, max = 14):
108         self.can_path = None
109         self.history = {}
110         self.history_order = []
111         self.max = max
112         self.readpath = readpath
113         self.filesizesha1 = None
114
115         if readpath:
116             try:
117                 f = open(readpath + "/Index")
118                 x = f.readline()
119
120                 def read_hashs(ind, f, self, x=x):
121                     while 1:
122                         x = f.readline()
123                         if not x or x[0] != " ": break
124                         l = x.split()
125                         if not self.history.has_key(l[2]):
126                             self.history[l[2]] = [None,None]
127                             self.history_order.append(l[2])
128                         self.history[l[2]][ind] = (l[0], int(l[1]))
129                     return x
130
131                 while x:
132                     l = x.split()
133
134                     if len(l) == 0:
135                         x = f.readline()
136                         continue
137
138                     if l[0] == "SHA1-History:":
139                         x = read_hashs(0,f,self)
140                         continue
141
142                     if l[0] == "SHA1-Patches:":
143                         x = read_hashs(1,f,self)
144                         continue
145
146                     if l[0] == "Canonical-Name:" or l[0]=="Canonical-Path:":
147                         self.can_path = l[1]
148
149                     if l[0] == "SHA1-Current:" and len(l) == 3:
150                         self.filesizesha1 = (l[1], int(l[2]))
151
152                     x = f.readline()
153
154             except IOError:
155                 0
156
157     def dump(self, out=sys.stdout):
158         if self.can_path:
159             out.write("Canonical-Path: %s\n" % (self.can_path))
160
161         if self.filesizesha1:
162             out.write("SHA1-Current: %s %7d\n" % (self.filesizesha1))
163
164         hs = self.history
165         l = self.history_order[:]
166
167         cnt = len(l)
168         if cnt > self.max:
169             for h in l[:cnt-self.max]:
170                 tryunlink("%s/%s.gz" % (self.readpath, h))
171                 del hs[h]
172             l = l[cnt-self.max:]
173             self.history_order = l[:]
174
175         out.write("SHA1-History:\n")
176         for h in l:
177             out.write(" %s %7d %s\n" % (hs[h][0][0], hs[h][0][1], h))
178         out.write("SHA1-Patches:\n")
179         for h in l:
180             out.write(" %s %7d %s\n" % (hs[h][1][0], hs[h][1][1], h))
181
182 def create_temp_file(r):
183     f = tempfile.TemporaryFile()
184     while 1:
185         x = r.readline()
186         if not x: break
187         f.write(x)
188     r.close()
189     del x,r
190     f.flush()
191     f.seek(0)
192     return f
193
194 def sizesha1(f):
195     size = os.fstat(f.fileno())[6]
196     f.seek(0)
197     sha1sum = apt_pkg.sha1sum(f)
198     return (sha1sum, size)
199
200 def genchanges(Options, outdir, oldfile, origfile, maxdiffs = 14):
201     if Options.has_key("NoAct"):
202         return
203
204     patchname = Options["PatchName"]
205
206     # origfile = /path/to/Packages
207     # oldfile  = ./Packages
208     # newfile  = ./Packages.tmp
209     # difffile = outdir/patchname
210     # index   => outdir/Index
211
212     # (outdir, oldfile, origfile) = argv
213
214     newfile = oldfile + ".new"
215     difffile = "%s/%s" % (outdir, patchname)
216
217     upd = Updates(outdir, int(maxdiffs))
218     (oldext, oldstat) = smartstat(oldfile)
219     (origext, origstat) = smartstat(origfile)
220     if not origstat:
221         print "%s: doesn't exist" % (origfile)
222         return
223     if not oldstat:
224         print "%s: initial run" % (origfile)
225         os.link(origfile + origext, oldfile + origext)
226         return
227
228     if oldstat[1:3] == origstat[1:3]:
229         print "%s: hardlink unbroken, assuming unchanged" % (origfile)
230         return
231
232     oldf = smartopen(oldfile)
233     oldsizesha1 = sizesha1(oldf)
234
235     # should probably early exit if either of these checks fail
236     # alternatively (optionally?) could just trim the patch history
237
238     if upd.filesizesha1:
239         if upd.filesizesha1 != oldsizesha1:
240             print "info: old file " + oldfile + " changed! %s %s => %s %s" % (upd.filesizesha1 + oldsizesha1)
241
242     if Options.has_key("CanonicalPath"): upd.can_path=Options["CanonicalPath"]
243
244     if os.path.exists(newfile): os.unlink(newfile)
245     smartlink(origfile, newfile)
246     newf = open(newfile, "r")
247     newsizesha1 = sizesha1(newf)
248     newf.close()
249
250     if newsizesha1 == oldsizesha1:
251         os.unlink(newfile)
252         oldf.close()
253         print "%s: unchanged" % (origfile)
254     else:
255         if not os.path.isdir(outdir):
256             os.mkdir(outdir)
257
258         w = os.popen("diff --ed - %s | gzip -c -9 > %s.gz" %
259                      (newfile, difffile), "w")
260         pipe_file(oldf, w)
261         oldf.close()
262
263         difff = smartopen(difffile)
264         difsizesha1 = sizesha1(difff)
265         difff.close()
266
267         upd.history[patchname] = (oldsizesha1, difsizesha1)
268         upd.history_order.append(patchname)
269
270         upd.filesizesha1 = newsizesha1
271
272         os.unlink(oldfile + oldext)
273         os.link(origfile + origext, oldfile + origext)
274         os.unlink(newfile)
275
276         f = open(outdir + "/Index", "w")
277         upd.dump(f)
278         f.close()
279
280
281 def main():
282     global Cnf, Options, Logger
283
284     os.umask(0002)
285
286     Cnf = utils.get_conf()
287     Arguments = [ ('h', "help", "Generate-Index-Diffs::Options::Help"),
288                   ('c', None, "Generate-Index-Diffs::Options::CanonicalPath", "hasArg"),
289                   ('p', "patchname", "Generate-Index-Diffs::Options::PatchName", "hasArg"),
290                   ('r', "rootdir", "Generate-Index-Diffs::Options::RootDir", "hasArg"),
291                   ('d', "tmpdir", "Generate-Index-Diffs::Options::TempDir", "hasArg"),
292                   ('m', "maxdiffs", "Generate-Index-Diffs::Options::MaxDiffs", "hasArg"),
293                   ('n', "n-act", "Generate-Index-Diffs::Options::NoAct"),
294                 ]
295     suites = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
296     Options = Cnf.SubTree("Generate-Index-Diffs::Options")
297     if Options.has_key("Help"): usage()
298
299     maxdiffs = Options.get("MaxDiffs::Default", "14")
300     maxpackages = Options.get("MaxDiffs::Packages", maxdiffs)
301     maxcontents = Options.get("MaxDiffs::Contents", maxdiffs)
302     maxsources = Options.get("MaxDiffs::Sources", maxdiffs)
303
304     if not Options.has_key("PatchName"):
305         format = "%Y-%m-%d-%H%M.%S"
306         Options["PatchName"] = time.strftime( format )
307
308     AptCnf = apt_pkg.newConfiguration()
309     apt_pkg.ReadConfigFileISC(AptCnf,utils.which_apt_conf_file())
310
311     if Options.has_key("RootDir"): Cnf["Dir::Root"] = Options["RootDir"]
312
313     if not suites:
314         suites = Cnf.SubTree("Suite").List()
315
316     for suitename in suites:
317         print "Processing: " + suite
318         SuiteBlock = Cnf.SubTree("Suite::" + suitename)
319
320         suiteobj = get_suite(suitename.lower())
321
322         # Use the canonical version of the suite name
323         suite = suite.suite_name
324
325         if suiteobj.untouchable:
326             print "Skipping: " + suite + " (untouchable)"
327             continue
328
329         architectures = get_suite_architectures(suite, skipall=True)
330
331         if SuiteBlock.has_key("Components"):
332             components = SuiteBlock.ValueList("Components")
333         else:
334             components = []
335
336         suite_suffix = Cnf.Find("Dinstall::SuiteSuffix")
337         if components and suite_suffix:
338             longsuite = suite + "/" + suite_suffix
339         else:
340             longsuite = suite
341
342         tree = SuiteBlock.get("Tree", "dists/%s" % (longsuite))
343
344         if AptCnf.has_key("tree::%s" % (tree)):
345             sections = AptCnf["tree::%s::Sections" % (tree)].split()
346         elif AptCnf.has_key("bindirectory::%s" % (tree)):
347             sections = AptCnf["bindirectory::%s::Sections" % (tree)].split()
348         else:
349             aptcnf_filename = os.path.basename(utils.which_apt_conf_file())
350             print "ALERT: suite %s not in %s, nor untouchable!" % (suite, aptcnf_filename)
351             continue
352
353         for archobj in architectures:
354             architecture = archobj.arch_string
355
356             if architecture != "source":
357                 # Process Contents
358                 file = "%s/Contents-%s" % (Cnf["Dir::Root"] + tree,
359                         architecture)
360                 storename = "%s/%s_contents_%s" % (Options["TempDir"], suite, architecture)
361                 genchanges(Options, file + ".diff", storename, file, \
362                   Cnf.get("Suite::%s::Generate-Index-Diffs::MaxDiffs::Contents" % (suite), maxcontents))
363
364             # use sections instead of components since dak.conf
365             # treats "foo/bar main" as suite "foo", suitesuffix "bar" and
366             # component "bar/main". suck.
367
368             for component in sections:
369                 if architecture == "source":
370                     longarch = architecture
371                     packages = "Sources"
372                     maxsuite = maxsources
373                 else:
374                     longarch = "binary-%s"% (architecture)
375                     packages = "Packages"
376                     maxsuite = maxpackages
377
378                 file = "%s/%s/%s/%s" % (Cnf["Dir::Root"] + tree,
379                            component, longarch, packages)
380                 storename = "%s/%s_%s_%s" % (Options["TempDir"], suite, component, architecture)
381                 genchanges(Options, file + ".diff", storename, file, \
382                   Cnf.get("Suite::%s::Generate-Index-Diffs::MaxDiffs::%s" % (suite, packages), maxsuite))
383
384 ################################################################################
385
386 if __name__ == '__main__':
387     main()