<?php
/**
 * Plugin Name: AI Summarise Button
 * Description: Gutenberg block + shortcode to open the current page in your chosen AI for summarising (ChatGPT basic, or Advanced provider picker). Includes alignment & color controls.
 * Version: 2.9.6
 * Author: You
 * License: GPLv2 or later
 * License URI: https://www.gnu.org/licenses/gpl-2.0.html
 * Requires PHP: 7.2
 */

if ( ! defined( 'ABSPATH' ) ) { exit; }

define( 'AISUBU_VER', '2.9.6' );
define( 'AISUBU_URL', plugin_dir_url( __FILE__ ) );
define( 'AISUBU_PATH', plugin_dir_path( __FILE__ ) );

if ( ! function_exists( 'aisubu_provider_url' ) ) {
  function aisubu_provider_url( $provider, $q ) {
    $base = 'https://chatgpt.com/?hints=search&q=' . rawurlencode( $q );
    if ( $provider === 'perplexity' ) { $base = 'https://www.perplexity.ai/?q=' . rawurlencode( $q ); }
    elseif ( $provider === 'copilot' ) { $base = 'https://copilot.microsoft.com/?q=' . rawurlencode( $q ); }
    elseif ( $provider === 'claude' ) { $base = 'https://claude.ai/new'; }
    elseif ( $provider === 'gemini' ) { $base = 'https://gemini.google.com/app'; }
    return $base;
  }
}

if ( ! function_exists( 'aisubu_build_q' ) ) {
  function aisubu_build_q( $title, $url, $attrs ) {
    $prefix = isset( $attrs['qPrefix'] ) ? $attrs['qPrefix'] : 'Read and summarise this';
    $noun   = isset( $attrs['noun'] ) ?  $attrs['noun']  : 'page';
    $suffix = isset( $attrs['qSuffix'] ) ? $attrs['qSuffix'] : 'so I can ask questions about it.';
    $title  = trim( wp_strip_all_tags( $title ) );
    return sprintf( '%s "%s" %s from %s %s', $prefix, $title, $noun, $url, $suffix );
  }
}

if ( ! function_exists( 'aisubu_enqueue_shared' ) ) {
  function aisubu_enqueue_shared() {
    wp_register_style( 'aisubu-frontend', AISUBU_URL . 'inc/frontend.css', array(), AISUBU_VER );
    wp_enqueue_style( 'aisubu-frontend' );

    $svg = plugins_url( 'inc/ai-logo.svg', __FILE__ );
    $png = plugins_url( 'inc/ai-logo.png', __FILE__ );
    $inline  = '.ai-summarise-logo{background-image:url("' . esc_url( $svg ) . '")}';
    $inline .= '.aisubu-no-svg .ai-summarise-logo{background-image:url("' . esc_url( $png ) . '")}';
    wp_add_inline_style( 'aisubu-frontend', $inline );

    wp_register_script( 'aisubu-frontend', AISUBU_URL . 'inc/frontend.js', array(), AISUBU_VER, true );
    wp_enqueue_script( 'aisubu-frontend' );
  }
}

add_action( 'wp_enqueue_scripts', 'aisubu_enqueue_shared' );
add_action( 'enqueue_block_editor_assets', 'aisubu_enqueue_shared' );

