#!/usr/bin/env python3

# tftp.py - PVC Cluster Auto-bootstrap TFTP preparation libraries
# Part of the Parallel Virtual Cluster (PVC) system
#
#    Copyright (C) 2018-2021 Joshua M. Boniface <joshua@boniface.me>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, version 3.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.
#
###############################################################################

import os.path
import git
import yaml


def build_tftp_repository(config):
    # Generate an installer config
    build_cmd = f"{config['ansible_path']}/pvc-installer/buildpxe.sh -o {config['tftp_root_path']} -u {config['deploy_username']}"
    print(f"Building TFTP contents via pvc-installer command: {build_cmd}")
    os.system(build_cmd)


def init_tftp(config):
    """
    Prepare a TFTP root
    """
    if not os.path.exists(config['tftp_root_path']):
        print("First run: building TFTP root and contents - this will take some time!")
        os.makedirs(config['tftp_root_path'])
        os.makedirs(config['tftp_host_path'])
        
        build_tftp_repository(config)