#!/usr/bin/env python3
import json, os, shutil

DIR = os.getcwd()

def get_json(path):
    return json.load(open(DIR + path))

def update_and_upgrade(ctx):
    os.system('apt-get update -y')
    os.system('apt-get upgrade -y')
    os.system('apt-get dist-upgrade -y')


def apt_install_core(ctx):
    for package in get_json('/apt/apt-installs-core.json'):
        os.system("apt-get install {} -y".format(package))


def add_apt_keys(ctx):
    for key in get_json('/apt/apt-keys.json'):
        os.system("wget -O - {} | apt-key add -".format(key))


def add_apt_sources(ctx):
    os.system('apt/add-apt-sources.sh')


def add_apt_repos(ctx):
    for repo in get_json('/apt/apt-repos.json'):
        os.system("add-apt-repository {} -y".format(repo))


def apt_install_extras(ctx):
    for package in get_json('/apt/apt-installs-extra.json'):
        os.system("apt-get install {} -y".format(package))


def run_custom_installs(ctx):
    os.system('apt/custom-installs.sh')


def install_atom_packages(ctx):
    packages = " ".join(get_json('/atom/packages.json'))
    os.system("apm install {}".format(packages))


def install_configs(ctx):
    shutil.copyfile(DIR + "/config/terminator.conf", "~/.config/terminator/config")


if __name__ == '__main__':
    print("Please run this file using waf, not directly.")
    exit(1)