#!/usr/bin/python

# -*- coding: utf-8 -*-
# Copyright (c) 2024 OpenStack Foundation
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.
# Included from: https://github.com/ceph/ceph-ansible/blob/master/module_utils/ca_common.py
#

import os
import datetime

def generate_ceph_cmd(sub_cmd, args, spec_path, user_key=None, cluster='ceph',
                      user='client.admin', container_image=None,
                      interactive=False):
    '''
    Generate 'ceph' command line to execute
    '''
    ceph_config_path = '/etc/ceph'
    if not user_key:
        user_key = '{}/{}.{}.keyring'.format(ceph_config_path, cluster, user)

    cmd = pre_generate_ceph_cmd(container_image=container_image,
                                interactive=interactive,
                                spec_path=spec_path)
    base_cmd = [
        '-n',
        user,
        '-k',
        user_key,
        '--cluster',
        cluster
    ]
    base_cmd.extend(sub_cmd)
    cmd.extend(base_cmd + args)

    return cmd


def container_exec(binary, container_image, spec_path=None, interactive=False):
    '''
    Build the docker CLI to run a command inside a container
    '''

    container_binary = os.getenv('CEPH_CONTAINER_BINARY')
    command_exec = [container_binary, 'run']

    fsid = ''
    if interactive:
        command_exec.extend(['--interactive'])

    if 'CEPH_FSID' in os.environ:
        fsid = os.getenv('CEPH_FSID')
    ceph_config_path = '/etc/ceph'
    if 'CEPH_CONF' in os.environ:
        ceph_config_path = os.getenv('CEPH_CONF')
    if fsid:
        path = '/var/lib/ceph/{}/config'.format(fsid)
        if os.path.exists(path):
            ceph_config_path = path
    command_exec.extend(['--rm',
                         '--net=host',
                         '--ipc=host',
                         '-v', '{}:/etc/ceph:z'.format(ceph_config_path),
                         '-v', '/var/lib/ceph/:/var/lib/ceph/:z',
                         '-v', '/var/log/ceph/:/var/log/ceph/:z'])

    if spec_path is not None and len(spec_path) > 0:
        command_exec.extend(['-v', '{}:{}:z'.format(spec_path, spec_path)])

    command_exec.extend(['--entrypoint=' + binary, container_image])

    return command_exec


def is_containerized():
    '''
    Check if we are running on a containerized cluster
    '''

    if 'CEPH_CONTAINER_IMAGE' in os.environ:
        container_image = os.getenv('CEPH_CONTAINER_IMAGE')
    else:
        container_image = None

    return container_image


def pre_generate_ceph_cmd(container_image=None, interactive=False, spec_path=None):
    '''
    Generate ceph prefix comaand
    '''
    if container_image:
        cmd = container_exec('ceph', container_image, spec_path=spec_path, interactive=interactive)
    else:
        cmd = ['ceph']

    return cmd


def exec_command(module, cmd, stdin=None):
    '''
    Execute command(s)
    '''

    binary_data = False
    if stdin:
        binary_data = True
    rc, out, err = module.run_command(cmd, data=stdin, binary_data=binary_data)

    return rc, cmd, out, err


def exit_module(module, out, rc, cmd, err, startd, changed=False):
    endd = datetime.datetime.now()
    delta = endd - startd

    result = dict(
        cmd=cmd,
        start=str(startd),
        end=str(endd),
        delta=str(delta),
        rc=rc,
        stdout=out.rstrip("\r\n"),
        stderr=err.rstrip("\r\n"),
        changed=changed,
    )
    module.exit_json(**result)


def fatal(message, module):
    '''
    Report a fatal error and exit
    '''

    if module:
        module.fail_json(msg=message, rc=1)
    else:
        raise(Exception(message))