import glob
import os
import re
import rpm
import sys

ts = rpm.TransactionSet()

packages = []
for arg in sys.argv[1:]:
	packages += glob.glob(arg)

erlang_provides = {}
erlang_requires = {}

package_names = []
package_headers = {}

##
## Custom tweaks begins here
##

# These packages should be marked as noarch
package_noarch = ["emacs-erlang", "emacs-erlang-el", "erlang-doc", "xemacs-erlang", "xemacs-erlang-el"]

# These are additional Requires which cannot be picked up automatically (yet).
# TODO these should be added automatically
package_additional_requires = {
		"emacs-erlang": ["emacs-common-erlang = %{version}-%{release}", "emacs(bin) >= %{_emacs_version}" ],
		"emacs-erlang-el": ["emacs-erlang = %{version}-%{release}"],
		"erlang-dialyzer": ["graphviz"],
		"erlang-erl_interface": ["%{name}-erts%{?_isa} = %{version}-%{release}"],
		# This library is dlopened so it can't be picked up automatically by the RPM
		# dependency checker
		"erlang-erts": ["lksctp-tools"],
		"erlang-gs": ["tk"],
		# FIXME stores files/links in jpackage-utils' directory "erlang-ic": ["jpackage-utils"],
		# FIXME stores files/links in jpackage-utils' directory "erlang-jinterface": ["%{name}-erts%{?_isa} = %{version}-%{release}, "jpackage-utils"],
		"erlang-jinterface": ["%{name}-erts%{?_isa} = %{version}-%{release}"],
		"erlang-wx": ["mesa-libGL", "mesa-libGLU"],
		"xemacs-erlang": ["emacs-common-erlang = %{version}-%{release}", "xemacs(bin) >= %{_xemacs_version}"],
		"xemacs-erlang-el": ["xemacs-erlang = %{version}-%{release}"]
		}

package_additional_buildrequires = {
		"erlang-diameter": ["ed"],
		"erlang-gs": ["tcl-devel", "tk-devel"],
		# in EPEL6 on arches different from %{ix86} x86_64 we have to
		# use java-devel-gcj, so technically this requirement makes it
		# impossible to build Java support there. (Un)Fortunately we
		# already dropped full support for EPEL6
		"erlang-ic": ["java-devel"],
		"erlang-jinterface": ["java-devel"],
		"erlang-odbc": ["unixODBC-devel"],
		"erlang-wx": ["wxGTK-devel"],
		}

package_additional_obsoletes = {
		"erlang-erts": ["erlang-appmon", "erlang-docbuilder", "erlang-inviso", "erlang-pman", "erlang-toolbar", "erlang-tv"],
		}

##
## Custom tweaks ends here
##

# To match 'erlang(asn1ct_eval_ext:transform_to_EXTERNAL1994/1)'
prog = re.compile("^erlang(.*:.*/\d+)")
rpmmask = re.compile(".*\.rpm")

# iterate over all rpms
for package in sorted([p for p in packages if rpmmask.match(p)]):
	fd = os.open(package, os.O_RDONLY)
	h = ts.hdrFromFdno(fd)
	os.close(fd)

	rpm_name = h[rpm.RPMTAG_NAME]

	# Let's calculate provides
	# We'll create dictionary to speedup future use
	# FIXME duplicate provides
	local_provides = [p for p in h[rpm.RPMTAG_PROVIDENAME] if prog.match(p)]
	erlang_provides.update({k: rpm_name for k in local_provides})

	# Let's calculate requires
	erlang_requires.update({rpm_name: [r for r in h[rpm.RPMTAG_REQUIRENAME] if prog.match(r) and not r in local_provides]})

	package_headers.update({rpm_name: h})

	package_names.append(rpm_name)

namemask = re.compile("^erlang-[a-zA-Z0-9_]*$")

print "### BEGIN OF AUTOGENERATED LIST ###"
print ""

for name in package_names:
	h = package_headers[name]

	if name == "erlang-doc":
		# Additional ifdef just for this sub-package (see below)
		print "%if %{with doc}"

	if namemask.match(name):
		print "%%package %s" % name.replace("erlang-", "")
	else:
		print "%%package -n %s" % name

	print "Summary: %s" % h[rpm.RPMTAG_SUMMARY]
	print "Group: %s" % h[rpm.RPMTAG_GROUP]

	rawdeps = [erlang_provides.get(r, "Error: missing requires: %s" % r) for r in erlang_requires[name]]

	deps = set()
	dependency_add = deps.add
	[x for x in rawdeps if not (x in deps or dependency_add(x))]

	# Additional BuildRequires (if any):
	for br in package_additional_buildrequires.get(name, []):
		print "BuildRequires: %s" % br

	# Add basic autogenerated Requires:
	for r in sorted(deps):
		#Requires: %{name}-stdlib%{?_isa} = %{version}-%{release}
		print "Requires: %s%%{?_isa} = %%{version}-%%{release}" % r.replace("erlang-", "%{name}-")

	# Add custom additional Requires (if any):
	for r in package_additional_requires.get(name, []):
		print "Requires: %s" % r

	# Add custom additional Obsoletes (if any):
	for os in package_additional_obsoletes.get(name, []):
		print "Obsoletes: %s" % os

	if name in package_noarch:
		print "BuildArch: noarch"

	print ""
	if namemask.match(name):
		print "%%description %s" % name.replace("erlang-", "")
	else:
		print "%%description -n %s" % name
	print "%s" % h[rpm.RPMTAG_DESCRIPTION]

	if name == "erlang-doc":
		# Additional ifdef just for this sub-package (see above)
		print "%endif"

	print ""

print "### END OF AUTOGENERATED LIST ###"