#! /bin/python3

import os, errno
import apt_pkg
import apt
import json

_APT_CONFIG_PIN="""Package: {PACKAGES}
Pin: release a=experimental   
Pin-Priority: 500
"""

def silentremove(filename):
    try:
        os.remove(filename)
    except OSError as e: # this would be "except OSError, e:" before Python 2.6
        if e.errno != errno.ENOENT: # errno.ENOENT = no such file or directory
            raise # re-raise exception if a different error occurred    

current_path = os.path.dirname(os.path.realpath(__file__))

srcnames_files = [open(current_path + "/package_srcnames/" + filename) for filename in  os.listdir(current_path + "/package_srcnames/")]
srcname_lines = []
pkgname_lines = []
cache = apt.Cache()
for file in srcnames_files:
    for line in file.readlines():
        srcname = line.strip()
        srcname_lines.append(srcname)
        srcrecords = apt_pkg.SourceRecords()
        srcrec = srcrecords.lookup(srcname)
        if srcrec:
            bins = srcrecords.binaries
            for bin in bins:
                pkgname_lines.append(bin)
    file.close()

with open (current_path + "/package_srcnames/package_pkgnames_overrides") as file:
    lines = file.readlines()
    for line in lines:
        pkgname_lines.append(line.strip())
    file.close()

src_data = {
    'source_names': [source_name for source_name in srcname_lines],
}

pkg_data = {
    'package_names': [pkg_name for pkg_name in pkgname_lines]
}

apt_pin_packages = ""

for pkg in pkgname_lines:
    apt_pin_packages += (pkg + " ")
    apt_pin_packages += (pkg + "t64 ")

silentremove("0-debian-exp-overrides")
with open("0-debian-exp-overrides", "w") as file:
    debian_exp_override_file = _APT_CONFIG_PIN.format(
        PACKAGES=apt_pin_packages,
    )
    file.write(debian_exp_override_file)

silentremove("exp_src_names.json")
with open("exp_src_names.json", "w") as twitterDataFile:
    twitterDataFile.write(
        json.dumps(src_data, indent=4)
    )

silentremove("exp_pkg_names.json")
with open("exp_pkg_names.json", "w") as twitterDataFile:
    twitterDataFile.write(
        json.dumps(pkg_data, indent=4)
    )