<?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\Model\Content;

use ExpressionEngine\Service\Model\Model;
use ExpressionEngine\Service\Model\VariableColumnModel;
use ExpressionEngine\Model\Content\Display\DefaultLayout;
use ExpressionEngine\Model\Content\Display\FieldDisplay;
use ExpressionEngine\Model\Content\Display\LayoutInterface;

/**
 * create: new ChannelEntry()->getForm();
 * existing: $entry->fill($data)->getForm();
 * set data: $entry->title = "Foo"; $entry->getForm();
 * mass set: $entry->set(array); $entry->getForm();
 */
abstract class ContentModel extends VariableColumnModel
{
    protected static $_events = array(
        'afterSave',
        'afterInsert',
        'afterUpdate',
        'beforeDelete'
    );

    protected $_field_facades;
    protected $_field_was_saved = array();
    protected $_custom_fields_loaded = false;

    /**
     * A link back to the owning Structure object.
     *
     * @return  Structure   A link back to the Structure object that defines
     *                      this Content's structure.
     */
    abstract public function getStructure();

    /**
     * Get the prefix for custom fields. Typically custom fields are
     * stored as 'field_id_#' where # is the field id.
     *
     * @return String Custom field column prefix
     */
    public function getCustomFieldPrefix()
    {
        return 'field_id_';
    }

    /**
     * Optionally return an array of default fields.
     *
     * @return array of field definitions
     */
    protected function getDefaultFields()
    {
        return array();
    }

    /**
     * Make sure we update our custom fields when save is triggered
     */
    public function onAfterSave()
    {
        foreach ($this->_field_was_saved as $field) {
            $field->setContentId($this->getId());
            $field->postSave();
        }

        $this->_field_was_saved = array();
    }

    public function onAfterInsert()
    {
        $this->saveFieldData($this->getValues());
    }

    public function onAfterUpdate($changed)
    {
        $this->saveFieldData($changed);
    }

    /**
     * Cascade the delete to the fieldtypes
     */
    public function onBeforeDelete()
    {
        // These models don't yet support any fieldtypes that implement delete(),
        // skip for now for performance
        if (! in_array($this->getName(), ['ee:Member', 'ee:Category'])) {
            foreach ($this->getCustomFields() as $field) {
                $field->delete();
            }
        }

        $this->deleteFieldData();
    }

    /**
     * Check if a custom field of $name exists
     */
    public function hasCustomField($name)
    {
        if (strpos($name, $this->getCustomFieldPrefix()) !== 0) {
            $default_fields = $this->getDefaultFields();

            return array_key_exists($name, $default_fields);
        }

        $this->usesCustomFields();

        if (! isset($this->_field_facades)) {
            return false;
        }

        return array_key_exists($name, $this->_field_facades);
    }

    /**
     * Get a list of all custom field facades
     */
    public function getCustomFields()
    {
        $this->usesCustomFields();

        return $this->_field_facades ?: array();
    }

    /**
     * Get a custom field facade
     */
    public function getCustomField($name)
    {
        $this->usesCustomFields();

        return $this->_field_facades[$name];
    }

    /**
    * Get a list of all custom field names
    */
    public function getCustomFieldNames()
    {
        $this->usesCustomFields();

        return array_keys($this->_field_facades);
    }

    /**
     * Get the layout for this content.
     */
    public function getDisplay(LayoutInterface $layout = null)
    {
        $this->usesCustomFields();

        $fields = array_map(
            function ($field) {
                return new FieldDisplay($field);
            },
            $this->getCustomFields()
        );

        $layout = $layout ?: new DefaultLayout();

        return $layout->transform($fields);
    }

    /**
     * Batch update properties
     *
     * Safely updates any properties that might exist,
     * passing them through the getters along the way.
     *
     * @param array $data Data to update
     * @return $this
     */
    public function set(array $data = array())
    {
        $this->usesCustomFields();

        return parent::set($data);
    }

    /**
     * Make sure that calls to fill() also apply to custom fields
     */
    public function fill(array $data = array())
    {
        parent::fill($data);

        if ($this->_custom_fields_loaded) {
            $this->fillCustomFields($data);
        }

        return $this;
    }

