#!/usr/bin/python
# Mock apt module for Docker testing

import subprocess

from ansible.module_utils.basic import AnsibleModule


def main():
    module = AnsibleModule(
        argument_spec={
            'name': {'type': 'list', 'aliases': ['pkg', 'package']},
            'state': {'type': 'str', 'default': 'present', 'choices': ['present', 'absent', 'latest', 'build-dep', 'fixed']},
            'update_cache': {'type': 'bool', 'default': False},
            'cache_valid_time': {'type': 'int', 'default': 0},
            'install_recommends': {'type': 'bool'},
            'force': {'type': 'bool', 'default': False},
            'allow_unauthenticated': {'type': 'bool', 'default': False},
            'allow_downgrade': {'type': 'bool', 'default': False},
            'allow_change_held_packages': {'type': 'bool', 'default': False},
            'dpkg_options': {'type': 'str', 'default': 'force-confdef,force-confold'},
            'autoremove': {'type': 'bool', 'default': False},
            'purge': {'type': 'bool', 'default': False},
            'force_apt_get': {'type': 'bool', 'default': False},
        },
        supports_check_mode=True
    )

    name = module.params['name']
    state = module.params['state']
    update_cache = module.params['update_cache']

    result = {
        'changed': False,
        'cache_updated': False,
        'cache_update_time': 0
    }

    # Log the operation
    with open('/var/log/mock-apt-module.log', 'a') as f:
        f.write(f"apt module called: name={name}, state={state}, update_cache={update_cache}\n")

    # Handle cache update
    if update_cache:
        # In Docker, apt-get update was already run in entrypoint
        # Just pretend it succeeded
        result['cache_updated'] = True
        result['cache_update_time'] = 1754231778  # Fixed timestamp
        result['changed'] = True

    # Handle package installation/removal
    if name:
        packages = name if isinstance(name, list) else [name]

        # Check which packages are already installed
        installed_packages = []
        for pkg in packages:
            # Use dpkg to check if package is installed
            check_cmd = ['dpkg', '-s', pkg]
            rc = subprocess.run(check_cmd, capture_output=True)
            if rc.returncode == 0:
                installed_packages.append(pkg)

        if state in ['present', 'latest']:
            # Check if we need to install anything
            missing_packages = [p for p in packages if p not in installed_packages]

            if missing_packages:
                # Log what we would install
                with open('/var/log/mock-apt-module.log', 'a') as f:
                    f.write(f"Would install packages: {missing_packages}\n")

                # For our test purposes, these packages are pre-installed in Docker
                # Just report success
                result['changed'] = True
                result['stdout'] = f"Mock: Packages {missing_packages} are already available"
                result['stderr'] = ""
            else:
                result['stdout'] = "All packages are already installed"

        elif state == 'absent':
            # Check if we need to remove anything
            present_packages = [p for p in packages if p in installed_packages]

            if present_packages:
                result['changed'] = True
                result['stdout'] = f"Mock: Would remove packages {present_packages}"
            else:
                result['stdout'] = "No packages to remove"

    # Always report success for our testing
    module.exit_json(**result)

if __name__ == '__main__':
    main()