lbrypress/classes/LBRY_Speech.php
2022-02-08 17:41:59 -06:00

256 lines
9 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
*/
defined('ABSPATH') || die(); // Exit if accessed directly
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);
}
// 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_SPEECH_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
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
$speech_channel = get_option(LBRY_SPEECH_SETTINGS)[LBRY_SPEECH_CHANNEL];
$speech_pw = LBRY()->admin->get_speech_pw();
if (!empty($speech_channel) && !empty($speech_pw)) {
$params['channelName'] = '@' . $speech_channel;
$params['channelPassword'] = $speech_pw;
}
$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);
$meta[LBRY_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) {
error_log('Failed to upload asset with ID ' . $media->id . ' to supplied speech URL.');
LBRY()->daemon->logger->log('Speech Upload', 'Failed to upload asset with ID ' . $media->id . ' to supplied speech URL. Message | ' . $e->getMessage());
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();
$content = get_post_field('post_content', $post_id);
$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(LBRY_SPEECH_ASSET_URL, $meta) && $meta[LBRY_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_SPEECH_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;
}
}