#!/usr/bin/python

# Copyright Red Hat, Inc.
# Apache License Version 2.0 (see LICENSE

from __future__ import absolute_import, division, print_function

__metaclass__ = type

DOCUMENTATION = r"""
---
module: generate_make_tasks

short_description: Generate per Makefile target a dedicated task file

description:
- Generate per Makefile target a dedicated task file

options:
  install_yamls_path:
    description:
    - Absolute path to install_yamls repository.
    required: True
    type: str
  output_directory:
    description:
    - Absolute path to the output directory. It must exists.
    required: True
    type: str

author:
  - Chandan Kumar (@raukadah)
  - Cedric Jeanneret (@cjeanner)
"""

EXAMPLES = r"""
- name: Create output directory
  ansible.builtin.file:
    path: "{{ ansible_user_dir }}/make_installyamls/tasks"
    state: directory
    mode: "0644"

- name: Generate make tasks
  generate_make_tasks:
    install_yamls_path: "{{ ansible_user_dir }}/src/github.com/openstack-k8s-operators/install_yamls/"
    output_directory: "{{ ansible_user_dir }}/make_installyamls/tasks"
"""  # noqa

RETURN = r"""
success:
    description: Status of the module
    type: bool
    returned: always
    sample: True
"""

from ansible.module_utils.basic import AnsibleModule

import os


MAKE_TMPL = """---
- name: Debug make_%(target)s_env
  when: make_%(target)s_env is defined
  ansible.builtin.debug:
    var: make_%(target)s_env
- name: Debug make_%(target)s_params
  when: make_%(target)s_params is defined
  ansible.builtin.debug:
    var: make_%(target)s_params
- name: Run %(target)s
  retries: "{{ make_%(target)s_retries | default(omit) }}"
  delay: "{{ make_%(target)s_delay | default(omit) }}"
  until: "{{ make_%(target)s_until | default(true) }}"
  register: "make_%(target)s_status"
  cifmw.general.ci_script:
    output_dir: "{{ cifmw_basedir|default(ansible_user_dir ~ '/ci-framework-data') }}/artifacts"
    chdir: "%(chdir)s"
    script: "make %(target)s"
    dry_run: "{{ make_%(target)s_dryrun|default(false)|bool }}"
    extra_args: "{{ dict((make_%(target)s_env|default({})), **(make_%(target)s_params|default({}))) }}"
"""  # noqa

NO_OUTDIR = "Directory %s does not exist. Please create it."
NO_MAKEFILE = "No makefile was parsed. Please check install_yamls path (%s)."


def main():
    mod_args = {
        "install_yamls_path": {"required": True, "type": "str"},
        "output_directory": {"required": True, "type": "str"},
    }
    module = AnsibleModule(argument_spec=mod_args, supports_check_mode=False)

    result = {
        "success": False,
        "changed": False,
        "debug": {},
    }
    install_yamls = module.params.get("install_yamls_path")
    output_directory = module.params.get("output_directory")

    # Check for output dir
    if not os.path.isdir(output_directory):
        result["error"] = NO_OUTDIR % output_directory
        module.fail_json(msg=result["error"])

    # Read the content of the makefiles and extract targets
    targets = []
    for makefiles in ["Makefile", "devsetup/Makefile"]:
        makefile_path = os.path.join(install_yamls, makefiles)
        result["debug"][makefile_path] = []
        if os.path.exists(makefile_path):
            makefile_chdir = os.path.dirname(makefile_path)
            with open(makefile_path, "r") as f:
                targets = [
                    line.split(".PHONY:")[1].strip()
                    for line in f.readlines()
                    if line.startswith(".PHONY:")
                ]
            result["debug"][makefile_path] = targets

            # Ensure we parsed some Makefiles
            if len(targets) == 0:
                error = NO_MAKEFILE % install_yamls
                module.fail_json(msg=error, **result)

            # Generate playbooks
            for target in targets:
                output = f"make_{target}.yml"
                with open(os.path.join(output_directory, output), "w") as fs:
                    fs.write(MAKE_TMPL % {"target": target, "chdir": makefile_chdir})

        else:
            error = NO_MAKEFILE % install_yamls
            module.fail_json(msg=error, **result)

    result["success"] = True
    module.exit_json(**result)


if __name__ == "__main__":
    main()