<?php

declare(strict_types=1);

namespace DS82\ThreeDPrintShowcase\Viewer;

use DS82\ThreeDPrintShowcase\Core\Settings;
use DS82\ThreeDPrintShowcase\Materials\Material;
use DS82\ThreeDPrintShowcase\Materials\MaterialRepository;

defined('ABSPATH') || exit;

final class ShortcodeViewer
{
    public function __construct(private readonly MaterialRepository $materials) {}

    public function register(): void
    {
        add_shortcode('ds82_3d_print_showcase', [$this, 'shortcode']);
        add_shortcode('ds82_3d_user_viewer', [$this, 'user_viewer_shortcode']);
        add_filter('script_loader_tag', [$this, 'module_script_tag'], 10, 3);
    }

    public function shortcode(array $attributes = []): string
    {
        $attributes = shortcode_atts([
            'model_url' => '', 'model_id' => 0, 'height' => 0, 'material' => '',
            'fallback_url' => '', 'auto_rotate' => '', 'force_selector' => 'no',
        ], $attributes, 'ds82_3d_print_showcase');

        $modelUrl = esc_url_raw((string) $attributes['model_url']);
        $modelId = absint($attributes['model_id']);
        $fallbackUrl = esc_url_raw((string) $attributes['fallback_url']);
        $attachmentPath = '';
        if (!$modelUrl && $modelId) {
            $modelUrl = (string) wp_get_attachment_url($modelId);
            $attachmentPath = (string) get_attached_file($modelId);
        }
        $urlPath = (string) wp_parse_url($modelUrl, PHP_URL_PATH);
        $formatSource = $attachmentPath !== '' ? $attachmentPath : $urlPath;
        $modelFormat = $modelUrl !== '' ? strtolower((string) pathinfo($formatSource, PATHINFO_EXTENSION)) : 'default';
        $supportedFormats = ['stl', '3mf', 'glb', 'gltf'];
        if ($modelUrl !== '' && !in_array($modelFormat, $supportedFormats, true)) {
            $message = sprintf(
                /* translators: 1: attachment ID, 2: detected file extension */
                __('Attachment ID %1$d does not point to a supported 3D model. Detected format: %2$s. Supported formats are STL, 3MF, GLB and GLTF.', '3d-print-showcase'),
                $modelId,
                $modelFormat !== '' ? strtoupper($modelFormat) : __('unknown', '3d-print-showcase')
            );
            return $this->renderFallbackOrError($fallbackUrl, $message, $attributes);
        }

        $settings = Settings::all();
        $height = absint($attributes['height']) ?: (int) $settings['height'];

        // The material attribute acts as an optional filter. When omitted, all
        // published materials from the active provider are available. A single
        // identifier keeps backwards compatibility and becomes the default.
        $requestedMaterialIds = $this->parseMaterialIdentifiers((string) $attributes['material']);
        $availableMaterials = $this->resolveMaterials($requestedMaterialIds);
        $material = $availableMaterials ? reset($availableMaterials) : null;
        $colors = $material ? $material->colors() : [(string) $settings['default_model_color']];
        $autoRotate = $attributes['auto_rotate'] === '' ? $settings['auto_rotate'] === 'yes' : in_array(strtolower((string) $attributes['auto_rotate']), ['1', 'yes', 'true'], true);
        $instanceId = 'ds82-3dps-shortcode-' . wp_rand(1000, 999999);
        $config = [
            'viewerId' => $instanceId,
            'productId' => 0,
            'modelUrl' => $modelUrl, 'modelFormat' => $modelFormat,
            'height' => min(1200, max(240, $height)), 'backgroundColor' => $material?->backgroundColor ?: (string) $settings['background_color'],
            'globalBackgroundColor' => (string) $settings['background_color'],
            'defaultMaterial' => $material?->identifier() ?? '',
            'materialEndpoint' => rest_url('ds82-3dps/v1/materials/'),
            'materials' => array_map(static fn (Material $item): array => $item->jsonSerialize(), $availableMaterials),
            'defaultModelColor' => $colors[0] ?? (string) $settings['default_model_color'], 'colors' => $colors,
            'colorMode' => $material?->colorMode ?? 'single', 'colorModeType' => $material?->colorModeType ?? 'standard',
            'appearanceType' => $material?->appearanceType ?? 'standard', 'roughness' => $material?->roughness ?? 0.55,
            'metalness' => $material?->metalness ?? 0.0, 'opacity' => $material?->opacity ?? 1.0,
            'textureUrl' => $material?->textureUrl ?? '', 'normalMapUrl' => $material?->normalMapUrl ?? '',
            'fallbackImageUrl' => $fallbackUrl, 'autoRotate' => $autoRotate,
            'fullscreen' => $settings['fullscreen'] === 'yes', 'lazyLoading' => $settings['lazy_loading'] === 'yes',
            'maxPixelRatio' => (float) $settings['max_pixel_ratio'], 'fpsLimit' => (int) $settings['fps_limit'],
            'strings' => [
                'loading' => __('Loading 3D model…', '3d-print-showcase'),
                'materialLoading' => __('Loading selected color…', '3d-print-showcase'),
                'error' => __('The 3D model could not be displayed.', '3d-print-showcase'),
                'webglError' => __('WebGL is not supported by this browser.', '3d-print-showcase'),
                'slicerProject3mf' => __('This 3MF file is a slicer project (Bambu Studio, OrcaSlicer, PrusaSlicer, or similar), not a standard 3D model. Export the model as a standard 3MF, GLB, or STL file.', '3d-print-showcase'),
            ],
        ];
        $this->enqueue_assets(!empty($availableMaterials) && count($availableMaterials) > 1);
        $json = wp_json_encode($config, JSON_HEX_TAG | JSON_HEX_APOS | JSON_HEX_AMP | JSON_HEX_QUOT | JSON_UNESCAPED_SLASHES);
        $disclaimer = $settings['disclaimer_enabled'] === 'yes' ? (string) $settings['disclaimer_text'] : '';
        ob_start(); ?>
        <section id="<?php echo esc_attr($instanceId); ?>" class="ds82-3dps-viewer" style="--ds82-3dps-height:<?php echo esc_attr((string) $config['height']); ?>px;--ds82-3dps-background:<?php echo esc_attr((string) $config['backgroundColor']); ?>" data-ds82-3dps-model-url="<?php echo esc_url((string) $config['modelUrl']); ?>" data-ds82-3dps-model-format="<?php echo esc_attr((string) $config['modelFormat']); ?>" aria-label="<?php esc_attr_e('Interactive 3D preview', '3d-print-showcase'); ?>">
            <script type="application/json" class="ds82-3dps-viewer__config"><?php echo wp_kses_post((string) $json); ?></script>
            <div class="ds82-3dps-viewer__stage"><div class="ds82-3dps-viewer__canvas" tabindex="0"></div><div class="ds82-3dps-viewer__loading" role="status"><span class="ds82-3dps-spinner" aria-hidden="true"></span><span><?php esc_html_e('Loading 3D model…', '3d-print-showcase'); ?></span></div><div class="ds82-3dps-viewer__error" role="alert" hidden></div>
            <?php if ($config['fallbackImageUrl']) : ?><img class="ds82-3dps-viewer__fallback" src="<?php echo esc_url((string) $config['fallbackImageUrl']); ?>" alt="" hidden><?php endif; ?>
            <div class="ds82-3dps-viewer__toolbar" role="toolbar" aria-label="<?php esc_attr_e('3D viewer controls', '3d-print-showcase'); ?>"><button type="button" data-action="reset" aria-label="<?php esc_attr_e('Reset view', '3d-print-showcase'); ?>">↺</button><button type="button" data-action="rotate" aria-pressed="<?php echo $config['autoRotate'] ? 'true' : 'false'; ?>" aria-label="<?php esc_attr_e('Toggle auto rotation', '3d-print-showcase'); ?>">⟳</button><?php if ($config['fullscreen']) : ?><button type="button" data-action="fullscreen" aria-label="<?php esc_attr_e('Fullscreen', '3d-print-showcase'); ?>">⛶</button><?php endif; ?></div></div>
            <?php if ($disclaimer !== '') : ?><p class="ds82-3dps-viewer__disclaimer"><?php echo esc_html($disclaimer); ?></p><?php endif; ?>
        </section>
        <?php if ((('yes' === (string) $settings['appearance_switching']) || in_array(strtolower((string) $attributes['force_selector']), ['1', 'yes', 'true'], true)) && count($availableMaterials) > 1) : ?>
            <?php echo $this->renderMaterialSelector($availableMaterials, (string) $material?->identifier(), $instanceId, $settings); // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped ?>
        <?php endif; ?>
        <?php return (string) ob_get_clean();
    }


