import logging
import subprocess

from borgmatic.logger import get_logger

logger = get_logger(__name__)


def execute_and_log_output(full_command, output_log_level):
    process = subprocess.Popen(full_command, stdout=None, stderr=subprocess.PIPE)

    while process.poll() is None:
        line = process.stderr.readline().rstrip().decode()
        if line.startswith('borg: error:'):
            logger.error(line)
        else:
            logger.log(output_log_level, line)

    remaining_output = process.stderr.read().rstrip().decode()
    if remaining_output:
        logger.info(remaining_output)

    exit_code = process.poll()
    if exit_code != 0:
        raise subprocess.CalledProcessError(exit_code, full_command)


def execute_command(full_command, output_log_level=logging.INFO):
    '''
    Execute the given command (a sequence of command/argument strings) and log its output at the
    given log level. If output log level is None, instead capture and return the output.
    '''
    logger.debug(' '.join(full_command))

    if output_log_level is None:
        output = subprocess.check_output(full_command)
        return output.decode() if output is not None else None
    else:
        execute_and_log_output(full_command, output_log_level)