2013-04-03 12:13:08 +00:00
< ? php
/**
* Name : Retrieve Feed Content
* Description : Follow the permalink of RSS / Atom feed items and replace the summary with the full content .
* Version : 0.2
* Author : Matthew Exon < http :// mat . exon . name >
*/
function retriever_install () {
register_hook ( 'plugin_settings' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings' );
register_hook ( 'plugin_settings_post' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings_post' );
register_hook ( 'post_remote' , 'addon/retriever/retriever.php' , 'retriever_post_remote_hook' );
register_hook ( 'contact_photo_menu' , 'addon/retriever/retriever.php' , 'retriever_contact_photo_menu' );
register_hook ( 'cron' , 'addon/retriever/retriever.php' , 'retriever_cron' );
$schema = file_get_contents ( dirname ( __file__ ) . '/database.sql' );
$arr = explode ( ';' , $schema );
foreach ( $arr as $a ) {
$r = q ( $a );
}
$r = q ( " SELECT `id` FROM `pconfig` WHERE `cat` LIKE 'retriever_%%' " );
if ( count ( $r ) || ( get_config ( 'retriever' , 'dbversion' ) == '0.1' )) {
$retrievers = array ();
$r = q ( " SELECT SUBSTRING(`cat`, 10) AS `contact`, `k`, `v` FROM `pconfig` WHERE `cat` LIKE 'retriever%%' " );
foreach ( $r as $rr ) {
$retrievers [ $rr [ 'contact' ]][ $rr [ 'k' ]] = $rr [ 'v' ];
}
foreach ( $retrievers as $k => $v ) {
$rr = q ( " SELECT `uid` FROM `contact` WHERE `id` = %d " , intval ( $k ));
$uid = $rr [ 0 ][ 'uid' ];
$v [ 'images' ] = 'on' ;
q ( " INSERT INTO `retriever_rule` (`uid`, `contact-id`, `data`) VALUES (%d, %d, '%s') " ,
intval ( $uid ), intval ( $k ), dbesc ( json_encode ( $v )));
}
q ( " DELETE FROM `pconfig` WHERE `cat` LIKE 'retriever%%' " );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.2' ) {
q ( " ALTER TABLE `retriever_resource` DROP COLUMN `retriever` " );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.3' ) {
q ( " ALTER TABLE `retriever_item` MODIFY COLUMN `item-uri` varchar(800) CHARACTER SET ascii NOT NULL " );
q ( " ALTER TABLE `retriever_resource` MODIFY COLUMN `url` varchar(800) CHARACTER SET ascii NOT NULL " );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.4' ) {
q ( " ALTER TABLE `retriever_item` ADD COLUMN `finished` tinyint(1) unsigned NOT NULL DEFAULT '0' " );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.5' ) {
q ( 'ALTER TABLE `retriever_resource` CHANGE `created` `created` timestamp NOT NULL DEFAULT now()' );
q ( 'ALTER TABLE `retriever_resource` CHANGE `completed` `completed` timestamp NULL DEFAULT NULL' );
q ( 'ALTER TABLE `retriever_resource` CHANGE `last-try` `last-try` timestamp NULL DEFAULT NULL' );
q ( 'ALTER TABLE `retriever_item` DROP KEY `all`' );
q ( 'ALTER TABLE `retriever_item` ADD KEY `all` (`item-uri`, `item-uid`, `contact-id`)' );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.6' ) {
q ( 'ALTER TABLE `retriever_item` CONVERT TO CHARACTER SET utf8 COLLATE utf8_bin' );
q ( 'ALTER TABLE `retriever_item` CHANGE `item-uri` `item-uri` varchar(800) CHARACTER SET ascii COLLATE ascii_bin NOT NULL' );
q ( 'ALTER TABLE `retriever_resource` CONVERT TO CHARACTER SET utf8 COLLATE utf8_bin' );
q ( 'ALTER TABLE `retriever_resource` CHANGE `url` `url` varchar(800) CHARACTER SET ascii COLLATE ascii_bin NOT NULL' );
q ( 'ALTER TABLE `retriever_rule` CONVERT TO CHARACTER SET utf8 COLLATE utf8_bin' );
}
if ( get_config ( 'retriever' , 'dbversion' ) == '0.7' ) {
$r = q ( " SELECT `id`, `data` FROM `retriever_rule` " );
foreach ( $r as $rr ) {
logger ( 'retriever_install: retriever ' . $rr [ 'id' ] . ' old config ' . $rr [ 'data' ], LOGGER_DATA );
$data = json_decode ( $rr [ 'data' ], true );
if ( $data [ 'pattern' ]) {
$matches = array ();
if ( preg_match ( " / \ /(.*) \ // " , $data [ 'pattern' ], $matches )) {
$data [ 'pattern' ] = $matches [ 1 ];
}
}
if ( $data [ 'match' ]) {
$include = array ();
foreach ( explode ( '|' , $data [ 'match' ]) as $component ) {
$matches = array ();
if ( preg_match ( " /([A-Za-z][A-Za-z0-9]*) \ [@([A-Za-z][a-z0-9]*)='([^']*)' \ ]/ " , $component , $matches )) {
$include [] = array (
'element' => $matches [ 1 ],
'attribute' => $matches [ 2 ],
'value' => $matches [ 3 ]);
}
if ( preg_match ( " /([A-Za-z][A-Za-z0-9]*) \ [contains(concat(' ',normalize-space(@class),' '),' ([^ ']+) ')]/ " , $component , $matches )) {
$include [] = array (
'element' => $matches [ 1 ],
'attribute' => $matches [ 2 ],
'value' => $matches [ 3 ]);
}
}
$data [ 'include' ] = $include ;
unset ( $data [ 'match' ]);
}
if ( $data [ 'remove' ]) {
$exclude = array ();
foreach ( explode ( '|' , $data [ 'remove' ]) as $component ) {
$matches = array ();
if ( preg_match ( " /([A-Za-z][A-Za-z0-9]*) \ [@([A-Za-z][a-z0-9]*)='([^']*)' \ ]/ " , $component , $matches )) {
$exclude [] = array (
'element' => $matches [ 1 ],
'attribute' => $matches [ 2 ],
'value' => $matches [ 3 ]);
}
if ( preg_match ( " /([A-Za-z][A-Za-z0-9]*) \ [contains(concat(' ',normalize-space(@class),' '),' ([^ ']+) ')]/ " , $component , $matches )) {
$exclude [] = array (
'element' => $matches [ 1 ],
'attribute' => $matches [ 2 ],
'value' => $matches [ 3 ]);
}
}
$data [ 'exclude' ] = $exclude ;
unset ( $data [ 'remove' ]);
}
$r = q ( 'UPDATE `retriever_rule` SET `data` = "%s" WHERE `id` = %d' , dbesc ( json_encode ( $data )), $rr [ 'id' ]);
logger ( 'retriever_install: retriever ' . $rr [ 'id' ] . ' new config ' . json_encode ( $data ), LOGGER_DATA );
}
}
set_config ( 'retriever' , 'dbversion' , '0.8' );
}
function retriever_uninstall () {
unregister_hook ( 'plugin_settings' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings' );
unregister_hook ( 'plugin_settings_post' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings_post' );
unregister_hook ( 'post_remote' , 'addon/retriever/retriever.php' , 'retriever_post_remote_hook' );
unregister_hook ( 'plugin_settings' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings' );
unregister_hook ( 'plugin_settings_post' , 'addon/retriever/retriever.php' , 'retriever_plugin_settings_post' );
unregister_hook ( 'contact_photo_menu' , 'addon/retriever/retriever.php' , 'retriever_contact_photo_menu' );
unregister_hook ( 'cron' , 'addon/retriever/retriever.php' , 'retriever_cron' );
}
function retriever_module () {}
function retriever_cron ( $a , $b ) {
// 100 is a nice sane number. Maybe this should be configurable.
retriever_retrieve_items ( 100 );
retriever_tidy ();
}
$retriever_item_count = 0 ;
function retriever_retrieve_items ( $max_items ) {
global $retriever_item_count ;
$retriever_schedule = array ( array ( 1 , 'minute' ),
array ( 10 , 'minute' ),
array ( 1 , 'hour' ),
array ( 1 , 'day' ),
array ( 2 , 'day' ),
array ( 1 , 'week' ),
array ( 1 , 'month' ));
$schedule_clauses = array ();
for ( $i = 0 ; $i < count ( $retriever_schedule ); $i ++ ) {
$num = $retriever_schedule [ $i ][ 0 ];
$unit = $retriever_schedule [ $i ][ 1 ];
array_push ( $schedule_clauses ,
'(`num-tries` = ' . $i . ' AND TIMESTAMPADD(' . dbesc ( $unit ) .
', ' . intval ( $num ) . ', `last-try`) < now())' );
}
$retrieve_items = $max_items - $retriever_item_count ;
do {
$r = q ( " SELECT * FROM `retriever_resource` WHERE `completed` IS NULL AND (`last-try` IS NULL OR %s) ORDER BY `last-try` ASC LIMIT %d " ,
dbesc ( implode ( $schedule_clauses , ' OR ' )),
intval ( $retrieve_items ));
if ( count ( $r ) == 0 ) {
break ;
}
foreach ( $r as $rr ) {
retrieve_resource ( $rr );
$retriever_item_count ++ ;
}
$retrieve_items = $max_items - $retriever_item_count ;
}
while ( $retrieve_items > 0 );
/* Look for items that are waiting even though the resource has
* completed . This usually happens because we ' ve been asked to
* retrospectively apply a config change . It could also happen
* due to a cron job dying or something . */
$r = q ( " SELECT retriever_resource.`id` as resource, retriever_item.`id` as item FROM retriever_resource, retriever_item, retriever_rule WHERE retriever_item.`finished` = 0 AND retriever_item.`resource` = retriever_resource.`id` AND retriever_resource.`completed` IS NOT NULL AND retriever_item.`contact-id` = retriever_rule.`contact-id` AND retriever_item.`item-uid` = retriever_rule.`uid` LIMIT %d " ,
intval ( $retrieve_items ));
if ( ! $r ) {
$r = array ();
}
foreach ( $r as $rr ) {
$resource = q ( " SELECT * FROM retriever_resource WHERE `id` = %d " , $rr [ 'resource' ]);
$retriever_item = retriever_get_retriever_item ( $rr [ 'item' ]);
if ( ! $retriever_item ) {
logger ( 'retriever_retrieve_items: no retriever item with id ' . $rr [ 'item' ]);
continue ;
}
$item = retriever_get_item ( $retriever_item );
if ( ! $item ) {
logger ( 'retriever_retrieve_items: no item ' . $retriever_item [ 'item-uri' ]);
continue ;
}
$retriever = get_retriever ( $item [ 'contact-id' ], $item [ 'uid' ]);
if ( ! $retriever ) {
logger ( 'retriever_retrieve_items: no retriever for item ' .
$retriever_item [ 'item-uri' ] . ' ' . $retriever_item [ 'uid' ] . ' ' . $item [ 'contact-id' ]);
continue ;
}
retriever_apply_completed_resource_to_item ( $retriever , $item , $resource [ 0 ]);
q ( " UPDATE `retriever_item` SET `finished` = 1 WHERE id = %d " ,
intval ( $retriever_item [ 'id' ]));
retriever_check_item_completed ( $item );
}
}
function retriever_tidy () {
q ( " DELETE FROM retriever_resource WHERE completed IS NOT NULL AND completed < DATE_SUB(now(), INTERVAL 1 WEEK) " );
q ( " DELETE FROM retriever_resource WHERE completed IS NULL AND created < DATE_SUB(now(), INTERVAL 3 MONTH) " );
$r = q ( " SELECT retriever_item.id FROM retriever_item LEFT OUTER JOIN retriever_resource ON (retriever_item.resource = retriever_resource.id) WHERE retriever_resource.id is null " );
foreach ( $r as $rr ) {
q ( 'DELETE FROM retriever_item WHERE id = %d' , intval ( $rr [ 'id' ]));
}
}
function retrieve_resource ( $resource ) {
logger ( 'retrieve_resource: ' . ( $resource [ 'num-tries' ] + 1 ) .
' attempt at resource ' . $resource [ 'id' ] . ' ' . $resource [ 'url' ], LOGGER_DEBUG );
q ( " UPDATE `retriever_resource` SET `last-try` = now(), `num-tries` = `num-tries` + 1 WHERE id = %d " ,
intval ( $resource [ 'id' ]));
$data = fetch_url ( $resource [ 'url' ], $resource [ 'binary' ], $resource [ 'type' ]);
$resource [ 'type' ] = get_app () -> get_curl_content_type ();
if ( $data ) {
$resource [ 'data' ] = $data ;
q ( " UPDATE `retriever_resource` SET `completed` = now(), `data` = '%s', `type` = '%s' WHERE id = %d " ,
dbesc ( $data ), dbesc ( $resource [ 'type' ]), intval ( $resource [ 'id' ]));
retriever_resource_completed ( $resource );
}
}
function get_retriever ( $contact_id , $uid , $create = false ) {
$r = q ( " SELECT * FROM `retriever_rule` WHERE `contact-id` = %d AND `uid` = %d " ,
intval ( $contact_id ), intval ( $uid ));
if ( count ( $r )) {
$r [ 0 ][ 'data' ] = json_decode ( $r [ 0 ][ 'data' ], true );
return $r [ 0 ];
}
if ( $create ) {
q ( " INSERT INTO `retriever_rule` (`uid`, `contact-id`) VALUES (%d, %d) " ,
intval ( $uid ), intval ( $contact_id ));
$r = q ( " SELECT * FROM `retriever_rule` WHERE `contact-id` = %d AND `uid` = %d " ,
intval ( $contact_id ), intval ( $uid ));
return $r [ 0 ];
}
}
function retriever_get_retriever_item ( $id ) {
$retriever_items = q ( " SELECT * FROM `retriever_item` WHERE id = %d " , intval ( $id ));
if ( count ( $retriever_items ) != 1 ) {
logger ( 'retriever_get_retriever_item: unable to find retriever_item ' . $id , LOGGER_NORMAL );
return ;
}
return $retriever_items [ 0 ];
}
function retriever_get_item ( $retriever_item ) {
$items = q ( " SELECT * FROM `item` WHERE `uri` = '%s' AND `uid` = %d AND `contact-id` = %d " ,
dbesc ( $retriever_item [ 'item-uri' ]),
intval ( $retriever_item [ 'item-uid' ]),
intval ( $retriever_item [ 'contact-id' ]));
if ( count ( $items ) != 1 ) {
logger ( 'retriever_get_item: unexpected number of results ' .
count ( $items ) . " when searching for item $uri $uid $cid " , LOGGER_NORMAL );
return ;
}
return $items [ 0 ];
}
function retriever_item_completed ( $retriever_item_id , $resource ) {
logger ( 'retriever_item_completed: id ' . $retriever_item_id . ' url ' . $resource [ 'url' ], LOGGER_DEBUG );
$retriever_item = retriever_get_retriever_item ( $retriever_item_id );
if ( ! $retriever_item ) {
return ;
}
2013-04-29 06:26:29 +00:00
// Note: the retriever might be null. Doesn't matter.
2013-04-03 12:13:08 +00:00
$retriever = get_retriever ( $retriever_item [ 'contact-id' ], $retriever_item [ 'item-uid' ]);
$item = retriever_get_item ( $retriever_item );
if ( ! $item ) {
return ;
}
retriever_apply_completed_resource_to_item ( $retriever , $item , $resource );
q ( " UPDATE `retriever_item` SET `finished` = 1 WHERE id = %d " ,
intval ( $retriever_item [ 'id' ]));
retriever_check_item_completed ( $item );
}
function retriever_resource_completed ( $resource ) {
logger ( 'retriever_resource_completed: id ' . $resource [ 'id' ] . ' url ' . $resource [ 'url' ], LOGGER_DEBUG );
$r = q ( " SELECT `id` FROM `retriever_item` WHERE `resource` = %d " , $resource [ 'id' ]);
foreach ( $r as $rr ) {
retriever_item_completed ( $rr [ 'id' ], $resource );
}
}
function apply_retrospective ( $retriever , $num ) {
$r = q ( " SELECT * FROM `item` WHERE `contact-id` = %d ORDER BY `received` DESC LIMIT %d " ,
intval ( $retriever [ 'contact-id' ]), intval ( $num ));
foreach ( $r as $item ) {
q ( 'UPDATE `item` SET `visible` = 0 WHERE `id` = %d' , $item [ 'id' ]);
retriever_on_item_insert ( $retriever , $item );
}
}
function retriever_on_item_insert ( $retriever , & $item ) {
if ( ! $retriever || ! $retriever [ 'id' ]) {
logger ( 'retriever_on_item_insert: No retriever supplied' , LOGGER_NORMAL );
return ;
}
if ( ! $retriever [ " data " ][ 'enable' ] == " on " ) {
return ;
}
if ( $retriever [ " data " ][ 'pattern' ]) {
$url = preg_replace ( '/' . $retriever [ " data " ][ 'pattern' ] . '/' , $retriever [ " data " ][ 'replace' ], $item [ 'plink' ]);
logger ( 'retriever_on_item_insert: Changed ' . $item [ 'plink' ] . ' to ' . $url , LOGGER_DATA );
}
else {
$url = $item [ 'plink' ];
}
$resource = add_retriever_resource ( $url );
$retriever_item_id = add_retriever_item ( $item , $resource );
}
function add_retriever_resource ( $url , $binary = false ) {
logger ( 'add_retriever_resource: ' . $url , LOGGER_DEBUG );
$r = q ( " SELECT * FROM `retriever_resource` WHERE `url` = '%s' " , dbesc ( $url ));
$resource = $r [ 0 ];
if ( count ( $r )) {
logger ( 'add_retriever_resource: Resource ' . $url . ' already requested' , LOGGER_DEBUG );
return $r [ 0 ];
}
else {
q ( " INSERT INTO `retriever_resource` (`binary`, `url`) " .
" VALUES (%d, '%s') " , intval ( $binary ? 1 : 0 ), dbesc ( $url ));
$r = q ( " SELECT * FROM `retriever_resource` WHERE `url` = '%s' " , dbesc ( $url ));
return $r [ 0 ];
}
}
function add_retriever_item ( & $item , $resource ) {
logger ( 'add_retriever_item: ' . $resource [ 'url' ] . ' for ' . $item [ 'uri' ] . ' ' . $item [ 'uid' ] . ' ' . $item [ 'contact-id' ], LOGGER_DEBUG );
q ( " INSERT INTO `retriever_item` (`item-uri`, `item-uid`, `contact-id`, `resource`) " .
" VALUES ('%s', %d, %d, %d) " ,
dbesc ( $item [ 'uri' ]), intval ( $item [ 'uid' ]), intval ( $item [ 'contact-id' ]), intval ( $resource [ " id " ]));
$r = q ( " SELECT id FROM `retriever_item` WHERE " .
" `item-uri` = '%s' AND `item-uid` = %d AND `contact-id` = %d AND `resource` = %d ORDER BY id DESC " ,
dbesc ( $item [ 'uri' ]), intval ( $item [ 'uid' ]), intval ( $item [ 'contact-id' ]), intval ( $resource [ 'id' ]));
if ( ! count ( $r )) {
logger ( " add_retriever_item: couldn't create retriever item for " .
$item [ 'uri' ] . ' ' . $item [ 'uid' ] . ' ' . $item [ 'contact-id' ],
LOGGER_NORMAL );
return ;
}
logger ( 'add_retriever_item: created retriever_item ' . $r [ 0 ][ 'id' ] . ' for item ' . $item [ 'uri' ] . ' ' . $item [ 'uid' ] . ' ' . $item [ 'contact-id' ], LOGGER_DEBUG );
return $r [ 0 ][ 'id' ];
}
function retriever_get_encoding ( $resource ) {
$matches = array ();
if ( preg_match ( '/charset=(.*)/' , $resource [ 'type' ], $matches )) {
return trim ( array_pop ( $matches ));
}
return 'utf-8' ;
}
function retriever_construct_xpath ( $spec ) {
if ( gettype ( $spec ) != " array " ) {
return ;
}
$components = array ();
foreach ( $spec as $clause ) {
if ( ! $clause [ 'attribute' ]) {
$components [] = $clause [ 'element' ];
continue ;
}
if ( $clause [ 'attribute' ] === ' class ' ) {
$components [] =
$clause [ 'element' ] .
" [contains(concat(' ', normalize-space(@class), ' '), ' " .
$clause [ 'value' ] . " ')] " ;
}
else {
$components [] =
$clause [ 'element' ] . '[@' .
$clause [ 'attribute' ] . " =' " .
$clause [ 'value' ] . " '] " ;
}
}
// It would be better to do this in smarty3 in extract.tpl
return implode ( '|' , $components );
}
function retriever_apply_dom_filter ( $retriever , & $item , $resource ) {
logger ( 'retriever_apply_dom_filter: applying XSLT to ' . $item [ 'id' ] . ' ' . $item [ 'plink' ], LOGGER_DEBUG );
require_once ( 'include/html2bbcode.php' );
if ( ! $retriever [ 'data' ][ 'include' ]) {
return ;
}
if ( ! $resource [ 'data' ]) {
logger ( 'retriever_apply_dom_filter: no text to work with' , LOGGER_NORMAL );
return ;
}
$encoding = retriever_get_encoding ( $resource );
logger ( '@@@ item type ' . $resource [ 'type' ] . ' encoding ' . $encoding );
$extracter_template = get_markup_template ( 'extract.tpl' , 'addon/retriever/' );
$doc = new DOMDocument ( '1.0' , 'utf-8' );
if ( strpos ( $resource [ 'type' ], 'html' ) !== false ) {
@ $doc -> loadHTML ( $resource [ 'data' ]);
}
else {
$doc -> loadXML ( $resource [ 'data' ]);
}
logger ( '@@@ actual encoding of document is ' . $doc -> encoding );
$components = parse_url ( $item [ 'plink' ]);
$rooturl = $components [ 'scheme' ] . " :// " . $components [ 'host' ];
$dirurl = $rooturl . dirname ( $components [ 'path' ]) . " / " ;
$params = array ( '$include' => retriever_construct_xpath ( $retriever [ 'data' ][ 'include' ]),
'$exclude' => retriever_construct_xpath ( $retriever [ 'data' ][ 'exclude' ]),
'$pageurl' => $item [ 'plink' ],
'$dirurl' => $dirurl ,
'$rooturl' => $rooturl );
$xslt = replace_macros ( $extracter_template , $params );
$xmldoc = new DOMDocument ();
$xmldoc -> loadXML ( $xslt );
$xp = new XsltProcessor ();
$xp -> importStylesheet ( $xmldoc );
$transformed = $xp -> transformToXML ( $doc );
$item [ 'body' ] = html2bbcode ( $transformed );
if ( ! strlen ( $item [ 'body' ])) {
logger ( 'retriever_apply_dom_filter retriever ' . $retriever [ 'id' ] . ' item ' . $item [ 'id' ] . ': output was empty' , LOGGER_NORMAL );
return ;
}
$item [ 'body' ] .= " \n \n " . t ( 'Retrieved' ) . ' ' . date ( " Y-m-d " ) . ': [url=' ;
$item [ 'body' ] .= $item [ 'plink' ];
$item [ 'body' ] .= ']' . $item [ 'plink' ] . '[/url]' ;
q ( " UPDATE `item` SET `body` = '%s', `edited` = '%s' WHERE `id` = %d " ,
dbesc ( $item [ 'body' ]), dbesc ( datetime_convert ( 'UTC' , 'UTC' )), intval ( $item [ 'id' ]));
}
function retrieve_images ( & $item ) {
$matches1 = array ();
preg_match_all ( " / \ [img \ =([0-9]*)x([0-9]*) \ ](.*?) \ [ \ /img \ ]/ism " , $item [ " body " ], $matches1 );
$matches2 = array ();
preg_match_all ( " / \ [img \ ](.*?) \ [ \ /img \ ]/ism " , $item [ " body " ], $matches2 );
$matches = array_merge ( $matches1 [ 3 ], $matches2 [ 1 ]);
logger ( 'retrieve_images: found ' . count ( $matches ) . ' images for item ' . $item [ 'uri' ] . ' ' . $item [ 'uid' ] . ' ' . $item [ 'contact-id' ], LOGGER_DEBUG );
foreach ( $matches as $url ) {
if ( strpos ( $url , get_app () -> get_baseurl ()) === FALSE ) {
$resource = add_retriever_resource ( $url , true );
if ( ! $resource [ 'completed' ]) {
add_retriever_item ( $item , $resource );
}
else {
retriever_transform_images ( $item , $resource );
}
}
}
}
function retriever_check_item_completed ( & $item )
{
$r = q ( 'SELECT count(*) FROM retriever_item WHERE `item-uri` = "%s" ' .
'AND `item-uid` = %d AND `contact-id` = %d AND `finished` = 0' ,
dbesc ( $item [ 'uri' ]), intval ( $item [ 'uid' ]),
intval ( $item [ 'contact-id' ]));
$waiting = $r [ 0 ][ 'count(*)' ];
logger ( 'retriever_check_item_completed: item ' . $item [ 'uri' ] . ' ' . $item [ 'uid' ]
. ' ' . $item [ 'contact-id' ] . ' waiting for ' . $waiting . ' resources' , LOGGER_DEBUG );
$old_visible = $item [ 'visible' ];
$item [ 'visible' ] = $waiting ? 0 : 1 ;
if (( $item [ 'id' ] > 0 ) && ( $old_visible != $item [ 'visible' ])) {
logger ( 'retriever_check_item_completed: changing visible flag to ' . $item [ 'visible' ] . ' and invoking notifier ("edit_post", ' . $item [ 'id' ] . ')' , LOGGER_DEBUG );
q ( " UPDATE `item` SET `visible` = %d, `edited` = '%s' WHERE `id` = %d " ,
intval ( $item [ 'visible' ]),
dbesc ( datetime_convert ( 'UTC' , 'UTC' )),
intval ( $item [ 'id' ]));
proc_run ( 'php' , " include/notifier.php " , 'edit_post' , $item [ 'id' ]);
}
}
function retriever_apply_completed_resource_to_item ( $retriever , & $item , $resource ) {
logger ( 'retriever_apply_completed_resource_to_item: retriever ' .
( $retriever ? $retriever [ 'id' ] : 'none' ) .
' resource ' . $resource [ 'url' ] . ' plink ' . $item [ 'plink' ], LOGGER_DEBUG );
if ( strpos ( $resource [ 'type' ], 'image' ) !== false ) {
retriever_transform_images ( $item , $resource );
}
if ( ! $retriever ) {
return ;
}
if (( strpos ( $resource [ 'type' ], 'html' ) !== false ) ||
( strpos ( $resource [ 'type' ], 'xml' ) !== false )) {
retriever_apply_dom_filter ( $retriever , $item , $resource );
if ( $retriever [ " data " ][ 'images' ] ) {
retrieve_images ( $item );
}
}
}
function retriever_store_photo ( $item , & $resource ) {
$hash = photo_new_resource ();
if ( class_exists ( 'Imagick' )) {
try {
$image = new Imagick ();
$image -> readImageBlob ( $resource [ 'data' ]);
$resource [ 'width' ] = $image -> getImageWidth ();
$resource [ 'height' ] = $image -> getImageHeight ();
}
catch ( Exception $e ) {
logger ( " ImageMagick couldn't process image " . $resource [ 'id' ] . " " . $resource [ 'url' ] . ' length ' . strlen ( $resource [ 'data' ]) . ': ' . $e -> getMessage (), LOGGER_DEBUG );
return false ;
}
}
if ( ! array_key_exists ( 'width' , $resource )) {
$image = @ imagecreatefromstring ( $resource [ 'data' ]);
if ( $image === false ) {
logger ( " Couldn't process image " . $resource [ 'id' ] . " " . $resource [ 'url' ], LOGGER_DEBUG );
return false ;
}
$resource [ 'width' ] = imagesx ( $image );
$resource [ 'height' ] = imagesy ( $image );
imagedestroy ( $image );
}
$url_components = parse_url ( $resource [ 'url' ]);
$filename = basename ( $url_components [ 'path' ]);
if ( ! strlen ( $filename )) {
$filename = 'image' ;
}
$r = q ( " INSERT INTO `photo`
( `uid` , `contact-id` , `guid` , `resource-id` , `created` , `edited` , `filename` , `type` , `album` , `height` , `width` , `datasize` , `data` )
VALUES ( % d , % d , '%s' , '%s' , '%s' , '%s' , '%s' , '%s' , '%s' , % d , % d , % d , '%s' ) " ,
intval ( $item [ 'item-uid' ]),
intval ( $item [ 'contact-id' ]),
dbesc ( get_guid ()),
dbesc ( $hash ),
dbesc ( datetime_convert ()),
dbesc ( datetime_convert ()),
dbesc ( $filename ),
dbesc ( $resource [ 'type' ]),
dbesc ( 'Retrieved Images' ),
intval ( $resource [ 'height' ]),
intval ( $resource [ 'width' ]),
intval ( strlen ( $resource [ 'data' ])),
dbesc ( $resource [ 'data' ])
);
return $hash ;
}
function retriever_transform_images ( & $item , $resource ) {
require_once ( 'include/Photo.php' );
if ( ! $resource [ " data " ]) {
logger ( 'retriever_transform_images: no data available for '
. $resource [ 'id' ] . ' ' . $resource [ 'url' ], LOGGER_NORMAL );
return ;
}
$hash = retriever_store_photo ( $item , $resource );
if ( $hash === false ) {
logger ( 'retriever_transform_images: unable to store photo '
. $resource [ 'id' ] . ' ' . $resource [ 'url' ], LOGGER_NORMAL );
return ;
}
$new_url = get_app () -> get_baseurl () . '/photo/' . $hash ;
logger ( 'retriever_transform_images: replacing ' . $resource [ 'url' ] . ' with ' .
$new_url . ' in item ' . $item [ 'plink' ], LOGGER_DEBUG );
$transformed = str_replace ( $resource [ " url " ], $new_url , $item [ 'body' ]);
if ( $transformed === $item [ 'body' ]) {
return ;
}
$item [ 'body' ] = $transformed ;
q ( " UPDATE `item` SET `edited` = '%s', `body` = '%s' WHERE `plink` = '%s' AND `uid` = %d AND `contact-id` = %d " ,
dbesc ( datetime_convert ( 'UTC' , 'UTC' )),
dbesc ( $item [ 'body' ]),
dbesc ( $item [ 'plink' ]),
intval ( $item [ 'uid' ]),
intval ( $item [ 'contact-id' ]));
}
function retriever_content ( $a ) {
if ( ! local_user ()) {
$a -> page [ 'content' ] .= " <p>Please log in</p> " ;
return ;
}
if ( $a -> argv [ 1 ] === 'help' ) {
$feeds = q ( " SELECT `id`, `name`, `thumb` FROM contact WHERE `uid` = %d AND `network` = 'feed' " ,
local_user ());
foreach ( $feeds as $k => $v ) {
$feeds [ $k ][ 'url' ] = $a -> get_baseurl () . '/retriever/' . $v [ 'id' ];
}
$template = get_markup_template ( '/help.tpl' , 'addon/retriever/' );
$a -> page [ 'content' ] .= replace_macros ( $template , array (
'$config' => $a -> get_baseurl () . '/settings/addon' ,
'$feeds' => $feeds ));
return ;
}
if ( $a -> argv [ 1 ]) {
$retriever = get_retriever ( $a -> argv [ 1 ], local_user (), false );
if ( x ( $_POST [ " id " ])) {
$retriever = get_retriever ( $a -> argv [ 1 ], local_user (), true );
$retriever [ " data " ] = array ();
foreach ( array ( 'pattern' , 'replace' , 'enable' , 'images' ) as $setting ) {
if ( x ( $_POST [ 'retriever_' . $setting ])) {
$retriever [ " data " ][ $setting ] = $_POST [ 'retriever_' . $setting ];
}
}
foreach ( $_POST as $k => $v ) {
if ( preg_match ( " /retriever-(include|exclude)-( \ d+)-(element|attribute|value)/ " , $k , $matches )) {
$retriever [ 'data' ][ $matches [ 1 ]][ intval ( $matches [ 2 ])][ $matches [ 3 ]] = $v ;
}
}
// You've gotta have an element, even if it's just "*"
foreach ( $retriever [ 'data' ][ 'include' ] as $k => $clause ) {
if ( ! $clause [ 'element' ]) {
unset ( $retriever [ 'data' ][ 'include' ][ $k ]);
}
}
foreach ( $retriever [ 'data' ][ 'exclude' ] as $k => $clause ) {
if ( ! $clause [ 'element' ]) {
unset ( $retriever [ 'data' ][ 'exclude' ][ $k ]);
}
}
q ( " UPDATE `retriever_rule` SET `data`='%s' WHERE `id` = %d " ,
dbesc ( json_encode ( $retriever [ " data " ])), intval ( $retriever [ " id " ]));
$a -> page [ 'content' ] .= " <p><b>Settings Updated " ;
if ( x ( $_POST [ " retriever_retrospective " ])) {
apply_retrospective ( $retriever , $_POST [ " retriever_retrospective " ]);
$a -> page [ 'content' ] .= " and retrospectively applied to " . $_POST [ " apply " ] . " posts " ;
}
$a -> page [ 'content' ] .= " .</p></b> " ;
}
$template = get_markup_template ( '/rule-config.tpl' , 'addon/retriever/' );
$a -> page [ 'content' ] .= replace_macros ( $template , array (
'$enable' => array (
'retriever_enable' ,
t ( 'Enabled' ),
$retriever [ 'data' ][ 'enable' ]),
'$pattern' => array (
'retriever_pattern' ,
t ( 'URL Pattern' ),
$retriever [ " data " ][ 'pattern' ],
t ( 'Regular expression matching part of the URL to replace' )),
'$replace' => array (
'retriever_replace' ,
t ( 'URL Replace' ),
$retriever [ " data " ][ 'replace' ],
t ( 'Text to replace matching part of above regular expression' )),
'$images' => array (
'retriever_images' ,
t ( 'Download Images' ),
$retriever [ 'data' ][ 'images' ]),
'$retrospective' => array (
'retriever_retrospective' ,
t ( 'Retrospectively Apply' ),
'0' ,
t ( 'Reapply the rules to this number of posts' )),
'$title' => t ( 'Retrieve Feed Content' ),
'$help' => $a -> get_baseurl () . '/retriever/help' ,
'$submit' => t ( 'Submit' ),
'$id' => ( $retriever [ " id " ] ? $retriever [ " id " ] : " create " ),
'$tag_t' => t ( 'Tag' ),
'$attribute_t' => t ( 'Attribute' ),
'$value_t' => t ( 'Value' ),
'$add_t' => t ( 'Add' ),
'$remove_t' => t ( 'Remove' ),
'$include_t' => t ( 'Include' ),
'$include' => $retriever [ 'data' ][ 'include' ],
'$exclude_t' => t ( 'Exclude' ),
'$exclude' => $retriever [ " data " ][ 'exclude' ]));
return ;
}
}
function retriever_contact_photo_menu ( $a , & $args ) {
if ( ! $args ) {
return ;
}
if ( $args [ " contact " ][ " network " ] == " feed " ) {
$args [ " menu " ][ 'retriever' ] = array ( t ( 'Retriever' ), $a -> get_baseurl () . '/retriever/' . $args [ " contact " ][ 'id' ]);
}
}
function retriever_post_remote_hook ( & $a , & $item ) {
logger ( 'retriever_post_remote_hook: ' . $item [ 'uri' ] . ' ' . $item [ 'uid' ] . ' ' . $item [ 'contact-id' ], LOGGER_DEBUG );
$retriever = get_retriever ( $item [ 'contact-id' ], $item [ " uid " ], false );
if ( $retriever ) {
retriever_on_item_insert ( $retriever , $item );
}
else {
if ( get_pconfig ( $item [ " uid " ], 'retriever' , 'all_photos' )) {
retrieve_images ( $item , null );
}
}
retriever_check_item_completed ( $item );
}
function retriever_plugin_settings ( & $a , & $s ) {
$all_photos = get_pconfig ( local_user (), 'retriever' , 'all_photos' );
$all_photos_mu = ( $all_photos == 'on' ) ? ' checked="true"' : '' ;
$template = get_markup_template ( '/settings.tpl' , 'addon/retriever/' );
$s .= replace_macros ( $template , array (
'$submit' => t ( 'Submit' ),
'$title' => t ( 'Retriever Settings' ),
'$help' => $a -> get_baseurl () . '/retriever/help' ,
'$all_photos' => $all_photos_mu ,
'$all_photos_t' => t ( 'All Photos' )));
}
function retriever_plugin_settings_post ( $a , $post ) {
if ( $_POST [ 'all_photos' ]) {
set_pconfig ( local_user (), 'retriever' , 'all_photos' , $_POST [ 'all_photos' ]);
}
else {
del_pconfig ( local_user (), 'retriever' , 'all_photos' );
}
}