from __future__ import absolute_import, division, print_function

__metaclass__ = type

import os
import yaml

from ansible.plugins.lookup import LookupBase
from ansible.errors import AnsibleError


class LookupModule(LookupBase):
    def run(self, terms, variables=None, **kwargs):
        application_id = terms[0]
        base_gid = kwargs.get("base_gid", 10000)
        roles_dir = kwargs.get("roles_dir", "roles")

        if not os.path.isdir(roles_dir):
            raise AnsibleError(f"Roles directory '{roles_dir}' not found")

        matched_roles = []

        for root, dirs, files in os.walk(roles_dir):
            if os.path.basename(root) == "vars" and "main.yml" in files:
                vars_path = os.path.join(root, "main.yml")
                try:
                    with open(vars_path, "r") as f:
                        data = yaml.safe_load(f) or {}
                        app_id = data.get("application_id")
                        if app_id:
                            matched_roles.append((app_id, vars_path))
                except Exception as e:
                    raise AnsibleError(f"Error parsing {vars_path}: {e}")

        # sort alphabetically by application_id
        sorted_ids = sorted(app_id for app_id, _ in matched_roles)

        try:
            index = sorted_ids.index(application_id)
        except ValueError:
            raise AnsibleError(
                f"Application ID '{application_id}' not found in any role"
            )

        return [base_gid + index]