<?php declare(strict_types=1);

namespace Grav\Plugin\Form;

use Twig\Extension\AbstractExtension;
use Twig\TwigFunction;
use function is_string;

/**
 * Class GravExtension
 * @package Grav\Common\Twig\Extension
 */
class TwigExtension extends AbstractExtension
{
    /**
     * Return a list of all functions.
     *
     * @return array
     */
    public function getFunctions(): array
    {
        return [
            new TwigFunction('prepare_form_fields', [$this, 'prepareFormFields']),
            new TwigFunction('prepare_form_field', [$this, 'prepareFormField']),
            new TwigFunction('include_form_field', [$this, 'includeFormField']),
        ];
    }

    /**
     * Filters form fields for the current parent.
     *
     * @param array $fields Form fields
     * @param string|null $parent Parent field name if available
     * @return array
     */
    public function prepareFormFields($fields, $parent = null): array
    {
        $list = [];

        if (is_iterable($fields)) {
            foreach ($fields as $name => $field) {
                $field = $this->prepareFormField($field, $name, $parent);
                if ($field) {
                    $list[$field['name']] = $field;
                }
            }
        }

        return $list;
    }

    /**
     * Filters field name by changing dot notation into array notation.
     *
     * @param array $field Form field
     * @param string|int|null $name Field name (defaults to field.name)
     * @param string|null $parent Parent field name if available
     * @param array|null $options List of options to override
     * @return array|null
     */
    public function prepareFormField($field, $name = null, $parent = null, array $options = []): ?array
    {
        // Make sure that the field is a valid form field type and is not being ignored.
        if (empty($field['type']) || ($field['validate']['ignore'] ?? false)) {
            return null;
        }

        // Check if we have just a list of fields (no name given).
        if (is_int($name)) {
            // Look at the field.name and if not set, fall back to the key.
            $name = (string)($field['name'] ?? $name);
        }

        // Make sure that the field has a name.
        $name = $name ?? $field['name'] ?? null;
        if (!is_string($name) || $name === '') {
            return null;
        }

        // Prefix name with the parent name if needed.
        if (str_starts_with($name, '.')) {
            $name = $parent ? $parent . $name : (string)substr($name, 1);
        } elseif (isset($options['key'])) {
            $name = str_replace('*', $options['key'], $name);
        }

        unset($options['key']);

        // Loop through options
        foreach ($options as $key => $option) {
            $field[$key] = $option;
        }

        // Always set field name.
        $field['name'] = $name;

        return $field;
    }

    /**
     * @param string $type
     * @param string|null $layout
     * @param string|null $default
     * @return string[]
     */
    public function includeFormField(string $type, string $layout = null, string $default = null): array
    {
        $list = [
            "forms/fields/{$type}/{$layout}-{$type}.html.twig",
            "forms/fields/{$type}/{$type}.html.twig",
        ];
        if ($default) {
            $list = array_merge(
                $list,
                [
                    "forms/fields/{$default}/{$layout}-{$default}.html.twig",
                    "forms/fields/{$default}/{$default}.html.twig",
                ]
            );
        }

        return $list;
    }
}
