| @@ -5,12 +5,12 @@ | ||
| 5 | 5 | class ScreenshotHelper { |
| 6 | 6 | |
| 7 | 7 | /** |
| 8 | 8 | * @param object $request |
| 9 | - * @param string $url | |
| 9 | + * @param string $url Fallback url. | |
| 10 | 10 | * @return array |
| 11 | 11 | */ |
| 12 | - static public function get_for_request( $request, $url, $logged_in = true ) { | |
| 12 | + static public function get_for_rest_request( $request, $url, $logged_in = true ) { | |
| 13 | 13 | $screenshot = $request->get_param( 'screenshot' ); |
| 14 | 14 | |
| 15 | 15 | if ( $screenshot ) { |
| 16 | 16 | return [ |
| @@ -18,9 +18,34 @@ | ||
| 18 | 18 | 'data' => $screenshot, |
| 19 | 19 | ]; |
| 20 | 20 | } |
| 21 | 21 | |
| 22 | - $response = wp_remote_get( | |
| 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( | |
| 23 | 48 | $url, [ |
| 24 | 49 | 'cookies' => $logged_in ? $_COOKIE : [], |
| 25 | 50 | ] |
| 26 | 51 | ); |