lbrypress/classes/LBRY_Speech.php

261 lines
9.2 KiB
PHP

<?php
/**
* Connects to an spee.ch style server to host assets via the LBRY Protocol
*
* Visit https://github.com/lbryio/spee.ch for more info
*
* @package LBRYPress
*/
class LBRY_Speech
{
/**
* HTML Parser
* @var LBRY_Speech_Parser
*/
private $parser = null;
public function __construct()
{
$this->parser = new LBRY_Speech_Parser();
if (is_admin()) {
add_action('save_post', array($this, 'upload_media'), 10, 2);
} else {
// Replace the image srcsets
add_filter('wp_calculate_image_srcset', array($this->parser, 'replace_image_srcset'), 10, 5);
// Core filter for lots of image source calls
add_filter('wp_get_attachment_image_src', array($this->parser, 'replace_attachment_image_src'), 10, 3);
// Replace any left over urls with speech urls
add_filter('the_content', array($this->parser, 'replace_urls_with_speech'));
}
}
/**
* Uploads assets to the speech server
* @param int $post_id The ID of the post to
* @return bool True if successful, false if not or if no Speech URL available
*/
public function upload_media($post_id, $post)
{
// Only check post_type of Post
if ('post' !== $post->post_type) {
return false;
}
$speech_url = get_option(LBRY_SETTINGS)[LBRY_SPEECH];
// Die if we don't have a spee.ch url
if (!$speech_url || $speech_url === '') {
return false;
}
$all_media = $this->find_media($post_id);
// IDEA: Notify user if post save time will take a while, may be a concern for request timeouts
if ($all_media) {
$requests = array();
// Build all the Curl Requests
foreach ($all_media as $media) {
$params = array(
'name' => $media->name,
'file' => $media->file,
'title' => $media->title,
'type' => $media->type
);
// Pull Channel and Password from config file for now
// COMBAK: This will change in the future
if (LBRY_SPEECH_CHANNEL && LBRY_SPEECH_CHANNEL_PASSWORD) {
$params['channelName'] = LBRY_SPEECH_CHANNEL;
$params['channelPassword'] = LBRY_SPEECH_CHANNEL_PASSWORD;
}
$ch = $this->build_request('publish', $params);
$requests[] = array(
'request' => $ch,
'media' => $media
);
}
// Init the curl multi handle
$mh = curl_multi_init();
// Add each request to the multi handle
foreach ($requests as $request) {
curl_multi_add_handle($mh, $request['request']);
}
// Execute all requests simultaneously
$running = null;
do {
curl_multi_exec($mh, $running);
} while ($running);
// Close the handles
foreach ($requests as $request) {
curl_multi_remove_handle($mh, $request['request']);
}
curl_multi_close($mh);
// Run through responses, and upload meta as necessary
foreach ($requests as $request) {
$result = json_decode(curl_multi_getcontent($request['request']));
$media = $request['media'];
$response_code = curl_getinfo($request['request'], CURLINFO_RESPONSE_CODE);
try {
// check we got a success code
if ($response_code != '200') {
if (!empty($result) && !$result->success && $result->message) {
throw new \Exception("API Issue with message: " . $result->message);
} else {
throw new \Exception("Speech URL Connection Issue | Code: " . $response_code, 1);
}
}
// Update image meta
if ($result && $result->success) {
$meta = wp_get_attachment_metadata($media->id);
if ($media->image_size) {
$meta['sizes'][$media->image_size]['speech_asset_url'] = $result->data->serveUrl;
} else {
$meta['speech_asset_url'] = $result->data->serveUrl;
}
wp_update_attachment_metadata($media->id, $meta);
} else { // Something unhandled happened here
throw new \Exception("Unknown Speech Upload issue for asset");
}
} catch (\Exception $e) {
$image_size = $media->image_size ? $media->image_size : 'full';
error_log('Failed to upload asset with ID ' . $media->id . ' for size ' . $image_size . ' to supplied speech URL.');
error_log($e->getMessage());
}
}
}
}
/**
* Finds all media attached to a post
* @param int $post_id The post to search
* @return array An array of Speech Media Objects
*/
private function find_media($post_id)
{
$all_media = array();
// Get content and put into a DOMDocument
$content = get_post_field('post_content', $post_id);
if (!$content) {
return $all_media;
}
$images = $this->parser->scrape_images($content);
// Get all the image ID's
$image_ids = array();
foreach ($images as $image) {
$new_id = $this->parser->get_attachment_id_from_tag('image', $image);
if ($new_id) {
$image_ids[] = $new_id;
}
}
// Don't forget the featured image
if ($featured_id = get_post_thumbnail_id($post_id)) {
$image_ids = array_merge($image_ids, array($featured_id));
}
// Throw each image into a media object
foreach ($image_ids as $attachment_id) {
// Create main image media object
$meta = wp_get_attachment_metadata($attachment_id);
// If we don't have meta, get out because none of this will work
if (!$meta) {
break;
}
if (!$this->is_published($meta)) {
$all_media[] = new LBRY_Speech_Media($attachment_id);
}
// COMBAK: find a way to make this more efficient?
// Create a media object for each image size
// Get images sizes for this attachment, as not all image sizes implemented
$image_sizes = wp_get_attachment_metadata($attachment_id)['sizes'];
foreach ($image_sizes as $size => $meta) {
if (!$this->is_published($meta)) {
$all_media[] = new LBRY_Speech_Media($attachment_id, array('image_size' => $size));
}
}
}
$videos = $this->parser->scrape_videos($content);
$video_ids = array();
foreach ($videos as $video) {
$new_id = $this->parser->get_attachment_id_from_tag('mp4', $video);
if ($new_id) {
$video_ids[] = $new_id;
}
}
// Parse video tags based on wordpress shortcode for local embedds
foreach ($video_ids as $attachment_id) {
$meta = wp_get_attachment_metadata($attachment_id);
if (!$this->is_published($meta)) {
$all_media[] = new LBRY_Speech_Media($attachment_id);
}
}
return $all_media;
}
/**
* Checks meta to see if a spee.ch url exists
* @param array $meta An array of meta which would possibly contain a speech_asset_url
* @return boolean Whether or not its published to speech
*/
public function is_published($meta)
{
if (key_exists('speech_asset_url', $meta) && $meta['speech_asset_url'] !== '') {
return true;
}
return false;
}
/**
* Builds a cURL request to the Speech URL
* @param string $method The method to call on the Speech API
* @param array $params The Parameters to send the Speech API Call
* @return string The cURL object pointer
*/
private function build_request($method, $params = array())
{
$speech_url = get_option(LBRY_SETTINGS)[LBRY_SPEECH];
// Die if no URL
if (!$speech_url) {
return false;
}
$address = $speech_url . '/api/claim/' . $method;
// Send it via curl
$ch = curl_init();
curl_setopt($ch, CURLOPT_URL, $address);
curl_setopt($ch, CURLOPT_POST, true);
curl_setopt($ch, CURLOPT_POSTFIELDS, $params);
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
curl_setopt($ch, CURLOPT_AUTOREFERER, false);
curl_setopt($ch, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
curl_setopt($ch, CURLOPT_SAFE_UPLOAD, true);
curl_setopt($ch, CURLOPT_HEADER, false);
return $ch;
}
}