    public function user_viewer_shortcode(array $attributes = []): string
    {
        $settings = Settings::all();
        if ('yes' !== (string) ($settings['user_viewer_enabled'] ?? 'yes')) {
            return current_user_can('manage_options') ? '<p>' . esc_html__('The User 3D Viewer is disabled in plugin settings.', '3d-print-showcase') . '</p>' : '';
        }

        $all = $this->resolveMaterials([]);
        $selected = array_map('strval', (array) ($settings['user_viewer_materials'] ?? []));
        $mode = (string) ($settings['user_viewer_material_mode'] ?? 'all');
        if ('include' === $mode) {
            $allowed = array_intersect_key($all, array_flip($selected));
        } elseif ('exclude' === $mode) {
            $allowed = array_diff_key($all, array_flip($selected));
        } else {
            $allowed = $all;
        }
        $materialIds = implode(',', array_keys($allowed));
        $viewer = $this->shortcode([
            'height' => absint($attributes['height'] ?? 0),
            'material' => $materialIds,
            'auto_rotate' => (string) ($attributes['auto_rotate'] ?? ''),
            'force_selector' => 'yes',
        ]);
        $formats = array_values(array_intersect(['stl', 'glb', '3mf'], (array) ($settings['user_viewer_formats'] ?? ['stl', 'glb'])));
        $accept = implode(',', array_map(static fn(string $format): string => '.' . $format, $formats));
        $maxSize = min(200, max(1, absint($settings['user_viewer_max_size'] ?? 25)));
        wp_enqueue_script('ds82-3dps-user-viewer', DS82_3DPS_URL . 'assets/js/frontend/user-viewer.js', [], DS82_3DPS_VERSION, true);

        ob_start(); ?>
        <section class="ds82-3dps-user-viewer" data-max-size="<?php echo esc_attr((string) $maxSize); ?>" data-formats="<?php echo esc_attr(implode(',', $formats)); ?>">
            <?php echo $viewer; // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped ?>
            <div class="ds82-3dps-user-viewer__upload">
                <h3><?php esc_html_e('Preview your own 3D model in our available colors', '3d-print-showcase'); ?></h3>
                <p><?php esc_html_e('Upload your own 3D model and preview it using the colors and materials available in this viewer.', '3d-print-showcase'); ?></p>
                <p><?php esc_html_e('Choose a model from your device. Processing takes place only in your browser; the file is not sent to the server.', '3d-print-showcase'); ?></p>
                <label class="ds82-3dps-user-viewer__button"><span><?php esc_html_e('Choose 3D model', '3d-print-showcase'); ?></span><input type="file" accept="<?php echo esc_attr($accept); ?>"></label>
                <span class="ds82-3dps-user-viewer__filename" aria-live="polite"></span>
                <p class="ds82-3dps-user-viewer__limits"><?php echo esc_html(sprintf(__('Allowed formats: %1$s. Maximum size: %2$d MB.', '3d-print-showcase'), strtoupper(implode(', ', $formats)), $maxSize)); ?></p>
                <p class="ds82-3dps-user-viewer__message" role="alert" hidden></p>
            </div>
        </section>
        <?php return (string) ob_get_clean();
    }

