<?php
/**
 * Batch 7 — WPBakery Page Builder (js_composer) tools (Pro-equivalent, net-new).
 *
 * Independent implementation against WPBakery's public API + its shortcode content
 * model. GPL-2.0-or-later. See NOTICE.md.
 *
 * Stays INERT unless WPBakery (WPB_VC_VERSION) is active, so it adds nothing on sites
 * without it and lights up automatically on sites that run WPBakery.
 * Registered via the asktam_mcp_register_tools hook (glob auto-loader).
 */

namespace AskTamMCP\Extensions;

defined('ABSPATH') || exit;

class Batch7 {

    public static function register($registry): void {
        if (!defined('WPB_VC_VERSION')) return; // WPBakery-only batch.

        $def = function (string $name, string $desc, array $props, array $required, string $scope, string $method) {
            $schema = ['type' => 'object', 'properties' => $props];
            if ($required) $schema['required'] = $required;
            return ['name' => $name, 'description' => $desc, 'scope' => $scope,
                    'handler' => 'AskTamMCP\\Extensions\\Batch7::' . $method, 'inputSchema' => $schema];
        };

        $registry->register($def('wpbakery_detect', 'Report whether WPBakery (js_composer) is active and its version.', [], [], 'read', 'detect'));

        $registry->register($def('wpbakery_list_elements', 'List registered WPBakery elements (shortcode base, name, category) from the WPBMap registry.',
            ['category' => ['type' => 'string'], 'per_page' => ['type' => 'integer', 'default' => 300]], [], 'read', 'listElements'));

        $registry->register($def('wpbakery_get_content', 'Get a post\'s WPBakery content: the raw shortcode markup plus the list of top-level element tags in order.',
            ['post_id' => ['type' => 'integer']], ['post_id'], 'read', 'getContent'));

        $registry->register($def('wpbakery_add_element', 'Append a WPBakery element to a post. Provide raw shortcode in "shortcode", or build one from tag + attrs + content. mode=append (default) or replace.',
            ['post_id' => ['type' => 'integer'], 'shortcode' => ['type' => 'string'], 'tag' => ['type' => 'string'],
             'attrs' => ['type' => 'object'], 'content' => ['type' => 'string'],
             'mode' => ['type' => 'string', 'enum' => ['append', 'replace'], 'default' => 'append']],
            ['post_id'], 'write', 'addElement'));

        $registry->register($def('wpbakery_replace_text', 'Find and replace a literal string within a post\'s WPBakery content.',
            ['post_id' => ['type' => 'integer'], 'search' => ['type' => 'string'], 'replace' => ['type' => 'string']],
            ['post_id', 'search', 'replace'], 'write', 'replaceText'));
    }

    public static function detect(array $args): array {
        return ['active' => defined('WPB_VC_VERSION'), 'version' => defined('WPB_VC_VERSION') ? WPB_VC_VERSION : null];
    }

    public static function listElements(array $args): array {
        if (!class_exists('WPBMap')) throw new \RuntimeException('WPBMap registry unavailable.');
        $cat = isset($args['category']) ? (string) $args['category'] : '';
        $per = (int) ($args['per_page'] ?? 300);
        $all = \WPBMap::getAllShortCodes();
        $rows = [];
        foreach ((array) $all as $base => $meta) {
            $c = '';
            if (is_array($meta)) $c = is_array($meta['category'] ?? null) ? implode(',', $meta['category']) : (string) ($meta['category'] ?? '');
            if ($cat !== '' && stripos($c, $cat) === false) continue;
            $rows[] = ['base' => $base, 'name' => is_array($meta) ? (string) ($meta['name'] ?? $base) : $base, 'category' => $c];
        }
        return ['count' => count($rows), 'elements' => array_slice($rows, 0, $per)];
    }

    public static function getContent(array $args): array {
        $id = (int) ($args['post_id'] ?? 0);
        $post = $id ? get_post($id) : null;
        if (!$post) throw new \RuntimeException("Post not found: {$id}");
        $raw = (string) $post->post_content;
        preg_match_all('/\[(vc_[a-z0-9_]+)/i', $raw, $m);
        return ['post_id' => $id, 'raw' => $raw, 'top_level_tags' => $m[1] ?? [], 'tag_count' => count($m[1] ?? [])];
    }

    private static function buildShortcode(array $args): string {
        if (!empty($args['shortcode'])) return (string) $args['shortcode'];
        $tag = sanitize_key((string) ($args['tag'] ?? ''));
        if ($tag === '') throw new \RuntimeException('Provide "shortcode" or "tag".');
        $attrs = '';
        foreach ((array) ($args['attrs'] ?? []) as $k => $v) {
            $attrs .= ' ' . sanitize_key((string) $k) . '="' . esc_attr((string) $v) . '"';
        }
        $content = (string) ($args['content'] ?? '');
        return "[{$tag}{$attrs}]" . $content . "[/{$tag}]";
    }

    public static function addElement(array $args): array {
        $id = (int) ($args['post_id'] ?? 0);
        $post = $id ? get_post($id) : null;
        if (!$post) throw new \RuntimeException("Post not found: {$id}");
        $sc = self::buildShortcode($args);
        $mode = (($args['mode'] ?? 'append') === 'replace') ? 'replace' : 'append';
        $new = $mode === 'replace' ? $sc : rtrim((string) $post->post_content) . "\n" . $sc;
        $res = wp_update_post(['ID' => $id, 'post_content' => $new], true);
        if (is_wp_error($res)) throw new \RuntimeException($res->get_error_message());
        return ['post_id' => $id, 'mode' => $mode, 'added' => $sc];
    }

    public static function replaceText(array $args): array {
        $id = (int) ($args['post_id'] ?? 0);
        $post = $id ? get_post($id) : null;
        if (!$post) throw new \RuntimeException("Post not found: {$id}");
        $search = (string) ($args['search'] ?? '');
        if ($search === '') throw new \RuntimeException('search is required.');
        $replace = (string) ($args['replace'] ?? '');
        $content = (string) $post->post_content;
        $count = 0;
        $new = str_replace($search, $replace, $content, $count);
        if ($count === 0) return ['post_id' => $id, 'replacements' => 0, 'note' => 'Search string not found.'];
        $res = wp_update_post(['ID' => $id, 'post_content' => $new], true);
        if (is_wp_error($res)) throw new \RuntimeException($res->get_error_message());
        return ['post_id' => $id, 'replacements' => $count];
    }
}