From: Zac Medico <zmedico@gentoo.org>
Date: Fri, 28 Jul 2006 04:58:45 +0000 (-0000)
Subject: Add a --force option for `ebuild digest` so that the user doesn't have to manually... 
X-Git-Tag: v2.1.1~129
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=83dda008fba7d8dad213de5636b1847338ac8737;p=portage.git

Add a --force option for `ebuild digest` so that the user doesn't have to manually remove the Manifest and files/digest-* files when upstream changes the identity of distfiles.  See bug #141843.

svn path=/main/trunk/; revision=4041
---

diff --git a/bin/ebuild b/bin/ebuild
index a62585a5e..64375298a 100755
--- a/bin/ebuild
+++ b/bin/ebuild
@@ -12,9 +12,9 @@ if len(sys.argv) <= 2:
 	sys.exit(1)
 
 
-(opts, pargs) = getopt.getopt(sys.argv[1:], '', ['debug'])
+opts, pargs = getopt.getopt(sys.argv[1:], '', ['debug', 'force'])
 debug = ("--debug",'') in opts
-
+force = ("--force",'') in opts
 
 if "merge" in pargs:
 	print "Disabling noauto in features... merge disables it. (qmerge doesn't)"
@@ -79,10 +79,31 @@ if len(pargs) > 1 and "config" in pargs:
 	print "config must be called on it's own, not combined with any other phase"
 	sys.exit(1)
 
+def discard_digests(myebuild, mysettings, mydbapi):
+	"""Discard all distfiles digests for the given ebuild.  This is useful when
+	upstream has changed the identity of the distfiles and the user would
+	otherwise have to manually remove the Manifest and files/digest-* files in
+	order to ensure correct results."""
+	pkgdir = os.path.dirname(myebuild)
+	cat, pkg = pkgdir.split(os.sep)[-2:]
+	cpv = cat + "/" + os.path.basename(myebuild)[:-7]
+	from portage_manifest import Manifest
+	mf = Manifest(pkgdir, mysettings["DISTDIR"],
+		fetchlist_dict=portage.FetchlistDict(pkgdir, mysettings, mydbapi))
+	mf.create([], assumeDistHashesSometimes=True, assumeDistHashesAlways=True)
+	distfiles = mydbapi.getfetchlist(cpv, mysettings=mysettings, all=True)[1]
+	for myfile in distfiles:
+		try:
+			del mf.fhashdict["DIST"][myfile]
+		except KeyError:
+			pass
+	mf.write()
 
 for arg in pargs:
 	try:
 		tmpsettings = portage.config(clone=portage.settings)
+		if arg == "digest" and force:
+			discard_digests(ebuild, tmpsettings, portage.portdb)
 		a = portage.doebuild(ebuild, arg, portage.root, tmpsettings, debug=debug, cleanup=("noauto" not in portage.features), tree=mytree)
 	except KeyboardInterrupt:
 		print "Interrupted."
diff --git a/pym/portage_manifest.py b/pym/portage_manifest.py
index ee9b4b833..8fbe6bc62 100644
--- a/pym/portage_manifest.py
+++ b/pym/portage_manifest.py
@@ -219,15 +219,14 @@ class Manifest(object):
 		for cpv in cpvlist:
 			dname = os.path.join(self.pkgdir, "files", "digest-%s" % self._catsplit(cpv)[1])
 			distlist = self._getCpvDistfiles(cpv)
-			have_all_checksums = True
+			missing_digests = set()
 			for f in distlist:
 				if f not in self.fhashdict["DIST"] or len(self.fhashdict["DIST"][f]) == 0:
-					have_all_checksums = False
-					break
-			if not have_all_checksums:
-				# We don't have all the required checksums to generate a proper
-				# digest, so we have to skip this cpv.
-				continue
+					missing_digests.add(f)
+			if missing_digests:
+				# This allows us to force remove of stale digests for the
+				# ebuild --force digest option.
+				distlist = [f for f in distlist if f not in missing_digests]
 			update_digest = True
 			if not force:
 				try:
@@ -236,13 +235,15 @@ class Manifest(object):
 					f.close()
 					if len(old_data) == 1 and "DIST" in old_data:
 						new_data = self._getDigestData(distlist)
-						for myfile in new_data["DIST"]:
-							for hashname in new_data["DIST"][myfile].keys():
-								if hashname != "size" and \
-								hashname not in portage_const.MANIFEST1_HASH_FUNCTIONS:
-									del new_data["DIST"][myfile][hashname]
-						if new_data["DIST"] == old_data["DIST"]:
-							update_digest = False
+						if "DIST" in new_data:
+							for myfile in new_data["DIST"]:
+								for hashname in \
+									new_data["DIST"][myfile].keys():
+									if hashname != "size" and hashname not in \
+										portage_const.MANIFEST1_HASH_FUNCTIONS:
+										del new_data["DIST"][myfile][hashname]
+							if new_data["DIST"] == old_data["DIST"]:
+								update_digest = False
 				except (IOError, OSError), e:
 					if errno.ENOENT == e.errno:
 						pass