    /** @return string[] */
    private function parseMaterialIdentifiers(string $raw): array
    {
        if (trim($raw) === '') {
            return [];
        }

        return array_values(array_unique(array_filter(array_map(
            static fn (string $identifier): string => sanitize_text_field(trim($identifier)),
            preg_split('/[,|]+/', $raw) ?: []
        ))));
    }

    /** @param string[] $requested @return array<string, Material> */
    private function resolveMaterials(array $requested): array
    {
        $items = [];
        if (!$requested) {
            foreach ($this->materials->all() as $identifier => $material) {
                if ($material instanceof Material) {
                    $items[(string) $identifier] = $material;
                }
            }
            return $items;
        }

        foreach ($requested as $identifier) {
            $material = $this->materials->find($identifier);
            if ($material) {
                $items[$material->identifier()] = $material;
            }
        }
        return $items;
    }

    private function swatchStyle(array $colors, string $type): string
    {
        $colors = array_values(array_filter(array_map('sanitize_hex_color', $colors))) ?: ['#b7b7b7'];
        if (1 === count($colors)) {
            return 'background:' . $colors[0] . ';';
        }
        $type = strtolower($type);
        if (str_contains($type, 'layered') || str_contains($type, 'vrstven')) {
            $stops = [];
            $bandCount = 12;
            foreach (range(0, $bandCount - 1) as $band) {
                $color = $colors[$band % count($colors)];
                $start = round(($band / $bandCount) * 100, 2);
                $end = round((($band + 1) / $bandCount) * 100, 2);
                $stops[] = $color . ' ' . $start . '%';
                $stops[] = $color . ' ' . $end . '%';
            }
            return 'background:linear-gradient(to top,' . implode(',', $stops) . ');';
        }
        if (str_contains($type, 'duhov') || str_contains($type, 'rainbow')) {
            return 'background:linear-gradient(to top,' . implode(',', $colors) . ');';
        }
        if (str_contains($type, 'přech') || str_contains($type, 'gradient')) {
            return 'background:linear-gradient(to right,' . implode(',', $colors) . ');';
        }
        return 'background:conic-gradient(' . implode(',', $colors) . ');';
    }

