xref: /dokuwiki/inc/indexer.php (revision ffec100965184c6bf045f0096101768def108b43)
1b4ce25e9SAndreas Gohr<?php
2b4ce25e9SAndreas Gohr/**
3fcd3bb7cSAndreas Gohr * Functions to create the fulltext search index
4b4ce25e9SAndreas Gohr *
5b4ce25e9SAndreas Gohr * @license    GPL 2 (http://www.gnu.org/licenses/gpl.html)
6b4ce25e9SAndreas Gohr * @author     Andreas Gohr <andi@splitbrain.org>
700803e56STom N Harris * @author     Tom N Harris <tnharris@whoopdedo.org>
8b4ce25e9SAndreas Gohr */
9b4ce25e9SAndreas Gohr
10fa8adffeSAndreas Gohrif(!defined('DOKU_INC')) die('meh.');
11b4ce25e9SAndreas Gohr
127c2ef4e8STom N Harris// Version tag used to force rebuild on upgrade
13*ffec1009SMichael Hamanndefine('INDEXER_VERSION', 6);
147c2ef4e8STom N Harris
1533815ce2SChris Smith// set the minimum token length to use in the index (note, this doesn't apply to numeric tokens)
16d3fb3219SAndreas Gohrif (!defined('IDX_MINWORDLENGTH')) define('IDX_MINWORDLENGTH',2);
1733815ce2SChris Smith
1893a60ad2SAndreas Gohr// Asian characters are handled as words. The following regexp defines the
1993a60ad2SAndreas Gohr// Unicode-Ranges for Asian characters
2093a60ad2SAndreas Gohr// Ranges taken from http://en.wikipedia.org/wiki/Unicode_block
2193a60ad2SAndreas Gohr// I'm no language expert. If you think some ranges are wrongly chosen or
2293a60ad2SAndreas Gohr// a range is missing, please contact me
23d5b23302STom N Harrisdefine('IDX_ASIAN1','[\x{0E00}-\x{0E7F}]'); // Thai
24d5b23302STom N Harrisdefine('IDX_ASIAN2','['.
25d5b23302STom N Harris                   '\x{2E80}-\x{3040}'.  // CJK -> Hangul
26d5b23302STom N Harris                   '\x{309D}-\x{30A0}'.
27a0c5c349STom N Harris                   '\x{30FD}-\x{31EF}\x{3200}-\x{D7AF}'.
2893a60ad2SAndreas Gohr                   '\x{F900}-\x{FAFF}'.  // CJK Compatibility Ideographs
2993a60ad2SAndreas Gohr                   '\x{FE30}-\x{FE4F}'.  // CJK Compatibility Forms
30ae6c4ec0SDanny Lin                   "\xF0\xA0\x80\x80-\xF0\xAA\x9B\x9F". // CJK Extension B
31ae6c4ec0SDanny Lin                   "\xF0\xAA\x9C\x80-\xF0\xAB\x9C\xBF". // CJK Extension C
32ae6c4ec0SDanny Lin                   "\xF0\xAB\x9D\x80-\xF0\xAB\xA0\x9F". // CJK Extension D
33ae6c4ec0SDanny Lin                   "\xF0\xAF\xA0\x80-\xF0\xAF\xAB\xBF". // CJK Compatibility Supplement
3493a60ad2SAndreas Gohr                   ']');
35d5b23302STom N Harrisdefine('IDX_ASIAN3','['.                // Hiragana/Katakana (can be two characters)
36d5b23302STom N Harris                   '\x{3042}\x{3044}\x{3046}\x{3048}'.
37d5b23302STom N Harris                   '\x{304A}-\x{3062}\x{3064}-\x{3082}'.
38d5b23302STom N Harris                   '\x{3084}\x{3086}\x{3088}-\x{308D}'.
39d5b23302STom N Harris                   '\x{308F}-\x{3094}'.
40d5b23302STom N Harris                   '\x{30A2}\x{30A4}\x{30A6}\x{30A8}'.
41d5b23302STom N Harris                   '\x{30AA}-\x{30C2}\x{30C4}-\x{30E2}'.
42d5b23302STom N Harris                   '\x{30E4}\x{30E6}\x{30E8}-\x{30ED}'.
43d5b23302STom N Harris                   '\x{30EF}-\x{30F4}\x{30F7}-\x{30FA}'.
44d5b23302STom N Harris                   ']['.
45d5b23302STom N Harris                   '\x{3041}\x{3043}\x{3045}\x{3047}\x{3049}'.
46d5b23302STom N Harris                   '\x{3063}\x{3083}\x{3085}\x{3087}\x{308E}\x{3095}-\x{309C}'.
47d5b23302STom N Harris                   '\x{30A1}\x{30A3}\x{30A5}\x{30A7}\x{30A9}'.
48d5b23302STom N Harris                   '\x{30C3}\x{30E3}\x{30E5}\x{30E7}\x{30EE}\x{30F5}\x{30F6}\x{30FB}\x{30FC}'.
49d5b23302STom N Harris                   '\x{31F0}-\x{31FF}'.
50d5b23302STom N Harris                   ']?');
51699b8a0bSAndreas Gohrdefine('IDX_ASIAN', '(?:'.IDX_ASIAN1.'|'.IDX_ASIAN2.'|'.IDX_ASIAN3.')');
5293a60ad2SAndreas Gohr
53b4ce25e9SAndreas Gohr/**
547c2ef4e8STom N Harris * Version of the indexer taking into consideration the external tokenizer.
557c2ef4e8STom N Harris * The indexer is only compatible with data written by the same version.
567c2ef4e8STom N Harris *
578cd4c12fSAndreas Gohr * @triggers INDEXER_VERSION_GET
58d0d6fe1bSTom N Harris * Plugins that modify what gets indexed should hook this event and
59d0d6fe1bSTom N Harris * add their version info to the event data like so:
60d0d6fe1bSTom N Harris *     $data[$plugin_name] = $plugin_version;
61d0d6fe1bSTom N Harris *
627c2ef4e8STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
638605afb1SMichael Hamann * @author Michael Hamann <michael@content-space.de>
647c2ef4e8STom N Harris */
657c2ef4e8STom N Harrisfunction idx_get_version(){
66d0d6fe1bSTom N Harris    static $indexer_version = null;
67d0d6fe1bSTom N Harris    if ($indexer_version == null) {
688605afb1SMichael Hamann        $version = INDEXER_VERSION;
698605afb1SMichael Hamann
70d0d6fe1bSTom N Harris        // DokuWiki version is included for the convenience of plugins
71d0d6fe1bSTom N Harris        $data = array('dokuwiki'=>$version);
728605afb1SMichael Hamann        trigger_event('INDEXER_VERSION_GET', $data, null, false);
73d0d6fe1bSTom N Harris        unset($data['dokuwiki']); // this needs to be first
74d0d6fe1bSTom N Harris        ksort($data);
75d0d6fe1bSTom N Harris        foreach ($data as $plugin=>$vers)
76d0d6fe1bSTom N Harris            $version .= '+'.$plugin.'='.$vers;
77d0d6fe1bSTom N Harris        $indexer_version = $version;
78d0d6fe1bSTom N Harris    }
79d0d6fe1bSTom N Harris    return $indexer_version;
807c2ef4e8STom N Harris}
817c2ef4e8STom N Harris
827c2ef4e8STom N Harris/**
83d5b23302STom N Harris * Measure the length of a string.
84d5b23302STom N Harris * Differs from strlen in handling of asian characters.
85d5b23302STom N Harris *
86d5b23302STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
87d5b23302STom N Harris */
88d5b23302STom N Harrisfunction wordlen($w){
89d5b23302STom N Harris    $l = strlen($w);
90d5b23302STom N Harris    // If left alone, all chinese "words" will get put into w3.idx
91d5b23302STom N Harris    // So the "length" of a "word" is faked
924b9792c6STom N Harris    if(preg_match_all('/[\xE2-\xEF]/',$w,$leadbytes)) {
934b9792c6STom N Harris        foreach($leadbytes[0] as $b)
944b9792c6STom N Harris            $l += ord($b) - 0xE1;
954b9792c6STom N Harris    }
96d5b23302STom N Harris    return $l;
97d5b23302STom N Harris}
98d5b23302STom N Harris
99d5b23302STom N Harris/**
10000803e56STom N Harris * Class that encapsulates operations on the indexer database.
101579b0f7eSTNHarris *
102579b0f7eSTNHarris * @author Tom N Harris <tnharris@whoopdedo.org>
103579b0f7eSTNHarris */
10400803e56STom N Harrisclass Doku_Indexer {
1053d2ce006SMichael Hamann    /**
1063d2ce006SMichael Hamann     * @var array $pidCache Cache for getPID()
1073d2ce006SMichael Hamann     */
1083d2ce006SMichael Hamann    protected $pidCache = array();
109579b0f7eSTNHarris
110579b0f7eSTNHarris    /**
11100803e56STom N Harris     * Adds the contents of a page to the fulltext index
112dd35e9c9SAndreas Gohr     *
11300803e56STom N Harris     * The added text replaces previous words for the same page.
11400803e56STom N Harris     * An empty value erases the page.
11500803e56STom N Harris     *
11600803e56STom N Harris     * @param string    $page   a page name
11700803e56STom N Harris     * @param string    $text   the body of the page
11800803e56STom N Harris     * @return boolean          the function completed successfully
11900803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
120dd35e9c9SAndreas Gohr     * @author Andreas Gohr <andi@splitbrain.org>
121dd35e9c9SAndreas Gohr     */
12200803e56STom N Harris    public function addPageWords($page, $text) {
12380fb93f6STom N Harris        if (!$this->lock())
1249b41be24STom N Harris            return "locked";
12500803e56STom N Harris
12600803e56STom N Harris        // load known documents
12703aafe1cSMichael Hamann        $pid = $this->getPIDNoLock($page);
1285981eb09STom N Harris        if ($pid === false) {
12980fb93f6STom N Harris            $this->unlock();
13000803e56STom N Harris            return false;
13100803e56STom N Harris        }
13200803e56STom N Harris
13300803e56STom N Harris        $pagewords = array();
13400803e56STom N Harris        // get word usage in page
13580fb93f6STom N Harris        $words = $this->getPageWords($text);
13600803e56STom N Harris        if ($words === false) {
13780fb93f6STom N Harris            $this->unlock();
13800803e56STom N Harris            return false;
13900803e56STom N Harris        }
14000803e56STom N Harris
14100803e56STom N Harris        if (!empty($words)) {
14200803e56STom N Harris            foreach (array_keys($words) as $wlen) {
14380fb93f6STom N Harris                $index = $this->getIndex('i', $wlen);
14400803e56STom N Harris                foreach ($words[$wlen] as $wid => $freq) {
14500803e56STom N Harris                    $idx = ($wid<count($index)) ? $index[$wid] : '';
14680fb93f6STom N Harris                    $index[$wid] = $this->updateTuple($idx, $pid, $freq);
14700803e56STom N Harris                    $pagewords[] = "$wlen*$wid";
14800803e56STom N Harris                }
14980fb93f6STom N Harris                if (!$this->saveIndex('i', $wlen, $index)) {
15080fb93f6STom N Harris                    $this->unlock();
15100803e56STom N Harris                    return false;
15200803e56STom N Harris                }
15300803e56STom N Harris            }
15400803e56STom N Harris        }
15500803e56STom N Harris
15600803e56STom N Harris        // Remove obsolete index entries
15780fb93f6STom N Harris        $pageword_idx = $this->getIndexKey('pageword', '', $pid);
15800803e56STom N Harris        if ($pageword_idx !== '') {
15900803e56STom N Harris            $oldwords = explode(':',$pageword_idx);
16000803e56STom N Harris            $delwords = array_diff($oldwords, $pagewords);
16100803e56STom N Harris            $upwords = array();
16200803e56STom N Harris            foreach ($delwords as $word) {
16300803e56STom N Harris                if ($word != '') {
16400803e56STom N Harris                    list($wlen,$wid) = explode('*', $word);
16500803e56STom N Harris                    $wid = (int)$wid;
16600803e56STom N Harris                    $upwords[$wlen][] = $wid;
16700803e56STom N Harris                }
16800803e56STom N Harris            }
16900803e56STom N Harris            foreach ($upwords as $wlen => $widx) {
17080fb93f6STom N Harris                $index = $this->getIndex('i', $wlen);
17100803e56STom N Harris                foreach ($widx as $wid) {
17280fb93f6STom N Harris                    $index[$wid] = $this->updateTuple($index[$wid], $pid, 0);
17300803e56STom N Harris                }
17480fb93f6STom N Harris                $this->saveIndex('i', $wlen, $index);
17500803e56STom N Harris            }
17600803e56STom N Harris        }
17700803e56STom N Harris        // Save the reverse index
17800803e56STom N Harris        $pageword_idx = join(':', $pagewords);
17980fb93f6STom N Harris        if (!$this->saveIndexKey('pageword', '', $pid, $pageword_idx)) {
18080fb93f6STom N Harris            $this->unlock();
18100803e56STom N Harris            return false;
18200803e56STom N Harris        }
18300803e56STom N Harris
18480fb93f6STom N Harris        $this->unlock();
185dd35e9c9SAndreas Gohr        return true;
186dd35e9c9SAndreas Gohr    }
187dd35e9c9SAndreas Gohr
188dd35e9c9SAndreas Gohr    /**
18900803e56STom N Harris     * Split the words in a page and add them to the index.
19044ca0adfSAndreas Gohr     *
191b9d8cc1eSTom N Harris     * @param string    $text   content of the page
192b9d8cc1eSTom N Harris     * @return array            list of word IDs and number of times used
19344ca0adfSAndreas Gohr     * @author Andreas Gohr <andi@splitbrain.org>
19417f42b01SChris Smith     * @author Christopher Smith <chris@jalakai.co.uk>
19500803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
196b4ce25e9SAndreas Gohr     */
19780fb93f6STom N Harris    protected function getPageWords($text) {
19844ca0adfSAndreas Gohr
19900803e56STom N Harris        $tokens = $this->tokenizer($text);
20017f42b01SChris Smith        $tokens = array_count_values($tokens);  // count the frequency of each token
20117f42b01SChris Smith
20217f42b01SChris Smith        $words = array();
2034e1bf408STom N Harris        foreach ($tokens as $w=>$c) {
204d5b23302STom N Harris            $l = wordlen($w);
205579b0f7eSTNHarris            if (isset($words[$l])){
2064e1bf408STom N Harris                $words[$l][$w] = $c + (isset($words[$l][$w]) ? $words[$l][$w] : 0);
20717f42b01SChris Smith            }else{
2084e1bf408STom N Harris                $words[$l] = array($w => $c);
20917f42b01SChris Smith            }
21017f42b01SChris Smith        }
21117f42b01SChris Smith
212579b0f7eSTNHarris        // arrive here with $words = array(wordlen => array(word => frequency))
213e5e50383SMichael Hamann        $word_idx_modified = false;
214b4ce25e9SAndreas Gohr        $index = array();   //resulting index
215579b0f7eSTNHarris        foreach (array_keys($words) as $wlen) {
21680fb93f6STom N Harris            $word_idx = $this->getIndex('w', $wlen);
217579b0f7eSTNHarris            foreach ($words[$wlen] as $word => $freq) {
21800803e56STom N Harris                $wid = array_search($word, $word_idx);
21900803e56STom N Harris                if ($wid === false) {
220d5b23302STom N Harris                    $wid = count($word_idx);
22100803e56STom N Harris                    $word_idx[] = $word;
222e5e50383SMichael Hamann                    $word_idx_modified = true;
223b4ce25e9SAndreas Gohr                }
224579b0f7eSTNHarris                if (!isset($index[$wlen]))
225579b0f7eSTNHarris                    $index[$wlen] = array();
226579b0f7eSTNHarris                $index[$wlen][$wid] = $freq;
22744ca0adfSAndreas Gohr            }
22800803e56STom N Harris            // save back the word index
22980fb93f6STom N Harris            if ($word_idx_modified && !$this->saveIndex('w', $wlen, $word_idx))
23044ca0adfSAndreas Gohr                return false;
23144ca0adfSAndreas Gohr        }
232b4ce25e9SAndreas Gohr
233b4ce25e9SAndreas Gohr        return $index;
234b4ce25e9SAndreas Gohr    }
235b4ce25e9SAndreas Gohr
23644ca0adfSAndreas Gohr    /**
237e1e1a7e0SMichael Hamann     * Add/update keys to/of the metadata index.
23844ca0adfSAndreas Gohr     *
23900803e56STom N Harris     * Adding new keys does not remove other keys for the page.
24000803e56STom N Harris     * An empty value will erase the key.
24100803e56STom N Harris     * The $key parameter can be an array to add multiple keys. $value will
24200803e56STom N Harris     * not be used if $key is an array.
24344ca0adfSAndreas Gohr     *
24400803e56STom N Harris     * @param string    $page   a page name
24500803e56STom N Harris     * @param mixed     $key    a key string or array of key=>value pairs
24600803e56STom N Harris     * @param mixed     $value  the value or list of values
24700803e56STom N Harris     * @return boolean          the function completed successfully
24800803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
249e1e1a7e0SMichael Hamann     * @author Michael Hamann <michael@content-space.de>
25044ca0adfSAndreas Gohr     */
25100803e56STom N Harris    public function addMetaKeys($page, $key, $value=null) {
25200803e56STom N Harris        if (!is_array($key)) {
25300803e56STom N Harris            $key = array($key => $value);
25400803e56STom N Harris        } elseif (!is_null($value)) {
25500803e56STom N Harris            // $key is array, but $value is not null
25600803e56STom N Harris            trigger_error("array passed to addMetaKeys but value is not null", E_USER_WARNING);
25700803e56STom N Harris        }
25800803e56STom N Harris
25980fb93f6STom N Harris        if (!$this->lock())
260e1e1a7e0SMichael Hamann            return "locked";
261b4ce25e9SAndreas Gohr
262488dd6ceSAndreas Gohr        // load known documents
26303aafe1cSMichael Hamann        $pid = $this->getPIDNoLock($page);
26400803e56STom N Harris        if ($pid === false) {
26580fb93f6STom N Harris            $this->unlock();
266579b0f7eSTNHarris            return false;
26744ca0adfSAndreas Gohr        }
26800803e56STom N Harris
269c1209673STom N Harris        // Special handling for titles so the index file is simpler
270c1209673STom N Harris        if (array_key_exists('title', $key)) {
271c1209673STom N Harris            $value = $key['title'];
272c1209673STom N Harris            if (is_array($value))
273c1209673STom N Harris                $value = $value[0];
27480fb93f6STom N Harris            $this->saveIndexKey('title', '', $pid, $value);
275c1209673STom N Harris            unset($key['title']);
276c1209673STom N Harris        }
277c1209673STom N Harris
27800803e56STom N Harris        foreach ($key as $name => $values) {
27900803e56STom N Harris            $metaname = idx_cleanName($name);
28080fb93f6STom N Harris            $this->addIndexKey('metadata', '', $metaname);
281b9d8cc1eSTom N Harris            $metaidx = $this->getIndex($metaname.'_i', '');
282b9d8cc1eSTom N Harris            $metawords = $this->getIndex($metaname.'_w', '');
28300803e56STom N Harris            $addwords = false;
284e1e1a7e0SMichael Hamann
285e1e1a7e0SMichael Hamann            if (!is_array($values)) $values = array($values);
286e1e1a7e0SMichael Hamann
287b9d8cc1eSTom N Harris            $val_idx = $this->getIndexKey($metaname.'_p', '', $pid);
288e1e1a7e0SMichael Hamann            if ($val_idx != '') {
289e1e1a7e0SMichael Hamann                $val_idx = explode(':', $val_idx);
290e1e1a7e0SMichael Hamann                // -1 means remove, 0 keep, 1 add
291e1e1a7e0SMichael Hamann                $val_idx = array_combine($val_idx, array_fill(0, count($val_idx), -1));
292e1e1a7e0SMichael Hamann            } else {
293e1e1a7e0SMichael Hamann                $val_idx = array();
294e1e1a7e0SMichael Hamann            }
295e1e1a7e0SMichael Hamann
29600803e56STom N Harris            foreach ($values as $val) {
29700803e56STom N Harris                $val = (string)$val;
29800803e56STom N Harris                if ($val !== "") {
29900803e56STom N Harris                    $id = array_search($val, $metawords);
30000803e56STom N Harris                    if ($id === false) {
30100803e56STom N Harris                        $id = count($metawords);
30200803e56STom N Harris                        $metawords[$id] = $val;
30300803e56STom N Harris                        $addwords = true;
304d5b23302STom N Harris                    }
305e1e1a7e0SMichael Hamann                    // test if value is already in the index
306e1e1a7e0SMichael Hamann                    if (isset($val_idx[$id]) && $val_idx[$id] <= 0)
307e1e1a7e0SMichael Hamann                        $val_idx[$id] = 0;
308e1e1a7e0SMichael Hamann                    else // else add it
309e1e1a7e0SMichael Hamann                        $val_idx[$id] = 1;
31044ca0adfSAndreas Gohr                }
311579b0f7eSTNHarris            }
312e1e1a7e0SMichael Hamann
31300803e56STom N Harris            if ($addwords)
31480fb93f6STom N Harris                $this->saveIndex($metaname.'_w', '', $metawords);
315e1e1a7e0SMichael Hamann            $vals_changed = false;
316e1e1a7e0SMichael Hamann            foreach ($val_idx as $id => $action) {
317e1e1a7e0SMichael Hamann                if ($action == -1) {
31880fb93f6STom N Harris                    $metaidx[$id] = $this->updateTuple($metaidx[$id], $pid, 0);
319e1e1a7e0SMichael Hamann                    $vals_changed = true;
320e1e1a7e0SMichael Hamann                    unset($val_idx[$id]);
321e1e1a7e0SMichael Hamann                } elseif ($action == 1) {
32280fb93f6STom N Harris                    $metaidx[$id] = $this->updateTuple($metaidx[$id], $pid, 1);
323e1e1a7e0SMichael Hamann                    $vals_changed = true;
324e1e1a7e0SMichael Hamann                }
325e1e1a7e0SMichael Hamann            }
326e1e1a7e0SMichael Hamann
327e1e1a7e0SMichael Hamann            if ($vals_changed) {
32880fb93f6STom N Harris                $this->saveIndex($metaname.'_i', '', $metaidx);
329e1e1a7e0SMichael Hamann                $val_idx = implode(':', array_keys($val_idx));
33080fb93f6STom N Harris                $this->saveIndexKey($metaname.'_p', '', $pid, $val_idx);
331b6344591STom N Harris            }
332e1e1a7e0SMichael Hamann
33300803e56STom N Harris            unset($metaidx);
33400803e56STom N Harris            unset($metawords);
335a0c5c349STom N Harris        }
336e1e1a7e0SMichael Hamann
33780fb93f6STom N Harris        $this->unlock();
338579b0f7eSTNHarris        return true;
33944ca0adfSAndreas Gohr    }
34044ca0adfSAndreas Gohr
34144ca0adfSAndreas Gohr    /**
342af73bba6SMichael Hamann     * Rename a page in the search index without changing the indexed content. This function doesn't check if the
343af73bba6SMichael Hamann     * old or new name exists in the filesystem. It returns an error if the old page isn't in the page list of the
344af73bba6SMichael Hamann     * indexer and it deletes all previously indexed content of the new page.
3455eb9e867SMichael Hamann     *
3465eb9e867SMichael Hamann     * @param string $oldpage The old page name
3475eb9e867SMichael Hamann     * @param string $newpage The new page name
3485eb9e867SMichael Hamann     * @return string|bool If the page was successfully renamed, can be a message in the case of an error
3495eb9e867SMichael Hamann     */
3505eb9e867SMichael Hamann    public function renamePage($oldpage, $newpage) {
3515eb9e867SMichael Hamann        if (!$this->lock()) return 'locked';
3525eb9e867SMichael Hamann
3535eb9e867SMichael Hamann        $pages = $this->getPages();
3545eb9e867SMichael Hamann
3555eb9e867SMichael Hamann        $id = array_search($oldpage, $pages);
3565eb9e867SMichael Hamann        if ($id === false) {
3575eb9e867SMichael Hamann            $this->unlock();
3585eb9e867SMichael Hamann            return 'page is not in index';
3595eb9e867SMichael Hamann        }
3605eb9e867SMichael Hamann
3615eb9e867SMichael Hamann        $new_id = array_search($newpage, $pages);
3625eb9e867SMichael Hamann        if ($new_id !== false) {
3635eb9e867SMichael Hamann            // make sure the page is not in the index anymore
364bc27f3e2SMichael Hamann            if ($this->deletePageNoLock($newpage) !== true) {
365bc27f3e2SMichael Hamann                return false;
366bc27f3e2SMichael Hamann            }
3675eb9e867SMichael Hamann
3685eb9e867SMichael Hamann            $pages[$new_id] = 'deleted:'.time().rand(0, 9999);
3695eb9e867SMichael Hamann        }
3705eb9e867SMichael Hamann
3715eb9e867SMichael Hamann        $pages[$id] = $newpage;
3725eb9e867SMichael Hamann
3735eb9e867SMichael Hamann        // update index
3745eb9e867SMichael Hamann        if (!$this->saveIndex('page', '', $pages)) {
3755eb9e867SMichael Hamann            $this->unlock();
3765eb9e867SMichael Hamann            return false;
3775eb9e867SMichael Hamann        }
3785eb9e867SMichael Hamann
3795eb9e867SMichael Hamann        // reset the pid cache
3805eb9e867SMichael Hamann        $this->pidCache = array();
3815eb9e867SMichael Hamann
3825eb9e867SMichael Hamann        $this->unlock();
3835eb9e867SMichael Hamann        return true;
3845eb9e867SMichael Hamann    }
3855eb9e867SMichael Hamann
3865eb9e867SMichael Hamann    /**
3875eb9e867SMichael Hamann     * Renames a meta value in the index. This doesn't change the meta value in the pages, it assumes that all pages
3885eb9e867SMichael Hamann     * will be updated.
3895eb9e867SMichael Hamann     *
3905eb9e867SMichael Hamann     * @param string $key       The metadata key of which a value shall be changed
3915eb9e867SMichael Hamann     * @param string $oldvalue  The old value that shall be renamed
3925eb9e867SMichael Hamann     * @param string $newvalue  The new value to which the old value shall be renamed, can exist (then values will be merged)
3935eb9e867SMichael Hamann     * @return bool|string      If renaming the value has been successful, false or error message on error.
3945eb9e867SMichael Hamann     */
3955eb9e867SMichael Hamann    public function renameMetaValue($key, $oldvalue, $newvalue) {
3965eb9e867SMichael Hamann        if (!$this->lock()) return 'locked';
3975eb9e867SMichael Hamann
3985eb9e867SMichael Hamann        // change the relation references index
3995eb9e867SMichael Hamann        $metavalues = $this->getIndex($key, '_w');
4005eb9e867SMichael Hamann        $oldid = array_search($oldvalue, $metavalues);
4015eb9e867SMichael Hamann        if ($oldid !== false) {
4025eb9e867SMichael Hamann            $newid = array_search($newvalue, $metavalues);
4035eb9e867SMichael Hamann            if ($newid !== false) {
4045eb9e867SMichael Hamann                // free memory
4055eb9e867SMichael Hamann                unset ($metavalues);
4065eb9e867SMichael Hamann
4075eb9e867SMichael Hamann                // okay, now we have two entries for the same value. we need to merge them.
4089e64ca0dSMichael Hamann                $indexline = $this->getIndexKey($key.'_i', '', $oldid);
4095eb9e867SMichael Hamann                if ($indexline != '') {
4109e64ca0dSMichael Hamann                    $newindexline = $this->getIndexKey($key.'_i', '', $newid);
4119e64ca0dSMichael Hamann                    $pagekeys     = $this->getIndex($key.'_p', '');
4125eb9e867SMichael Hamann                    $parts = explode(':', $indexline);
4135eb9e867SMichael Hamann                    foreach ($parts as $part) {
4145eb9e867SMichael Hamann                        list($id, $count) = explode('*', $part);
4155eb9e867SMichael Hamann                        $newindexline =  $this->updateTuple($newindexline, $id, $count);
4165eb9e867SMichael Hamann
4175eb9e867SMichael Hamann                        $keyline = explode(':', $pagekeys[$id]);
4185eb9e867SMichael Hamann                        // remove old meta value
4195eb9e867SMichael Hamann                        $keyline = array_diff($keyline, array($oldid));
4205eb9e867SMichael Hamann                        // add new meta value when not already present
4215eb9e867SMichael Hamann                        if (!in_array($newid, $keyline)) {
4225eb9e867SMichael Hamann                            array_push($keyline, $newid);
4235eb9e867SMichael Hamann                        }
4245eb9e867SMichael Hamann                        $pagekeys[$id] = implode(':', $keyline);
4255eb9e867SMichael Hamann                    }
4269e64ca0dSMichael Hamann                    $this->saveIndex($key.'_p', '', $pagekeys);
4275eb9e867SMichael Hamann                    unset($pagekeys);
4289e64ca0dSMichael Hamann                    $this->saveIndexKey($key.'_i', '', $oldid, '');
4299e64ca0dSMichael Hamann                    $this->saveIndexKey($key.'_i', '', $newid, $newindexline);
4305eb9e867SMichael Hamann                }
4315eb9e867SMichael Hamann            } else {
4325eb9e867SMichael Hamann                $metavalues[$oldid] = $newvalue;
4339e64ca0dSMichael Hamann                if (!$this->saveIndex($key.'_w', '', $metavalues)) {
4345eb9e867SMichael Hamann                    $this->unlock();
4355eb9e867SMichael Hamann                    return false;
4365eb9e867SMichael Hamann                }
4375eb9e867SMichael Hamann            }
4385eb9e867SMichael Hamann        }
4395eb9e867SMichael Hamann
4405eb9e867SMichael Hamann        $this->unlock();
4415eb9e867SMichael Hamann        return true;
4425eb9e867SMichael Hamann    }
44325adeb91SMichael Hamann
4445eb9e867SMichael Hamann    /**
44500803e56STom N Harris     * Remove a page from the index
44644ca0adfSAndreas Gohr     *
44700803e56STom N Harris     * Erases entries in all known indexes.
44844ca0adfSAndreas Gohr     *
44900803e56STom N Harris     * @param string    $page   a page name
45000803e56STom N Harris     * @return boolean          the function completed successfully
45100803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
45244ca0adfSAndreas Gohr     */
45300803e56STom N Harris    public function deletePage($page) {
45480fb93f6STom N Harris        if (!$this->lock())
455bbc85ee4STom N Harris            return "locked";
456bbc85ee4STom N Harris
45725adeb91SMichael Hamann        $result = $this->deletePageNoLock($page);
45825adeb91SMichael Hamann
45925adeb91SMichael Hamann        $this->unlock();
46025adeb91SMichael Hamann
46125adeb91SMichael Hamann        return $result;
46225adeb91SMichael Hamann    }
46325adeb91SMichael Hamann
46425adeb91SMichael Hamann    /**
46525adeb91SMichael Hamann     * Remove a page from the index without locking the index, only use this function if the index is already locked
46625adeb91SMichael Hamann     *
46725adeb91SMichael Hamann     * Erases entries in all known indexes.
46825adeb91SMichael Hamann     *
46925adeb91SMichael Hamann     * @param string    $page   a page name
47025adeb91SMichael Hamann     * @return boolean          the function completed successfully
47125adeb91SMichael Hamann     * @author Tom N Harris <tnharris@whoopdedo.org>
47225adeb91SMichael Hamann     */
47325adeb91SMichael Hamann    protected function deletePageNoLock($page) {
474bbc85ee4STom N Harris        // load known documents
47503aafe1cSMichael Hamann        $pid = $this->getPIDNoLock($page);
4765981eb09STom N Harris        if ($pid === false) {
477bbc85ee4STom N Harris            return false;
478bbc85ee4STom N Harris        }
479bbc85ee4STom N Harris
480bbc85ee4STom N Harris        // Remove obsolete index entries
48180fb93f6STom N Harris        $pageword_idx = $this->getIndexKey('pageword', '', $pid);
482bbc85ee4STom N Harris        if ($pageword_idx !== '') {
483bbc85ee4STom N Harris            $delwords = explode(':',$pageword_idx);
484bbc85ee4STom N Harris            $upwords = array();
485bbc85ee4STom N Harris            foreach ($delwords as $word) {
486bbc85ee4STom N Harris                if ($word != '') {
487bbc85ee4STom N Harris                    list($wlen,$wid) = explode('*', $word);
488bbc85ee4STom N Harris                    $wid = (int)$wid;
489bbc85ee4STom N Harris                    $upwords[$wlen][] = $wid;
490bbc85ee4STom N Harris                }
491bbc85ee4STom N Harris            }
492bbc85ee4STom N Harris            foreach ($upwords as $wlen => $widx) {
49380fb93f6STom N Harris                $index = $this->getIndex('i', $wlen);
494bbc85ee4STom N Harris                foreach ($widx as $wid) {
49580fb93f6STom N Harris                    $index[$wid] = $this->updateTuple($index[$wid], $pid, 0);
496bbc85ee4STom N Harris                }
49780fb93f6STom N Harris                $this->saveIndex('i', $wlen, $index);
498bbc85ee4STom N Harris            }
499bbc85ee4STom N Harris        }
500bbc85ee4STom N Harris        // Save the reverse index
50180fb93f6STom N Harris        if (!$this->saveIndexKey('pageword', '', $pid, "")) {
502bbc85ee4STom N Harris            return false;
503bbc85ee4STom N Harris        }
504bbc85ee4STom N Harris
50580fb93f6STom N Harris        $this->saveIndexKey('title', '', $pid, "");
50680fb93f6STom N Harris        $keyidx = $this->getIndex('metadata', '');
5070604da34STom N Harris        foreach ($keyidx as $metaname) {
50880fb93f6STom N Harris            $val_idx = explode(':', $this->getIndexKey($metaname.'_p', '', $pid));
50980fb93f6STom N Harris            $meta_idx = $this->getIndex($metaname.'_i', '');
5100604da34STom N Harris            foreach ($val_idx as $id) {
511c55c59e3SMichael Hamann                if ($id === '') continue;
51280fb93f6STom N Harris                $meta_idx[$id] = $this->updateTuple($meta_idx[$id], $pid, 0);
5130604da34STom N Harris            }
51480fb93f6STom N Harris            $this->saveIndex($metaname.'_i', '', $meta_idx);
51580fb93f6STom N Harris            $this->saveIndexKey($metaname.'_p', '', $pid, '');
5160604da34STom N Harris        }
517bbc85ee4STom N Harris
518bbc85ee4STom N Harris        return true;
519d5b23302STom N Harris    }
52044ca0adfSAndreas Gohr
521d5b23302STom N Harris    /**
5223cf3c7d6SMichael Hamann     * Clear the whole index
5233cf3c7d6SMichael Hamann     *
5243cf3c7d6SMichael Hamann     * @return bool If the index has been cleared successfully
5253cf3c7d6SMichael Hamann     */
5263cf3c7d6SMichael Hamann    public function clear() {
5273cf3c7d6SMichael Hamann        global $conf;
5283cf3c7d6SMichael Hamann
5293cf3c7d6SMichael Hamann        if (!$this->lock()) return false;
5303cf3c7d6SMichael Hamann
5313cf3c7d6SMichael Hamann        @unlink($conf['indexdir'].'/page.idx');
5323cf3c7d6SMichael Hamann        @unlink($conf['indexdir'].'/title.idx');
5333cf3c7d6SMichael Hamann        @unlink($conf['indexdir'].'/pageword.idx');
5343cf3c7d6SMichael Hamann        @unlink($conf['indexdir'].'/metadata.idx');
5353cf3c7d6SMichael Hamann        $dir = @opendir($conf['indexdir']);
5363cf3c7d6SMichael Hamann        if($dir!==false){
5373cf3c7d6SMichael Hamann            while(($f = readdir($dir)) !== false){
5383cf3c7d6SMichael Hamann                if(substr($f,-4)=='.idx' &&
5393cf3c7d6SMichael Hamann                    (substr($f,0,1)=='i' || substr($f,0,1)=='w'
5403cf3c7d6SMichael Hamann                        || substr($f,-6)=='_w.idx' || substr($f,-6)=='_i.idx' || substr($f,-6)=='_p.idx'))
5413cf3c7d6SMichael Hamann                    @unlink($conf['indexdir']."/$f");
5423cf3c7d6SMichael Hamann            }
5433cf3c7d6SMichael Hamann        }
5443cf3c7d6SMichael Hamann        @unlink($conf['indexdir'].'/lengths.idx');
5453cf3c7d6SMichael Hamann
5463cf3c7d6SMichael Hamann        // clear the pid cache
5473cf3c7d6SMichael Hamann        $this->pidCache = array();
5483cf3c7d6SMichael Hamann
5493cf3c7d6SMichael Hamann        $this->unlock();
5503cf3c7d6SMichael Hamann        return true;
5513cf3c7d6SMichael Hamann    }
5523cf3c7d6SMichael Hamann
5533cf3c7d6SMichael Hamann    /**
55400803e56STom N Harris     * Split the text into words for fulltext search
555d5b23302STom N Harris     *
55600803e56STom N Harris     * TODO: does this also need &$stopwords ?
557d5b23302STom N Harris     *
5588cd4c12fSAndreas Gohr     * @triggers INDEXER_TEXT_PREPARE
5598cd4c12fSAndreas Gohr     * This event allows plugins to modify the text before it gets tokenized.
5608cd4c12fSAndreas Gohr     * Plugins intercepting this event should also intercept INDEX_VERSION_GET
5618cd4c12fSAndreas Gohr     *
56200803e56STom N Harris     * @param string    $text   plain text
56300803e56STom N Harris     * @param boolean   $wc     are wildcards allowed?
56400803e56STom N Harris     * @return array            list of words in the text
565d5b23302STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
566d5b23302STom N Harris     * @author Andreas Gohr <andi@splitbrain.org>
567d5b23302STom N Harris     */
56800803e56STom N Harris    public function tokenizer($text, $wc=false) {
56900803e56STom N Harris        $wc = ($wc) ? '' : '\*';
57000803e56STom N Harris        $stopwords =& idx_get_stopwords();
57100803e56STom N Harris
5728cd4c12fSAndreas Gohr        // prepare the text to be tokenized
5738cd4c12fSAndreas Gohr        $evt = new Doku_Event('INDEXER_TEXT_PREPARE', $text);
5748cd4c12fSAndreas Gohr        if ($evt->advise_before(true)) {
57500803e56STom N Harris            if (preg_match('/[^0-9A-Za-z ]/u', $text)) {
57600803e56STom N Harris                // handle asian chars as single words (may fail on older PHP version)
57700803e56STom N Harris                $asia = @preg_replace('/('.IDX_ASIAN.')/u', ' \1 ', $text);
57800803e56STom N Harris                if (!is_null($asia)) $text = $asia; // recover from regexp falure
57922952965SYoBoY            }
58022952965SYoBoY        }
5818cd4c12fSAndreas Gohr        $evt->advise_after();
5828cd4c12fSAndreas Gohr        unset($evt);
5838cd4c12fSAndreas Gohr
584f77fc90dSMichael Hamann        $text = strtr($text,
5854f0030ddSAndreas Gohr                       array(
5864f0030ddSAndreas Gohr                           "\r" => ' ',
5874f0030ddSAndreas Gohr                           "\n" => ' ',
5884f0030ddSAndreas Gohr                           "\t" => ' ',
5894f0030ddSAndreas Gohr                           "\xC2\xAD" => '', //soft-hyphen
5904f0030ddSAndreas Gohr                       )
5914f0030ddSAndreas Gohr                     );
59200803e56STom N Harris        if (preg_match('/[^0-9A-Za-z ]/u', $text))
59300803e56STom N Harris            $text = utf8_stripspecials($text, ' ', '\._\-:'.$wc);
59422952965SYoBoY
59500803e56STom N Harris        $wordlist = explode(' ', $text);
5965f27cb0eSMichael Hamann        foreach ($wordlist as $i => $word) {
5975f27cb0eSMichael Hamann            $wordlist[$i] = (preg_match('/[^0-9A-Za-z]/u', $word)) ?
59800803e56STom N Harris                utf8_strtolower($word) : strtolower($word);
5995f27cb0eSMichael Hamann        }
6005f27cb0eSMichael Hamann
6015f27cb0eSMichael Hamann        foreach ($wordlist as $i => $word) {
602675bf41fSTom N Harris            if ((!is_numeric($word) && strlen($word) < IDX_MINWORDLENGTH)
603675bf41fSTom N Harris              || array_search($word, $stopwords) !== false)
604675bf41fSTom N Harris                unset($wordlist[$i]);
60522952965SYoBoY        }
606675bf41fSTom N Harris        return array_values($wordlist);
60722952965SYoBoY    }
60822952965SYoBoY
60922952965SYoBoY    /**
61003aafe1cSMichael Hamann     * Get the numeric PID of a page
61103aafe1cSMichael Hamann     *
61203aafe1cSMichael Hamann     * @param string $page The page to get the PID for
61303aafe1cSMichael Hamann     * @return bool|int The page id on success, false on error
61403aafe1cSMichael Hamann     */
61503aafe1cSMichael Hamann    public function getPID($page) {
6163d2ce006SMichael Hamann        // return PID without locking when it is in the cache
6173d2ce006SMichael Hamann        if (isset($this->pidCache[$page])) return $this->pidCache[$page];
6183d2ce006SMichael Hamann
61903aafe1cSMichael Hamann        if (!$this->lock())
62003aafe1cSMichael Hamann            return false;
62103aafe1cSMichael Hamann
62203aafe1cSMichael Hamann        // load known documents
62303aafe1cSMichael Hamann        $pid = $this->getPIDNoLock($page);
62403aafe1cSMichael Hamann        if ($pid === false) {
62503aafe1cSMichael Hamann            $this->unlock();
62603aafe1cSMichael Hamann            return false;
62703aafe1cSMichael Hamann        }
62803aafe1cSMichael Hamann
62903aafe1cSMichael Hamann        $this->unlock();
63003aafe1cSMichael Hamann        return $pid;
63103aafe1cSMichael Hamann    }
63203aafe1cSMichael Hamann
63303aafe1cSMichael Hamann    /**
63403aafe1cSMichael Hamann     * Get the numeric PID of a page without locking the index.
63503aafe1cSMichael Hamann     * Only use this function when the index is already locked.
63603aafe1cSMichael Hamann     *
63703aafe1cSMichael Hamann     * @param string $page The page to get the PID for
63803aafe1cSMichael Hamann     * @return bool|int The page id on success, false on error
63903aafe1cSMichael Hamann     */
64003aafe1cSMichael Hamann    protected function getPIDNoLock($page) {
6413d2ce006SMichael Hamann        // avoid expensive addIndexKey operation for the most recently requested pages by using a cache
6423d2ce006SMichael Hamann        if (isset($this->pidCache[$page])) return $this->pidCache[$page];
6433d2ce006SMichael Hamann        $pid = $this->addIndexKey('page', '', $page);
6443d2ce006SMichael Hamann        // limit cache to 10 entries by discarding the oldest element as in DokuWiki usually only the most recently
6453d2ce006SMichael Hamann        // added item will be requested again
6463d2ce006SMichael Hamann        if (count($this->pidCache) > 10) array_shift($this->pidCache);
6473d2ce006SMichael Hamann        $this->pidCache[$page] = $pid;
6483d2ce006SMichael Hamann        return $pid;
64903aafe1cSMichael Hamann    }
65003aafe1cSMichael Hamann
65103aafe1cSMichael Hamann    /**
65203aafe1cSMichael Hamann     * Get the page id of a numeric PID
65303aafe1cSMichael Hamann     *
65403aafe1cSMichael Hamann     * @param int $pid The PID to get the page id for
65503aafe1cSMichael Hamann     * @return string The page id
65603aafe1cSMichael Hamann     */
65703aafe1cSMichael Hamann    public function getPageFromPID($pid) {
65803aafe1cSMichael Hamann        return $this->getIndexKey('page', '', $pid);
65903aafe1cSMichael Hamann    }
66003aafe1cSMichael Hamann
66103aafe1cSMichael Hamann    /**
66200803e56STom N Harris     * Find pages in the fulltext index containing the words,
663579b0f7eSTNHarris     *
66400803e56STom N Harris     * The search words must be pre-tokenized, meaning only letters and
66500803e56STom N Harris     * numbers with an optional wildcard
666579b0f7eSTNHarris     *
66700803e56STom N Harris     * The returned array will have the original tokens as key. The values
66800803e56STom N Harris     * in the returned list is an array with the page names as keys and the
669b00bd361STom N Harris     * number of times that token appears on the page as value.
67000803e56STom N Harris     *
671e3ab6fc5SMichael Hamann     * @param array  $tokens list of words to search for
67200803e56STom N Harris     * @return array         list of page names with usage counts
67300803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
67400803e56STom N Harris     * @author Andreas Gohr <andi@splitbrain.org>
675579b0f7eSTNHarris     */
6769b41be24STom N Harris    public function lookup(&$tokens) {
67700803e56STom N Harris        $result = array();
67880fb93f6STom N Harris        $wids = $this->getIndexWords($tokens, $result);
67900803e56STom N Harris        if (empty($wids)) return array();
68000803e56STom N Harris        // load known words and documents
68180fb93f6STom N Harris        $page_idx = $this->getIndex('page', '');
68200803e56STom N Harris        $docs = array();
68300803e56STom N Harris        foreach (array_keys($wids) as $wlen) {
68400803e56STom N Harris            $wids[$wlen] = array_unique($wids[$wlen]);
68580fb93f6STom N Harris            $index = $this->getIndex('i', $wlen);
68600803e56STom N Harris            foreach($wids[$wlen] as $ixid) {
68700803e56STom N Harris                if ($ixid < count($index))
68880fb93f6STom N Harris                    $docs["$wlen*$ixid"] = $this->parseTuples($page_idx, $index[$ixid]);
689d5b23302STom N Harris            }
690d5b23302STom N Harris        }
69100803e56STom N Harris        // merge found pages into final result array
69200803e56STom N Harris        $final = array();
69300803e56STom N Harris        foreach ($result as $word => $res) {
69400803e56STom N Harris            $final[$word] = array();
69500803e56STom N Harris            foreach ($res as $wid) {
696952ab260SMichael Hamann                // handle the case when ($ixid < count($index)) has been false
697952ab260SMichael Hamann                // and thus $docs[$wid] hasn't been set.
698952ab260SMichael Hamann                if (!isset($docs[$wid])) continue;
69900803e56STom N Harris                $hits = &$docs[$wid];
70000803e56STom N Harris                foreach ($hits as $hitkey => $hitcnt) {
70100803e56STom N Harris                    // make sure the document still exists
70200803e56STom N Harris                    if (!page_exists($hitkey, '', false)) continue;
70300803e56STom N Harris                    if (!isset($final[$word][$hitkey]))
70400803e56STom N Harris                        $final[$word][$hitkey] = $hitcnt;
70500803e56STom N Harris                    else
70600803e56STom N Harris                        $final[$word][$hitkey] += $hitcnt;
707d5b23302STom N Harris                }
708579b0f7eSTNHarris            }
709579b0f7eSTNHarris        }
71000803e56STom N Harris        return $final;
711579b0f7eSTNHarris    }
712579b0f7eSTNHarris
713579b0f7eSTNHarris    /**
71400803e56STom N Harris     * Find pages containing a metadata key.
715d5b23302STom N Harris     *
71600803e56STom N Harris     * The metadata values are compared as case-sensitive strings. Pass a
71700803e56STom N Harris     * callback function that returns true or false to use a different
718b00bd361STom N Harris     * comparison function. The function will be called with the $value being
719b00bd361STom N Harris     * searched for as the first argument, and the word in the index as the
7201538718dSTom N Harris     * second argument. The function preg_match can be used directly if the
7211538718dSTom N Harris     * values are regexes.
722d5b23302STom N Harris     *
72300803e56STom N Harris     * @param string    $key    name of the metadata key to look for
7241538718dSTom N Harris     * @param string    $value  search term to look for, must be a string or array of strings
72500803e56STom N Harris     * @param callback  $func   comparison function
726b00bd361STom N Harris     * @return array            lists with page names, keys are query values if $value is array
72700803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
728cd763a5bSMichael Hamann     * @author Michael Hamann <michael@content-space.de>
72900803e56STom N Harris     */
730b00bd361STom N Harris    public function lookupKey($key, &$value, $func=null) {
731f078bb00STom N Harris        if (!is_array($value))
732f078bb00STom N Harris            $value_array = array($value);
733f078bb00STom N Harris        else
734f078bb00STom N Harris            $value_array =& $value;
735cd763a5bSMichael Hamann
736c1209673STom N Harris        // the matching ids for the provided value(s)
737c1209673STom N Harris        $value_ids = array();
738c1209673STom N Harris
739c1209673STom N Harris        $metaname = idx_cleanName($key);
740c1209673STom N Harris
741c1209673STom N Harris        // get all words in order to search the matching ids
742c1209673STom N Harris        if ($key == 'title') {
74380fb93f6STom N Harris            $words = $this->getIndex('title', '');
744c1209673STom N Harris        } else {
745b9d8cc1eSTom N Harris            $words = $this->getIndex($metaname.'_w', '');
746c1209673STom N Harris        }
747c1209673STom N Harris
748f078bb00STom N Harris        if (!is_null($func)) {
7491538718dSTom N Harris            foreach ($value_array as $val) {
750f078bb00STom N Harris                foreach ($words as $i => $word) {
7511538718dSTom N Harris                    if (call_user_func_array($func, array($val, $word)))
752c1209673STom N Harris                        $value_ids[$i][] = $val;
753f078bb00STom N Harris                }
754f078bb00STom N Harris            }
755f078bb00STom N Harris        } else {
756f078bb00STom N Harris            foreach ($value_array as $val) {
757f078bb00STom N Harris                $xval = $val;
758b6d540bdSTom N Harris                $caret = '^';
759b6d540bdSTom N Harris                $dollar = '$';
760f078bb00STom N Harris                // check for wildcards
761f078bb00STom N Harris                if (substr($xval, 0, 1) == '*') {
762f078bb00STom N Harris                    $xval = substr($xval, 1);
763b6d540bdSTom N Harris                    $caret = '';
764f078bb00STom N Harris                }
765f078bb00STom N Harris                if (substr($xval, -1, 1) == '*') {
766f078bb00STom N Harris                    $xval = substr($xval, 0, -1);
767b6d540bdSTom N Harris                    $dollar = '';
768f078bb00STom N Harris                }
769b6d540bdSTom N Harris                if (!$caret || !$dollar) {
770f078bb00STom N Harris                    $re = $caret.preg_quote($xval, '/').$dollar;
771f078bb00STom N Harris                    foreach(array_keys(preg_grep('/'.$re.'/', $words)) as $i)
772c1209673STom N Harris                        $value_ids[$i][] = $val;
773cd763a5bSMichael Hamann                } else {
774f078bb00STom N Harris                    if (($i = array_search($val, $words)) !== false)
775c1209673STom N Harris                        $value_ids[$i][] = $val;
776cd763a5bSMichael Hamann                }
777cd763a5bSMichael Hamann            }
778cd763a5bSMichael Hamann        }
779cd763a5bSMichael Hamann
780cd763a5bSMichael Hamann        unset($words); // free the used memory
781cd763a5bSMichael Hamann
7827233c152SMichael Hamann        // initialize the result so it won't be null
783c1209673STom N Harris        $result = array();
7847233c152SMichael Hamann        foreach ($value_array as $val) {
7857233c152SMichael Hamann            $result[$val] = array();
7867233c152SMichael Hamann        }
7877233c152SMichael Hamann
78880fb93f6STom N Harris        $page_idx = $this->getIndex('page', '');
789cd763a5bSMichael Hamann
790c1209673STom N Harris        // Special handling for titles
791c1209673STom N Harris        if ($key == 'title') {
792c1209673STom N Harris            foreach ($value_ids as $pid => $val_list) {
793c1209673STom N Harris                $page = $page_idx[$pid];
794c1209673STom N Harris                foreach ($val_list as $val) {
795c1209673STom N Harris                    $result[$val][] = $page;
796c1209673STom N Harris                }
797c1209673STom N Harris            }
798c1209673STom N Harris        } else {
799c1209673STom N Harris            // load all lines and pages so the used lines can be taken and matched with the pages
800b9d8cc1eSTom N Harris            $lines = $this->getIndex($metaname.'_i', '');
801c1209673STom N Harris
802c1209673STom N Harris            foreach ($value_ids as $value_id => $val_list) {
803cd763a5bSMichael Hamann                // parse the tuples of the form page_id*1:page2_id*1 and so on, return value
804cd763a5bSMichael Hamann                // is an array with page_id => 1, page2_id => 1 etc. so take the keys only
80580fb93f6STom N Harris                $pages = array_keys($this->parseTuples($page_idx, $lines[$value_id]));
806c1209673STom N Harris                foreach ($val_list as $val) {
807c1209673STom N Harris                    $result[$val] = array_merge($result[$val], $pages);
808c1209673STom N Harris                }
809c1209673STom N Harris            }
810cd763a5bSMichael Hamann        }
811f078bb00STom N Harris        if (!is_array($value)) $result = $result[$value];
812cd763a5bSMichael Hamann        return $result;
81300803e56STom N Harris    }
81400803e56STom N Harris
81500803e56STom N Harris    /**
81600803e56STom N Harris     * Find the index ID of each search term.
81700803e56STom N Harris     *
81800803e56STom N Harris     * The query terms should only contain valid characters, with a '*' at
81900803e56STom N Harris     * either the beginning or end of the word (or both).
82000803e56STom N Harris     * The $result parameter can be used to merge the index locations with
82100803e56STom N Harris     * the appropriate query term.
82200803e56STom N Harris     *
823e3ab6fc5SMichael Hamann     * @param array  $words  The query terms.
824e3ab6fc5SMichael Hamann     * @param array  $result Set to word => array("length*id" ...)
825d5b23302STom N Harris     * @return array         Set to length => array(id ...)
826d5b23302STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
827d5b23302STom N Harris     */
82880fb93f6STom N Harris    protected function getIndexWords(&$words, &$result) {
829d5b23302STom N Harris        $tokens = array();
830d5b23302STom N Harris        $tokenlength = array();
831d5b23302STom N Harris        $tokenwild = array();
832d5b23302STom N Harris        foreach ($words as $word) {
833d5b23302STom N Harris            $result[$word] = array();
834b6d540bdSTom N Harris            $caret = '^';
835b6d540bdSTom N Harris            $dollar = '$';
836d5b23302STom N Harris            $xword = $word;
837d5b23302STom N Harris            $wlen = wordlen($word);
838d5b23302STom N Harris
839d5b23302STom N Harris            // check for wildcards
840d5b23302STom N Harris            if (substr($xword, 0, 1) == '*') {
841d5b23302STom N Harris                $xword = substr($xword, 1);
842b6d540bdSTom N Harris                $caret = '';
843d5b23302STom N Harris                $wlen -= 1;
844d5b23302STom N Harris            }
845d5b23302STom N Harris            if (substr($xword, -1, 1) == '*') {
846d5b23302STom N Harris                $xword = substr($xword, 0, -1);
847b6d540bdSTom N Harris                $dollar = '';
848d5b23302STom N Harris                $wlen -= 1;
849d5b23302STom N Harris            }
850b6d540bdSTom N Harris            if ($wlen < IDX_MINWORDLENGTH && $caret && $dollar && !is_numeric($xword))
85100803e56STom N Harris                continue;
85200803e56STom N Harris            if (!isset($tokens[$xword]))
853d5b23302STom N Harris                $tokenlength[$wlen][] = $xword;
854b6d540bdSTom N Harris            if (!$caret || !$dollar) {
855f078bb00STom N Harris                $re = $caret.preg_quote($xword, '/').$dollar;
85600803e56STom N Harris                $tokens[$xword][] = array($word, '/'.$re.'/');
85700803e56STom N Harris                if (!isset($tokenwild[$xword]))
85800803e56STom N Harris                    $tokenwild[$xword] = $wlen;
85900803e56STom N Harris            } else {
860d5b23302STom N Harris                $tokens[$xword][] = array($word, null);
861d5b23302STom N Harris            }
86200803e56STom N Harris        }
863d5b23302STom N Harris        asort($tokenwild);
86400803e56STom N Harris        // $tokens = array( base word => array( [ query term , regexp ] ... ) ... )
865d5b23302STom N Harris        // $tokenlength = array( base word length => base word ... )
866d5b23302STom N Harris        // $tokenwild = array( base word => base word length ... )
867d5b23302STom N Harris        $length_filter = empty($tokenwild) ? $tokenlength : min(array_keys($tokenlength));
86880fb93f6STom N Harris        $indexes_known = $this->indexLengths($length_filter);
869d5b23302STom N Harris        if (!empty($tokenwild)) sort($indexes_known);
870d5b23302STom N Harris        // get word IDs
871d5b23302STom N Harris        $wids = array();
872d5b23302STom N Harris        foreach ($indexes_known as $ixlen) {
87380fb93f6STom N Harris            $word_idx = $this->getIndex('w', $ixlen);
874d5b23302STom N Harris            // handle exact search
875d5b23302STom N Harris            if (isset($tokenlength[$ixlen])) {
876d5b23302STom N Harris                foreach ($tokenlength[$ixlen] as $xword) {
87700803e56STom N Harris                    $wid = array_search($xword, $word_idx);
87800803e56STom N Harris                    if ($wid !== false) {
879d5b23302STom N Harris                        $wids[$ixlen][] = $wid;
880d5b23302STom N Harris                        foreach ($tokens[$xword] as $w)
881d5b23302STom N Harris                            $result[$w[0]][] = "$ixlen*$wid";
882d5b23302STom N Harris                    }
883d5b23302STom N Harris                }
884d5b23302STom N Harris            }
885d5b23302STom N Harris            // handle wildcard search
886d5b23302STom N Harris            foreach ($tokenwild as $xword => $wlen) {
887d5b23302STom N Harris                if ($wlen >= $ixlen) break;
888d5b23302STom N Harris                foreach ($tokens[$xword] as $w) {
889d5b23302STom N Harris                    if (is_null($w[1])) continue;
890d5b23302STom N Harris                    foreach(array_keys(preg_grep($w[1], $word_idx)) as $wid) {
891d5b23302STom N Harris                        $wids[$ixlen][] = $wid;
892d5b23302STom N Harris                        $result[$w[0]][] = "$ixlen*$wid";
893d5b23302STom N Harris                    }
894d5b23302STom N Harris                }
895d5b23302STom N Harris            }
896d5b23302STom N Harris        }
897d5b23302STom N Harris        return $wids;
898d5b23302STom N Harris    }
899d5b23302STom N Harris
900d5b23302STom N Harris    /**
90100803e56STom N Harris     * Return a list of all pages
902c1209673STom N Harris     * Warning: pages may not exist!
903488dd6ceSAndreas Gohr     *
90400803e56STom N Harris     * @param string    $key    list only pages containing the metadata key (optional)
90500803e56STom N Harris     * @return array            list of page names
90600803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
90700803e56STom N Harris     */
90800803e56STom N Harris    public function getPages($key=null) {
90980fb93f6STom N Harris        $page_idx = $this->getIndex('page', '');
91000803e56STom N Harris        if (is_null($key)) return $page_idx;
911c1209673STom N Harris
912c1209673STom N Harris        $metaname = idx_cleanName($key);
913c1209673STom N Harris
914c1209673STom N Harris        // Special handling for titles
915c1209673STom N Harris        if ($key == 'title') {
91680fb93f6STom N Harris            $title_idx = $this->getIndex('title', '');
917c1209673STom N Harris            array_splice($page_idx, count($title_idx));
918c1209673STom N Harris            foreach ($title_idx as $i => $title)
919c1209673STom N Harris                if ($title === "") unset($page_idx[$i]);
920675bf41fSTom N Harris            return array_values($page_idx);
921c1209673STom N Harris        }
922c1209673STom N Harris
923c1209673STom N Harris        $pages = array();
924b9d8cc1eSTom N Harris        $lines = $this->getIndex($metaname.'_i', '');
925c1209673STom N Harris        foreach ($lines as $line) {
92680fb93f6STom N Harris            $pages = array_merge($pages, $this->parseTuples($page_idx, $line));
927c1209673STom N Harris        }
928c1209673STom N Harris        return array_keys($pages);
92900803e56STom N Harris    }
93000803e56STom N Harris
93100803e56STom N Harris    /**
93200803e56STom N Harris     * Return a list of words sorted by number of times used
93300803e56STom N Harris     *
93400803e56STom N Harris     * @param int       $min    bottom frequency threshold
93500803e56STom N Harris     * @param int       $max    upper frequency limit. No limit if $max<$min
936e3ab6fc5SMichael Hamann     * @param int       $minlen minimum length of words to count
93700803e56STom N Harris     * @param string    $key    metadata key to list. Uses the fulltext index if not given
93800803e56STom N Harris     * @return array            list of words as the keys and frequency as values
93900803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
94000803e56STom N Harris     */
941b8c040dbSTom N Harris    public function histogram($min=1, $max=0, $minlen=3, $key=null) {
942175193d2STom N Harris        if ($min < 1)
943175193d2STom N Harris            $min = 1;
944175193d2STom N Harris        if ($max < $min)
945175193d2STom N Harris            $max = 0;
946175193d2STom N Harris
947175193d2STom N Harris        $result = array();
948175193d2STom N Harris
949175193d2STom N Harris        if ($key == 'title') {
95080fb93f6STom N Harris            $index = $this->getIndex('title', '');
951175193d2STom N Harris            $index = array_count_values($index);
952175193d2STom N Harris            foreach ($index as $val => $cnt) {
953b8c040dbSTom N Harris                if ($cnt >= $min && (!$max || $cnt <= $max) && strlen($val) >= $minlen)
954175193d2STom N Harris                    $result[$val] = $cnt;
955175193d2STom N Harris            }
956175193d2STom N Harris        }
957175193d2STom N Harris        elseif (!is_null($key)) {
958175193d2STom N Harris            $metaname = idx_cleanName($key);
95980fb93f6STom N Harris            $index = $this->getIndex($metaname.'_i', '');
960175193d2STom N Harris            $val_idx = array();
961175193d2STom N Harris            foreach ($index as $wid => $line) {
96280fb93f6STom N Harris                $freq = $this->countTuples($line);
9639a9b579aSMichael Hamann                if ($freq >= $min && (!$max || $freq <= $max))
964175193d2STom N Harris                    $val_idx[$wid] = $freq;
965175193d2STom N Harris            }
966175193d2STom N Harris            if (!empty($val_idx)) {
96780fb93f6STom N Harris                $words = $this->getIndex($metaname.'_w', '');
9689a9b579aSMichael Hamann                foreach ($val_idx as $wid => $freq) {
9699a9b579aSMichael Hamann                    if (strlen($words[$wid]) >= $minlen)
970175193d2STom N Harris                        $result[$words[$wid]] = $freq;
971175193d2STom N Harris                }
972175193d2STom N Harris            }
9739a9b579aSMichael Hamann        }
974175193d2STom N Harris        else {
975175193d2STom N Harris            $lengths = idx_listIndexLengths();
976175193d2STom N Harris            foreach ($lengths as $length) {
977b8c040dbSTom N Harris                if ($length < $minlen) continue;
97880fb93f6STom N Harris                $index = $this->getIndex('i', $length);
979175193d2STom N Harris                $words = null;
980175193d2STom N Harris                foreach ($index as $wid => $line) {
98180fb93f6STom N Harris                    $freq = $this->countTuples($line);
982175193d2STom N Harris                    if ($freq >= $min && (!$max || $freq <= $max)) {
983175193d2STom N Harris                        if ($words === null)
98480fb93f6STom N Harris                            $words = $this->getIndex('w', $length);
985175193d2STom N Harris                        $result[$words[$wid]] = $freq;
986175193d2STom N Harris                    }
987175193d2STom N Harris                }
988175193d2STom N Harris            }
989175193d2STom N Harris        }
990175193d2STom N Harris
991175193d2STom N Harris        arsort($result);
992175193d2STom N Harris        return $result;
99300803e56STom N Harris    }
99400803e56STom N Harris
99500803e56STom N Harris    /**
99600803e56STom N Harris     * Lock the indexer.
99700803e56STom N Harris     *
99800803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
99900803e56STom N Harris     */
100080fb93f6STom N Harris    protected function lock() {
100100803e56STom N Harris        global $conf;
100200803e56STom N Harris        $status = true;
1003f77fc90dSMichael Hamann        $run = 0;
100400803e56STom N Harris        $lock = $conf['lockdir'].'/_indexer.lock';
100500803e56STom N Harris        while (!@mkdir($lock, $conf['dmode'])) {
100600803e56STom N Harris            usleep(50);
1007f77fc90dSMichael Hamann            if(is_dir($lock) && time()-@filemtime($lock) > 60*5){
1008f77fc90dSMichael Hamann                // looks like a stale lock - remove it
1009f77fc90dSMichael Hamann                if (!@rmdir($lock)) {
1010f77fc90dSMichael Hamann                    $status = "removing the stale lock failed";
1011f77fc90dSMichael Hamann                    return false;
101200803e56STom N Harris                } else {
1013f77fc90dSMichael Hamann                    $status = "stale lock removed";
1014f77fc90dSMichael Hamann                }
1015f77fc90dSMichael Hamann            }elseif($run++ == 1000){
1016f77fc90dSMichael Hamann                // we waited 5 seconds for that lock
101700803e56STom N Harris                return false;
101800803e56STom N Harris            }
101900803e56STom N Harris        }
102000803e56STom N Harris        if ($conf['dperm'])
102100803e56STom N Harris            chmod($lock, $conf['dperm']);
102200803e56STom N Harris        return $status;
102300803e56STom N Harris    }
102400803e56STom N Harris
102500803e56STom N Harris    /**
102600803e56STom N Harris     * Release the indexer lock.
102700803e56STom N Harris     *
102800803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
102900803e56STom N Harris     */
103080fb93f6STom N Harris    protected function unlock() {
103100803e56STom N Harris        global $conf;
103200803e56STom N Harris        @rmdir($conf['lockdir'].'/_indexer.lock');
103300803e56STom N Harris        return true;
103400803e56STom N Harris    }
103500803e56STom N Harris
103600803e56STom N Harris    /**
103700803e56STom N Harris     * Retrieve the entire index.
103800803e56STom N Harris     *
1039b9d8cc1eSTom N Harris     * The $suffix argument is for an index that is split into
1040b9d8cc1eSTom N Harris     * multiple parts. Different index files should use different
1041b9d8cc1eSTom N Harris     * base names.
1042b9d8cc1eSTom N Harris     *
1043b9d8cc1eSTom N Harris     * @param string    $idx    name of the index
1044b9d8cc1eSTom N Harris     * @param string    $suffix subpart identifier
1045b9d8cc1eSTom N Harris     * @return array            list of lines without CR or LF
104600803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
104700803e56STom N Harris     */
104880fb93f6STom N Harris    protected function getIndex($idx, $suffix) {
104900803e56STom N Harris        global $conf;
105000803e56STom N Harris        $fn = $conf['indexdir'].'/'.$idx.$suffix.'.idx';
1051d64516f5SMichael Hamann        if (!@file_exists($fn)) return array();
1052d64516f5SMichael Hamann        return file($fn, FILE_IGNORE_NEW_LINES);
105300803e56STom N Harris    }
105400803e56STom N Harris
105500803e56STom N Harris    /**
105600803e56STom N Harris     * Replace the contents of the index with an array.
105700803e56STom N Harris     *
1058b9d8cc1eSTom N Harris     * @param string    $idx    name of the index
1059b9d8cc1eSTom N Harris     * @param string    $suffix subpart identifier
1060e3ab6fc5SMichael Hamann     * @param array     $lines  list of lines without LF
1061e3ab6fc5SMichael Hamann     * @return bool             If saving succeeded
106200803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
106300803e56STom N Harris     */
106480fb93f6STom N Harris    protected function saveIndex($idx, $suffix, &$lines) {
106500803e56STom N Harris        global $conf;
106600803e56STom N Harris        $fn = $conf['indexdir'].'/'.$idx.$suffix;
106700803e56STom N Harris        $fh = @fopen($fn.'.tmp', 'w');
106800803e56STom N Harris        if (!$fh) return false;
106900803e56STom N Harris        fwrite($fh, join("\n", $lines));
10702e1018bcSMichael Hamann        if (!empty($lines))
10712e1018bcSMichael Hamann            fwrite($fh, "\n");
107200803e56STom N Harris        fclose($fh);
107300803e56STom N Harris        if (isset($conf['fperm']))
107400803e56STom N Harris            chmod($fn.'.tmp', $conf['fperm']);
107500803e56STom N Harris        io_rename($fn.'.tmp', $fn.'.idx');
107600803e56STom N Harris        return true;
107700803e56STom N Harris    }
107800803e56STom N Harris
107900803e56STom N Harris    /**
108000803e56STom N Harris     * Retrieve a line from the index.
108100803e56STom N Harris     *
1082b9d8cc1eSTom N Harris     * @param string    $idx    name of the index
1083b9d8cc1eSTom N Harris     * @param string    $suffix subpart identifier
1084b9d8cc1eSTom N Harris     * @param int       $id     the line number
1085b9d8cc1eSTom N Harris     * @return string           a line with trailing whitespace removed
108600803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
108700803e56STom N Harris     */
108880fb93f6STom N Harris    protected function getIndexKey($idx, $suffix, $id) {
108900803e56STom N Harris        global $conf;
109000803e56STom N Harris        $fn = $conf['indexdir'].'/'.$idx.$suffix.'.idx';
109100803e56STom N Harris        if (!@file_exists($fn)) return '';
109200803e56STom N Harris        $fh = @fopen($fn, 'r');
109300803e56STom N Harris        if (!$fh) return '';
109400803e56STom N Harris        $ln = -1;
109500803e56STom N Harris        while (($line = fgets($fh)) !== false) {
109600803e56STom N Harris            if (++$ln == $id) break;
109700803e56STom N Harris        }
109800803e56STom N Harris        fclose($fh);
109900803e56STom N Harris        return rtrim((string)$line);
110000803e56STom N Harris    }
110100803e56STom N Harris
110200803e56STom N Harris    /**
110300803e56STom N Harris     * Write a line into the index.
110400803e56STom N Harris     *
1105b9d8cc1eSTom N Harris     * @param string    $idx    name of the index
1106b9d8cc1eSTom N Harris     * @param string    $suffix subpart identifier
1107b9d8cc1eSTom N Harris     * @param int       $id     the line number
1108b9d8cc1eSTom N Harris     * @param string    $line   line to write
1109e3ab6fc5SMichael Hamann     * @return bool             If saving succeeded
111000803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
111100803e56STom N Harris     */
111280fb93f6STom N Harris    protected function saveIndexKey($idx, $suffix, $id, $line) {
111300803e56STom N Harris        global $conf;
111400803e56STom N Harris        if (substr($line, -1) != "\n")
111500803e56STom N Harris            $line .= "\n";
111600803e56STom N Harris        $fn = $conf['indexdir'].'/'.$idx.$suffix;
111700803e56STom N Harris        $fh = @fopen($fn.'.tmp', 'w');
1118dbb771bbSAdrian Lang        if (!$fh) return false;
111900803e56STom N Harris        $ih = @fopen($fn.'.idx', 'r');
112000803e56STom N Harris        if ($ih) {
112100803e56STom N Harris            $ln = -1;
112200803e56STom N Harris            while (($curline = fgets($ih)) !== false) {
112300803e56STom N Harris                fwrite($fh, (++$ln == $id) ? $line : $curline);
112400803e56STom N Harris            }
11254373c7b5SMichael Hamann            if ($id > $ln) {
11264373c7b5SMichael Hamann                while ($id > ++$ln)
11274373c7b5SMichael Hamann                    fwrite($fh, "\n");
112800803e56STom N Harris                fwrite($fh, $line);
11294373c7b5SMichael Hamann            }
113000803e56STom N Harris            fclose($ih);
113100803e56STom N Harris        } else {
11324373c7b5SMichael Hamann            $ln = -1;
11334373c7b5SMichael Hamann            while ($id > ++$ln)
11344373c7b5SMichael Hamann                fwrite($fh, "\n");
113500803e56STom N Harris            fwrite($fh, $line);
113600803e56STom N Harris        }
113700803e56STom N Harris        fclose($fh);
113800803e56STom N Harris        if (isset($conf['fperm']))
113900803e56STom N Harris            chmod($fn.'.tmp', $conf['fperm']);
114000803e56STom N Harris        io_rename($fn.'.tmp', $fn.'.idx');
114100803e56STom N Harris        return true;
114200803e56STom N Harris    }
114300803e56STom N Harris
114400803e56STom N Harris    /**
114500803e56STom N Harris     * Retrieve or insert a value in the index.
114600803e56STom N Harris     *
1147b9d8cc1eSTom N Harris     * @param string    $idx    name of the index
1148b9d8cc1eSTom N Harris     * @param string    $suffix subpart identifier
1149b9d8cc1eSTom N Harris     * @param string    $value  line to find in the index
115003aafe1cSMichael Hamann     * @return int|bool          line number of the value in the index or false if writing the index failed
115100803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
115200803e56STom N Harris     */
115380fb93f6STom N Harris    protected function addIndexKey($idx, $suffix, $value) {
115480fb93f6STom N Harris        $index = $this->getIndex($idx, $suffix);
115500803e56STom N Harris        $id = array_search($value, $index);
115600803e56STom N Harris        if ($id === false) {
115700803e56STom N Harris            $id = count($index);
115800803e56STom N Harris            $index[$id] = $value;
115980fb93f6STom N Harris            if (!$this->saveIndex($idx, $suffix, $index)) {
116000803e56STom N Harris                trigger_error("Failed to write $idx index", E_USER_ERROR);
116100803e56STom N Harris                return false;
116200803e56STom N Harris            }
116300803e56STom N Harris        }
116400803e56STom N Harris        return $id;
116500803e56STom N Harris    }
116600803e56STom N Harris
1167e3ab6fc5SMichael Hamann    /**
116800803e56STom N Harris     * Get the list of lengths indexed in the wiki.
116900803e56STom N Harris     *
117000803e56STom N Harris     * Read the index directory or a cache file and returns
117100803e56STom N Harris     * a sorted array of lengths of the words used in the wiki.
117200803e56STom N Harris     *
117300803e56STom N Harris     * @author YoBoY <yoboy.leguesh@gmail.com>
117400803e56STom N Harris     */
117580fb93f6STom N Harris    protected function listIndexLengths() {
1176b1720e5cSMichael Hamann        return idx_listIndexLengths();
117700803e56STom N Harris    }
117800803e56STom N Harris
117900803e56STom N Harris    /**
118000803e56STom N Harris     * Get the word lengths that have been indexed.
118100803e56STom N Harris     *
118200803e56STom N Harris     * Reads the index directory and returns an array of lengths
118300803e56STom N Harris     * that there are indices for.
118400803e56STom N Harris     *
118500803e56STom N Harris     * @author YoBoY <yoboy.leguesh@gmail.com>
118600803e56STom N Harris     */
118780fb93f6STom N Harris    protected function indexLengths($filter) {
118800803e56STom N Harris        global $conf;
118900803e56STom N Harris        $idx = array();
119000803e56STom N Harris        if (is_array($filter)) {
119100803e56STom N Harris            // testing if index files exist only
119200803e56STom N Harris            $path = $conf['indexdir']."/i";
119300803e56STom N Harris            foreach ($filter as $key => $value) {
119400803e56STom N Harris                if (@file_exists($path.$key.'.idx'))
119500803e56STom N Harris                    $idx[] = $key;
119600803e56STom N Harris            }
119700803e56STom N Harris        } else {
119800803e56STom N Harris            $lengths = idx_listIndexLengths();
119900803e56STom N Harris            foreach ($lengths as $key => $length) {
120000803e56STom N Harris                // keep all the values equal or superior
120100803e56STom N Harris                if ((int)$length >= (int)$filter)
120200803e56STom N Harris                    $idx[] = $length;
120300803e56STom N Harris            }
120400803e56STom N Harris        }
120500803e56STom N Harris        return $idx;
120600803e56STom N Harris    }
120700803e56STom N Harris
120800803e56STom N Harris    /**
120900803e56STom N Harris     * Insert or replace a tuple in a line.
121000803e56STom N Harris     *
121100803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
121200803e56STom N Harris     */
121380fb93f6STom N Harris    protected function updateTuple($line, $id, $count) {
121400803e56STom N Harris        $newLine = $line;
121500803e56STom N Harris        if ($newLine !== '')
121600803e56STom N Harris            $newLine = preg_replace('/(^|:)'.preg_quote($id,'/').'\*\d*/', '', $newLine);
121700803e56STom N Harris        $newLine = trim($newLine, ':');
121800803e56STom N Harris        if ($count) {
1219d64516f5SMichael Hamann            if (strlen($newLine) > 0)
122000803e56STom N Harris                return "$id*$count:".$newLine;
122100803e56STom N Harris            else
122200803e56STom N Harris                return "$id*$count".$newLine;
122300803e56STom N Harris        }
122400803e56STom N Harris        return $newLine;
122500803e56STom N Harris    }
122600803e56STom N Harris
122700803e56STom N Harris    /**
122800803e56STom N Harris     * Split a line into an array of tuples.
122900803e56STom N Harris     *
123000803e56STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
123100803e56STom N Harris     * @author Andreas Gohr <andi@splitbrain.org>
123200803e56STom N Harris     */
123380fb93f6STom N Harris    protected function parseTuples(&$keys, $line) {
123400803e56STom N Harris        $result = array();
123500803e56STom N Harris        if ($line == '') return $result;
123600803e56STom N Harris        $parts = explode(':', $line);
123700803e56STom N Harris        foreach ($parts as $tuple) {
1238675bf41fSTom N Harris            if ($tuple === '') continue;
123900803e56STom N Harris            list($key, $cnt) = explode('*', $tuple);
1240d64516f5SMichael Hamann            if (!$cnt) continue;
124100803e56STom N Harris            $key = $keys[$key];
124200803e56STom N Harris            if (!$key) continue;
124300803e56STom N Harris            $result[$key] = $cnt;
124400803e56STom N Harris        }
124500803e56STom N Harris        return $result;
124600803e56STom N Harris    }
1247175193d2STom N Harris
1248175193d2STom N Harris    /**
1249175193d2STom N Harris     * Sum the counts in a list of tuples.
1250175193d2STom N Harris     *
1251175193d2STom N Harris     * @author Tom N Harris <tnharris@whoopdedo.org>
1252175193d2STom N Harris     */
125380fb93f6STom N Harris    protected function countTuples($line) {
1254175193d2STom N Harris        $freq = 0;
1255175193d2STom N Harris        $parts = explode(':', $line);
1256175193d2STom N Harris        foreach ($parts as $tuple) {
1257675bf41fSTom N Harris            if ($tuple === '') continue;
1258175193d2STom N Harris            list($pid, $cnt) = explode('*', $tuple);
1259175193d2STom N Harris            $freq += (int)$cnt;
1260175193d2STom N Harris        }
1261175193d2STom N Harris        return $freq;
1262175193d2STom N Harris    }
126300803e56STom N Harris}
126400803e56STom N Harris
126500803e56STom N Harris/**
126600803e56STom N Harris * Create an instance of the indexer.
126700803e56STom N Harris *
1268b3d1090eSMichael Hamann * @return Doku_Indexer               a Doku_Indexer
126900803e56STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
127000803e56STom N Harris */
12719b41be24STom N Harrisfunction idx_get_indexer() {
12724f708321SMichael Hamann    static $Indexer;
12731421e548SMichael Hamann    if (!isset($Indexer)) {
127400803e56STom N Harris        $Indexer = new Doku_Indexer();
127500803e56STom N Harris    }
127600803e56STom N Harris    return $Indexer;
127700803e56STom N Harris}
127800803e56STom N Harris
127900803e56STom N Harris/**
128000803e56STom N Harris * Returns words that will be ignored.
128100803e56STom N Harris *
128200803e56STom N Harris * @return array                list of stop words
128300803e56STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
128400803e56STom N Harris */
128500803e56STom N Harrisfunction & idx_get_stopwords() {
128600803e56STom N Harris    static $stopwords = null;
128700803e56STom N Harris    if (is_null($stopwords)) {
128800803e56STom N Harris        global $conf;
128900803e56STom N Harris        $swfile = DOKU_INC.'inc/lang/'.$conf['lang'].'/stopwords.txt';
129000803e56STom N Harris        if(@file_exists($swfile)){
129100803e56STom N Harris            $stopwords = file($swfile, FILE_IGNORE_NEW_LINES);
129200803e56STom N Harris        }else{
129300803e56STom N Harris            $stopwords = array();
129400803e56STom N Harris        }
129500803e56STom N Harris    }
129600803e56STom N Harris    return $stopwords;
129700803e56STom N Harris}
129800803e56STom N Harris
129900803e56STom N Harris/**
130000803e56STom N Harris * Adds/updates the search index for the given page
130100803e56STom N Harris *
130200803e56STom N Harris * Locking is handled internally.
130300803e56STom N Harris *
130400803e56STom N Harris * @param string        $page   name of the page to index
13059b41be24STom N Harris * @param boolean       $verbose    print status messages
1306d041f8dbSMichael Hamann * @param boolean       $force  force reindexing even when the index is up to date
130700803e56STom N Harris * @return boolean              the function completed successfully
130800803e56STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
130900803e56STom N Harris */
1310d041f8dbSMichael Hamannfunction idx_addPage($page, $verbose=false, $force=false) {
13119b41be24STom N Harris    $idxtag = metaFN($page,'.indexed');
1312a23ac4d7SMichael Hamann    // check if page was deleted but is still in the index
1313bbc85ee4STom N Harris    if (!page_exists($page)) {
1314bbc85ee4STom N Harris        if (!@file_exists($idxtag)) {
1315bbc85ee4STom N Harris            if ($verbose) print("Indexer: $page does not exist, ignoring".DOKU_LF);
1316bbc85ee4STom N Harris            return false;
1317bbc85ee4STom N Harris        }
1318bbc85ee4STom N Harris        $Indexer = idx_get_indexer();
1319bbc85ee4STom N Harris        $result = $Indexer->deletePage($page);
1320bbc85ee4STom N Harris        if ($result === "locked") {
1321bbc85ee4STom N Harris            if ($verbose) print("Indexer: locked".DOKU_LF);
1322bbc85ee4STom N Harris            return false;
1323bbc85ee4STom N Harris        }
1324bbc85ee4STom N Harris        @unlink($idxtag);
1325bbc85ee4STom N Harris        return $result;
1326bbc85ee4STom N Harris    }
1327a23ac4d7SMichael Hamann
1328a23ac4d7SMichael Hamann    // check if indexing needed
1329a23ac4d7SMichael Hamann    if(!$force && @file_exists($idxtag)){
1330a23ac4d7SMichael Hamann        if(trim(io_readFile($idxtag)) == idx_get_version()){
1331a23ac4d7SMichael Hamann            $last = @filemtime($idxtag);
1332a23ac4d7SMichael Hamann            if($last > @filemtime(wikiFN($page))){
1333a23ac4d7SMichael Hamann                if ($verbose) print("Indexer: index for $page up to date".DOKU_LF);
1334a23ac4d7SMichael Hamann                return false;
1335a23ac4d7SMichael Hamann            }
1336a23ac4d7SMichael Hamann        }
1337a23ac4d7SMichael Hamann    }
1338a23ac4d7SMichael Hamann
133965aa8490SMichael Hamann    $indexenabled = p_get_metadata($page, 'internal index', METADATA_RENDER_UNLIMITED);
1340bbc85ee4STom N Harris    if ($indexenabled === false) {
1341bbc85ee4STom N Harris        $result = false;
1342bbc85ee4STom N Harris        if (@file_exists($idxtag)) {
1343bbc85ee4STom N Harris            $Indexer = idx_get_indexer();
1344bbc85ee4STom N Harris            $result = $Indexer->deletePage($page);
1345bbc85ee4STom N Harris            if ($result === "locked") {
1346bbc85ee4STom N Harris                if ($verbose) print("Indexer: locked".DOKU_LF);
1347bbc85ee4STom N Harris                return false;
1348bbc85ee4STom N Harris            }
1349bbc85ee4STom N Harris            @unlink($idxtag);
1350bbc85ee4STom N Harris        }
1351bbc85ee4STom N Harris        if ($verbose) print("Indexer: index disabled for $page".DOKU_LF);
1352bbc85ee4STom N Harris        return $result;
1353bbc85ee4STom N Harris    }
1354bbc85ee4STom N Harris
135503aafe1cSMichael Hamann    $Indexer = idx_get_indexer();
135603aafe1cSMichael Hamann    $pid = $Indexer->getPID($page);
135703aafe1cSMichael Hamann    if ($pid === false) {
135803aafe1cSMichael Hamann        if ($verbose) print("Indexer: getting the PID failed for $page".DOKU_LF);
135903aafe1cSMichael Hamann        return false;
136003aafe1cSMichael Hamann    }
136100803e56STom N Harris    $body = '';
136239d6fd30SMichael Hamann    $metadata = array();
136365aa8490SMichael Hamann    $metadata['title'] = p_get_metadata($page, 'title', METADATA_RENDER_UNLIMITED);
136465aa8490SMichael Hamann    if (($references = p_get_metadata($page, 'relation references', METADATA_RENDER_UNLIMITED)) !== null)
136539d6fd30SMichael Hamann        $metadata['relation_references'] = array_keys($references);
1366a424180eSMichael Hamann    else
1367a424180eSMichael Hamann        $metadata['relation_references'] = array();
1368*ffec1009SMichael Hamann
1369*ffec1009SMichael Hamann    if (($media = p_get_metadata($page, 'relation media', METADATA_RENDER_UNLIMITED)) !== null)
1370*ffec1009SMichael Hamann        $metadata['relation_media'] = array_keys($media);
1371*ffec1009SMichael Hamann    else
1372*ffec1009SMichael Hamann        $metadata['relation_media'] = array();
1373*ffec1009SMichael Hamann
137403aafe1cSMichael Hamann    $data = compact('page', 'body', 'metadata', 'pid');
137500803e56STom N Harris    $evt = new Doku_Event('INDEXER_PAGE_ADD', $data);
137639d6fd30SMichael Hamann    if ($evt->advise_before()) $data['body'] = $data['body'] . " " . rawWiki($page);
137700803e56STom N Harris    $evt->advise_after();
137800803e56STom N Harris    unset($evt);
137939d6fd30SMichael Hamann    extract($data);
138000803e56STom N Harris
13819b41be24STom N Harris    $result = $Indexer->addPageWords($page, $body);
1382e1e1a7e0SMichael Hamann    if ($result === "locked") {
13839b41be24STom N Harris        if ($verbose) print("Indexer: locked".DOKU_LF);
13849b41be24STom N Harris        return false;
13859b41be24STom N Harris    }
1386320f489aSMichael Hamann
1387320f489aSMichael Hamann    if ($result) {
138839d6fd30SMichael Hamann        $result = $Indexer->addMetaKeys($page, $metadata);
1389320f489aSMichael Hamann        if ($result === "locked") {
1390320f489aSMichael Hamann            if ($verbose) print("Indexer: locked".DOKU_LF);
1391320f489aSMichael Hamann            return false;
1392320f489aSMichael Hamann        }
1393320f489aSMichael Hamann    }
1394320f489aSMichael Hamann
13959b41be24STom N Harris    if ($result)
13969b41be24STom N Harris        io_saveFile(metaFN($page,'.indexed'), idx_get_version());
13979b41be24STom N Harris    if ($verbose) {
13989b41be24STom N Harris        print("Indexer: finished".DOKU_LF);
13999b41be24STom N Harris        return true;
14009b41be24STom N Harris    }
14019b41be24STom N Harris    return $result;
140200803e56STom N Harris}
140300803e56STom N Harris
140400803e56STom N Harris/**
140500803e56STom N Harris * Find tokens in the fulltext index
140600803e56STom N Harris *
140700803e56STom N Harris * Takes an array of words and will return a list of matching
140800803e56STom N Harris * pages for each one.
1409488dd6ceSAndreas Gohr *
141063773904SAndreas Gohr * Important: No ACL checking is done here! All results are
141163773904SAndreas Gohr *            returned, regardless of permissions
141263773904SAndreas Gohr *
1413e3ab6fc5SMichael Hamann * @param array      $words  list of words to search for
141400803e56STom N Harris * @return array             list of pages found, associated with the search terms
1415488dd6ceSAndreas Gohr */
14169b41be24STom N Harrisfunction idx_lookup(&$words) {
14179b41be24STom N Harris    $Indexer = idx_get_indexer();
141800803e56STom N Harris    return $Indexer->lookup($words);
1419488dd6ceSAndreas Gohr}
1420488dd6ceSAndreas Gohr
1421488dd6ceSAndreas Gohr/**
142200803e56STom N Harris * Split a string into tokens
1423488dd6ceSAndreas Gohr *
1424488dd6ceSAndreas Gohr */
142500803e56STom N Harrisfunction idx_tokenizer($string, $wc=false) {
14269b41be24STom N Harris    $Indexer = idx_get_indexer();
142700803e56STom N Harris    return $Indexer->tokenizer($string, $wc);
1428488dd6ceSAndreas Gohr}
142900803e56STom N Harris
143000803e56STom N Harris/* For compatibility */
1431488dd6ceSAndreas Gohr
1432f5eb7cf0SAndreas Gohr/**
143300803e56STom N Harris * Read the list of words in an index (if it exists).
1434f5eb7cf0SAndreas Gohr *
14354e1bf408STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
1436f5eb7cf0SAndreas Gohr */
143700803e56STom N Harrisfunction idx_getIndex($idx, $suffix) {
14381c07b9e6STom N Harris    global $conf;
143900803e56STom N Harris    $fn = $conf['indexdir'].'/'.$idx.$suffix.'.idx';
144000803e56STom N Harris    if (!@file_exists($fn)) return array();
144100803e56STom N Harris    return file($fn);
144200803e56STom N Harris}
1443f5eb7cf0SAndreas Gohr
144400803e56STom N Harris/**
144500803e56STom N Harris * Get the list of lengths indexed in the wiki.
144600803e56STom N Harris *
144700803e56STom N Harris * Read the index directory or a cache file and returns
144800803e56STom N Harris * a sorted array of lengths of the words used in the wiki.
144900803e56STom N Harris *
145000803e56STom N Harris * @author YoBoY <yoboy.leguesh@gmail.com>
145100803e56STom N Harris */
145200803e56STom N Harrisfunction idx_listIndexLengths() {
145300803e56STom N Harris    global $conf;
145400803e56STom N Harris    // testing what we have to do, create a cache file or not.
145500803e56STom N Harris    if ($conf['readdircache'] == 0) {
145600803e56STom N Harris        $docache = false;
14571c07b9e6STom N Harris    } else {
145800803e56STom N Harris        clearstatcache();
145900803e56STom N Harris        if (@file_exists($conf['indexdir'].'/lengths.idx')
146000803e56STom N Harris        && (time() < @filemtime($conf['indexdir'].'/lengths.idx') + $conf['readdircache'])) {
146100803e56STom N Harris            if (($lengths = @file($conf['indexdir'].'/lengths.idx', FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES)) !== false) {
146200803e56STom N Harris                $idx = array();
146300803e56STom N Harris                foreach ($lengths as $length) {
146400803e56STom N Harris                    $idx[] = (int)$length;
146500803e56STom N Harris                }
146600803e56STom N Harris                return $idx;
1467f5eb7cf0SAndreas Gohr            }
14681c07b9e6STom N Harris        }
146900803e56STom N Harris        $docache = true;
147000803e56STom N Harris    }
14714e1bf408STom N Harris
147200803e56STom N Harris    if ($conf['readdircache'] == 0 || $docache) {
147300803e56STom N Harris        $dir = @opendir($conf['indexdir']);
147400803e56STom N Harris        if ($dir === false)
147500803e56STom N Harris            return array();
147626f7dbf5SMichael Hamann        $idx = array();
147700803e56STom N Harris        while (($f = readdir($dir)) !== false) {
147800803e56STom N Harris            if (substr($f, 0, 1) == 'i' && substr($f, -4) == '.idx') {
147900803e56STom N Harris                $i = substr($f, 1, -4);
148000803e56STom N Harris                if (is_numeric($i))
148100803e56STom N Harris                    $idx[] = (int)$i;
148200803e56STom N Harris            }
148300803e56STom N Harris        }
148400803e56STom N Harris        closedir($dir);
148500803e56STom N Harris        sort($idx);
148600803e56STom N Harris        // save this in a file
148700803e56STom N Harris        if ($docache) {
148800803e56STom N Harris            $handle = @fopen($conf['indexdir'].'/lengths.idx', 'w');
148900803e56STom N Harris            @fwrite($handle, implode("\n", $idx));
149000803e56STom N Harris            @fclose($handle);
149100803e56STom N Harris        }
149200803e56STom N Harris        return $idx;
149300803e56STom N Harris    }
149400803e56STom N Harris
149500803e56STom N Harris    return array();
149600803e56STom N Harris}
149700803e56STom N Harris
149800803e56STom N Harris/**
149900803e56STom N Harris * Get the word lengths that have been indexed.
150000803e56STom N Harris *
150100803e56STom N Harris * Reads the index directory and returns an array of lengths
150200803e56STom N Harris * that there are indices for.
150300803e56STom N Harris *
150400803e56STom N Harris * @author YoBoY <yoboy.leguesh@gmail.com>
150500803e56STom N Harris */
150600803e56STom N Harrisfunction idx_indexLengths($filter) {
150700803e56STom N Harris    global $conf;
150800803e56STom N Harris    $idx = array();
150900803e56STom N Harris    if (is_array($filter)) {
151000803e56STom N Harris        // testing if index files exist only
151100803e56STom N Harris        $path = $conf['indexdir']."/i";
151200803e56STom N Harris        foreach ($filter as $key => $value) {
151300803e56STom N Harris            if (@file_exists($path.$key.'.idx'))
151400803e56STom N Harris                $idx[] = $key;
151500803e56STom N Harris        }
1516f5eb7cf0SAndreas Gohr    } else {
151700803e56STom N Harris        $lengths = idx_listIndexLengths();
151800803e56STom N Harris        foreach ($lengths as $key => $length) {
151900803e56STom N Harris            // keep all the values equal or superior
152000803e56STom N Harris            if ((int)$length >= (int)$filter)
152100803e56STom N Harris                $idx[] = $length;
1522f5eb7cf0SAndreas Gohr        }
152300803e56STom N Harris    }
152400803e56STom N Harris    return $idx;
1525f5eb7cf0SAndreas Gohr}
1526f5eb7cf0SAndreas Gohr
152700803e56STom N Harris/**
152800803e56STom N Harris * Clean a name of a key for use as a file name.
152900803e56STom N Harris *
153000803e56STom N Harris * Romanizes non-latin characters, then strips away anything that's
153100803e56STom N Harris * not a letter, number, or underscore.
153200803e56STom N Harris *
153300803e56STom N Harris * @author Tom N Harris <tnharris@whoopdedo.org>
153400803e56STom N Harris */
153500803e56STom N Harrisfunction idx_cleanName($name) {
153600803e56STom N Harris    $name = utf8_romanize(trim((string)$name));
153700803e56STom N Harris    $name = preg_replace('#[ \./\\:-]+#', '_', $name);
153800803e56STom N Harris    $name = preg_replace('/[^A-Za-z0-9_]/', '', $name);
153900803e56STom N Harris    return strtolower($name);
1540f5eb7cf0SAndreas Gohr}
1541f5eb7cf0SAndreas Gohr
154200803e56STom N Harris//Setup VIM: ex: et ts=4 :
1543