#!/usr/bin/env python3

# host.py - PVC Cluster Auto-bootstrap host 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/>.
#
###############################################################################

from celery.utils.log import get_task_logger

import pvcbootstrapd.lib.db as db


logger = get_task_logger(__name__)


def installer_init(config, cspec, data):
    bmc_macaddr = data['bmc_macaddr']
    bmc_ipaddr = data['bmc_ipaddr']
    host_macaddr = data['host_macaddr']
    host_ipaddr = data['host_ipaddr']
    cspec_cluster = cspec['bootstrap'][bmc_macaddr]['node']['cluster']
    cspec_hostname = cspec['bootstrap'][bmc_macaddr]['node']['hostname']
    cspec_nid = int(''.join(filter(str.isdigit, cspec_hostname)))

    cluster = db.get_cluster(config, name=cspec_cluster)
    if cluster is None:
        cluster = db.add_cluster(config, cspec_cluster, "provisioning")
    logger.debug(cluster)

    node = db.get_node(config, cspec_cluster, name=cspec_hostname)
    if node is None:
        node = db.add_node(config, cspec_cluster, "installing", cspec_hostname, cspec_nid, bmc_macaddr, bmc_ipaddr, host_macaddr, host_ipaddr)
    else:
        node = db.update_node_addresses(config, cspec_cluster, cspec_hostname, bmc_macaddr, bmc_ipaddr, host_macaddr, host_ipaddr)
    logger.debug(node)


def installer_complete(config, cspec, data):
    bmc_macaddr = data['bmc_macaddr']
    cspec_hostname = cspec['bootstrap'][bmc_macaddr]['node']['hostname']
    cspec_cluster = cspec['bootstrap'][bmc_macaddr]['node']['cluster']

    node = db.update_node_state(config, cspec_cluster, cspec_hostname, "installed")
    logger.debug(node)


def set_boot_state(config, cspec, data, state):
    bmc_macaddr = data['bmc_macaddr']
    bmc_ipaddr = data['bmc_ipaddr']
    host_macaddr = data['host_macaddr']
    host_ipaddr = data['host_ipaddr']
    cspec_cluster = cspec['bootstrap'][bmc_macaddr]['node']['cluster']
    cspec_hostname = cspec['bootstrap'][bmc_macaddr]['node']['hostname']

    node = db.update_node_addresses(config, cspec_cluster, cspec_hostname, bmc_macaddr, bmc_ipaddr, host_macaddr, host_ipaddr)
    node = db.update_node_state(config, cspec_cluster, cspec_hostname, state)
    logger.debug(node)