    /** @param array<string, Material> $materials */
    private function renderMaterialSelector(array $materials, string $default, string $viewerId, array $settings): string
    {
        $displayMode = (string) ($settings['appearance_display'] ?? 'labels');
        $tooltips = 'yes' === (string) ($settings['appearance_tooltips'] ?? 'yes');
        ob_start(); ?>
        <?php
        $defaultMaterial = $materials[$default];
        $defaultColors = array_values(array_filter(array_map('sanitize_hex_color', $defaultMaterial->colors()))) ?: ['#b7b7b7'];
        $defaultStyle = $this->swatchStyle($defaultColors, (string) $defaultMaterial->colorModeType);
        ?>
        <fieldset class="ds82-3dps-material-selector<?php echo 'swatches' === $displayMode ? ' is-swatches-only' : ''; ?>" data-product-id="0" data-viewer-id="<?php echo esc_attr($viewerId); ?>" data-remember="no">
            <legend class="screen-reader-text"><?php esc_html_e('Choose an appearance', '3d-print-showcase'); ?></legend>
            <input type="hidden" name="ds82_3dps_material" value="<?php echo esc_attr($default); ?>">
            <button type="button" class="ds82-3dps-material-selector__trigger" aria-expanded="false">
                <span class="ds82-3dps-material-selector__trigger-swatch" style="<?php echo esc_attr($defaultStyle); ?>" aria-hidden="true"></span>
                <span class="ds82-3dps-material-selector__trigger-label"><?php esc_html_e('Appearance', '3d-print-showcase'); ?>:</span>
                <span class="ds82-3dps-material-selector__selected"><?php echo esc_html($defaultMaterial->label()); ?></span>
                <span class="ds82-3dps-material-selector__trigger-arrow" aria-hidden="true">▾</span>
            </button>
            <div class="ds82-3dps-material-selector__panel" hidden>
                <div class="ds82-3dps-material-selector__header">
                    <strong><?php esc_html_e('Choose an appearance', '3d-print-showcase'); ?></strong>
                    <button type="button" class="ds82-3dps-material-selector__close" aria-label="<?php esc_attr_e('Close appearance selection', '3d-print-showcase'); ?>">×</button>
                </div>
                <label class="ds82-3dps-material-selector__search-wrap">
                    <span class="screen-reader-text"><?php esc_html_e('Search appearances', '3d-print-showcase'); ?></span>
                    <input type="search" class="ds82-3dps-material-selector__search" placeholder="<?php esc_attr_e('Search appearances…', '3d-print-showcase'); ?>">
                </label>
                <div class="ds82-3dps-material-selector__grid">
                    <?php foreach ($materials as $identifier => $material) :
                        $colors = array_values(array_filter(array_map('sanitize_hex_color', $material->colors())));
                        $colors = $colors ?: ['#b7b7b7'];
                        $style = $this->swatchStyle($colors, (string) $material->colorModeType);
                        $selected = $identifier === $default;
                    ?>
                        <button type="button" class="ds82-3dps-material-option<?php echo $selected ? ' is-selected' : ''; ?>" data-material-id="<?php echo esc_attr((string) $identifier); ?>" data-material-label="<?php echo esc_attr($material->label()); ?>" aria-pressed="<?php echo $selected ? 'true' : 'false'; ?>" title="<?php echo $tooltips ? esc_attr($material->label()) : ''; ?>">
                            <span class="ds82-3dps-material-option__swatch" style="<?php echo esc_attr($style); ?>" aria-hidden="true"></span>
                            <span class="ds82-3dps-material-option__text"><strong><?php echo esc_html($material->display_id()); ?></strong><span><?php echo esc_html($material->name); ?></span></span>
                        </button>
                    <?php endforeach; ?>
                </div>
                <p class="ds82-3dps-material-selector__empty" hidden><?php esc_html_e('No matching appearances were found.', '3d-print-showcase'); ?></p>
            </div>
        </fieldset>
        <?php return (string) ob_get_clean();
    }