if ( ! function_exists( 'aisubu_render_button' ) ) {
  function aisubu_render_button( $attributes ) {
    $defaults = array(
      'mode'        => 'basic',
      'buttonText'  => 'Summarise with AI',
      'qPrefix'     => 'Read and summarise this',
      'qSuffix'     => 'so I can ask questions about it.',
      'noun'        => 'page',
      'provider'    => 'chatgpt',
      'bgColor'     => '',
      'textColor'   => '',
      'borderColor' => '',
      'borderRadius'=> 8,
      'paddingX'    => 10,
      'paddingY'    => 6,
      'isFullWidth' => false,
      'iconPosition'=> 'left',
      'justify'     => 'left',
      'logoSize'    => 15,
      'maxWidth'    => 200,
      'maxHeight'   => 32,
      'preset'      => 'chatgpt',
      'openInNewTab'=> true,
      'relNoFollow' => true,
      'align'       => '',
    );
    $a = wp_parse_args( $attributes, $defaults );

    $title = function_exists( 'wp_get_document_title' ) ? wp_get_document_title() : get_bloginfo( 'name' );
    $url   = home_url( '/' );
    if ( function_exists( 'is_singular' ) && is_singular() ) {
      $t = get_the_title(); if ( $t ) $title = $t;
      $p = get_permalink(); if ( $p ) $url = $p;
    }
    $q = aisubu_build_q( $title, $url, array( 'qPrefix' => $a['qPrefix'], 'qSuffix' => $a['qSuffix'], 'noun' => $a['noun'] ) );

    $bg = $a['bgColor']; $fg = $a['textColor']; $bd = $a['borderColor'];
    if ( $a['preset'] !== 'custom' && ( ! $bg && ! $fg && ! $bd ) ) {
      if ( $a['preset'] === 'chatgpt' ) { $bg = '#10A37F'; $fg = '#ffffff'; $bd = '#0D8667'; }
      elseif ( $a['preset'] === 'outline' ) { $bg = '#ffffff'; $fg = '#111827'; $bd = '#D1D5DB'; }
      elseif ( $a['preset'] === 'dark' ) { $bg = '#111827'; $fg = '#ffffff'; $bd = '#1F2937'; }
      elseif ( $a['preset'] === 'blue' ) { $bg = '#2563EB'; $fg = '#ffffff'; $bd = '#1E40AF'; }
    }
    if ( ! $bg ) $bg = '#10A37F'; if ( ! $fg ) $fg = '#ffffff'; if ( ! $bd ) $bd = 'rgba(0,0,0,.08)';

    $rel = 'noopener'; if ( $a['relNoFollow'] ) $rel .= ' nofollow';
    $target = $a['openInNewTab'] ? '_blank' : '_self';

    $jc = 'flex-start'; $ta = 'left'; $auto = '';
    if ( $a['justify'] === 'center' ) { $jc='center'; $ta='center'; $auto='margin-left:auto;margin-right:auto;'; }
    elseif ( $a['justify'] === 'right' ) { $jc='flex-end'; $ta='right'; $auto='margin-left:auto;'; }
    $wrap_style = 'width:100%;display:flex;align-items:center;gap:.5rem;margin:.5rem 0;justify-content:'.$jc.';text-align:'.$ta.';';

    $btn_style  = $auto.'display:inline-flex;align-items:center;gap:0;border:1px solid '.esc_attr($bd).';border-radius:'.intval($a['borderRadius']).'px;'
                . 'background:'.esc_attr($bg).';color:'.esc_attr($fg).';text-decoration:none;line-height:1;'
                . 'max-width:'.intval($a['maxWidth']).'px;height:'.intval($a['maxHeight']).'px;'
                . 'padding:'.intval($a['paddingY']).'px '.intval($a['paddingX']).'px;white-space:nowrap;overflow:hidden;';
    if ( $a['isFullWidth'] ) $btn_style .= 'width:100%;justify-content:center;';

    $icon_class = 'ai-summarise-logo' . ( $a['iconPosition'] === 'right' ? ' right' : '' );
    $icon_html  = '<span class="ai-summarise-logo-wrap"><span class="'. esc_attr($icon_class) .'" style="width:'.intval($a['logoSize']).'px;height:'.intval($a['logoSize']).'px;line-height:0;vertical-align:middle;"></span></span>';
    $label_html = '<span class="asb-label" style="line-height:1;text-overflow:ellipsis;overflow:hidden">'.esc_html($a['buttonText']).'</span>';
    $inner      = ( $a['iconPosition'] === 'right' ) ? ( $label_html.$icon_html ) : ( $icon_html.$label_html );

    if ( $a['mode'] === 'basic' ) {
      $href = aisubu_provider_url( 'chatgpt', $q );
      return '<div class="asb-wrap" style="'. $wrap_style .'"><a class="asb-button" style="'. $btn_style .'" target="'. esc_attr($target) .'" rel="'. esc_attr($rel) .'" href="'. esc_url($href) .'">'. $inner .'</a></div>';
    } else {
      return '<div class="asb-wrap" style="'. $wrap_style .'"><a href="#" class="asb-button" data-mode="advanced" data-q="'. esc_attr($q) .'" data-default="'. esc_attr($a['provider']) .'" style="'. $btn_style .'" target="'. esc_attr($target) .'" rel="'. esc_attr($rel) .'">'. $inner .'</a></div>';
    }
  }
}

