#!/usr/bin/python3
#
# SPDX-License-Identifier: Apache-2.0
#

"""
Contains helper functions that will configure basic system settings.
"""

from consts.timeout import HostTimeout
from utils import serial
from utils.install_log import LOG
from helper import host_helper


def update_platform_cpus(stream, hostname, cpu_num=5):
    """
    Update platform CPU allocation.
    """

    LOG.info("Allocating %s CPUs for use by the %s platform.", cpu_num, hostname)
    serial.send_bytes(
        stream,
        "\nsource /etc/platform/openrc; system host-cpu-modify " 
        f"{hostname} -f platform -p0 {cpu_num}",
        prompt="keystone",
        timeout=300,
    )


def set_dns(stream, dns_ip):
    """
    Perform DNS configuration on the system.
    """

    LOG.info("Configuring DNS to %s.", dns_ip)
    serial.send_bytes(
        stream,
        "source /etc/platform/openrc; system dns-modify "
        f"nameservers={dns_ip}",
        prompt="keystone",
    )


def config_controller(stream, password):
    """
    Configure controller-0 using optional arguments
    """

    serial.send_bytes(
        stream,
        "ansible-playbook /usr/share/ansible/stx-ansible/playbooks/bootstrap.yml",
        expect_prompt=False,
    )
    host_helper.check_password(stream, password=password)
    ret = serial.expect_bytes(stream, "~$", timeout=HostTimeout.LAB_CONFIG)
    if ret != 0:
        LOG.info("Configuration failed. Exiting installer.")
        raise Exception("Configcontroller failed")  # pylint: disable=E0012, W0719