<?php
/**
 * Plugin Name: 3D Viewer Block
 * Description: Display interactive 3D models on the web
 * Version: 1.0.0
 * Author: bPlugins LLC
 * Author URI: http://bplugins.com
 * License: GPLv3
 * License URI: https://www.gnu.org/licenses/gpl-3.0.txt
 * Text Domain: 3d-viewer-block
 */

// ABS PATH
if ( !defined( 'ABSPATH' ) ) { exit; }

// Constant
define( 'TDVB_PLUGIN_VERSION', 'localhost' === $_SERVER['HTTP_HOST'] ? time() : '1.0.0' );
define( 'TDVB_ASSETS_DIR', plugin_dir_url( __FILE__ ) . 'assets/' );

// Generate Styles
class TDVBStyleGenerator {
    public static $styles = [];
    public static function addStyle( $selector, $styles ){
        if( array_key_exists( $selector, self::$styles ) ){
           self::$styles[$selector] = wp_parse_args( self::$styles[$selector], $styles );
        }else { self::$styles[$selector] = $styles; }
    }
    public static function renderStyle(){
        $output = '';
        foreach( self::$styles as $selector => $style ){
            $new = '';
            foreach( $style as $property => $value ){
                if( $value == '' ){ $new .= $property; }else { $new .= " $property: $value;"; }
            }
            $output .= "$selector { $new }";
        }
        return $output;
    }
}

// 3D Viewer Block
class tdvb3DViewerBlock{
    protected static $_instance = null;

    function __construct(){
        add_action( 'enqueue_block_assets', [$this, 'enqueueBlockAssets'] );
        add_action( 'init', [$this, 'register'] );

        add_filter( 'upload_mimes', [$this, 'uploadMimes'] );
        if ( version_compare( $GLOBALS['wp_version'], '5.1' ) >= 0 ) {
            add_filter( 'wp_check_filetype_and_ext', [$this, 'wpCheckFiletypeAndExt'], 10, 5 );
        } else { add_filter( 'wp_check_filetype_and_ext', [$this, 'wpCheckFiletypeAndExt'], 10, 4 ); }
    }

    public static function instance(){
        if( self::$_instance === null ){
            self::$_instance = new self();
        }
        return self::$_instance;
    }

    function enqueueBlockAssets(){
        echo wp_get_script_tag( [
            'src'  => TDVB_ASSETS_DIR . 'js/model-viewer.min.js',
            'type' => 'module',
        ] );
    }

    function register() {
        wp_register_script( 'tdvb_editor_script', plugins_url( 'dist/editor.js', __FILE__ ), [ 'wp-blob', 'wp-block-editor', 'wp-blocks', 'wp-components', 'wp-compose', 'wp-data', 'wp-element', 'wp-html-entities', 'wp-i18n', 'wp-rich-text' ], TDVB_PLUGIN_VERSION, false ); // Backend Script
        wp_register_style( 'tdvb_editor_style', plugins_url( 'dist/editor.css', __FILE__ ), [ 'wp-edit-blocks' ], TDVB_PLUGIN_VERSION ); // Backend Style
        wp_register_script( 'tdvb_script', plugins_url( 'dist/script.js', __FILE__ ), [ 'jquery' ], TDVB_PLUGIN_VERSION, true ); // Frontend Script
        wp_register_style( 'tdvb_style', plugins_url( 'dist/style.css', __FILE__ ), [ 'wp-editor' ], TDVB_PLUGIN_VERSION ); // Both Style

        register_block_type( 'tdvb/td-viewer', [
            'editor_script' => 'tdvb_editor_script',
            'editor_style'  => 'tdvb_editor_style',
            'script'        => 'tdvb_script',
            'style'         => 'tdvb_style',
            'render_callback' => [$this, 'render']
        ] ); // Register Block

        wp_set_script_translations( 'tdvb_editor_script', '3d-viewer-block', plugin_dir_path( __FILE__ ) . 'languages' ); // Translate
    }

    function render( $attributes ){
        extract( $attributes );
        $align = $align ?? '';
        $cId = $cId ?? '';
        $alignment = $alignment ?? 'center';
        $sourceType = $sourceType ?? 'upload';
        $model = $model ?? [ 'id' => null, 'url' => '', 'title' => '' ];
        $modelLink = $modelLink ?? 'https://modelviewer.dev/shared-assets/models/shishkebab.glb';
        $width = $width ?? '100%';
        $height = $height ?? '350px';
        $isTouchMove = $isTouchMove ?? true;
        $isZoom = $isZoom ?? true;
        $loadingType = $loadingType ?? 'auto';

        $tDViewerStyle = new TDVBStyleGenerator(); // Generate Styles
        $tDViewerStyle::addStyle( "#tdvb3DViewerBlock-$cId .tdvb3DViewerBlock", [
            'text-align' => $alignment
        ] );
        $tDViewerStyle::addStyle( "#tdvb3DViewerBlock-$cId .tdvb3DViewerBlock .tdvbModel", [
            'width' => '0px' === $width || '0%' === $width || '0em' === $width ? 'auto' : $width,
            'height' => '0px' === $height || '0%' === $height || '0em' === $height ? '350px' : $height,
        ] );

        $jsonData = wp_json_encode( [ 'isTouchMove' => $isTouchMove, 'isZoom' => $isZoom ] );

        ob_start(); ?>
        <div class='wp-block-tdvb-td-viewer <?php echo 'align' . esc_attr( $align ); ?>' id='tdvb3DViewerBlock-<?php echo esc_attr( $cId ) ?>' data-model='<?php echo esc_attr( $jsonData ); ?>'>
            <style><?php echo wp_kses( $tDViewerStyle::renderStyle(), [] ) ?></style>

            <div class='tdvb3DViewerBlock'>
                <model-viewer
                    class='tdvbModel'
                    src='<?php echo esc_url( 'upload' === $sourceType ? $model['url'] : $modelLink ); ?>'
                    alt='<?php echo esc_attr( 'upload' === $sourceType ? $model['title'] : '' ); ?>'
                    camera-controls
                    loading='<?php echo esc_attr( $loadingType ); ?>'
                    auto-rotate
                ></model-viewer>
            </div> <!-- 3D Viewer Block -->
        </div>

        <?php $tDViewerStyle::$styles = []; // Empty styles
        return ob_get_clean();
    } // Render

    //Allow some additional file types for upload
    function uploadMimes( $mimes ) {
        // New allowed mime types.
        $mimes['glb'] = 'model/gltf-binary';
        $mimes['gltf'] = 'model/gltf-binary';
        return $mimes;
    }
    function wpCheckFiletypeAndExt( $data, $file, $filename, $mimes, $real_mime=null ){
        // If file extension is 2 or more 
        $f_sp = explode( '.', $filename );
        $f_exp_count = count( $f_sp );

        if( $f_exp_count <= 1 ){
            return $data;
        }else{
            $f_name = $f_sp[0];
            $ext = $f_sp[$f_exp_count - 1];
        }

        if( $ext == 'glb' || $ext == 'gltf' ){
            $type = 'model/gltf-binary';
            $proper_filename = '';
            return compact('ext', 'type', 'proper_filename');
        }else {
            return $data;
        }
    }
}
tdvb3DViewerBlock::instance();