# Copyright 2023 Red Hat, Inc. Jake Hunsaker <jhunsake@redhat.com>

# This file is part of the sos project: https://github.com/sosreport/sos
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# version 2 of the GNU General Public License.
#
# See the LICENSE file in the source distribution for further information.

import os

from sos.cleaner.preppers import SoSPrepper


class KeywordPrepper(SoSPrepper):
    """
    Prepper to handle keywords passed to cleaner via either the `--keywords`
    or `--keyword-file` options.
    """

    name = 'keyword'

    def _get_items_for_keyword(self, archive):
        items = []
        for kw in self.opts.keywords:
            items.append(kw)
        if self.opts.keyword_file and os.path.exists(self.opts.keyword_file):
            with open(self.opts.keyword_file, 'r') as kwf:
                items.extend(kwf.read().splitlines())

        for item in items:
            self.regex_items['keyword'].add(item)

        return items

# vim: set et ts=4 sw=4 :