if ( ! function_exists( 'aisubu_shortcode' ) ) {
  function aisubu_shortcode( $atts ) {
    $map = array(
      'mode'=>'basic','text'=>'Summarise with AI','preset'=>'chatgpt','align'=>'left','width'=>'200','height'=>'32','logosize'=>'15',
      'qprefix'=>'Read and summarise this','qsuffix'=>'so I can ask questions about it.','noun'=>'page',
      'provider'=>'chatgpt','newtab'=>'1','nofollow'=>'1','bgcolor'=>'','textcolor'=>'','bordercolor'=>''
    );
    $atts = shortcode_atts( $map, $atts, 'aisubu_summarise_button' );
    $attributes = array(
      'mode'        => $atts['mode'],
      'buttonText'  => $atts['text'],
      'preset'      => $atts['preset'],
      'justify'     => $atts['align'],
      'maxWidth'    => intval($atts['width']),
      'maxHeight'   => intval($atts['height']),
      'logoSize'    => intval($atts['logosize']),
      'qPrefix'     => $atts['qprefix'],
      'qSuffix'     => $atts['qsuffix'],
      'noun'        => $atts['noun'],
      'provider'    => $atts['provider'],
      'openInNewTab'=> ( $atts['newtab'] === '1' ),
      'relNoFollow' => ( $atts['nofollow'] === '1' ),
      'bgColor'     => $atts['bgcolor'],
      'textColor'   => $atts['textcolor'],
      'borderColor' => $atts['bordercolor'],
    );
    return aisubu_render_button( $attributes );
  }
  add_shortcode( 'aisubu_summarise_button', 'aisubu_shortcode' );
}

if ( ! function_exists( 'aisubu_register_block' ) ) {
  function aisubu_register_block() {
    if ( ! function_exists( 'register_block_type' ) ) return;

    wp_register_script(
      'aisubu-block',
      plugins_url( 'block.js', __FILE__ ),
      array( 'wp-blocks','wp-element','wp-components','wp-block-editor','wp-i18n' ),
      AISUBU_VER,
      true
    );

    register_block_type( 'ai/summarise-button', array(
      'api_version'    => 2,
      'editor_script'  => 'aisubu-block',
      'render_callback'=> 'aisubu_render_button',
      'attributes'     => array(
        'mode' => array('type'=>'string','default'=>'basic'),
        'buttonText' => array('type'=>'string','default'=>'Summarise with AI'),
        'preset' => array('type'=>'string','default'=>'chatgpt'),
        'justify' => array('type'=>'string','default'=>'left'),
        'align' => array('type'=>'string'),
        'maxWidth' => array('type'=>'number','default'=>200),
        'maxHeight'=> array('type'=>'number','default'=>32),
        'logoSize' => array('type'=>'number','default'=>15),
        'qPrefix' => array('type'=>'string','default'=>'Read and summarise this'),
        'qSuffix' => array('type'=>'string','default'=>'so I can ask questions about it.'),
        'noun' => array('type'=>'string','default'=>'page'),
        'provider' => array('type'=>'string','default'=>'chatgpt'),
        'openInNewTab'=> array('type'=>'boolean','default'=>true),
        'relNoFollow' => array('type'=>'boolean','default'=>true),
        'iconPosition'=> array('type'=>'string','default'=>'left'),
        'isFullWidth' => array('type'=>'boolean','default'=>false),
        'borderRadius'=> array('type'=>'number','default'=>8),
        'paddingX' => array('type'=>'number','default'=>10),
        'paddingY' => array('type'=>'number','default'=>6),
        'bgColor' => array('type'=>'string','default'=>''),
        'textColor' => array('type'=>'string','default'=>''),
        'borderColor' => array('type'=>'string','default'=>''),
      ),
      'supports'       => array(
        'html' => false,
        'align' => array( 'left', 'center', 'right' )
      )
    ) );
  }
  add_action( 'init', 'aisubu_register_block' );
}