Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

namespace StringLocator;

[2] Fix | Delete

[3] Fix | Delete

class Directory_Iterator {

[4] Fix | Delete

[5] Fix | Delete

/**

[6] Fix | Delete

* Directory that will be searched.

[7] Fix | Delete

[8] Fix | Delete

* @var

[9] Fix | Delete

[10] Fix | Delete

private $directory;

[11] Fix | Delete

[12] Fix | Delete

/**

[13] Fix | Delete

* The search term being used.

[14] Fix | Delete

[15] Fix | Delete

* @var

[16] Fix | Delete

[17] Fix | Delete

private $search;

[18] Fix | Delete

[19] Fix | Delete

/**

[20] Fix | Delete

* A check if regex searches are enabled or not.

[21] Fix | Delete

[22] Fix | Delete

* @var

[23] Fix | Delete

[24] Fix | Delete

private $regex;

[25] Fix | Delete

[26] Fix | Delete

/**

[27] Fix | Delete

* DirectoryIterator constructor.

[28] Fix | Delete

[29] Fix | Delete

* @param $directory

[30] Fix | Delete

* @param $search

[31] Fix | Delete

* @param $regex

[32] Fix | Delete

[33] Fix | Delete

* @return array

[34] Fix | Delete

[35] Fix | Delete

public function __construct( $directory, $search, $regex ) {

[36] Fix | Delete

$this->directory = $directory;

[37] Fix | Delete

$this->search = $search;

[38] Fix | Delete

$this->regex = $regex;

[39] Fix | Delete

}

[40] Fix | Delete

[41] Fix | Delete

/**

[42] Fix | Delete

* Build the folder structure.

[43] Fix | Delete

[44] Fix | Delete

* @return array

[45] Fix | Delete

[46] Fix | Delete

public function get_structure() {

[47] Fix | Delete

$scan_path = $this->prepare_scan_path( $this->directory );

[48] Fix | Delete

[49] Fix | Delete

if ( is_file( $scan_path->path ) ) {

[50] Fix | Delete

$files = array( $scan_path->path );

[51] Fix | Delete

} else {

[52] Fix | Delete

$files = $this->ajax_scan_path( $scan_path->path );

[53] Fix | Delete

}

[54] Fix | Delete

[55] Fix | Delete

[56] Fix | Delete

* Make sure each chunk of file arrays never exceeds 500 files

[57] Fix | Delete

* This is to prevent the SQL string from being too large and crashing everything

[58] Fix | Delete

[59] Fix | Delete

$back_compat_filter = apply_filters( 'string-locator-files-per-array', 500 ); //phpcs:ignore WordPress.NamingConventions.ValidHookName.UseUnderscores

[60] Fix | Delete

[61] Fix | Delete

$file_chunks = array_chunk( $files, apply_filters( 'string_locator_files_per_array', $back_compat_filter ), true );

[62] Fix | Delete

[63] Fix | Delete

$store = (object) array(

[64] Fix | Delete

'scan_path' => $scan_path,

[65] Fix | Delete

'search' => $this->search,

[66] Fix | Delete

'directory' => $this->directory,

[67] Fix | Delete

'chunks' => count( $file_chunks ),

[68] Fix | Delete

'regex' => $this->regex,

[69] Fix | Delete

);

[70] Fix | Delete

[71] Fix | Delete

$response = array(

[72] Fix | Delete

'total' => count( $files ),

[73] Fix | Delete

'current' => 0,

[74] Fix | Delete

'directory' => $scan_path,

[75] Fix | Delete

'chunks' => count( $file_chunks ),

[76] Fix | Delete

'regex' => $this->regex,

[77] Fix | Delete

);

[78] Fix | Delete

[79] Fix | Delete

set_transient( 'string-locator-search-overview', $store );

[80] Fix | Delete

update_option( 'string-locator-search-history', array(), false );

[81] Fix | Delete

[82] Fix | Delete

foreach ( $file_chunks as $count => $file_chunk ) {

[83] Fix | Delete

set_transient( 'string-locator-search-files-' . $count, $file_chunk );

[84] Fix | Delete

}

[85] Fix | Delete

[86] Fix | Delete

return $response;

[87] Fix | Delete

}

[88] Fix | Delete

[89] Fix | Delete

/**

[90] Fix | Delete

* Parse the search option to determine what kind of search we are performing and what directory to start in.

[91] Fix | Delete

[92] Fix | Delete

* @param string $option The search-type identifier.

[93] Fix | Delete

[94] Fix | Delete

* @return bool|object

[95] Fix | Delete

[96] Fix | Delete

private function prepare_scan_path( $option ) {

[97] Fix | Delete

$data = array(

[98] Fix | Delete

'path' => '',

[99] Fix | Delete

'type' => '',

[100] Fix | Delete

'slug' => '',

[101] Fix | Delete

);

[102] Fix | Delete

[103] Fix | Delete

switch ( true ) {

[104] Fix | Delete

case ( 't--' === $option ):

[105] Fix | Delete

$data['path'] = WP_CONTENT_DIR . '/themes/';

[106] Fix | Delete

$data['type'] = 'theme';

[107] Fix | Delete

break;

[108] Fix | Delete

case ( strlen( $option ) > 3 && 't-' === substr( $option, 0, 2 ) ):

[109] Fix | Delete

$data['path'] = WP_CONTENT_DIR . '/themes/' . substr( $option, 2 );

[110] Fix | Delete

$data['type'] = 'theme';

[111] Fix | Delete

$data['slug'] = substr( $option, 2 );

[112] Fix | Delete

break;

[113] Fix | Delete

case ( 'p--' === $option ):

[114] Fix | Delete

$data['path'] = WP_CONTENT_DIR . '/plugins/';

[115] Fix | Delete

$data['type'] = 'plugin';

[116] Fix | Delete

break;

[117] Fix | Delete

case ( 'mup--' === $option ):

[118] Fix | Delete

$data['path'] = WP_CONTENT_DIR . '/mu-plugins/';

[119] Fix | Delete

$data['type'] = 'mu-plugin';

[120] Fix | Delete

break;

[121] Fix | Delete

case ( strlen( $option ) > 3 && 'p-' === substr( $option, 0, 2 ) ):

[122] Fix | Delete

$slug = explode( '/', substr( $option, 2 ) );

[123] Fix | Delete

[124] Fix | Delete

$data['path'] = WP_CONTENT_DIR . '/plugins/' . $slug[0];

[125] Fix | Delete

$data['type'] = 'plugin';

[126] Fix | Delete

$data['slug'] = $slug[0];

[127] Fix | Delete

break;

[128] Fix | Delete

case ( 'core' === $option ):

[129] Fix | Delete

$data['path'] = ABSPATH;

[130] Fix | Delete

$data['type'] = 'core';

[131] Fix | Delete

break;

[132] Fix | Delete

case ( 'wp-content' === $option ):

[133] Fix | Delete

$data['path'] = WP_CONTENT_DIR;

[134] Fix | Delete

$data['type'] = 'core';

[135] Fix | Delete

break;

[136] Fix | Delete

}

[137] Fix | Delete

[138] Fix | Delete

if ( empty( $data['path'] ) ) {

[139] Fix | Delete

return false;

[140] Fix | Delete

}

[141] Fix | Delete

[142] Fix | Delete

return (object) $data;

[143] Fix | Delete

}

[144] Fix | Delete

[145] Fix | Delete

private function ajax_scan_path( $path ) {

[146] Fix | Delete

$files = array();

[147] Fix | Delete

[148] Fix | Delete

$paths = new \RecursiveIteratorIterator(

[149] Fix | Delete

new \RecursiveDirectoryIterator( $path ),

[150] Fix | Delete

\RecursiveIteratorIterator::SELF_FIRST

[151] Fix | Delete

);

[152] Fix | Delete

[153] Fix | Delete

foreach ( $paths as $name => $location ) {

[154] Fix | Delete

if ( is_dir( $location->getPathname() ) ) {

[155] Fix | Delete

continue;

[156] Fix | Delete

}

[157] Fix | Delete

[158] Fix | Delete

$files[] = $location->getPathname();

[159] Fix | Delete

}

[160] Fix | Delete

[161] Fix | Delete

return $files;

[162] Fix | Delete

}

[163] Fix | Delete

[164] Fix | Delete

}

[165] Fix | Delete

[166] Fix | Delete