<?php

[0] Fix | Delete

/**

[1] Fix | Delete

* The avatar cache class

[2] Fix | Delete

*

[3] Fix | Delete

* @since 3.0

[4] Fix | Delete

* @package LiteSpeed

[5] Fix | Delete

* @subpackage LiteSpeed/inc

[6] Fix | Delete

* @author LiteSpeed Technologies <info@litespeedtech.com>

[7] Fix | Delete

*/

[8] Fix | Delete

namespace LiteSpeed;

[9] Fix | Delete

[10] Fix | Delete

defined('WPINC') || exit();

[11] Fix | Delete

[12] Fix | Delete

class Avatar extends Base

[13] Fix | Delete

{

[14] Fix | Delete

const TYPE_GENERATE = 'generate';

[15] Fix | Delete

[16] Fix | Delete

private $_conf_cache_ttl;

[17] Fix | Delete

private $_tb;

[18] Fix | Delete

[19] Fix | Delete

private $_avatar_realtime_gen_dict = array();

[20] Fix | Delete

protected $_summary;

[21] Fix | Delete

[22] Fix | Delete

/**

[23] Fix | Delete

* Init

[24] Fix | Delete

*

[25] Fix | Delete

* @since 1.4

[26] Fix | Delete

*/

[27] Fix | Delete

public function __construct()

[28] Fix | Delete

{

[29] Fix | Delete

if (!$this->conf(self::O_DISCUSS_AVATAR_CACHE)) {

[30] Fix | Delete

return;

[31] Fix | Delete

}

[32] Fix | Delete

[33] Fix | Delete

Debug2::debug2('[Avatar] init');

[34] Fix | Delete

[35] Fix | Delete

$this->_tb = $this->cls('Data')->tb('avatar');

[36] Fix | Delete

[37] Fix | Delete

$this->_conf_cache_ttl = $this->conf(self::O_DISCUSS_AVATAR_CACHE_TTL);

[38] Fix | Delete

[39] Fix | Delete

add_filter('get_avatar_url', array($this, 'crawl_avatar'));

[40] Fix | Delete

[41] Fix | Delete

$this->_summary = self::get_summary();

[42] Fix | Delete

}

[43] Fix | Delete

[44] Fix | Delete

/**

[45] Fix | Delete

* Check if need db table or not

[46] Fix | Delete

*

[47] Fix | Delete

* @since 3.0

[48] Fix | Delete

* @access public

[49] Fix | Delete

*/

[50] Fix | Delete

public function need_db()

[51] Fix | Delete

{

[52] Fix | Delete

if ($this->conf(self::O_DISCUSS_AVATAR_CACHE)) {

[53] Fix | Delete

return true;

[54] Fix | Delete

}

[55] Fix | Delete

[56] Fix | Delete

return false;

[57] Fix | Delete

}

[58] Fix | Delete

/**

[59] Fix | Delete

* Get gravatar URL from DB and regenerate

[60] Fix | Delete

*

[61] Fix | Delete

* @since 3.0

[62] Fix | Delete

* @access public

[63] Fix | Delete

*/

[64] Fix | Delete

public function serve_static($md5)

[65] Fix | Delete

{

[66] Fix | Delete

global $wpdb;

[67] Fix | Delete

[68] Fix | Delete

Debug2::debug('[Avatar] is avatar request');

[69] Fix | Delete

[70] Fix | Delete

if (strlen($md5) !== 32) {

[71] Fix | Delete

Debug2::debug('[Avatar] wrong md5 ' . $md5);

[72] Fix | Delete

return;

[73] Fix | Delete

}

[74] Fix | Delete

[75] Fix | Delete

$q = "SELECT url FROM `$this->_tb` WHERE md5=%s";

[76] Fix | Delete

$url = $wpdb->get_var($wpdb->prepare($q, $md5));

[77] Fix | Delete

[78] Fix | Delete

if (!$url) {

[79] Fix | Delete

Debug2::debug('[Avatar] no matched url for md5 ' . $md5);

[80] Fix | Delete

return;

[81] Fix | Delete

}

[82] Fix | Delete

[83] Fix | Delete

$url = $this->_generate($url);

[84] Fix | Delete

[85] Fix | Delete

wp_redirect($url);

[86] Fix | Delete

exit();

[87] Fix | Delete

}

[88] Fix | Delete

[89] Fix | Delete

/**

[90] Fix | Delete

* Localize gravatar

[91] Fix | Delete

*

[92] Fix | Delete

* @since 3.0

[93] Fix | Delete

* @access public

[94] Fix | Delete

*/

[95] Fix | Delete

public function crawl_avatar($url)

[96] Fix | Delete

{

[97] Fix | Delete

if (!$url) {

[98] Fix | Delete

return $url;

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

// Check if its already in dict or not

[102] Fix | Delete

if (!empty($this->_avatar_realtime_gen_dict[$url])) {

[103] Fix | Delete

Debug2::debug2('[Avatar] already in dict [url] ' . $url);

[104] Fix | Delete

[105] Fix | Delete

return $this->_avatar_realtime_gen_dict[$url];

[106] Fix | Delete

}

[107] Fix | Delete

[108] Fix | Delete

$realpath = $this->_realpath($url);

[109] Fix | Delete

if (file_exists($realpath) && time() - filemtime($realpath) <= $this->_conf_cache_ttl) {

[110] Fix | Delete

Debug2::debug2('[Avatar] cache file exists [url] ' . $url);

[111] Fix | Delete

return $this->_rewrite($url, filemtime($realpath));

[112] Fix | Delete

}

[113] Fix | Delete

[114] Fix | Delete

if (!strpos($url, 'gravatar.com')) {

[115] Fix | Delete

return $url;

[116] Fix | Delete

}

[117] Fix | Delete

[118] Fix | Delete

// Send request

[119] Fix | Delete

if (!empty($this->_summary['curr_request']) && time() - $this->_summary['curr_request'] < 300) {

[120] Fix | Delete

Debug2::debug2('[Avatar] Bypass generating due to interval limit [url] ' . $url);

[121] Fix | Delete

return $url;

[122] Fix | Delete

}

[123] Fix | Delete

[124] Fix | Delete

// Generate immediately

[125] Fix | Delete

$this->_avatar_realtime_gen_dict[$url] = $this->_generate($url);

[126] Fix | Delete

[127] Fix | Delete

return $this->_avatar_realtime_gen_dict[$url];

[128] Fix | Delete

}

[129] Fix | Delete

[130] Fix | Delete

/**

[131] Fix | Delete

* Read last time generated info

[132] Fix | Delete

*

[133] Fix | Delete

* @since 3.0

[134] Fix | Delete

* @access public

[135] Fix | Delete

*/

[136] Fix | Delete

public function queue_count()

[137] Fix | Delete

{

[138] Fix | Delete

global $wpdb;

[139] Fix | Delete

[140] Fix | Delete

// If var not exists, mean table not exists // todo: not true

[141] Fix | Delete

if (!$this->_tb) {

[142] Fix | Delete

return false;

[143] Fix | Delete

}

[144] Fix | Delete

[145] Fix | Delete

$q = "SELECT COUNT(*) FROM `$this->_tb` WHERE dateline<" . (time() - $this->_conf_cache_ttl);

[146] Fix | Delete

return $wpdb->get_var($q);

[147] Fix | Delete

}

[148] Fix | Delete

[149] Fix | Delete

/**

[150] Fix | Delete

* Get the final URL of local avatar

[151] Fix | Delete

*

[152] Fix | Delete

* Check from db also

[153] Fix | Delete

*

[154] Fix | Delete

* @since 3.0

[155] Fix | Delete

*/

[156] Fix | Delete

private function _rewrite($url, $time = null)

[157] Fix | Delete

{

[158] Fix | Delete

return LITESPEED_STATIC_URL . '/avatar/' . $this->_filepath($url) . ($time ? '?ver=' . $time : '');

[159] Fix | Delete

}

[160] Fix | Delete

[161] Fix | Delete

/**

[162] Fix | Delete

* Generate realpath of the cache file

[163] Fix | Delete

*

[164] Fix | Delete

* @since 3.0

[165] Fix | Delete

* @access private

[166] Fix | Delete

*/

[167] Fix | Delete

private function _realpath($url)

[168] Fix | Delete

{

[169] Fix | Delete

return LITESPEED_STATIC_DIR . '/avatar/' . $this->_filepath($url);

[170] Fix | Delete

}

[171] Fix | Delete

[172] Fix | Delete

/**

[173] Fix | Delete

* Get filepath

[174] Fix | Delete

*

[175] Fix | Delete

* @since 4.0

[176] Fix | Delete

*/

[177] Fix | Delete

private function _filepath($url)

[178] Fix | Delete

{

[179] Fix | Delete

$filename = md5($url) . '.jpg';

[180] Fix | Delete

if (is_multisite()) {

[181] Fix | Delete

$filename = get_current_blog_id() . '/' . $filename;

[182] Fix | Delete

}

[183] Fix | Delete

return $filename;

[184] Fix | Delete

}

[185] Fix | Delete

[186] Fix | Delete

/**

[187] Fix | Delete

* Cron generation

[188] Fix | Delete

*

[189] Fix | Delete

* @since 3.0

[190] Fix | Delete

* @access public

[191] Fix | Delete

*/

[192] Fix | Delete

public static function cron($force = false)

[193] Fix | Delete

{

[194] Fix | Delete

global $wpdb;

[195] Fix | Delete

[196] Fix | Delete

$_instance = self::cls();

[197] Fix | Delete

if (!$_instance->queue_count()) {

[198] Fix | Delete

Debug2::debug('[Avatar] no queue');

[199] Fix | Delete

return;

[200] Fix | Delete

}

[201] Fix | Delete

[202] Fix | Delete

// For cron, need to check request interval too

[203] Fix | Delete

if (!$force) {

[204] Fix | Delete

if (!empty($_instance->_summary['curr_request']) && time() - $_instance->_summary['curr_request'] < 300) {

[205] Fix | Delete

Debug2::debug('[Avatar] curr_request too close');

[206] Fix | Delete

return;

[207] Fix | Delete

}

[208] Fix | Delete

}

[209] Fix | Delete

[210] Fix | Delete

$q = "SELECT url FROM `$_instance->_tb` WHERE dateline < %d ORDER BY id DESC LIMIT %d";

[211] Fix | Delete

$q = $wpdb->prepare($q, array(time() - $_instance->_conf_cache_ttl, apply_filters('litespeed_avatar_limit', 30)));

[212] Fix | Delete

[213] Fix | Delete

$list = $wpdb->get_results($q);

[214] Fix | Delete

Debug2::debug('[Avatar] cron job [count] ' . count($list));

[215] Fix | Delete

[216] Fix | Delete

foreach ($list as $v) {

[217] Fix | Delete

Debug2::debug('[Avatar] cron job [url] ' . $v->url);

[218] Fix | Delete

[219] Fix | Delete

$_instance->_generate($v->url);

[220] Fix | Delete

}

[221] Fix | Delete

}

[222] Fix | Delete

[223] Fix | Delete

/**

[224] Fix | Delete

* Remote generator

[225] Fix | Delete

*

[226] Fix | Delete

* @since 3.0

[227] Fix | Delete

* @access private

[228] Fix | Delete

*/

[229] Fix | Delete

private function _generate($url)

[230] Fix | Delete

{

[231] Fix | Delete

global $wpdb;

[232] Fix | Delete

[233] Fix | Delete

// Record the data

[234] Fix | Delete

[235] Fix | Delete

$file = $this->_realpath($url);

[236] Fix | Delete

[237] Fix | Delete

// Update request status

[238] Fix | Delete

self::save_summary(array('curr_request' => time()));

[239] Fix | Delete

[240] Fix | Delete

// Generate

[241] Fix | Delete

$this->_maybe_mk_cache_folder('avatar');

[242] Fix | Delete

[243] Fix | Delete

$response = wp_remote_get($url, array('timeout' => 180, 'stream' => true, 'filename' => $file));

[244] Fix | Delete

[245] Fix | Delete

Debug2::debug('[Avatar] _generate [url] ' . $url);

[246] Fix | Delete

[247] Fix | Delete

// Parse response data

[248] Fix | Delete

if (is_wp_error($response)) {

[249] Fix | Delete

$error_message = $response->get_error_message();

[250] Fix | Delete

file_exists($file) && unlink($file);

[251] Fix | Delete

Debug2::debug('[Avatar] failed to get: ' . $error_message);

[252] Fix | Delete

return $url;

[253] Fix | Delete

}

[254] Fix | Delete

[255] Fix | Delete

// Save summary data

[256] Fix | Delete

self::save_summary(array(

[257] Fix | Delete

'last_spent' => time() - $this->_summary['curr_request'],

[258] Fix | Delete

'last_request' => $this->_summary['curr_request'],

[259] Fix | Delete

'curr_request' => 0,

[260] Fix | Delete

));

[261] Fix | Delete

[262] Fix | Delete

// Update DB

[263] Fix | Delete

$md5 = md5($url);

[264] Fix | Delete

$q = "UPDATE `$this->_tb` SET dateline=%d WHERE md5=%s";

[265] Fix | Delete

$existed = $wpdb->query($wpdb->prepare($q, array(time(), $md5)));

[266] Fix | Delete

if (!$existed) {

[267] Fix | Delete

$q = "INSERT INTO `$this->_tb` SET url=%s, md5=%s, dateline=%d";

[268] Fix | Delete

$wpdb->query($wpdb->prepare($q, array($url, $md5, time())));

[269] Fix | Delete

}

[270] Fix | Delete

[271] Fix | Delete

Debug2::debug('[Avatar] saved avatar ' . $file);

[272] Fix | Delete

[273] Fix | Delete

return $this->_rewrite($url);

[274] Fix | Delete

}

[275] Fix | Delete

[276] Fix | Delete

/**

[277] Fix | Delete

* Handle all request actions from main cls

[278] Fix | Delete

*

[279] Fix | Delete

* @since 3.0

[280] Fix | Delete

* @access public

[281] Fix | Delete

*/

[282] Fix | Delete

public function handler()

[283] Fix | Delete

{

[284] Fix | Delete

$type = Router::verify_type();

[285] Fix | Delete

[286] Fix | Delete

switch ($type) {

[287] Fix | Delete

case self::TYPE_GENERATE:

[288] Fix | Delete

self::cron(true);

[289] Fix | Delete

break;

[290] Fix | Delete

[291] Fix | Delete

default:

[292] Fix | Delete

break;

[293] Fix | Delete

}

[294] Fix | Delete

[295] Fix | Delete

Admin::redirect();

[296] Fix | Delete

}

[297] Fix | Delete

}

[298] Fix | Delete

[299] Fix | Delete