banana/spool.inc.php

   1 <?php
   2 /********************************************************************************
   3 * include/spool.inc.php : spool subroutines
   4 * -----------------------
   5 *
   6 * This file is part of the banana distribution
   7 * Copyright: See COPYING files that comes with this distribution
   8 ********************************************************************************/
   9
  10 require_once dirname(__FILE__) . '/banana.inc.php';
  11 require_once dirname(__FILE__) . '/tree.inc.php';
  12
  13 define('BANANA_SPOOL_VERSION', '0.5.14');
  14
  15 /** Class spoolhead
  16  *  class used in thread overviews
  17  */
  18 class BananaSpoolHead
  19 {
  20     public $id;
  21     public $msgid;
  22
  23     /** date (timestamp) */
  24     public $date;
  25     /** subject */
  26     public $subject;
  27     /** author */
  28     public $from;
  29     public $name;
  30     public $color;
  31     /** reference of parent */
  32     public $parent = null;
  33     /** array of children */
  34     public $children = Array();
  35     /** true if post is read */
  36     public $isread;
  37     /** number of posts deeper in this branch of tree */
  38     public $desc;
  39     /**  same as desc, but counts only unread posts */
  40     public $descunread;
  41     /** last time the number of children has been updated */
  42     public $time = 0;
  43
  44     /** storage data */
  45     public $storage = array();
  46
  47     /** constructor
  48      * @param $_date INTEGER timestamp of post
  49      * @param $_subject STRING subject of post
  50      * @param $_from STRING author of post
  51      * @param $_desc INTEGER desc value (1 for a new post)
  52      * @param $_read BOOLEAN true if read
  53      * @param $_descunread INTEGER descunread value (0 for a new post)
  54      */
  55     public function __construct($id, array &$message)
  56     {
  57         $this->id         = $id;
  58         $this->msgid      = @$message['message-id'];
  59         $this->date       = $message['date'];
  60         $this->subject    = @$message['subject'];
  61         $this->from       = $message['from'];
  62         $this->color      = sprintf('#%06x', abs(crc32($this->from) % 0xffffff));
  63         $this->desc       = 1;
  64         $this->isread     = true;
  65         $this->descunread = 0;
  66         if (preg_match("/^([^ ]+@[^ ]+) \((.*)\)$/", $this->from, $regs)) {
  67             $this->name = $regs[2];
  68         }
  69         if (preg_match("/^\"?([^<>\"]+)\"? +<(.+@.+)>$/", $this->from, $regs)) {
  70             $this->name = preg_replace("/^'(.*)'$/", '\1', $regs[1]);
  71             $this->name = stripslashes($this->name);
  72         }
  73         if ($this->name) {
  74             $this->name =  preg_replace("/\\\(\(|\))/","\\1", $this->name);
  75         } else if (preg_match("/([^< ]+)@([^> ]+)/", $this->from, $regs)) {
  76             $this->name = $regs[1];
  77         } else {
  78             $this->name = 'Anonymous';
  79         }
  80     }
  81 }
  82
  83
  84 class BananaSpool
  85 {
  86     private $version;
  87     private $mode;
  88
  89     /**  group name */
  90     public $group;
  91     /**  spool */
  92     public $overview = array();
  93     /**  array msgid => msgnum */
  94     public $ids      = array();
  95     /** thread starts */
  96     public $roots    = array();
  97
  98     /** protocole specific data */
  99     public $storage = array();
 100
 101     private $unreadnb = 0;
 102
 103     /** constructor
 104      * @param $_group STRING group name
 105      * @param $_display INTEGER 1 => all posts, 2 => only threads with new posts
 106      * @param $_since INTEGER time stamp (used for read/unread)
 107      */
 108     protected function __construct($group)
 109     {
 110         $this->version    = BANANA_SPOOL_VERSION;
 111         $this->mode       = Banana::SPOOL_ALL;
 112         $this->group      = $group;
 113     }
 114
 115     public static function &getSpool($group, $since = 0, $clean = false)
 116     {
 117         if (!is_null(Banana::$spool) && Banana::$spool->group == $group) {
 118             $spool =& Banana::$spool;
 119         } else {
 120             $spool =& BananaSpool::readFromFile($group);
 121         }
 122         if (is_null($spool)) {
 123             $spool = new BananaSpool($group);
 124         }
 125         Banana::$spool =& $spool;
 126         $spool->build();
 127         if ($clean) {
 128             $spool->markAllAsRead();
 129         }
 130         $spool->updateUnread($since);
 131         return $spool;
 132     }
 133
 134     public static function getPath($file = null)
 135     {
 136         $path = Banana::$spool_root . '/' . Banana::$protocole->name() . '/' . Banana::$protocole->filename();
 137         if (!is_dir($path)) {
 138             if (file_exists($path)) {
 139                 @unlink($path);
 140             }
 141             mkdir($path, 0777, true);
 142         }
 143         return $path . '/' . $file;
 144     }
 145
 146     private static function spoolFilename()
 147     {
 148         return BananaSpool::getPath('spool');
 149     }
 150
 151     private static function &readFromFile($group)
 152     {
 153         $spool = null;
 154         $file = BananaSpool::spoolFilename();
 155         if (!file_exists($file)) {
 156             return $spool;
 157         }
 158         $spool =  unserialize(file_get_contents($file));
 159         if ($spool->version != BANANA_SPOOL_VERSION || $spool->mode != Banana::SPOOL_ALL) {
 160             $spool = null;
 161             return $spool;
 162         }
 163         $spool->markAllAsRead();
 164         return $spool;
 165     }
 166
 167     private function compare($a, $b)
 168     {
 169         return ($b->date - $a->date);
 170     }
 171
 172     private function saveToFile()
 173     {
 174         $file = BananaSpool::spoolFilename();
 175
 176         $this->roots = Array();
 177         foreach($this->overview as &$msg) {
 178             if (is_null($msg->parent)) {
 179                 $this->roots[] =& $msg;
 180             }
 181         }
 182         usort($this->roots, array($this, 'compare'));
 183
 184         if ($this->mode == Banana::SPOOL_ALL) {
 185             file_put_contents($file, serialize($this));
 186         }
 187     }
 188
 189     private function build()
 190     {
 191         $threshold = 0;
 192
 193         // Compute the range of indexes
 194         list($msgnum, $first, $last) = Banana::$protocole->getIndexes();
 195         if ($last < $first) {
 196             $threshold = $first + $msgnum - $last;
 197             $threshold = (int)(log($threshold)/log(2));
 198             $threshold = (2 ^ ($threshold + 1)) - 1;
 199         }
 200         if (Banana::$spool_max && Banana::$spool_max < $msgnum) {
 201             $first = $last - Banana::$spool_max;
 202             if ($first < 0) {
 203                 $first += $threshold;
 204             }
 205         }
 206         $clean  = $this->clean($first, $last, $msgnum);
 207         $update = $this->update($first, $last, $msgnum);
 208
 209         if ($clean || $update) {
 210             $this->saveToFile();
 211         }
 212     }
 213
 214     private function clean(&$first, &$last, $msgnum)
 215     {
 216         $do_save = false;
 217         if (!empty($this->overview)) {
 218             $mids = array_keys($this->overview);
 219             foreach ($mids as $id) {
 220                 if (($first <= $last && ($id < $first || $id > $last))
 221                         || ($first > $last && $id < $first && $id > $last)) {
 222                     $this->delid($id, false);
 223                     $do_save = true;
 224                 }
 225             }
 226             if (!empty($this->overview)) {
 227                 $first = max(array_keys($this->overview))+1;
 228             }
 229         }
 230         return $do_save;
 231     }
 232
 233     private function update(&$first, &$last, $msgnum)
 234     {
 235         if ($first > $last || !$msgnum) {
 236             return false;
 237         }
 238
 239         $messages =& Banana::$protocole->getMessageHeaders($first, $last,
 240             array('Date', 'Subject', 'From', 'Message-ID', 'References', 'In-Reply-To'));
 241
 242         // Build all the new Spool Heads
 243         $time = time();
 244         foreach ($messages as $id=>&$message) {
 245             if (!isset($this->overview[$id])) {
 246                 $this->overview[$id] = new BananaSpoolHead($id, $message);
 247                 $head =& $this->overview[$id];
 248                 $this->ids[$head->msgid] =& $head;
 249                 $head->time = $time;
 250             }
 251         }
 252
 253         // Build tree
 254         $null = null;
 255         foreach ($messages as $id=>&$message) {
 256             $msg         =& $this->overview[$id];
 257             $parents     =& $this->getReferences($message);
 258             while (!empty($parents) && ($msg->parent === $msg || is_null($msg->parent))) {
 259                 @$msg->parent =& array_pop($parents);
 260             }
 261
 262             if (!is_null($msg->parent)) {
 263                 $parent =& $msg->parent;
 264                 $parent->children[] =& $msg;
 265                 while (!is_null($parent)) {
 266                     $parent->desc += $msg->desc;
 267                     $parent->time  = $time;
 268                     $prev =& $parent;
 269                     if ($parent !== $parent->parent) {
 270                         $parent =& $parent->parent;
 271                     } else {
 272                         $parent =& $null;
 273                     }
 274                 }
 275             }
 276         }
 277         Banana::$protocole->updateSpool($messages);
 278         return true;
 279     }
 280
 281     public function updateUnread($since)
 282     {
 283         if (empty($since)) {
 284             return;
 285         }
 286
 287         $newpostsids = Banana::$protocole->getNewIndexes($since);
 288
 289         if (empty($newpostsids)) {
 290             return;
 291         }
 292
 293         $newpostsids = array_intersect($newpostsids, array_keys($this->ids));
 294         foreach ($newpostsids as $mid) {
 295             $overview =& $this->ids[$mid];
 296             if ($overview->isread) {
 297                 $overview->isread = false;
 298                 while (!is_null($overview)) {
 299                     $overview->descunread++;
 300                     $overview =& $overview->parent;
 301                 }
 302             }
 303         }
 304         $this->unreadnb += count($newpostsids);
 305     }
 306
 307     public function setMode($mode)
 308     {
 309         $this->mode = $mode;
 310         switch ($mode) {
 311           case Banana::SPOOL_UNREAD:
 312             $num = max(array_keys($this->overview));
 313             if ($this->overview[$num]->isread) {
 314                 break;
 315             }
 316             foreach ($this->roots as &$root) {
 317                 if ($root->descunread == 0) {
 318                     $this->killdesc($root->id);
 319                 }
 320             }
 321             break;
 322         }
 323     }
 324
 325     /** Fetch list of references
 326      */
 327     public function &getReferences(array &$refs)
 328     {
 329         $references = array();
 330         if (isset($refs['references'])) {
 331             $text = preg_split('/\s/', str_replace('><', '> <', $refs['references']));
 332             foreach ($text as $id=>&$value) {
 333                 if (isset($this->ids[$value])) {
 334                     $references[] =& $this->ids[$value];
 335                 }
 336             }
 337         } elseif (isset($refs['in-reply-to']) && isset($this->ids[$refs['in-reply-to']])) {
 338             $references[] =& $this->ids[$refs['in-reply-to']];
 339         }
 340         return $references;
 341     }
 342
 343     /** Get the tree associated to a given id
 344      */
 345     public function &getTree($id)
 346     {
 347         return BananaTree::build($id)->show();
 348     }
 349
 350     /** Mark the given id as read
 351      * @param id MSGNUM of post
 352      */
 353     public function markAsRead($id)
 354     {
 355         $overview =& $this->overview[$id];
 356         if (!$overview->isread) {
 357             $overview->isread = true;
 358             $this->unreadnb--;
 359             while (!is_null($overview)) {
 360                 $overview->descunread--;
 361                 $overview =& $overview->parent;
 362             }
 363         }
 364     }
 365
 366     /** Mark all unread messages as read
 367      */
 368     public function markAllAsRead(array &$array = null)
 369     {
 370         if (!$this->unreadnb) {
 371             return;
 372         }
 373         if (is_null($array) && !empty($this->roots)) {
 374             $array =& $this->roots;
 375         } elseif (is_null($array)) {
 376             return;
 377         }
 378         foreach ($array as &$msg) {
 379             if (!$msg->isread) {
 380                 $this->markAsRead($msg->id);
 381                 if (!$this->unreadnb) {
 382                     return;
 383                 }
 384             }
 385             if ($msg->descunread) {
 386                 $this->markAllAsRead($msg->children);
 387             }
 388         }
 389     }
 390
 391     /** kill post and childrens
 392      * @param $_id MSGNUM of post
 393      */
 394     private function killdesc($_id)
 395     {
 396         $overview =& $this->overview[$_id];
 397         $children =& $overview->children;
 398         if (sizeof($children)) {
 399             foreach ($children as &$c) {
 400                 $this->killdesc($c->id);
 401             }
 402         }
 403         unset($this->overview[$_id]);
 404         foreach ($this->roots as $k=>&$root) {
 405             if ($root === $overview) {
 406                 unset($this->roots[$k]);
 407                 break;
 408             }
 409         }
 410         unset($this->ids[$overview->msgid]);
 411         $overview = null;
 412     }
 413
 414     /** delete a post from overview
 415      * @param $_id MSGNUM of post
 416      */
 417     public function delid($_id, $write = true)
 418     {
 419         if (!isset($this->overview[$_id])) {
 420             return;
 421         }
 422
 423         $overview =& $this->overview[$_id];
 424         // Be sure it is not counted as unread
 425         if (!$overview->isread) {
 426             $this->markAsRead($_id);
 427         }
 428
 429         $parent =& $overview->parent;
 430
 431         // Remove from the message tree
 432         if (!is_null($parent)) {
 433             $time = time();
 434             foreach ($parent->children as $key=>&$child) {
 435                 if ($child === $overview) {
 436                     unset($parent->children[$key]);
 437                     break;
 438                 }
 439             }
 440             if (sizeof($overview->children)) {
 441                 foreach ($overview->children as &$child) {
 442                     $parent->children[] =& $child;
 443                     $child->time   = $time;
 444                     $child->parent =& $parent;
 445                 }
 446             }
 447             while (!is_null($parent)) {
 448                 $parent->desc--;
 449                 $parent->time = $time;
 450                 $parent =& $parent->parent;
 451             }
 452         }
 453
 454         // Remove all refenrences and assign null to the object
 455         unset($this->ids[$overview->msgid]);
 456         unset($this->overview[$_id]);
 457         BananaTree::kill($_id);
 458         foreach ($this->roots as $k=>&$root) {
 459             if ($root === $overview) {
 460                 unset($this->roots[$k]);
 461                 break;
 462             }
 463         }
 464         $overview = null;
 465
 466         if ($write) {
 467             $this->saveToFile();
 468         }
 469     }
 470
 471     public function formatDate(BananaSpoolHead $head)
 472     {
 473         $stamp  = $head->date;
 474         $today  = intval(time() / (24*3600));
 475         $dday   = intval($stamp / (24*3600));
 476
 477         if ($today == $dday) {
 478             $format = "%H:%M";
 479         } elseif ($today == 1 + $dday) {
 480             $format = _b_('hier')." %H:%M";
 481         } elseif ($today < 7 + $dday) {
 482             $format = '%a %H:%M';
 483         } elseif ($today < 90 + $dday) {
 484             $format = '%a %e %b';
 485         } else {
 486             $format = '%a %e %b %Y';
 487         }
 488         return strftime($format, $stamp);
 489     }
 490
 491     public function formatSubject(BananaSpoolHead $head)
 492     {
 493         $subject = $popup = $head->subject;
 494         $popup = $subject;
 495         if (function_exists('hook_formatDisplayHeader')) {
 496             list($subject, $link) = hook_formatDisplayHeader('subject', $subject, true);
 497         } else {
 498             $subject = banana_catchFormats(banana_entities(stripslashes($subject)));
 499             $link = null;
 500         }
 501         if (empty($subject)) {
 502             $subject = _b_('(pas de sujet)');
 503         }
 504         if (strlen($subject) > 100) {
 505             $subject = substr($subject, 0, 99) . '&hellip;';
 506         }
 507         if ($head->id !== Banana::$artid) {
 508             $subject = Banana::$page->makeLink(Array('group' => $this->group, 'artid' => $head->id,
 509                                                      'text'  => $subject, 'popup' => $popup));
 510         }
 511         return $subject . $link;
 512     }
 513
 514     public function formatFrom(BananaSpoolHead $head)
 515     {
 516         return BananaMessage::formatFrom($head->from);
 517     }
 518
 519     public function start()
 520     {
 521         if (Banana::$first) {
 522             return Banana::$first;
 523         } else {
 524             $first = array_search(Banana::$artid, $this->roots);
 525             return max(0, $first - Banana::$spool_tbefore);
 526         }
 527     }
 528
 529     public function context()
 530     {
 531         return Banana::$first ? Banana::$spool_tmax : Banana::$spool_tcontext;
 532     }
 533
 534     /** Return root message of the given thread
 535      * @param id INTEGER id of a message
 536      */
 537     public function &root($id)
 538     {
 539         $parent =& $this->overview[$id];
 540         while (!is_null($parent->parent)) {
 541             $parent =& $parent->parent;
 542         }
 543         return $parent;
 544     }
 545
 546     /** Return the last post id with the given subject
 547      * @param subject
 548      */
 549     public function getPostId($subject)
 550     {
 551         $subject = trim($subject);
 552         $id = max(array_keys($this->overview));
 553         while (isset($this->overview[$id])) {
 554             $test = $this->overview[$id]->subject;
 555             if (function_exists('hook_formatDisplayHeader')) {
 556                 $val = hook_formatDisplayHeader('subject', $test, true);
 557                 if (is_array($val)) {
 558                     $test = banana_html_entity_decode($val[0]);
 559                 } else {
 560                     $test = banana_html_entity_decode($val);
 561                 }
 562             }
 563             $test = trim($test);
 564             if ($test == $subject) {
 565                 return $id;
 566             }
 567             $id--;
 568         }
 569         return -1;
 570     }
 571
 572     /** Returns previous thread root index
 573      * @param id INTEGER message number
 574      */
 575     public function prevThread($id)
 576     {
 577         $root =& $this->root($id);
 578         $last = null;
 579         foreach ($this->roots as &$i) {
 580             if ($i === $root) {
 581                 return $last;
 582             }
 583             $last = $i->id;
 584         }
 585         return $last;
 586     }
 587
 588     /** Returns next thread root index
 589      * @param id INTEGER message number
 590      */
 591     public function nextThread($id)
 592     {
 593         $root =& $this->root($id);
 594         $ok   = false;
 595         foreach ($this->roots as &$i) {
 596             if ($ok) {
 597                 return $i->id;
 598             }
 599             if ($i === $root) {
 600                 $ok = true;
 601             }
 602         }
 603         return null;
 604     }
 605
 606     /** Return prev post in the thread
 607      * @param id INTEGER message number
 608      */
 609     public function prevPost($id)
 610     {
 611         $parent =& $this->overview[$id]->parent;
 612         if (is_null($parent)) {
 613             return null;
 614         }
 615         $last = $parent->id;
 616         foreach ($parent->children as &$child) {
 617             if ($child->id == $id) {
 618                 return $last;
 619             }
 620             $last = $child->id;
 621         }
 622         return null;
 623     }
 624
 625     /** Return next post in the thread
 626      * @param id INTEGER message number
 627      */
 628     public function nextPost($id)
 629     {
 630         $cur =& $this->overview[$id];
 631         if (count($cur->children) != 0) {
 632             return $cur->children[0]->id;
 633         }
 634
 635         $parent =& $cur;
 636         while (true) {
 637             $parent =& $cur->parent;
 638             if (is_null($parent)) {
 639                 return null;
 640             }
 641             $ok = false;
 642             foreach ($parent->children as &$child) {
 643                 if ($ok) {
 644                     return $child->id;
 645                 }
 646                 if ($child === $cur) {
 647                     $ok = true;
 648                 }
 649             }
 650             $cur =& $parent;
 651         }
 652         return null;
 653     }
 654
 655     /** Look for an unread message in the thread rooted by the message
 656      * @param id INTEGER message number
 657      */
 658     private function _nextUnread(BananaSpoolHead $cur)
 659     {
 660         if (!$cur->isread) {
 661             return $cur->id;
 662         }
 663         foreach ($cur->children as &$child) {
 664             $unread = $this->_nextUnread($child);
 665             if (!is_null($unread)) {
 666                 return $unread;
 667             }
 668         }
 669         return null;
 670     }
 671
 672     /** Find next unread message
 673      * @param id INTEGER message number
 674      */
 675     public function nextUnread($id = null)
 676     {
 677         if (!$this->unreadnb) {
 678             return null;
 679         }
 680
 681         if (!is_null($id)) {
 682             // Look in message children
 683             foreach ($this->overview[$id]->children as &$child) {
 684                 $next = $this->_nextUnread($child);
 685                 if (!is_null($next)) {
 686                     return $next;
 687                 }
 688             }
 689         }
 690
 691         // Look in current thread
 692         if (is_null($id)) {
 693             $cur = null;
 694         } else {
 695             $cur =& $this->overview[$id];
 696         }
 697         do {
 698             if (is_null($cur)) {
 699                 $parent =& $cur;
 700                 $ok     = true;
 701             } else {
 702                 $parent =& $cur->parent;
 703                 $ok     =  false;
 704             }
 705             if (!is_null($parent)) {
 706                 $array =& $parent->children;
 707             } else {
 708                 $array =& $this->roots;
 709             }
 710             foreach ($array as &$child) {
 711                 if ($ok) {
 712                     $next = $this->_nextUnread($child);
 713                     if (!is_null($next)) {
 714                         return $next;
 715                     }
 716                 }
 717                 if ($child === $cur) {
 718                     $ok = true;
 719                 }
 720             }
 721             $cur =& $parent;
 722         } while(!is_null($cur));
 723         return null;
 724     }
 725 }
 726 // vim:set et sw=4 sts=4 ts=4 enc=utf-8:
 727 ?>