Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

namespace StringLocator\REST;

[2] Fix | Delete

[3] Fix | Delete

use StringLocator\Base\REST;

[4] Fix | Delete

use StringLocator\Directory_Iterator;

[5] Fix | Delete

use StringLocator\String_Locator;

[6] Fix | Delete

[7] Fix | Delete

class Directory_Structure extends REST {

[8] Fix | Delete

[9] Fix | Delete

protected $rest_base = 'get-directory-structure';

[10] Fix | Delete

[11] Fix | Delete

public function __construct() {

[12] Fix | Delete

parent::__construct();

[13] Fix | Delete

}

[14] Fix | Delete

[15] Fix | Delete

public function register_rest_route() {

[16] Fix | Delete

register_rest_route(

[17] Fix | Delete

$this->namespace,

[18] Fix | Delete

$this->rest_base,

[19] Fix | Delete

array(

[20] Fix | Delete

'methods' => 'POST',

[21] Fix | Delete

'callback' => array( $this, 'get_structure' ),

[22] Fix | Delete

'permission_callback' => array( $this, 'permission_callback' ),

[23] Fix | Delete

)

[24] Fix | Delete

);

[25] Fix | Delete

}

[26] Fix | Delete

[27] Fix | Delete

public function permission_callback() {

[28] Fix | Delete

return current_user_can( String_Locator::$search_capability );

[29] Fix | Delete

}

[30] Fix | Delete

[31] Fix | Delete

public function get_structure( \WP_REST_Request $request ) {

[32] Fix | Delete

$short_circuit = apply_filters( 'string_locator_directory_iterator_short_circuit', array(), $request );

[33] Fix | Delete

[34] Fix | Delete

if ( ! empty( $short_circuit ) ) {

[35] Fix | Delete

return $short_circuit;

[36] Fix | Delete

}

[37] Fix | Delete

[38] Fix | Delete

$data = json_decode( $request->get_param( 'data' ) );

[39] Fix | Delete

[40] Fix | Delete

$directory = $data->directory;

[41] Fix | Delete

$search = $data->search;

[42] Fix | Delete

$regex = $data->regex;

[43] Fix | Delete

[44] Fix | Delete

// Validate the search path to avoid unintended directory traversal.

[45] Fix | Delete

if ( 0 !== validate_file( $directory ) ) {

[46] Fix | Delete

return new \WP_REST_Response(

[47] Fix | Delete

array(

[48] Fix | Delete

'success' => false,

[49] Fix | Delete

'data' => __( 'Invalid search source provided.', 'string-locator' ),

[50] Fix | Delete

[51] Fix | Delete

400

[52] Fix | Delete

);

[53] Fix | Delete

}

[54] Fix | Delete

[55] Fix | Delete

$iterator = new Directory_Iterator(

[56] Fix | Delete

$directory,

[57] Fix | Delete

$search,

[58] Fix | Delete

$regex

[59] Fix | Delete

);

[60] Fix | Delete

[61] Fix | Delete

return array(

[62] Fix | Delete

'success' => true,

[63] Fix | Delete

'data' => $iterator->get_structure(),

[64] Fix | Delete

);

[65] Fix | Delete

}

[66] Fix | Delete

[67] Fix | Delete

}

[68] Fix | Delete

[69] Fix | Delete

new Directory_Structure();

[70] Fix | Delete

[71] Fix | Delete