<?php
/**
 * This source file is part of the open source project
 * ExpressionEngine (https://expressionengine.com)
 *
 * @link      https://expressionengine.com/
 * @copyright Copyright (c) 2003-2022, Packet Tide, LLC (https://www.packettide.com)
 * @license   https://expressionengine.com/license Licensed under Apache License, Version 2.0
 */

namespace ExpressionEngine\Addons\Spam\Service;

/**
 * Spam Training
 */
class Training
{
    public $stop_words_path = 'spam/training/stopwords.txt';
    public $kernel;
    private $shm_id;

    /**
     * __construct
     *
     * @param string $kernel The name of the kernel to use
     * @access public
     * @return void
     */
    public function __construct($kernel)
    {
        $this->kernel = $this->getKernel($kernel);
    }

    /**
     * Close the shared memory segment if we're using it.
     *
     * @access public
     * @return void
     */
    public function __destruct()
    {
        if (! empty($this->shm_id)) {
            shmop_close($this->shm_id);
        }
    }

    /**
     * Load the classifier object from memory if available, otherwise construct
     * a new classifier from the database.
     *
     * @access public
     * @return The prepared classifier
     */
    public function loadClassifier($vectorizers)
    {
        $collection = ee('spam:Collection', $vectorizers);

        if (function_exists('shmop_open')) {
            // Generate System V IPC key to identify out shared memory segment
            $id = ftok(__FILE__, 't');

            // It's as if millions of Daniel Binghams suddenly cried out in terror
            $this->shm_id = @shmop_open($id, 'a', 0, 0);

            // first check if we already have a memory segment
            if ($this->shm_id === false) {
                // No memory segment, serialize and write classifier from database
                $classifier = $this->classifier($collection);
                $data = serialize($classifier);
                $size = strlen($data);
                $this->shm_id = shmop_open($id, 'c', 0644, $size);
                shmop_write($this->shm_id, $data, 0);

                return $classifier;
            } else {
                // Read from the memory segment and unserialize
                $size = shmop_size($this->shm_id);
                $data = shmop_read($this->shm_id, 0, $size);

                return unserialize($data);
            }
        } else {
            return $this->classifier($collection);
        }
    }

    /**
     * Deletes the shared memory segment containing our classifier
     *
     * @access public
     * @return void
     */
    public function deleteClassifier()
    {
        if (function_exists('shmop_open')) {
            if (! empty($this->shm_id) && is_int($this->shm_id)) {
                $shm_id = $this->shm_id;
                unset($this->shm_id);
            } else {
                $id = ftok(__FILE__, 't');
                $shm_id = @shmop_open($id, 'a', 0, 0);

                if ($shm_id === false) {
                    // No memory segment exists
                    return;
                }
            }

            shmop_delete($shm_id);
            shmop_close($shm_id);
        }
    }

    /**
     * Returns a new classifier based on our training data.
     *
     * @param  Vectorizer $collection
     * @access public
     * @return boolean
     */
    public function classifier($collection)
    {
        $stop_words = explode("\n", ee()->lang->load('spam/stopwords', null, true, false));

        // Grab the trained parameters
        $training = array(
            'spam' => $this->getParameters('spam'),
            'ham' => $this->getParameters('ham'),
        );

        return ee('spam:Classifier', $training, $collection, $stop_words);
    }

    /**
     * Returns an array of all the parameters for a class
     *
     * @param string The class name
     * @access private
     * @return array
     */
    private function getParameters($class)
    {
        $parameters = ee('Model')->get('spam:SpamParameter')
            ->fields('mean', 'variance')
            ->filter('class', $class)
            ->filter('kernel_id', $this->kernel->kernel_id)
            ->all();

        $result = array();

        foreach ($parameters as $parameter) {
            $result[] = ee('spam:Distribution', $parameter->mean, $parameter->variance);
        }

        return $result;
    }

    /**
     * Returns an array of document counts for every word in the training set
     *
     * @access public
     * @return array
     */
    public function getVocabulary()
    {
        $vocab = ee('Model')->get('spam:SpamVocabulary')
            ->fields('term', 'count')
            ->filter('kernel_id', $this->kernel->kernel_id)
            ->limit(ee()->config->item('spam_word_limit') ?: 5000)
            ->all();

        $result = array();

        foreach ($vocab as $word) {
            $result[$word->term] = $word->count;
        }

        return $result;
    }

    /**
     * Returns the total document count for the current kernel
     *
     * @access public
     * @return array
     */
    public function getDocumentCount()
    {
        return $this->kernel->count;
    }

    /**
     * Grab the appropriate kernel ID or insert a new one
     *
     * @param string $name The name of the kernel
     * @access private
     * @return int The kernel ID
     */
    private function getKernel($name)
    {
        $kernel = ee('Model')->get('spam:SpamKernel')
            ->filter('name', $name)
            ->first();

        if (empty($kernel)) {
            $kernel = ee('Model')->make('spam:SpamKernel', array('name' => $name));
            $kernel->save();
        }

        return $kernel;
    }
}

// EOF
