iranianpep/keyword-extractor

<?php

namespace KeywordExtractor\Modifiers\Filters;

use KeywordExtractor\Modifiers\AbstractModifier;

class WhitelistFilter extends AbstractModifier
{
    private $whitelist;

    public function __construct(array $whitelist)
    {
        $this->setWhitelist($whitelist);
    }

    public function modifyToken($token)
    {
        if (in_array($token, $this->getWhitelist())) {
            return $token;
        }

        return '';
    }

    /**
     * @return array
     */
    public function getWhitelist(): array
    {
        return $this->whitelist;
    }

    /**
     * @param array $whitelist
     */
    public function setWhitelist(array $whitelist): void
    {
        $this->whitelist = $whitelist;
    }
}