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             foreach ($parent->children as $key=>&$child) {
 434                 if ($child === $overview) {
 435                     unset($parent->children[$key]);
 436                     break;
 437                 }
 438             }
 439             if (sizeof($overview->children)) {
 440                 $parent->children = array_merge($parent->children, $overview->children);
 441                 foreach ($overview->children as &$child) {
 442                     $child->parent =& $parent;
 443                 }
 444             }
 445             $time = time();
 446             while (!is_null($parent)) {
 447                 $parent->desc--;
 448                 $parent->time = $time;
 449                 $parent =& $parent->parent;
 450             }
 451         }
 452
 453         // Remove all refenrences and assign null to the object
 454         unset($this->ids[$overview->msgid]);
 455         unset($this->overview[$_id]);
 456         BananaTree::kill($_id);
 457         foreach ($this->roots as $k=>&$root) {
 458             if ($root === $overview) {
 459                 unset($this->roots[$k]);
 460                 break;
 461             }
 462         }
 463         $overview = null;
 464
 465         if ($write) {
 466             $this->saveToFile();
 467         }
 468     }
 469
 470     public function formatDate(BananaSpoolHead &$head)
 471     {
 472         $stamp  = $head->date;
 473         $today  = intval(time() / (24*3600));
 474         $dday   = intval($stamp / (24*3600));
 475
 476         if ($today == $dday) {
 477             $format = "%H:%M";
 478         } elseif ($today == 1 + $dday) {
 479             $format = _b_('hier')." %H:%M";
 480         } elseif ($today < 7 + $dday) {
 481             $format = '%a %H:%M';
 482         } elseif ($today < 90 + $dday) {
 483             $format = '%a %e %b';
 484         } else {
 485             $format = '%a %e %b %Y';
 486         }
 487         return strftime($format, $stamp);
 488     }
 489
 490     public function formatSubject(BananaSpoolHead &$head)
 491     {
 492         $subject = $popup = $head->subject;
 493         $popup = $subject;
 494         if (function_exists('hook_formatDisplayHeader')) {
 495             list($subject, $link) = hook_formatDisplayHeader('subject', $subject, true);
 496         } else {
 497             $subject = banana_catchFormats(banana_entities(stripslashes($subject)));
 498             $link = null;
 499         }
 500         if (empty($subject)) {
 501             $subject = _b_('(pas de sujet)');
 502         }
 503         if ($head->id != Banana::$artid) {
 504             $subject = Banana::$page->makeLink(Array('group' => $this->group, 'artid' => $head->id,
 505                                                      'text'  => $subject, 'popup' => $popup));
 506         }
 507         return $subject . $link;
 508     }
 509
 510     public function formatFrom(BananaSpoolHead &$head)
 511     {
 512         return BananaMessage::formatFrom($head->from);
 513     }
 514
 515     public function start()
 516     {
 517         if (Banana::$first) {
 518             return Banana::$first;
 519         } else {
 520             $first = array_search(Banana::$artid, $this->roots);
 521             return max(0, $first - Banana::$spool_tbefore);
 522         }
 523     }
 524
 525     public function context()
 526     {
 527         return Banana::$first ? Banana::$spool_tmax : Banana::$spool_tcontext;
 528     }
 529
 530
 531     /** computes linear post index
 532      * @param $_id INTEGER MSGNUM of post
 533      * @return INTEGER linear index of post
 534      */
 535     public function getNdX($_id)
 536     {
 537         $ndx    = 1;
 538         $id_cur = $_id;
 539         while (true) {
 540             $id_parent = $this->overview[$id_cur]->parent;
 541             if (is_null($id_parent)) break;
 542             $pos = array_search($id_cur, $this->overview[$id_parent]->children);
 543
 544             for ($i = 0; $i < $pos ; $i++) {
 545                 $ndx += $this->overview[$this->overview[$id_parent]->children[$i]]->desc;
 546             }
 547             $ndx++; //noeud père
 548
 549             $id_cur = $id_parent;
 550         }
 551
 552         foreach ($this->roots as $i) {
 553             if ($i==$id_cur) {
 554                 break;
 555             }
 556             $ndx += $this->overview[$i]->desc;
 557         }
 558         return $ndx;
 559     }
 560
 561     /** Return root message of the given thread
 562      * @param id INTEGER id of a message
 563      */
 564     public function &root($id)
 565     {
 566         $parent =& $this->overview[$id];
 567         while (!is_null($parent->parent)) {
 568             $parent =& $parent->parent;
 569         }
 570         return $parent;
 571     }
 572
 573     /** Return the last post id with the given subject
 574      * @param subject
 575      */
 576     public function getPostId($subject)
 577     {
 578         $subject = trim($subject);
 579         $id = max(array_keys($this->overview));
 580         while (isset($this->overview[$id])) {
 581             $test = $this->overview[$id]->subject;
 582             if (function_exists('hook_formatDisplayHeader')) {
 583                 $val = hook_formatDisplayHeader('subject', $test, true);
 584                 if (is_array($val)) {
 585                     $test = banana_html_entity_decode($val[0]);
 586                 } else {
 587                     $test = banana_html_entity_decode($val);
 588                 }
 589             }
 590             $test = trim($test);
 591             if ($test == $subject) {
 592                 return $id;
 593             }
 594             $id--;
 595         }
 596         return -1;
 597     }
 598
 599     /** Returns previous thread root index
 600      * @param id INTEGER message number
 601      */
 602     public function prevThread($id)
 603     {
 604         $root =& $this->root($id);
 605         $last = null;
 606         foreach ($this->roots as &$i) {
 607             if ($i === $root) {
 608                 return $last;
 609             }
 610             $last = $i->id;
 611         }
 612         return $last;
 613     }
 614
 615     /** Returns next thread root index
 616      * @param id INTEGER message number
 617      */
 618     public function nextThread($id)
 619     {
 620         $root =& $this->root($id);
 621         $ok   = false;
 622         foreach ($this->roots as &$i) {
 623             if ($ok) {
 624                 return $i->id;
 625             }
 626             if ($i === $root) {
 627                 $ok = true;
 628             }
 629         }
 630         return null;
 631     }
 632
 633     /** Return prev post in the thread
 634      * @param id INTEGER message number
 635      */
 636     public function prevPost($id)
 637     {
 638         $parent =& $this->overview[$id]->parent;
 639         if (is_null($parent)) {
 640             return null;
 641         }
 642         $last = $parent->id;
 643         foreach ($parent->children as &$child) {
 644             if ($child->id == $id) {
 645                 return $last;
 646             }
 647             $last = $child->id;
 648         }
 649         return null;
 650     }
 651
 652     /** Return next post in the thread
 653      * @param id INTEGER message number
 654      */
 655     public function nextPost($id)
 656     {
 657         $cur =& $this->overview[$id];
 658         if (count($cur->children) != 0) {
 659             return $cur->children[0]->id;
 660         }
 661
 662         $parent =& $cur;
 663         while (true) {
 664             $parent =& $cur->parent;
 665             if (is_null($parent)) {
 666                 return null;
 667             }
 668             $ok = false;
 669             foreach ($parent->children as &$child) {
 670                 if ($ok) {
 671                     return $child->id;
 672                 }
 673                 if ($child === $cur) {
 674                     $ok = true;
 675                 }
 676             }
 677             $cur =& $parent;
 678         }
 679         return null;
 680     }
 681
 682     /** Look for an unread message in the thread rooted by the message
 683      * @param id INTEGER message number
 684      */
 685     private function _nextUnread(BananaSpoolHead &$cur)
 686     {
 687         if (!$cur->isread) {
 688             return $cur->id;
 689         }
 690         foreach ($cur->children as &$child) {
 691             $unread = $this->_nextUnread($child);
 692             if (!is_null($unread)) {
 693                 return $unread;
 694             }
 695         }
 696         return null;
 697     }
 698
 699     /** Find next unread message
 700      * @param id INTEGER message number
 701      */
 702     public function nextUnread($id = null)
 703     {
 704         if (!$this->unreadnb) {
 705             return null;
 706         }
 707
 708         if (!is_null($id)) {
 709             // Look in message children
 710             foreach ($this->overview[$id]->children as &$child) {
 711                 $next = $this->_nextUnread($child);
 712                 if (!is_null($next)) {
 713                     return $next;
 714                 }
 715             }
 716         }
 717
 718         // Look in current thread
 719         if (is_null($id)) {
 720             $cur = null;
 721         } else {
 722             $cur =& $this->overview[$id];
 723         }
 724         do {
 725             if (is_null($cur)) {
 726                 $parent =& $cur;
 727                 $ok     = true;
 728             } else {
 729                 $parent =& $cur->parent;
 730                 $ok     =  false;
 731             }
 732             if (!is_null($parent)) {
 733                 $array =& $parent->children;
 734             } else {
 735                 $array =& $this->roots;
 736             }
 737             foreach ($array as &$child) {
 738                 if ($ok) {
 739                     $next = $this->_nextUnread($child);
 740                     if (!is_null($next)) {
 741                         return $next;
 742                     }
 743                 }
 744                 if ($child === $cur) {
 745                     $ok = true;
 746                 }
 747             }
 748             $cur =& $parent;
 749         } while(!is_null($cur));
 750         return null;
 751     }
 752 }
 753 // vim:set et sw=4 sts=4 ts=4 enc=utf-8:
 754 ?>