#!/usr/bin/python

import re

from ansible.plugins.action import ActionBase
from ansible_collections.tensin.infra.plugins.module_utils import file

LOG_PATH = '/var/log/dpkg.log'


class ActionModule(ActionBase):
    def run(self, tmp=None, task_vars=None):
        super().run(tmp, task_vars)

        before = self._parse_dpkg_log(task_vars, default='')

        apt_result = self._execute_module(
            module_name='ansible.builtin.apt',
            module_args=self._task.args.get('apt_args', {}),
            task_vars=task_vars,
        )

        after = self._parse_dpkg_log(task_vars)

        apt_result['packages'] = after[len(before) :]
        return apt_result

    def _parse_dpkg_log(self, task_vars, default=None):
        contents = file.read_file(self, LOG_PATH, task_vars, default=default)
        lines = contents.splitlines()
        regex = re.compile(' install | upgrade | remove | purge ')
        lines = [line for line in lines if regex.search(line)]
        lines = [' '.join(line.split(' ')[2:]) for line in lines]
        return lines