    /**
     * Small change to custom fields to make sure the save() method
     * gets fired.
     */
    public function save()
    {
        foreach ($this->getCustomFields() as $field) {
            $name = $field->getName();
            if ($this->isDirty($name)) {
                $this->setRawProperty($name, $field->save($this));
                $this->_field_was_saved[] = $field;
            }
        }

        return parent::save();
    }

    /**
     * Setting a property needs to apply to custom fields
     */
    public function setProperty($name, $new_value)
    {
        if ($this->_custom_fields_loaded && $this->hasCustomField($name)) {
            $this->emit('beforeSetCustomField', $name, $new_value);

            $field = $this->getCustomField($name);
            $value = $field->getData(); // old value

            $this->backupIfChanging($name, $value, $new_value);

            $field->setData($new_value);

            $this->emit('afterSetCustomField', $name, $new_value);

            if (! parent::hasProperty($name)) {
                return $this;
            }

            $new_value = $field->getData();
        }

        return parent::setProperty($name, $new_value);
    }

    /**
     * Support method for the model validation mixin
     */
    public function getValidationRules()
    {
        $this->usesCustomFields();

        $rules = parent::getValidationRules();

        $facades = $this->getCustomFields();

        foreach ($facades as $name => $facade) {
            // If this field is hidden by conditional logic or it is being
            // edited but has not changed then we are not concerned with validation
            if (get_bool_from_string($facade->getHidden()) || (!$this->isNew() && !$this->isDirty($name))) {
                continue;
            }

            if (! isset($rules[$name])) {
                $rules[$name] = '';
            } else {
                $rules[$name] .= '|';
            }

            if ($facade->isRequired()) {
                $rules[$name] .= 'required|';
            }

            $rules[$name] .= "validateCustomField";
        }

        return $rules;
    }

    /**
     * Callback to validate custom fields
     */
    public function validateCustomField($key, $value, $params, $rule)
    {
        $field = $this->getCustomField($key);
        $result = $this->getCustomField($key)->validate($value);

        $this->setRawProperty($key, $field->getData());

        return $result;
    }

    /**
     * Ensures that custom fields are setup and their data is in sync.
     */
    protected function usesCustomFields()
    {
        if (! isset($this->_field_facades)) {
            $this->initializeCustomFields();
        }
    }

    /**
     * Populate the custom fields on fill()
     */
    protected function fillCustomFields(array $data = array())
    {
        $this->setDataOnCustomFields($data);
    }

    protected function setDataOnCustomFields(array $data = array())
    {
        foreach ($data as $name => $value) {
            // Optimization, skip if null
            if (! isset($value)) {
                continue;
            }

            if (strpos($name, 'field_ft_') !== false) {
                $name = str_replace('field_ft_', 'field_id_', $name);

                if ($this->hasCustomField($name)) {
                    $this->getCustomField($name)->setFormat($value);
                }

                continue;
            }

            if (strpos($name, 'field_dt_') !== false) {
                $name = str_replace('field_dt_', 'field_id_', $name);

                if ($this->hasCustomField($name)) {
                    $this->getCustomField($name)->setTimezone($value);
                }

                continue;
            }

            if ($this->hasCustomField($name)) {
                $this->getCustomField($name)->setData($value);
            }
        }
    }

    /**
     * Get the party started, grab the structure and setup everything.
     */
    protected function initializeCustomFields()
    {
        $this->_field_facades = array();
        $default_fields = $this->getDefaultFields();

        foreach ($default_fields as $id => $field) {
            $this->addFacade($id, $field);
        }

        $structure = $this->getStructure();

        // no structure yet? defer
        if (! $structure) {
            $this->_field_facades = null;

            return;
        }

        $native_fields = $this->getStructure()->getAllCustomFields();
        $native_prefix = $this->getCustomFieldPrefix();

        foreach ($native_fields as $field) {
            $settings = array_merge($field->getSettingsValues(), $field->toArray());

            $this->addFacade(
                $field->getId(),
                $settings,
                $native_prefix
            );
        }

        $this->setDataOnCustomFields($this->getValues());

        foreach ($this->_field_facades as $facade) {
            $facade->setContentType($this->getStructure()->getContentType());
        }

        $this->_custom_fields_loaded = true;
    }

