22

✏️ Editing: url-shortener.php

<?php
/**
 *
 * @package WPUpper Share Buttons
 * @author  Victor Freitas
 * @subpackage URL Shortener
 * @version 3.27
 */
if ( ! defined( 'ABSPATH' ) ) {
	 // Exit if accessed directly.
	exit( 0 );
}

class WPUSB_URL_Shortener {

private $permalink;

private $token;

const TABLE_NAME = 'wpusb_url_shortener';
	const API = 'https://api-ssl.bitly.com/v3/shorten';

public function __construct( $permalink, $token ) {
		$this->_set_permalink( $permalink );
		$this->_set_token( $token );
	}

private function _set_permalink( $permalink ) {
		$this->permalink = $permalink;
	}

private function _set_token( $token ) {
		$this->token = $token;
	}

/**
	 * Get url short cache
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param String $hash
	 * @return String
	 */
	public static function get_cache( $hash ) {
		global $wpdb;

$table        = $wpdb->prefix . self::TABLE_NAME;
		$current_time = current_time( 'timestamp' );
		$query        = $wpdb->prepare(
			"SELECT
				`short_url`
			 FROM
			 	`{$table}`
			 USE INDEX(`hash`)
			 WHERE
			 	`hash` = %s
			 	AND `expires` > %d
			",
			$hash,
			$current_time
		);
		$value = $wpdb->get_var( $query );

if ( empty( $value ) ) {
			return '';
		}

return rawurlencode( esc_url( $value ) );
	}

/**
	 * Set url short cache
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param String $hash
	 * @param Integer $post_id
	 * @param String $short_url
	 * @param Integer $expiration
	 * @return Integer
	 */
	public static function set_cache( $hash, $post_id, $short_url, $expiration = 0 ) {
		global $wpdb;

$expiration_time = ( current_time( 'timestamp' ) + intval( $expiration ) );

if ( self::_is_exists( $hash ) ) {
			return self::_update( $short_url, $expiration_time, $hash );
		}

return self::_insert( $post_id, $hash, $short_url, $expiration_time );
	}

/**
	 * Check hash exists
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param String $hash
	 * @return Integer
	 */
	private static function _is_exists( $hash ) {
		global $wpdb;

$table = $wpdb->prefix . self::TABLE_NAME;
		$query = $wpdb->prepare(
			"SELECT
				`post_id`
			 FROM
			 	`{$table}`
			 	USE INDEX(`hash`)
			 WHERE
			 	`hash` = %s
			",
			$hash
		);

return (int) $wpdb->get_var( $query );
	}

/**
	 * Update records in the table
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param String $short_url
	 * @param Integer $expiration_time
	 * @param String $hash
	 * @return Integer
	 */
	private static function _update( $short_url, $expiration_time, $hash ) {
		global $wpdb;

return $wpdb->update(
			$wpdb->prefix . self::TABLE_NAME,
			array(
				'short_url' => $short_url,
				'expires'   => $expiration_time,
			),
			array(
				'hash' => $hash,
			),
			array( '%s', '%d' ),
			array( '%s' )
		);
	}

/**
	 * Insert records in the table
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param Integer $post_id
	 * @param String $hash
	 * @param String $short_url
	 * @param Integer $expiration_time
	 * @return Integer
	 */
	private static function _insert( $post_id, $hash, $short_url, $expiration_time ) {
		global $wpdb;

$table = $wpdb->prefix . self::TABLE_NAME;

$wpdb->delete(
			$table, array(
				'post_id' => $post_id,
			)
		);

return $wpdb->insert(
			$table,
			array(
				'post_id'   => $post_id,
				'hash'      => $hash,
				'short_url' => $short_url,
				'expires'   => $expiration_time,
			),
			array( '%d', '%s', '%s', '%d' )
		);
	}

/**
	 * Generate URL short with curl from request API
	 *
	 * @since 3.27
	 * @global $wpdb
	 * @param Null
	 * @return Mixed False|String
	 */
	public function get_short() {
		if ( ! WPUSB_Utils::has_curl() || ! WPUSB_Utils::has_json() ) {
			return false;
		}

$value  = WPUSB_Utils::option( 'bitly_domain' );
		$domain = WPUSB_Utils::get_bitly_domain( $value );
		$params = array(
			'access_token' => $this->token,
			'longUrl'      => $this->permalink,
		);

if ( $domain ) {
			$params['domain'] = $domain;
		}

$url = esc_url_raw( add_query_arg( $params, self::API ) );
		$ch  = curl_init();

curl_setopt( $ch, CURLOPT_CONNECTTIMEOUT, 5 );
		curl_setopt( $ch, CURLOPT_TIMEOUT, 5 );
		curl_setopt( $ch, CURLOPT_URL, $url );
		curl_setopt( $ch, CURLOPT_RETURNTRANSFER, true );
		curl_setopt( $ch, CURLOPT_HEADER, false );

$response = curl_exec( $ch );

curl_close( $ch );

$response = WPUSB_Utils::json_decode( $response );

if ( isset( $response->status_code ) && 200 === intval( $response->status_code ) ) {
			return $response->data->url;
		}

return false;
	}
}

MR SKB