/** * Check if a file (or stream) is a valid image type * * @param string $file Can be a temporary filename (think $_FILE before saving to $filename) or a stream (string). Default is temporary filename * @param string $filename * @param bool $is_stream (optional) * * @return bool */ function is_valid_image($file, $filename, $is_stream = false) { $format = explode('.', $filename); $format = strtolower(end($format)); // Extension if (!('jpg' == $format || 'jpeg' == $format || 'gif' == $format || 'png' == $format)) { return false; } if ($is_stream) { $tmp_image_path = \Pressbooks\Utility\create_tmp_file(); file_put_contents($tmp_image_path, $file); $file = $tmp_image_path; } $type = @exif_imagetype($file); if (IMAGETYPE_JPEG == $type || IMAGETYPE_GIF == $type || IMAGETYPE_PNG == $type) { return true; } else { return false; } }
/** * Fetch a font with wp_remote_get(), save it to $fullpath with a unique name. * Will return an empty string if something went wrong. * * @param $url string * @param $fullpath string * * @return string filename */ protected function fetchAndSaveUniqueFont($url, $fullpath) { if (isset($this->fetchedFontCache[$url])) { return $this->fetchedFontCache[$url]; } $response = wp_remote_get($url, array('timeout' => $this->timeout)); // WordPress error? if (is_wp_error($response)) { // TODO: handle $response->get_error_message(); $this->fetchedFontCache[$url] = ''; return ''; } // Basename without query string $filename = explode('?', basename($url)); $filename = array_shift($filename); $filename = sanitize_file_name(urldecode($filename)); $filename = Sanitize\force_ascii($filename); $tmp_file = \Pressbooks\Utility\create_tmp_file(); file_put_contents($tmp_file, wp_remote_retrieve_body($response)); // TODO: Validate that this is actually a font // TODO: Refactor fetchAndSaveUniqueImage() and fetchAndSaveUniqueFont() into a single method, but "inject" different validation // Check for duplicates, save accordingly if (!file_exists("{$fullpath}/{$filename}")) { copy($tmp_file, "{$fullpath}/{$filename}"); } elseif (md5(file_get_contents($tmp_file)) != md5(file_get_contents("{$fullpath}/{$filename}"))) { $filename = wp_unique_filename($fullpath, $filename); copy($tmp_file, "{$fullpath}/{$filename}"); } $this->fetchedFontCache[$url] = $filename; return $filename; }
/** * Fetch an image with wp_remote_get(), save it to $fullpath with a unique name. * Will return an empty string if something went wrong. * * @param $url string * @param $fullpath string * * @return string filename */ protected function fetchAndSaveUniqueImage($url, $fullpath) { // Cheap cache static $already_done = array(); if (isset($already_done[$url])) { return $already_done[$url]; } $response = wp_remote_get($url, array('timeout' => $this->timeout)); // WordPress error? if (is_wp_error($response)) { // TODO: handle $response->get_error_message(); $already_done[$url] = ''; return ''; } // Basename without query string $filename = explode('?', basename($url)); // isolate latex image service from WP, add file extension if ('s.wordpress.com' == parse_url($url, PHP_URL_HOST) && 'latex.php' == $filename[0]) { $filename = md5(array_pop($filename)); // content-type = 'image/png' $type = explode('/', $response['headers']['content-type']); $type = array_pop($type); $filename = $filename . '.' . $type; } else { $filename = array_shift($filename); $filename = sanitize_file_name(urldecode($filename)); $filename = \Pressbooks\Sanitize\force_ascii($filename); } $tmp_file = \Pressbooks\Utility\create_tmp_file(); file_put_contents($tmp_file, wp_remote_retrieve_body($response)); if (!\Pressbooks\Image\is_valid_image($tmp_file, $filename)) { $already_done[$url] = ''; return ''; // Not an image } if ($this->compressImages) { $format = explode('.', $filename); $format = strtolower(end($format)); // Extension \Pressbooks\Image\resize_down($format, $tmp_file); } // Check for duplicates, save accordingly if (!file_exists("{$fullpath}/{$filename}")) { copy($tmp_file, "{$fullpath}/{$filename}"); } elseif (md5(file_get_contents($tmp_file)) != md5(file_get_contents("{$fullpath}/{$filename}"))) { $filename = wp_unique_filename($fullpath, $filename); copy($tmp_file, "{$fullpath}/{$filename}"); } $already_done[$url] = $filename; return $filename; }
function convert() { $this->outputPath = \Pressbooks\Utility\create_tmp_file(); return true; }
/** * Create a temporary file that automatically gets deleted on __sleep() * * @return string fullpath */ function createTmpFile() { return \Pressbooks\Utility\create_tmp_file(); }
/** * @covers \Pressbooks\Utility\create_tmp_file */ public function test_create_tmp_file() { $file = \Pressbooks\Utility\create_tmp_file(); $this->assertFileExists($file); file_put_contents($file, 'Hello world!'); $this->assertEquals('Hello world!', file_get_contents($file)); }
/** * Fetch a url with wp_remote_get(), save it to $fullpath with a unique name. * Will return an empty string if something went wrong. * * @param string $url * @param string $fullpath * @return string|array */ protected function fetchAndSaveUniqueMedia($url, $fullpath) { if (isset($this->fetchedMediaCache[$url])) { return $this->fetchedMediaCache[$url]; } $response = wp_remote_get($url, array('timeout' => $this->timeout)); // WordPress error? if (is_wp_error($response)) { try { // protocol relative urls handed to wp_remote_get will fail // try adding a protocol $protocol_relative = wp_parse_url($url); if (!isset($protocol_relative['scheme'])) { if (true === is_ssl()) { $url = 'https:' . $url; } else { $url = 'http:' . $url; } } $response = wp_remote_get($url, array('timeout' => $this->timeout)); if (is_wp_error($response)) { throw new \Exception('Bad URL: ' . $url); } } catch (\Exception $exc) { $this->fetchedImageCache[$url] = ''; error_log('\\PressBooks\\Export\\Epub3\\fetchAndSaveUniqueMedia wp_error on wp_remote_get() - ' . $response->get_error_message() . ' - ' . $exc->getMessage()); return ''; } } // Basename without query string $filename = explode('?', basename($url)); $filename = array_shift($filename); $filename = sanitize_file_name(urldecode($filename)); $filename = Sanitize\force_ascii($filename); $tmp_file = \Pressbooks\Utility\create_tmp_file(); file_put_contents($tmp_file, wp_remote_retrieve_body($response)); if (!\Pressbooks\Media\is_valid_media($tmp_file, $filename)) { $this->fetchedMediaCache[$url] = ''; return ''; // Not a valid media type } // Check for duplicates, save accordingly if (!file_exists("{$fullpath}/{$filename}")) { copy($tmp_file, "{$fullpath}/{$filename}"); } elseif (md5(file_get_contents($tmp_file)) != md5(file_get_contents("{$fullpath}/{$filename}"))) { $filename = wp_unique_filename($fullpath, $filename); copy($tmp_file, "{$fullpath}/{$filename}"); } $this->fetchedMediaCache[$url] = $filename; return $filename; }