    /**
     * Turn a field into a facade for more consistent access.
     */
    protected function addFacade($id, $info, $name_prefix = '')
    {
        $name = $name_prefix . $id;
        $format = null;

        if (array_key_exists('field_fmt', $info)) {
            $format = $info['field_fmt'];
        }

        if ($this->hasProperty('field_ft_' . $id)) {
            $format = $this->getProperty('field_ft_' . $id) ?: $format;
            $this->setProperty('field_ft_' . $id, $format);
        }

        $facade = new FieldFacade($id, $info);
        $facade->setName($name);
        $facade->setContentId($this->getId());

        if (isset($format)) {
            $facade->setFormat($format);
        }

        $this->_field_facades[$name] = $facade;
    }

    /**
     * Find all the fields that are stored in their own tables. For those that
     * are dirty (have changed) we update or insert the changes into their
     * tables. If the list of changed properties is not supplied we will get
     * the list of dirty properties.
     *
     * @param array $changed An associative array of class properties that have changed
     */
    public function saveFieldData($changed = null)
    {
        $dirty = ($changed) ?: $this->getDirty();

        // Optimization: if there are no dirty fields, there's nothing to do
        if (empty($dirty)) {
            return;
        }

        foreach ($this->getStructure()->getAllCustomFields() as $field) {
            // Skip this field if it is in `exp_channel_data`
            if ($field->legacy_field_data) {
                continue;
            }

            $values = array();

            foreach ($field->getColumnNames() as $column) {
                if (array_key_exists($column, $dirty)) {
                    $values[$column] = $this->$column;
                }
            }

            // Skip this field if neither it nor its meta data changed
            if (empty($values)) {
                continue;
            }

            $query = ee('Model/Datastore')->rawQuery();

            $key_column = $this->getPrimaryKey();

            $query->where($key_column, $this->getId());
            $query->from($field->getTableName());
            $result = $query->get();

            $query = ee('Model/Datastore')->rawQuery();

            // When a new entity is saved, this will be triggered by an
            // onAfterInsert event (else, we won't have id to link to).
            // The primary key can only be marked dirty on an insert event,
            // not an update.
            if ($result->num_rows()) {
                $query->set($values);
                $query->where($key_column, $this->getId());
                $query->update($field->getTableName());
            } else {
                // Don't try to insert null values in case they're not allowed,
                // fall back to column default instead
                $values = array_filter($values, function ($value) {
                    return ! is_null($value);
                });
                $values[$key_column] = $this->getId();
                $query->set($values);
                $query->insert($field->getTableName());
            }
        }
    }

    /**
     * Deletes entry data in the field tables.
     */
    protected function deleteFieldData()
    {
        $tables = array();

        foreach ($this->getStructure()->getAllCustomFields() as $field) {
            // Skip this field if it is in `exp_channel_data`
            if ($field->legacy_field_data) {
                continue;
            }

            $tables[] = $field->getTableName();
        }

        if (! empty($tables)) {
            ee('Model/Datastore')->rawQuery()
                ->where($this->getPrimaryKey(), $this->getId())
                ->delete($tables);
        }
    }

    /**
     * Mark a property or the entire entity as dirty.
     *
     * @param String $name Property name [optional]
     */
    public function markAsDirty($name = null)
    {
        if (!empty($name)) {
            if (! $this->hasBackup($name) && $this->hasCustomField($name)) {
                $this->setBackup($name, $this->getRawProperty($name));
            }
        } else {
            foreach ($this->getCustomFieldNames() as $field) {
                $this->setBackup($field, $this->getRawProperty($field));
            }
        }

        parent::markAsDirty($name);

        return $this;
    }

    /**
     * Recount stats for file usage
     *
     * @param array $file_ids
     * @return void
     */
    protected static function updateFilesTotalRecords($file_ids = [])
    {
        if (!empty($file_ids)) {
            $updateQuery = 'UPDATE exp_files SET total_records = (SELECT COUNT(exp_file_usage.file_id) FROM exp_file_usage WHERE exp_file_usage.file_id = exp_files.file_id AND exp_file_usage.file_id IN (' . implode(', ', $file_ids) . ')) WHERE exp_files.file_id IN (' . implode(', ', $file_ids) . ')';
            ee('db')->query($updateQuery);
        }
    }
}

// EOF
