| 1 |
<?php |
| 2 |
|
| 3 |
namespace FL\Assistant\Helpers; |
| 4 |
|
| 5 |
class ScreenshotHelper { |
| 6 |
|
| 7 |
/** |
| 8 |
* @param object $request |
| 9 |
* @param string $url Fallback url. |
| 10 |
* @return array |
| 11 |
*/ |
| 12 |
static public function get_for_rest_request( $request, $url, $logged_in = true ) { |
| 13 |
$screenshot = $request->get_param( 'screenshot' ); |
| 14 |
|
| 15 |
if ( $screenshot ) { |
| 16 |
return [ |
| 17 |
'type' => 'base64', |
| 18 |
'data' => $screenshot, |
| 19 |
]; |
| 20 |
} |
| 21 |
|
| 22 |
return self::get_for_url( $url, $logged_in ); |
| 23 |
} |
| 24 |
|
| 25 |
/** |
| 26 |
* @param string $url Fallback url. |
| 27 |
* @return array |
| 28 |
*/ |
| 29 |
static public function get_for_post_request( $url, $logged_in = true ) { |
| 30 |
$screenshot = isset( $_POST['screenshot'] ) ? $_POST['screenshot'] : null; |
| 31 |
|
| 32 |
if ( $screenshot ) { |
| 33 |
return [ |
| 34 |
'type' => 'base64', |
| 35 |
'data' => $screenshot, |
| 36 |
]; |
| 37 |
} |
| 38 |
|
| 39 |
return self::get_for_url( $url, $logged_in ); |
| 40 |
} |
| 41 |
|
| 42 |
/** |
| 43 |
* @param string $url |
| 44 |
* @return array |
| 45 |
*/ |
| 46 |
static public function get_for_url( $url, $logged_in = true ) { |
| 47 |
$response = wp_safe_remote_get( |
| 48 |
$url, [ |
| 49 |
'cookies' => $logged_in ? $_COOKIE : [], |
| 50 |
] |
| 51 |
); |
| 52 |
|
| 53 |
return [ |
| 54 |
'type' => 'html', |
| 55 |
'html' => wp_remote_retrieve_body( $response ), |
| 56 |
]; |
| 57 |
} |
| 58 |
} |
| 59 |
|
