Your IP : 18.118.2.240
<?php
/**
* The avatar cache class
*
* @since 3.0
* @package LiteSpeed
* @subpackage LiteSpeed/inc
* @author LiteSpeed Technologies <info@litespeedtech.com>
*/
namespace LiteSpeed;
defined( 'WPINC' ) || exit;
class Avatar extends Base {
const TYPE_GENERATE = 'generate';
private $_conf_cache_ttl;
private $_tb;
private $_avatar_realtime_gen_dict = array();
protected $_summary;
/**
* Init
*
* @since 1.4
*/
public function __construct() {
if ( ! $this->conf( self::O_DISCUSS_AVATAR_CACHE ) ) {
return;
}
Debug2::debug2( '[Avatar] init' );
$this->_tb = $this->cls( 'Data' )->tb( 'avatar' );
$this->_conf_cache_ttl = $this->conf( self::O_DISCUSS_AVATAR_CACHE_TTL );
add_filter( 'get_avatar_url', array( $this, 'crawl_avatar' ) );
$this->_summary = self::get_summary();
}
/**
* Check if need db table or not
*
* @since 3.0
* @access public
*/
public function need_db() {
if ( $this->conf( self::O_DISCUSS_AVATAR_CACHE ) ) {
return true;
}
return false;
}
/**
* Get gravatar URL from DB and regenarate
*
* @since 3.0
* @access public
*/
public function serve_static( $md5 ) {
global $wpdb;
Debug2::debug( '[Avatar] is avatar request' );
if ( strlen( $md5 ) !== 32 ) {
Debug2::debug( '[Avatar] wrong md5 ' . $md5 );
return;
}
$q = "SELECT url FROM `$this->_tb` WHERE md5=%s";
$url = $wpdb->get_var( $wpdb->prepare( $q, $md5 ) );
if ( ! $url ) {
Debug2::debug( '[Avatar] no matched url for md5 ' . $md5 );
return;
}
$url = $this->_generate( $url );
wp_redirect( $url );
exit;
}
/**
* Localize gravatar
*
* @since 3.0
* @access public
*/
public function crawl_avatar( $url ) {
if ( ! $url ) {
return $url;
}
// Check if its already in dict or not
if ( ! empty( $this->_avatar_realtime_gen_dict[ $url ] ) ) {
Debug2::debug2( '[Avatar] already in dict [url] ' . $url );
return $this->_avatar_realtime_gen_dict[ $url ];
}
$realpath = $this->_realpath( $url );
if ( file_exists( $realpath ) && time() - filemtime( $realpath ) <= $this->_conf_cache_ttl ) {
Debug2::debug2( '[Avatar] cache file exists [url] ' . $url );
return $this->_rewrite( $url, filemtime( $realpath ) );
}
if ( ! strpos( $url, 'gravatar.com' ) ) {
return $url;
}
// Send request
if ( ! empty( $this->_summary[ 'curr_request' ] ) && time() - $this->_summary[ 'curr_request' ] < 300 ) {
Debug2::debug2( '[Avatar] Bypass generating due to interval limit [url] ' . $url );
return $url;
}
// Generate immediately
$this->_avatar_realtime_gen_dict[ $url ] = $this->_generate( $url );
return $this->_avatar_realtime_gen_dict[ $url ];
}
/**
* Read last time generated info
*
* @since 3.0
* @access public
*/
public function queue_count() {
global $wpdb;
// If var not exists, mean table not exists // todo: not true
if ( ! $this->_tb ) {
return false;
}
$q = "SELECT COUNT(*) FROM `$this->_tb` WHERE dateline<" . ( time() - $this->_conf_cache_ttl );
return $wpdb->get_var( $q );
}
/**
* Get the final URL of local avatar
*
* Check from db also
*
* @since 3.0
*/
private function _rewrite( $url, $time = null ) {
return LITESPEED_STATIC_URL . '/avatar/' . $this->_filepath( $url ) . ( $time ? '?ver=' . $time : '' );
}
/**
* Generate realpath of the cache file
*
* @since 3.0
* @access private
*/
private function _realpath( $url ) {
return LITESPEED_STATIC_DIR . '/avatar/' . $this->_filepath( $url );
}
/**
* Get filepath
*
* @since 4.0
*/
private function _filepath( $url ) {
$filename = md5( $url ) . '.jpg';
if ( is_multisite() ) {
$filename = get_current_blog_id() . '/' . $filename;
}
return $filename;
}
/**
* Cron generation
*
* @since 3.0
* @access public
*/
public static function cron( $force = false ) {
global $wpdb;
$_instance = self::cls();
if ( ! $_instance->queue_count() ) {
Debug2::debug( '[Avatar] no queue' );
return;
}
// For cron, need to check request interval too
if ( ! $force ) {
if ( ! empty( $_instance->_summary[ 'curr_request' ] ) && time() - $_instance->_summary[ 'curr_request' ] < 300 ) {
Debug2::debug( '[Avatar] curr_request too close' );
return;
}
}
$q = "SELECT url FROM `$_instance->_tb` WHERE dateline < %d ORDER BY id DESC LIMIT %d";
$q = $wpdb->prepare( $q, array( time() - $_instance->_conf_cache_ttl, apply_filters( 'litespeed_avatar_limit', 30 ) ) );
$list = $wpdb->get_results( $q );
Debug2::debug( '[Avatar] cron job [count] ' . count( $list ) );
foreach ( $list as $v ) {
Debug2::debug( '[Avatar] cron job [url] ' . $v->url );
$_instance->_generate( $v->url );
}
}
/**
* Remote generator
*
* @since 3.0
* @access private
*/
private function _generate( $url ) {
global $wpdb;
// Record the data
$file = $this->_realpath( $url );
// Update request status
self::save_summary( array( 'curr_request' => time() ) );
// Generate
$this->_maybe_mk_cache_folder( 'avatar' );
$response = wp_remote_get( $url, array( 'timeout' => 180, 'stream' => true, 'filename' => $file ) );
Debug2::debug( '[Avatar] _generate [url] ' . $url );
// Parse response data
if ( is_wp_error( $response ) ) {
$error_message = $response->get_error_message();
file_exists( $file ) && unlink( $file );
Debug2::debug( '[Avatar] failed to get: ' . $error_message );
return $url;
}
// Save summary data
self::save_summary( array(
'last_spent' => time() - $this->_summary[ 'curr_request' ],
'last_request' => $this->_summary[ 'curr_request' ],
'curr_request' => 0,
) );
// Update DB
$md5 = md5( $url );
$q = "UPDATE `$this->_tb` SET dateline=%d WHERE md5=%s";
$existed = $wpdb->query( $wpdb->prepare( $q, array( time(), $md5 ) ) );
if ( ! $existed ) {
$q = "INSERT INTO `$this->_tb` SET url=%s, md5=%s, dateline=%d";
$wpdb->query( $wpdb->prepare( $q, array( $url, $md5, time() ) ) );
}
Debug2::debug( '[Avatar] saved avatar ' . $file );
return $this->_rewrite( $url );
}
/**
* Handle all request actions from main cls
*
* @since 3.0
* @access public
*/
public function handler() {
$type = Router::verify_type();
switch ( $type ) {
case self::TYPE_GENERATE :
self::cron( true );
break;
default:
break;
}
Admin::redirect();
}
}