<?php
/*
Plugin Name: AccessibilityPlus
Plugin URI: 
Description: Improve pagespeed insight or lighthouse accessibility scores by fixing recommended issues.
Version: 1.2
Author: easywpstuff
Author URI: https://easywpstuff.com
License: GNU General Public License v2 or later
*/


require_once('vendor/autoload.php');

// include option

include_once('inc/options.php');


// Name: Form elements do not have associated labels
function easywpstuff_assesplus_check_input_labels($html) {
  if (empty($html)) {
    return $html;
  }

  $dom = new \DiDom\Document($html);
  $inputs = $dom->find('input');

  foreach ($inputs as $input) {
    $parent = $input->parent();
    if ($parent && $parent->tag !== 'label') {
      if (!$input->hasAttribute('id') && !$input->hasAttribute('placeholder') && !$input->hasAttribute('aria-label') && !$input->hasAttribute('aria-labelledby')) {
        $inputType = $input->getAttribute('type');
        if ($inputType && !in_array($inputType, ['hidden', 'button', 'submit'])) {
          $input->setAttribute('aria-label', $inputType);
        }
      }
    }
  }

  return $dom->html();
}

// Links do not have a discernible name
function easywpstuff_assesplus_add_aria_labels_to_links($html) {
    // Remove closing tags for img and source elements to avoid parsing issues
    $html = str_replace('</img>', '', $html);
    $html = str_replace('</source>', '', $html);

    $dom = new \DiDom\Document($html);
    if (empty($dom)) {
       return $dom;
    }

    $links = $dom->find('a');
    $filteredHeaderLinks = array_filter($dom->find('header a'), function ($link) {
        return $link->has('i') || ($link->has('span') && trim($link->text()));
    });

    $links = array_merge($links, $filteredHeaderLinks);

    foreach ($links as $link) {
        $span = $link->first('span');
        $div = $link->first('div');
        $spnimg = $link->first('img');

        if ($span && !$div && !$spnimg && !$link->hasAttribute('aria-label')) {
            $span_text = trim($span->text());
            if ($span_text) {
                $link->setAttribute('aria-label', $span_text);
            }
        }

        $spanimg = $link->first('span img');
        if ($spanimg && !$link->hasAttribute('aria-label') && (!$spanimg->hasAttribute('alt') || !$spanimg->hasAttribute('title'))) {
            $link->setAttribute('aria-label', 'image');
        }

        if (!trim($link->text()) && !$link->hasAttribute('aria-label')) {
            $child_elements = $link->find('*');
            if (empty($child_elements)) {
                $link->setAttribute('aria-label', 'link');
            } else {
                $icon = $link->first('i');
                if ($icon) {
                    $link->setAttribute('aria-label', 'icon');
                } else {
                    $img = $link->first('img');
                    if ($img && !$img->hasAttribute('alt')) {
                        $link->setAttribute('aria-label', 'image');
                    } else if ($img && (!$img->hasAttribute('alt') || !$img->hasAttribute('title'))) {
                        $link->setAttribute('aria-label', 'image');
                    }
                    $svg = $link->first('svg');
                    if ($svg) {
                        $link->setAttribute('aria-label', 'svg');
                    }
                    if (!$svg && !$img) {
                        $link->setAttribute('aria-label', 'link');
                    }
                }
            }
        }
    }

    return $dom->html();
}


//Buttons do not have an accessible name

function easywpstuff_assesplus_add_aria_labels_to_buttons($html) {
  $dom = new \DiDom\Document($html);
  $buttons = $dom->find('button');

  foreach ($buttons as $button) {
    if (!trim($button->text()) && !$button->hasAttribute('aria-label')) {
      $child_elements = $button->find('*');
      if (empty($child_elements)) {
        $button->setAttribute('aria-label', 'button');
      } else {
        $button->setAttribute('aria-label', 'button');
      }
    }
    if (!$button->hasAttribute('aria-label')) {
      $svg = $button->first('svg');
      $path = $button->first('path');
      $span = $button->first('span');

      if ($svg && $path) {
        $button->setAttribute('aria-label', 'button');
      } elseif ($span) {
        $span_text = trim($span->text());
        if ($span_text) {
          $button->setAttribute('aria-label', $span_text);
        }
      }
    }
  }

  return $dom->html();
}

//[user-scalable="no"] is used in the <meta name="viewport"> element or the [maximum-scale] attribute is less than 5.
function easywpstuff_assesplus_modify_viewport_meta_tag($html) {
  $dom = new \DiDom\Document($html);
  $viewport_meta_tag = $dom->first('meta[name=viewport]');
  
  if ($viewport_meta_tag) {
    $content = $viewport_meta_tag->getAttribute('content');
    
    $content = preg_replace('/,\s*user-scalable=(?:no|0)/', '', $content);

    if (preg_match('/maximum-scale=([^,\s]*)/', $content, $matches)) {
      $maximum_scale = (float) $matches[1];
      if ($maximum_scale < 2) {
        $content = preg_replace('/maximum-scale=[^,\s]*/', 'maximum-scale=2', $content);
      }
    } else {
      $content .= ', maximum-scale=2';
    }

    $viewport_meta_tag->setAttribute('content', $content);
  }

  return $dom->html();
}