    private function renderFallbackOrError(string $fallbackUrl, string $message, array $attributes): string
    {
        if ($fallbackUrl !== '') {
            $settings = Settings::all();
            $height = absint($attributes['height'] ?? 0) ?: (int) $settings['height'];
            $height = min(1200, max(240, $height));
            $background = (string) ($settings['background_color'] ?? '#f7f7f7');

            wp_enqueue_style('ds82-3dps-frontend', DS82_3DPS_URL . 'assets/css/frontend.css', [], DS82_3DPS_VERSION);

            return sprintf(
                '<section class="ds82-3dps-viewer ds82-3dps-viewer--fallback-only" style="--ds82-3dps-height:%1$dpx;--ds82-3dps-background:%2$s"><div class="ds82-3dps-viewer__stage"><img class="ds82-3dps-viewer__fallback" src="%3$s" alt=""></div></section>',
                $height,
                esc_attr($background),
                esc_url($fallbackUrl)
            );
        }

        return current_user_can('manage_options')
            ? '<p class="ds82-3dps-viewer-error">' . esc_html($message) . '</p>'
            : '';
    }

    public function module_script_tag(string $tag, string $handle, string $src): string
    { return $handle === 'ds82-3dps-viewer' && !str_contains($tag, 'type=') ? str_replace('<script ', '<script type="module" ', $tag) : $tag; }

    private function enqueue_assets(bool $withSelector = false): void
    {
        wp_enqueue_style('ds82-3dps-frontend', DS82_3DPS_URL . 'assets/css/frontend.css', [], DS82_3DPS_VERSION);
        if ($withSelector) {
            wp_enqueue_script('ds82-3dps-material-selector', DS82_3DPS_URL . 'assets/js/frontend/material-selector.js', [], DS82_3DPS_VERSION, true);
        }
        if (function_exists('wp_enqueue_script_module')) wp_enqueue_script_module('ds82-3dps-viewer', DS82_3DPS_URL . 'assets/js/frontend/viewer.js', [], DS82_3DPS_VERSION);
        else wp_enqueue_script('ds82-3dps-viewer', DS82_3DPS_URL . 'assets/js/frontend/viewer.js', [], DS82_3DPS_VERSION, true);
    }
}