<?php

/**
 * Portuguese Stemmer
 *
 * Lightweight suffix-based stemmer for Portuguese.
 *
 * @package ProRank\SEO\Modules\Content\InternalLinking\Parser\MultiLanguage\Languages
 * @since   1.0.0
 */

declare(strict_types=1);

namespace ProRank\SEO\Modules\Content\InternalLinking\Parser\MultiLanguage\Languages;

use ProRank\SEO\Modules\Content\InternalLinking\Parser\MultiLanguage\Stemmer;

if ( ! defined( 'ABSPATH' ) ) {
    exit;
}

class PortugueseStemmer extends Stemmer {
    public function __construct() {
        parent::__construct('pt');
    }

    public function stem(string $word): string {
        $word = $this->normalize($word);
        if (strlen($word) <= 3) {
            return $word;
        }

        $suffixes = [
            'amentos', 'imentos', 'ações', 'acao', 'acoes', 'mente', 'amento',
            'imento', 'idades', 'idade', 'ismo', 'ista', 'istas', 'ivos', 'ivas',
            'ivo', 'iva', 'icos', 'icas', 'ico', 'ica', 'oso', 'osa', 'osos', 'osas',
            'ção', 'ções', 'cao', 'coes', 'es', 's',
        ];

        foreach ($suffixes as $suffix) {
            if (str_ends_with($word, $suffix) && strlen($word) > strlen($suffix) + 2) {
                return substr($word, 0, -strlen($suffix));
            }
        }

        return $word;
    }
}