function easywpstuff_assesplus_add_aria_labels_to_headers($html) {
    $dom = new \DiDom\Document($html);
    foreach ($dom->find('header div') as $div) {
        $role = $div->getAttribute('role');
        if ($role === 'link' && !$div->hasAttribute('aria-label')) {
            $div->setAttribute('aria-label', 'link');
        }
        if ($role === 'button' && !$div->hasAttribute('aria-label')) {
            $div->setAttribute('aria-label', 'button');
        }
        if ($role === 'menuitem' && !$div->hasAttribute('aria-label')) {
            $div->setAttribute('aria-label', 'menuitem');
        }
    }

    return $dom->html();
}

// <frame> or <iframe> elements do not have a title
function easywpstuff_assesplus_booster_add_to_iframes($html) {
    $dom = new \DiDom\Document($html);
    foreach ($dom->find('iframe') as $iframe) {
        if (!$iframe->hasAttribute('title')) {
            $iframe->setAttribute('title', 'iframe');
        }
    }

    return $dom->html();
}

//Links are not crawlable
function easywpstuff_assesplus_addHashToBlankLinks($html) {
    $dom = new \DiDom\Document($html);
    $blankLinks = $dom->find('a');
    foreach ($blankLinks as $link) {
        $href = $link->getAttribute('href');
        if (!$href || $href === '') {
            $link->setAttribute('href', '#');
        }
        if ($href === 'javascript: void(0);' || $href === 'javascript:void(0);') {
            $link->setAttribute('href', '#');
        }
    }

    return $dom->html();
}

//ARIA progressbar elements do not have accessible names.
function easywpstuff_assesplus_add_aria_label_to_progressbar($html) {
    $dom = new \DiDom\Document($html);
    foreach ($dom->find('[role=progressbar]') as $element) {
        $has_label = $element->hasAttribute('aria-label') || 
                     $element->hasAttribute('aria-labelledby') || 
                     $element->hasAttribute('title');

        if (!$has_label || (empty($element->getAttribute('aria-label')) && empty($element->getAttribute('aria-labelledby')) && empty($element->getAttribute('title')))) {
            $element->setAttribute('aria-label', 'progressbar');
        }
    }

    return $dom->html();
}
//Image elements do not have [alt] attributes
function easywpstuff_assesplus_addAltToImg($html) {
    $dom = new \DiDom\Document($html);
    $imgs = $dom->find('img');
    foreach ($imgs as $img) {
        if (!$img->getAttribute('alt')) {
            $alt = '';
            if ($img->hasAttribute('title')) {
                $alt = $img->getAttribute('title');
            } else {
                $src = $img->getAttribute('src');
                $alt = basename($src);
                $alt = preg_replace('/\d+/', '', $alt);
                $alt = preg_replace('/\.[^.]+$/', '', $alt);
                $alt = str_replace('-', ' ', $alt);
                $alt = str_replace('_', ' ', $alt);
            }
            $img->setAttribute('alt', $alt);
        }
    }
    return $dom->html();
}


// buffer if option is enabled
function easywpstuff_assesplus_bstr_template_redirect( $buffer ) {

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_no_labels'] == 1 ) {
		$buffer = easywpstuff_assesplus_check_input_labels( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_no_discernible_name'] == 1 ) {
	
		$buffer = easywpstuff_assesplus_add_aria_labels_to_links( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_no_accessible_name'] == 1 ) {
	
		$buffer = easywpstuff_assesplus_add_aria_labels_to_buttons( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_viewport_meta_tag'] == 1 ) {
		$buffer = easywpstuff_assesplus_modify_viewport_meta_tag( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_menuitem_accessible'] == 1 ) {
		$buffer = easywpstuff_assesplus_add_aria_labels_to_headers( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_booster_field_iframe_title_tag'] == 1 ) {
		$buffer = easywpstuff_assesplus_booster_add_to_iframes( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_add_aria_label_to_progressbar'] == 1 ) {
		$buffer = easywpstuff_assesplus_add_aria_label_to_progressbar( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_links_not_crawlable'] == 1 ) {
		$buffer = easywpstuff_assesplus_addHashToBlankLinks( $buffer );
	}

	if ( get_option( 'easywpstuff_assesplus_booster_options' )['easywpstuff_assesplus_img_donot_alt'] == 1 ) {
		$buffer = easywpstuff_assesplus_addAltToImg( $buffer );
	}

	return $buffer;
}

//add_action( 'template_redirect', 'easywpstuff_assesplus_bstr_template_redirect', 99 );

// run the function
add_action( 'template_redirect', 'easywpstuff_assesplus_loader', 99 );
function easywpstuff_assesplus_loader() {

     ob_start( 'easywpstuff_assesplus_bstr_template_redirect');
	
}

// added settings link
function easywpstuff_assesplus_bs_settings_link($links) {
  $settings_link = '<a href="' . admin_url( 'options-general.php?page=accessibilityplus' ) . '">' . __( 'Settings' ) . '</a>';
  array_unshift( $links, $settings_link );
  return $links;
}
add_filter( 'plugin_action_links_' . plugin_basename( __FILE__ ), 'easywpstuff_assesplus_bs_settings_link' );

// enqueue admin styles on option page
function easywpstuff_assesplus_admin_enqueue_plugin_styles( $hook ) {
  if ( 'settings_page_accessibilityplus' === $hook ) {
    wp_enqueue_style( 'lhbooster-styles', plugin_dir_url( __FILE__ ) . 'assets/style.css' );
  }
}
add_action( 'admin_enqueue_scripts', 'easywpstuff_assesplus_admin_enqueue_plugin_styles' );