xref: /dokuwiki/inc/fulltext.php (revision 8d0e286a89e3ad07963562359bfa5f9e974fd1fb)
1f5eb7cf0SAndreas Gohr<?php
2f5eb7cf0SAndreas Gohr/**
3f5eb7cf0SAndreas Gohr * DokuWiki fulltextsearch functions using the index
4f5eb7cf0SAndreas Gohr *
5f5eb7cf0SAndreas Gohr * @license    GPL 2 (http://www.gnu.org/licenses/gpl.html)
6f5eb7cf0SAndreas Gohr * @author     Andreas Gohr <andi@splitbrain.org>
7f5eb7cf0SAndreas Gohr */
8f5eb7cf0SAndreas Gohr
9fa8adffeSAndreas Gohrif(!defined('DOKU_INC')) die('meh.');
10f5eb7cf0SAndreas Gohr
11bd0293e7SAndreas Gohr/**
12bd0293e7SAndreas Gohr * create snippets for the first few results only
13bd0293e7SAndreas Gohr */
14bd0293e7SAndreas Gohrif(!defined('FT_SNIPPET_NUMBER')) define('FT_SNIPPET_NUMBER',15);
15f5eb7cf0SAndreas Gohr
16f5eb7cf0SAndreas Gohr/**
17f5eb7cf0SAndreas Gohr * The fulltext search
18f5eb7cf0SAndreas Gohr *
19f5eb7cf0SAndreas Gohr * Returns a list of matching documents for the given query
20506fa893SAndreas Gohr *
216840140fSChris Smith * refactored into ft_pageSearch(), _ft_pageSearch() and trigger_event()
226840140fSChris Smith *
2342ea7f44SGerrit Uitslag * @param string $query
2442ea7f44SGerrit Uitslag * @param array $highlight
2542ea7f44SGerrit Uitslag * @return array
26f5eb7cf0SAndreas Gohr */
27*8d0e286aSMichael Großefunction ft_pageSearch($query,&$highlight, $sort = 'hits'){
286840140fSChris Smith
2959bc3b48SGerrit Uitslag    $data = array();
306840140fSChris Smith    $data['query'] = $query;
316840140fSChris Smith    $data['highlight'] =& $highlight;
32*8d0e286aSMichael Große    $data['sort'] = $sort;
336840140fSChris Smith
346840140fSChris Smith    return trigger_event('SEARCH_QUERY_FULLPAGE', $data, '_ft_pageSearch');
356840140fSChris Smith}
36865c2687SKazutaka Miyasaka
37865c2687SKazutaka Miyasaka/**
38865c2687SKazutaka Miyasaka * Returns a list of matching documents for the given query
39865c2687SKazutaka Miyasaka *
40865c2687SKazutaka Miyasaka * @author Andreas Gohr <andi@splitbrain.org>
41865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
4242ea7f44SGerrit Uitslag *
4342ea7f44SGerrit Uitslag * @param array $data event data
4442ea7f44SGerrit Uitslag * @return array matching documents
45865c2687SKazutaka Miyasaka */
466840140fSChris Smithfunction _ft_pageSearch(&$data) {
479b41be24STom N Harris    $Indexer = idx_get_indexer();
489b41be24STom N Harris
49865c2687SKazutaka Miyasaka    // parse the given query
509b41be24STom N Harris    $q = ft_queryParser($Indexer, $data['query']);
51865c2687SKazutaka Miyasaka    $data['highlight'] = $q['highlight'];
526840140fSChris Smith
53865c2687SKazutaka Miyasaka    if (empty($q['parsed_ary'])) return array();
54506fa893SAndreas Gohr
55f5eb7cf0SAndreas Gohr    // lookup all words found in the query
569b41be24STom N Harris    $lookup = $Indexer->lookup($q['words']);
57f5eb7cf0SAndreas Gohr
58865c2687SKazutaka Miyasaka    // get all pages in this dokuwiki site (!: includes nonexistent pages)
59865c2687SKazutaka Miyasaka    $pages_all = array();
609b41be24STom N Harris    foreach ($Indexer->getPages() as $id) {
619b41be24STom N Harris        $pages_all[$id] = 0; // base: 0 hit
62f5eb7cf0SAndreas Gohr    }
63f5eb7cf0SAndreas Gohr
64865c2687SKazutaka Miyasaka    // process the query
65865c2687SKazutaka Miyasaka    $stack = array();
66865c2687SKazutaka Miyasaka    foreach ($q['parsed_ary'] as $token) {
67865c2687SKazutaka Miyasaka        switch (substr($token, 0, 3)) {
68865c2687SKazutaka Miyasaka            case 'W+:':
692f502d70SKazutaka Miyasaka            case 'W-:':
702f502d70SKazutaka Miyasaka            case 'W_:': // word
71865c2687SKazutaka Miyasaka                $word    = substr($token, 3);
72865c2687SKazutaka Miyasaka                $stack[] = (array) $lookup[$word];
73865c2687SKazutaka Miyasaka                break;
742f502d70SKazutaka Miyasaka            case 'P+:':
752f502d70SKazutaka Miyasaka            case 'P-:': // phrase
76865c2687SKazutaka Miyasaka                $phrase = substr($token, 3);
77865c2687SKazutaka Miyasaka                // since phrases are always parsed as ((W1)(W2)...(P)),
78865c2687SKazutaka Miyasaka                // the end($stack) always points the pages that contain
79865c2687SKazutaka Miyasaka                // all words in this phrase
80865c2687SKazutaka Miyasaka                $pages  = end($stack);
81865c2687SKazutaka Miyasaka                $pages_matched = array();
82865c2687SKazutaka Miyasaka                foreach(array_keys($pages) as $id){
83a7e8b43eSMichael Hamann                    $evdata = array(
84a7e8b43eSMichael Hamann                        'id' => $id,
85a7e8b43eSMichael Hamann                        'phrase' => $phrase,
86a7e8b43eSMichael Hamann                        'text' => rawWiki($id)
87a7e8b43eSMichael Hamann                    );
88a7e8b43eSMichael Hamann                    $evt = new Doku_Event('FULLTEXT_PHRASE_MATCH',$evdata);
89a7e8b43eSMichael Hamann                    if ($evt->advise_before() && $evt->result !== true) {
90a7e8b43eSMichael Hamann                        $text = utf8_strtolower($evdata['text']);
91865c2687SKazutaka Miyasaka                        if (strpos($text, $phrase) !== false) {
92a7e8b43eSMichael Hamann                            $evt->result = true;
93a7e8b43eSMichael Hamann                        }
94a7e8b43eSMichael Hamann                    }
95a7e8b43eSMichael Hamann                    $evt->advise_after();
96a7e8b43eSMichael Hamann                    if ($evt->result === true) {
97865c2687SKazutaka Miyasaka                        $pages_matched[$id] = 0; // phrase: always 0 hit
98865c2687SKazutaka Miyasaka                    }
99865c2687SKazutaka Miyasaka                }
100865c2687SKazutaka Miyasaka                $stack[] = $pages_matched;
101865c2687SKazutaka Miyasaka                break;
1022f502d70SKazutaka Miyasaka            case 'N+:':
1032f502d70SKazutaka Miyasaka            case 'N-:': // namespace
104de3383c6SMichael Große                $ns = cleanID(substr($token, 3)) . ':';
105865c2687SKazutaka Miyasaka                $pages_matched = array();
106865c2687SKazutaka Miyasaka                foreach (array_keys($pages_all) as $id) {
107865c2687SKazutaka Miyasaka                    if (strpos($id, $ns) === 0) {
108865c2687SKazutaka Miyasaka                        $pages_matched[$id] = 0; // namespace: always 0 hit
109865c2687SKazutaka Miyasaka                    }
110865c2687SKazutaka Miyasaka                }
111865c2687SKazutaka Miyasaka                $stack[] = $pages_matched;
112865c2687SKazutaka Miyasaka                break;
113865c2687SKazutaka Miyasaka            case 'AND': // and operation
114865c2687SKazutaka Miyasaka                list($pages1, $pages2) = array_splice($stack, -2);
115865c2687SKazutaka Miyasaka                $stack[] = ft_resultCombine(array($pages1, $pages2));
116865c2687SKazutaka Miyasaka                break;
117865c2687SKazutaka Miyasaka            case 'OR':  // or operation
118865c2687SKazutaka Miyasaka                list($pages1, $pages2) = array_splice($stack, -2);
119865c2687SKazutaka Miyasaka                $stack[] = ft_resultUnite(array($pages1, $pages2));
120865c2687SKazutaka Miyasaka                break;
121865c2687SKazutaka Miyasaka            case 'NOT': // not operation (unary)
122865c2687SKazutaka Miyasaka                $pages   = array_pop($stack);
123865c2687SKazutaka Miyasaka                $stack[] = ft_resultComplement(array($pages_all, $pages));
124a21136cdSAndreas Gohr                break;
125a21136cdSAndreas Gohr        }
126f5eb7cf0SAndreas Gohr    }
127865c2687SKazutaka Miyasaka    $docs = array_pop($stack);
128865c2687SKazutaka Miyasaka
129865c2687SKazutaka Miyasaka    if (empty($docs)) return array();
130865c2687SKazutaka Miyasaka
131865c2687SKazutaka Miyasaka    // check: settings, acls, existence
132865c2687SKazutaka Miyasaka    foreach (array_keys($docs) as $id) {
133865c2687SKazutaka Miyasaka        if (isHiddenPage($id) || auth_quickaclcheck($id) < AUTH_READ || !page_exists($id, '', false)) {
134865c2687SKazutaka Miyasaka            unset($docs[$id]);
135f5eb7cf0SAndreas Gohr        }
136f5eb7cf0SAndreas Gohr    }
137f5eb7cf0SAndreas Gohr
1381b48999cSMichael Große    $docs = _ft_filterResultsByTime($docs);
139*8d0e286aSMichael Große
140*8d0e286aSMichael Große    if ($data['sort'] === 'mtime') {
141*8d0e286aSMichael Große        uksort($docs, 'ft_pagemtimesorter');
142*8d0e286aSMichael Große    } else {
143865c2687SKazutaka Miyasaka        // sort docs by count
144f5eb7cf0SAndreas Gohr        arsort($docs);
145*8d0e286aSMichael Große    }
146f5eb7cf0SAndreas Gohr
147f5eb7cf0SAndreas Gohr    return $docs;
148f5eb7cf0SAndreas Gohr}
149f5eb7cf0SAndreas Gohr
150f5eb7cf0SAndreas Gohr/**
15154f4c056SAndreas Gohr * Returns the backlinks for a given page
15254f4c056SAndreas Gohr *
153320f489aSMichael Hamann * Uses the metadata index.
15407ff0babSMichael Hamann *
15507ff0babSMichael Hamann * @param string $id           The id for which links shall be returned
15607ff0babSMichael Hamann * @param bool   $ignore_perms Ignore the fact that pages are hidden or read-protected
15707ff0babSMichael Hamann * @return array The pages that contain links to the given page
15854f4c056SAndreas Gohr */
15907ff0babSMichael Hamannfunction ft_backlinks($id, $ignore_perms = false){
160320f489aSMichael Hamann    $result = idx_get_indexer()->lookupKey('relation_references', $id);
16154f4c056SAndreas Gohr
16263773904SAndreas Gohr    if(!count($result)) return $result;
16363773904SAndreas Gohr
16463773904SAndreas Gohr    // check ACL permissions
16563773904SAndreas Gohr    foreach(array_keys($result) as $idx){
16607ff0babSMichael Hamann        if(($ignore_perms !== true && (
16707ff0babSMichael Hamann                isHiddenPage($result[$idx]) || auth_quickaclcheck($result[$idx]) < AUTH_READ
16807ff0babSMichael Hamann            )) || !page_exists($result[$idx], '', false)){
16963773904SAndreas Gohr            unset($result[$idx]);
17063773904SAndreas Gohr        }
17163773904SAndreas Gohr    }
17263773904SAndreas Gohr
17354f4c056SAndreas Gohr    sort($result);
17454f4c056SAndreas Gohr    return $result;
17554f4c056SAndreas Gohr}
17654f4c056SAndreas Gohr
17754f4c056SAndreas Gohr/**
178a05e297aSAndreas Gohr * Returns the pages that use a given media file
179a05e297aSAndreas Gohr *
180ffec1009SMichael Hamann * Uses the relation media metadata property and the metadata index.
181a05e297aSAndreas Gohr *
182ffec1009SMichael Hamann * Note that before 2013-07-31 the second parameter was the maximum number of results and
183ffec1009SMichael Hamann * permissions were ignored. That's why the parameter is now checked to be explicitely set
184ffec1009SMichael Hamann * to true (with type bool) in order to be compatible with older uses of the function.
185ffec1009SMichael Hamann *
186ffec1009SMichael Hamann * @param string $id           The media id to look for
187ffec1009SMichael Hamann * @param bool   $ignore_perms Ignore hidden pages and acls (optional, default: false)
188ffec1009SMichael Hamann * @return array A list of pages that use the given media file
189a05e297aSAndreas Gohr */
190ffec1009SMichael Hamannfunction ft_mediause($id, $ignore_perms = false){
191ffec1009SMichael Hamann    $result = idx_get_indexer()->lookupKey('relation_media', $id);
192a05e297aSAndreas Gohr
193ffec1009SMichael Hamann    if(!count($result)) return $result;
194a05e297aSAndreas Gohr
195ffec1009SMichael Hamann    // check ACL permissions
196ffec1009SMichael Hamann    foreach(array_keys($result) as $idx){
197ffec1009SMichael Hamann        if(($ignore_perms !== true && (
198ffec1009SMichael Hamann                    isHiddenPage($result[$idx]) || auth_quickaclcheck($result[$idx]) < AUTH_READ
199ffec1009SMichael Hamann                )) || !page_exists($result[$idx], '', false)){
200ffec1009SMichael Hamann            unset($result[$idx]);
201a05e297aSAndreas Gohr        }
202a05e297aSAndreas Gohr    }
203a05e297aSAndreas Gohr
204a05e297aSAndreas Gohr    sort($result);
205a05e297aSAndreas Gohr    return $result;
206a05e297aSAndreas Gohr}
207a05e297aSAndreas Gohr
208a05e297aSAndreas Gohr
209a05e297aSAndreas Gohr
210a05e297aSAndreas Gohr/**
211506fa893SAndreas Gohr * Quicksearch for pagenames
212506fa893SAndreas Gohr *
213506fa893SAndreas Gohr * By default it only matches the pagename and ignores the
21480423ab6SAdrian Lang * namespace. This can be changed with the second parameter.
21580423ab6SAdrian Lang * The third parameter allows to search in titles as well.
216506fa893SAndreas Gohr *
2178d22f1e9SAndreas Gohr * The function always returns titles as well
2186840140fSChris Smith *
2198d22f1e9SAndreas Gohr * @triggers SEARCH_QUERY_PAGELOOKUP
220506fa893SAndreas Gohr * @author Andreas Gohr <andi@splitbrain.org>
2218d22f1e9SAndreas Gohr * @author Adrian Lang <lang@cosmocode.de>
22242ea7f44SGerrit Uitslag *
22342ea7f44SGerrit Uitslag * @param string $id        page id
22442ea7f44SGerrit Uitslag * @param bool   $in_ns     match against namespace as well?
22542ea7f44SGerrit Uitslag * @param bool   $in_title  search in title?
22642ea7f44SGerrit Uitslag * @return string[]
227506fa893SAndreas Gohr */
2288d22f1e9SAndreas Gohrfunction ft_pageLookup($id, $in_ns=false, $in_title=false){
2298d22f1e9SAndreas Gohr    $data = compact('id', 'in_ns', 'in_title');
2308d22f1e9SAndreas Gohr    $data['has_titles'] = true; // for plugin backward compatibility check
2316840140fSChris Smith    return trigger_event('SEARCH_QUERY_PAGELOOKUP', $data, '_ft_pageLookup');
2326840140fSChris Smith}
2336840140fSChris Smith
23442ea7f44SGerrit Uitslag/**
23542ea7f44SGerrit Uitslag * Returns list of pages as array(pageid => First Heading)
23642ea7f44SGerrit Uitslag *
23742ea7f44SGerrit Uitslag * @param array &$data event data
23842ea7f44SGerrit Uitslag * @return string[]
23942ea7f44SGerrit Uitslag */
2406840140fSChris Smithfunction _ft_pageLookup(&$data){
24180423ab6SAdrian Lang    // split out original parameters
2426840140fSChris Smith    $id = $data['id'];
243940f24fcSMichael Große    $Indexer = idx_get_indexer();
244940f24fcSMichael Große    $parsedQuery = ft_queryParser($Indexer, $id);
245940f24fcSMichael Große    if (count($parsedQuery['ns']) > 0) {
246940f24fcSMichael Große        $ns = cleanID($parsedQuery['ns'][0]) . ':';
247940f24fcSMichael Große        $id = implode(' ', $parsedQuery['highlight']);
248b0f6db0cSAdrian Lang    }
249b0f6db0cSAdrian Lang
2508d22f1e9SAndreas Gohr    $in_ns    = $data['in_ns'];
2518d22f1e9SAndreas Gohr    $in_title = $data['in_title'];
25280423ab6SAdrian Lang    $cleaned = cleanID($id);
2539b41be24STom N Harris
2549b41be24STom N Harris    $Indexer = idx_get_indexer();
2559b41be24STom N Harris    $page_idx = $Indexer->getPages();
2569b41be24STom N Harris
2579b41be24STom N Harris    $pages = array();
2585479a8c3SAndreas Gohr    if ($id !== '' && $cleaned !== '') {
2599b41be24STom N Harris        foreach ($page_idx as $p_id) {
2609b41be24STom N Harris            if ((strpos($in_ns ? $p_id : noNSorNS($p_id), $cleaned) !== false)) {
2619b41be24STom N Harris                if (!isset($pages[$p_id]))
26267c15eceSMichael Hamann                    $pages[$p_id] = p_get_first_heading($p_id, METADATA_DONT_RENDER);
263506fa893SAndreas Gohr            }
264506fa893SAndreas Gohr        }
265f078bb00STom N Harris        if ($in_title) {
266c66f16a3SMichael Hamann            foreach ($Indexer->lookupKey('title', $id, '_ft_pageLookupTitleCompare') as $p_id) {
267f078bb00STom N Harris                if (!isset($pages[$p_id]))
26867c15eceSMichael Hamann                    $pages[$p_id] = p_get_first_heading($p_id, METADATA_DONT_RENDER);
269f078bb00STom N Harris            }
270f078bb00STom N Harris        }
271d0bdf765SAdrian Lang    }
2720c074a52SMichael Hamann
273d0bdf765SAdrian Lang    if (isset($ns)) {
2740c074a52SMichael Hamann        foreach (array_keys($pages) as $p_id) {
2750c074a52SMichael Hamann            if (strpos($p_id, $ns) !== 0) {
2760c074a52SMichael Hamann                unset($pages[$p_id]);
277d0bdf765SAdrian Lang            }
278d0bdf765SAdrian Lang        }
279506fa893SAndreas Gohr    }
28063773904SAndreas Gohr
28180423ab6SAdrian Lang    // discard hidden pages
28280423ab6SAdrian Lang    // discard nonexistent pages
28363773904SAndreas Gohr    // check ACL permissions
28463773904SAndreas Gohr    foreach(array_keys($pages) as $idx){
28580423ab6SAdrian Lang        if(!isVisiblePage($idx) || !page_exists($idx) ||
28680423ab6SAdrian Lang           auth_quickaclcheck($idx) < AUTH_READ) {
28763773904SAndreas Gohr            unset($pages[$idx]);
28863773904SAndreas Gohr        }
28963773904SAndreas Gohr    }
29063773904SAndreas Gohr
2911b48999cSMichael Große    $pages = _ft_filterResultsByTime($pages);
2921b48999cSMichael Große
2933d2017d9SAdrian Lang    uksort($pages,'ft_pagesorter');
2948d22f1e9SAndreas Gohr    return $pages;
295506fa893SAndreas Gohr}
296506fa893SAndreas Gohr
2971b48999cSMichael Große
2981b48999cSMichael Große/**
2991b48999cSMichael Große * @param array $results search results in the form pageid => value
3001b48999cSMichael Große *
3011b48999cSMichael Große * @return array
3021b48999cSMichael Große */
3031b48999cSMichael Großefunction _ft_filterResultsByTime(array $results) {
3041b48999cSMichael Große    global $INPUT;
3051b48999cSMichael Große    if ($INPUT->has('after') || $INPUT->has('before')) {
3061b48999cSMichael Große        $after = $INPUT->str('after');
3071b48999cSMichael Große        $after = is_int($after) ? $after : strtotime($after);
3081b48999cSMichael Große
3091b48999cSMichael Große        $before = $INPUT->str('before');
3101b48999cSMichael Große        $before = is_int($before) ? $before : strtotime($before);
3111b48999cSMichael Große
3121b48999cSMichael Große        foreach ($results as $id => $value) {
3131b48999cSMichael Große            $mTime = filemtime(wikiFN($id));
3141b48999cSMichael Große            if ($after && $after > $mTime) {
3151b48999cSMichael Große                unset($results[$id]);
3161b48999cSMichael Große                continue;
3171b48999cSMichael Große            }
3181b48999cSMichael Große            if ($before && $before < $mTime) {
3191b48999cSMichael Große                unset($results[$id]);
3201b48999cSMichael Große            }
3211b48999cSMichael Große        }
3221b48999cSMichael Große    }
3231b48999cSMichael Große
3241b48999cSMichael Große    return $results;
3251b48999cSMichael Große}
3261b48999cSMichael Große
327506fa893SAndreas Gohr/**
328c66f16a3SMichael Hamann * Tiny helper function for comparing the searched title with the title
329c66f16a3SMichael Hamann * from the search index. This function is a wrapper around stripos with
330c66f16a3SMichael Hamann * adapted argument order and return value.
33142ea7f44SGerrit Uitslag *
33242ea7f44SGerrit Uitslag * @param string $search searched title
33342ea7f44SGerrit Uitslag * @param string $title  title from index
33442ea7f44SGerrit Uitslag * @return bool
335c66f16a3SMichael Hamann */
336c66f16a3SMichael Hamannfunction _ft_pageLookupTitleCompare($search, $title) {
337c66f16a3SMichael Hamann    return stripos($title, $search) !== false;
338c66f16a3SMichael Hamann}
339c66f16a3SMichael Hamann
340c66f16a3SMichael Hamann/**
341f31eb72bSAndreas Gohr * Sort pages based on their namespace level first, then on their string
342f31eb72bSAndreas Gohr * values. This makes higher hierarchy pages rank higher than lower hierarchy
343f31eb72bSAndreas Gohr * pages.
34442ea7f44SGerrit Uitslag *
34542ea7f44SGerrit Uitslag * @param string $a
34642ea7f44SGerrit Uitslag * @param string $b
34742ea7f44SGerrit Uitslag * @return int Returns < 0 if $a is less than $b; > 0 if $a is greater than $b, and 0 if they are equal.
348f31eb72bSAndreas Gohr */
349f31eb72bSAndreas Gohrfunction ft_pagesorter($a, $b){
350f31eb72bSAndreas Gohr    $ac = count(explode(':',$a));
351f31eb72bSAndreas Gohr    $bc = count(explode(':',$b));
352f31eb72bSAndreas Gohr    if($ac < $bc){
353f31eb72bSAndreas Gohr        return -1;
354f31eb72bSAndreas Gohr    }elseif($ac > $bc){
355f31eb72bSAndreas Gohr        return 1;
356f31eb72bSAndreas Gohr    }
357f31eb72bSAndreas Gohr    return strcmp ($a,$b);
358f31eb72bSAndreas Gohr}
359f31eb72bSAndreas Gohr
360f31eb72bSAndreas Gohr/**
361*8d0e286aSMichael Große * Sort pages by their mtime, from newest to oldest
362*8d0e286aSMichael Große *
363*8d0e286aSMichael Große * @param string $a
364*8d0e286aSMichael Große * @param string $b
365*8d0e286aSMichael Große *
366*8d0e286aSMichael Große * @return int Returns < 0 if $a is newer than $b, > 0 if $b is newer than $a and 0 if they are of the same age
367*8d0e286aSMichael Große */
368*8d0e286aSMichael Großefunction ft_pagemtimesorter($a, $b) {
369*8d0e286aSMichael Große    $mtimeA = filemtime(wikiFN($a));
370*8d0e286aSMichael Große    $mtimeB = filemtime(wikiFN($b));
371*8d0e286aSMichael Große    return $mtimeB - $mtimeA;
372*8d0e286aSMichael Große}
373*8d0e286aSMichael Große
374*8d0e286aSMichael Große/**
375506fa893SAndreas Gohr * Creates a snippet extract
376506fa893SAndreas Gohr *
377506fa893SAndreas Gohr * @author Andreas Gohr <andi@splitbrain.org>
37860e91a17SAndreas Gohr * @triggers FULLTEXT_SNIPPET_CREATE
37942ea7f44SGerrit Uitslag *
38042ea7f44SGerrit Uitslag * @param string $id page id
38142ea7f44SGerrit Uitslag * @param array $highlight
38242ea7f44SGerrit Uitslag * @return mixed
383506fa893SAndreas Gohr */
384546d3a99SAndreas Gohrfunction ft_snippet($id,$highlight){
385506fa893SAndreas Gohr    $text = rawWiki($id);
3864f0030ddSAndreas Gohr    $text = str_replace("\xC2\xAD",'',$text); // remove soft-hyphens
38760e91a17SAndreas Gohr    $evdata = array(
38860e91a17SAndreas Gohr            'id'        => $id,
38960e91a17SAndreas Gohr            'text'      => &$text,
39060e91a17SAndreas Gohr            'highlight' => &$highlight,
39160e91a17SAndreas Gohr            'snippet'   => '',
39260e91a17SAndreas Gohr            );
39360e91a17SAndreas Gohr
39460e91a17SAndreas Gohr    $evt = new Doku_Event('FULLTEXT_SNIPPET_CREATE',$evdata);
39560e91a17SAndreas Gohr    if ($evt->advise_before()) {
396ced0762eSchris        $match = array();
397ced0762eSchris        $snippets = array();
3989ee93076Schris        $utf8_offset = $offset = $end = 0;
399ced0762eSchris        $len = utf8_strlen($text);
4009ee93076Schris
401546d3a99SAndreas Gohr        // build a regexp from the phrases to highlight
4022237b4faSAndreas Gohr        $re1 = '('.join('|',array_map('ft_snippet_re_preprocess', array_map('preg_quote_cb',array_filter((array) $highlight)))).')';
403b571ff2dSChuck Kollars        $re2 = "$re1.{0,75}(?!\\1)$re1";
404b571ff2dSChuck Kollars        $re3 = "$re1.{0,45}(?!\\1)$re1.{0,45}(?!\\1)(?!\\2)$re1";
405546d3a99SAndreas Gohr
406b571ff2dSChuck Kollars        for ($cnt=4; $cnt--;) {
407b571ff2dSChuck Kollars            if (0) {
408b571ff2dSChuck Kollars            } else if (preg_match('/'.$re3.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
409b571ff2dSChuck Kollars            } else if (preg_match('/'.$re2.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
410b571ff2dSChuck Kollars            } else if (preg_match('/'.$re1.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
411b571ff2dSChuck Kollars            } else {
412b571ff2dSChuck Kollars                break;
413b571ff2dSChuck Kollars            }
414ced0762eSchris
415ced0762eSchris            list($str,$idx) = $match[0];
416ced0762eSchris
417ced0762eSchris            // convert $idx (a byte offset) into a utf8 character offset
418ced0762eSchris            $utf8_idx = utf8_strlen(substr($text,0,$idx));
419ced0762eSchris            $utf8_len = utf8_strlen($str);
420ced0762eSchris
421ced0762eSchris            // establish context, 100 bytes surrounding the match string
422ced0762eSchris            // first look to see if we can go 100 either side,
423ced0762eSchris            // then drop to 50 adding any excess if the other side can't go to 50,
424ced0762eSchris            $pre = min($utf8_idx-$utf8_offset,100);
425ced0762eSchris            $post = min($len-$utf8_idx-$utf8_len,100);
426ced0762eSchris
427ced0762eSchris            if ($pre>50 && $post>50) {
428ced0762eSchris                $pre = $post = 50;
429ced0762eSchris            } else if ($pre>50) {
430ced0762eSchris                $pre = min($pre,100-$post);
431ced0762eSchris            } else if ($post>50) {
432ced0762eSchris                $post = min($post, 100-$pre);
433ef3e3cddSMichael Hamann            } else if ($offset == 0) {
434ced0762eSchris                // both are less than 50, means the context is the whole string
43510ffc9ddSAndreas Gohr                // make it so and break out of this loop - there is no need for the
43610ffc9ddSAndreas Gohr                // complex snippet calculations
437ced0762eSchris                $snippets = array($text);
438ced0762eSchris                break;
439ced0762eSchris            }
440ced0762eSchris
44110ffc9ddSAndreas Gohr            // establish context start and end points, try to append to previous
44210ffc9ddSAndreas Gohr            // context if possible
4439ee93076Schris            $start = $utf8_idx - $pre;
444ced0762eSchris            $append = ($start < $end) ? $end : false;  // still the end of the previous context snippet
4459ee93076Schris            $end = $utf8_idx + $utf8_len + $post;      // now set it to the end of this context
446ced0762eSchris
447ced0762eSchris            if ($append) {
448ced0762eSchris                $snippets[count($snippets)-1] .= utf8_substr($text,$append,$end-$append);
449ced0762eSchris            } else {
450ced0762eSchris                $snippets[] = utf8_substr($text,$start,$end-$start);
451ced0762eSchris            }
452ced0762eSchris
453ced0762eSchris            // set $offset for next match attempt
45443d58b76SMichael Hamann            // continue matching after the current match
45543d58b76SMichael Hamann            // if the current match is not the longest possible match starting at the current offset
45643d58b76SMichael Hamann            // this prevents further matching of this snippet but for possible matches of length
45743d58b76SMichael Hamann            // smaller than match length + context (at least 50 characters) this match is part of the context
45843d58b76SMichael Hamann            $utf8_offset = $utf8_idx + $utf8_len;
45943d58b76SMichael Hamann            $offset = $idx + strlen(utf8_substr($text,$utf8_idx,$utf8_len));
4609ee93076Schris            $offset = utf8_correctIdx($text,$offset);
4619ee93076Schris        }
4629ee93076Schris
463ced0762eSchris        $m = "\1";
464b571ff2dSChuck Kollars        $snippets = preg_replace('/'.$re1.'/iu',$m.'$1'.$m,$snippets);
465b571ff2dSChuck Kollars        $snippet = preg_replace('/'.$m.'([^'.$m.']*?)'.$m.'/iu','<strong class="search_hit">$1</strong>',hsc(join('... ',$snippets)));
466bd2cb6fcSchris
46760e91a17SAndreas Gohr        $evdata['snippet'] = $snippet;
46860e91a17SAndreas Gohr    }
46960e91a17SAndreas Gohr    $evt->advise_after();
47060e91a17SAndreas Gohr    unset($evt);
47160e91a17SAndreas Gohr
47260e91a17SAndreas Gohr    return $evdata['snippet'];
473506fa893SAndreas Gohr}
474506fa893SAndreas Gohr
475506fa893SAndreas Gohr/**
47626eb848cSGina Haeussge * Wraps a search term in regex boundary checks.
47742ea7f44SGerrit Uitslag *
47842ea7f44SGerrit Uitslag * @param string $term
47942ea7f44SGerrit Uitslag * @return string
48026eb848cSGina Haeussge */
4812237b4faSAndreas Gohrfunction ft_snippet_re_preprocess($term) {
48235594613SKazutaka Miyasaka    // do not process asian terms where word boundaries are not explicit
48335594613SKazutaka Miyasaka    if(preg_match('/'.IDX_ASIAN.'/u',$term)){
48435594613SKazutaka Miyasaka        return $term;
48535594613SKazutaka Miyasaka    }
48635594613SKazutaka Miyasaka
4873161005dSAndreas Gohr    if (UTF8_PROPERTYSUPPORT) {
48884e581a6SAndreas Gohr        // unicode word boundaries
48984e581a6SAndreas Gohr        // see http://stackoverflow.com/a/2449017/172068
49084e581a6SAndreas Gohr        $BL = '(?<!\pL)';
49184e581a6SAndreas Gohr        $BR = '(?!\pL)';
4923161005dSAndreas Gohr    } else {
4933161005dSAndreas Gohr        // not as correct as above, but at least won't break
4943161005dSAndreas Gohr        $BL = '\b';
4953161005dSAndreas Gohr        $BR = '\b';
4963161005dSAndreas Gohr    }
4973161005dSAndreas Gohr
4982237b4faSAndreas Gohr    if(substr($term,0,2) == '\\*'){
4992237b4faSAndreas Gohr        $term = substr($term,2);
5002237b4faSAndreas Gohr    }else{
50184e581a6SAndreas Gohr        $term = $BL.$term;
5022237b4faSAndreas Gohr    }
5032237b4faSAndreas Gohr
5042237b4faSAndreas Gohr    if(substr($term,-2,2) == '\\*'){
5052237b4faSAndreas Gohr        $term = substr($term,0,-2);
5062237b4faSAndreas Gohr    }else{
50784e581a6SAndreas Gohr        $term = $term.$BR;
5082237b4faSAndreas Gohr    }
5098a803caeSAndreas Gohr
51084e581a6SAndreas Gohr    if($term == $BL || $term == $BR || $term == $BL.$BR) $term = '';
5112237b4faSAndreas Gohr    return $term;
51226eb848cSGina Haeussge}
51326eb848cSGina Haeussge
51426eb848cSGina Haeussge/**
515f5eb7cf0SAndreas Gohr * Combine found documents and sum up their scores
516f5eb7cf0SAndreas Gohr *
517f5eb7cf0SAndreas Gohr * This function is used to combine searched words with a logical
518f5eb7cf0SAndreas Gohr * AND. Only documents available in all arrays are returned.
519f5eb7cf0SAndreas Gohr *
520f5eb7cf0SAndreas Gohr * based upon PEAR's PHP_Compat function for array_intersect_key()
521f5eb7cf0SAndreas Gohr *
522f5eb7cf0SAndreas Gohr * @param array $args An array of page arrays
52342ea7f44SGerrit Uitslag * @return array
524f5eb7cf0SAndreas Gohr */
525f5eb7cf0SAndreas Gohrfunction ft_resultCombine($args){
526f5eb7cf0SAndreas Gohr    $array_count = count($args);
527134f4ab2SAndreas Gohr    if($array_count == 1){
528134f4ab2SAndreas Gohr        return $args[0];
529134f4ab2SAndreas Gohr    }
530134f4ab2SAndreas Gohr
531f5eb7cf0SAndreas Gohr    $result = array();
53209c27a6dSGuy Brand    if ($array_count > 1) {
533a21136cdSAndreas Gohr        foreach ($args[0] as $key => $value) {
534a21136cdSAndreas Gohr            $result[$key] = $value;
535f5eb7cf0SAndreas Gohr            for ($i = 1; $i !== $array_count; $i++) {
536a21136cdSAndreas Gohr                if (!isset($args[$i][$key])) {
537a21136cdSAndreas Gohr                    unset($result[$key]);
538a21136cdSAndreas Gohr                    break;
539f5eb7cf0SAndreas Gohr                }
540a21136cdSAndreas Gohr                $result[$key] += $args[$i][$key];
541f5eb7cf0SAndreas Gohr            }
542f5eb7cf0SAndreas Gohr        }
54309c27a6dSGuy Brand    }
544f5eb7cf0SAndreas Gohr    return $result;
545f5eb7cf0SAndreas Gohr}
546f5eb7cf0SAndreas Gohr
547f5eb7cf0SAndreas Gohr/**
548865c2687SKazutaka Miyasaka * Unites found documents and sum up their scores
549f5eb7cf0SAndreas Gohr *
550865c2687SKazutaka Miyasaka * based upon ft_resultCombine() function
551865c2687SKazutaka Miyasaka *
552865c2687SKazutaka Miyasaka * @param array $args An array of page arrays
55342ea7f44SGerrit Uitslag * @return array
55442ea7f44SGerrit Uitslag *
555865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
556865c2687SKazutaka Miyasaka */
557865c2687SKazutaka Miyasakafunction ft_resultUnite($args) {
558865c2687SKazutaka Miyasaka    $array_count = count($args);
559865c2687SKazutaka Miyasaka    if ($array_count === 1) {
560865c2687SKazutaka Miyasaka        return $args[0];
561865c2687SKazutaka Miyasaka    }
562865c2687SKazutaka Miyasaka
563865c2687SKazutaka Miyasaka    $result = $args[0];
564865c2687SKazutaka Miyasaka    for ($i = 1; $i !== $array_count; $i++) {
565865c2687SKazutaka Miyasaka        foreach (array_keys($args[$i]) as $id) {
566865c2687SKazutaka Miyasaka            $result[$id] += $args[$i][$id];
567865c2687SKazutaka Miyasaka        }
568865c2687SKazutaka Miyasaka    }
569865c2687SKazutaka Miyasaka    return $result;
570865c2687SKazutaka Miyasaka}
571865c2687SKazutaka Miyasaka
572865c2687SKazutaka Miyasaka/**
573865c2687SKazutaka Miyasaka * Computes the difference of documents using page id for comparison
574865c2687SKazutaka Miyasaka *
575865c2687SKazutaka Miyasaka * nearly identical to PHP5's array_diff_key()
576865c2687SKazutaka Miyasaka *
577865c2687SKazutaka Miyasaka * @param array $args An array of page arrays
57842ea7f44SGerrit Uitslag * @return array
57942ea7f44SGerrit Uitslag *
580865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
581865c2687SKazutaka Miyasaka */
582865c2687SKazutaka Miyasakafunction ft_resultComplement($args) {
583865c2687SKazutaka Miyasaka    $array_count = count($args);
584865c2687SKazutaka Miyasaka    if ($array_count === 1) {
585865c2687SKazutaka Miyasaka        return $args[0];
586865c2687SKazutaka Miyasaka    }
587865c2687SKazutaka Miyasaka
588865c2687SKazutaka Miyasaka    $result = $args[0];
589865c2687SKazutaka Miyasaka    foreach (array_keys($result) as $id) {
590865c2687SKazutaka Miyasaka        for ($i = 1; $i !== $array_count; $i++) {
591865c2687SKazutaka Miyasaka            if (isset($args[$i][$id])) unset($result[$id]);
592865c2687SKazutaka Miyasaka        }
593865c2687SKazutaka Miyasaka    }
594865c2687SKazutaka Miyasaka    return $result;
595865c2687SKazutaka Miyasaka}
596865c2687SKazutaka Miyasaka
597865c2687SKazutaka Miyasaka/**
598865c2687SKazutaka Miyasaka * Parses a search query and builds an array of search formulas
599865c2687SKazutaka Miyasaka *
600865c2687SKazutaka Miyasaka * @author Andreas Gohr <andi@splitbrain.org>
601865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
60242ea7f44SGerrit Uitslag *
60342ea7f44SGerrit Uitslag * @param Doku_Indexer $Indexer
60442ea7f44SGerrit Uitslag * @param string $query search query
60542ea7f44SGerrit Uitslag * @return array of search formulas
606f5eb7cf0SAndreas Gohr */
6079b41be24STom N Harrisfunction ft_queryParser($Indexer, $query){
608865c2687SKazutaka Miyasaka    /**
609865c2687SKazutaka Miyasaka     * parse a search query and transform it into intermediate representation
610865c2687SKazutaka Miyasaka     *
611865c2687SKazutaka Miyasaka     * in a search query, you can use the following expressions:
612865c2687SKazutaka Miyasaka     *
613865c2687SKazutaka Miyasaka     *   words:
614865c2687SKazutaka Miyasaka     *     include
615865c2687SKazutaka Miyasaka     *     -exclude
616865c2687SKazutaka Miyasaka     *   phrases:
617865c2687SKazutaka Miyasaka     *     "phrase to be included"
618865c2687SKazutaka Miyasaka     *     -"phrase you want to exclude"
619865c2687SKazutaka Miyasaka     *   namespaces:
620865c2687SKazutaka Miyasaka     *     @include:namespace (or ns:include:namespace)
621865c2687SKazutaka Miyasaka     *     ^exclude:namespace (or -ns:exclude:namespace)
622865c2687SKazutaka Miyasaka     *   groups:
623865c2687SKazutaka Miyasaka     *     ()
624865c2687SKazutaka Miyasaka     *     -()
625865c2687SKazutaka Miyasaka     *   operators:
626865c2687SKazutaka Miyasaka     *     and ('and' is the default operator: you can always omit this)
6277871d415SKazutaka Miyasaka     *     or  (or pipe symbol '|', lower precedence than 'and')
628865c2687SKazutaka Miyasaka     *
629865c2687SKazutaka Miyasaka     * e.g. a query [ aa "bb cc" @dd:ee ] means "search pages which contain
630865c2687SKazutaka Miyasaka     *      a word 'aa', a phrase 'bb cc' and are within a namespace 'dd:ee'".
631865c2687SKazutaka Miyasaka     *      this query is equivalent to [ -(-aa or -"bb cc" or -ns:dd:ee) ]
632865c2687SKazutaka Miyasaka     *      as long as you don't mind hit counts.
633865c2687SKazutaka Miyasaka     *
634865c2687SKazutaka Miyasaka     * intermediate representation consists of the following parts:
635865c2687SKazutaka Miyasaka     *
636865c2687SKazutaka Miyasaka     *   ( )           - group
637865c2687SKazutaka Miyasaka     *   AND           - logical and
638865c2687SKazutaka Miyasaka     *   OR            - logical or
639865c2687SKazutaka Miyasaka     *   NOT           - logical not
6402f502d70SKazutaka Miyasaka     *   W+:, W-:, W_: - word      (underscore: no need to highlight)
6412f502d70SKazutaka Miyasaka     *   P+:, P-:      - phrase    (minus sign: logically in NOT group)
6422f502d70SKazutaka Miyasaka     *   N+:, N-:      - namespace
643865c2687SKazutaka Miyasaka     */
644865c2687SKazutaka Miyasaka    $parsed_query = '';
645865c2687SKazutaka Miyasaka    $parens_level = 0;
646865c2687SKazutaka Miyasaka    $terms = preg_split('/(-?".*?")/u', utf8_strtolower($query), -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
647865c2687SKazutaka Miyasaka
648865c2687SKazutaka Miyasaka    foreach ($terms as $term) {
649865c2687SKazutaka Miyasaka        $parsed = '';
650865c2687SKazutaka Miyasaka        if (preg_match('/^(-?)"(.+)"$/u', $term, $matches)) {
651865c2687SKazutaka Miyasaka            // phrase-include and phrase-exclude
652865c2687SKazutaka Miyasaka            $not = $matches[1] ? 'NOT' : '';
6539b41be24STom N Harris            $parsed = $not.ft_termParser($Indexer, $matches[2], false, true);
654f5eb7cf0SAndreas Gohr        } else {
655865c2687SKazutaka Miyasaka            // fix incomplete phrase
656865c2687SKazutaka Miyasaka            $term = str_replace('"', ' ', $term);
657865c2687SKazutaka Miyasaka
658865c2687SKazutaka Miyasaka            // fix parentheses
659865c2687SKazutaka Miyasaka            $term = str_replace(')'  , ' ) ', $term);
660865c2687SKazutaka Miyasaka            $term = str_replace('('  , ' ( ', $term);
661865c2687SKazutaka Miyasaka            $term = str_replace('- (', ' -(', $term);
662865c2687SKazutaka Miyasaka
6637871d415SKazutaka Miyasaka            // treat pipe symbols as 'OR' operators
6647871d415SKazutaka Miyasaka            $term = str_replace('|', ' or ', $term);
6657871d415SKazutaka Miyasaka
666865c2687SKazutaka Miyasaka            // treat ideographic spaces (U+3000) as search term separators
667865c2687SKazutaka Miyasaka            // FIXME: some more separators?
668865c2687SKazutaka Miyasaka            $term = preg_replace('/[ \x{3000}]+/u', ' ',  $term);
669865c2687SKazutaka Miyasaka            $term = trim($term);
670865c2687SKazutaka Miyasaka            if ($term === '') continue;
671865c2687SKazutaka Miyasaka
672865c2687SKazutaka Miyasaka            $tokens = explode(' ', $term);
673865c2687SKazutaka Miyasaka            foreach ($tokens as $token) {
674865c2687SKazutaka Miyasaka                if ($token === '(') {
675865c2687SKazutaka Miyasaka                    // parenthesis-include-open
676865c2687SKazutaka Miyasaka                    $parsed .= '(';
677865c2687SKazutaka Miyasaka                    ++$parens_level;
678865c2687SKazutaka Miyasaka                } elseif ($token === '-(') {
679865c2687SKazutaka Miyasaka                    // parenthesis-exclude-open
680865c2687SKazutaka Miyasaka                    $parsed .= 'NOT(';
681865c2687SKazutaka Miyasaka                    ++$parens_level;
682865c2687SKazutaka Miyasaka                } elseif ($token === ')') {
683865c2687SKazutaka Miyasaka                    // parenthesis-any-close
684865c2687SKazutaka Miyasaka                    if ($parens_level === 0) continue;
685865c2687SKazutaka Miyasaka                    $parsed .= ')';
686865c2687SKazutaka Miyasaka                    $parens_level--;
687865c2687SKazutaka Miyasaka                } elseif ($token === 'and') {
688865c2687SKazutaka Miyasaka                    // logical-and (do nothing)
689865c2687SKazutaka Miyasaka                } elseif ($token === 'or') {
690865c2687SKazutaka Miyasaka                    // logical-or
691865c2687SKazutaka Miyasaka                    $parsed .= 'OR';
692865c2687SKazutaka Miyasaka                } elseif (preg_match('/^(?:\^|-ns:)(.+)$/u', $token, $matches)) {
693865c2687SKazutaka Miyasaka                    // namespace-exclude
6942f502d70SKazutaka Miyasaka                    $parsed .= 'NOT(N+:'.$matches[1].')';
695865c2687SKazutaka Miyasaka                } elseif (preg_match('/^(?:@|ns:)(.+)$/u', $token, $matches)) {
696865c2687SKazutaka Miyasaka                    // namespace-include
6972f502d70SKazutaka Miyasaka                    $parsed .= '(N+:'.$matches[1].')';
698865c2687SKazutaka Miyasaka                } elseif (preg_match('/^-(.+)$/', $token, $matches)) {
699865c2687SKazutaka Miyasaka                    // word-exclude
7009b41be24STom N Harris                    $parsed .= 'NOT('.ft_termParser($Indexer, $matches[1]).')';
701865c2687SKazutaka Miyasaka                } else {
702865c2687SKazutaka Miyasaka                    // word-include
7039b41be24STom N Harris                    $parsed .= ft_termParser($Indexer, $token);
704865c2687SKazutaka Miyasaka                }
705865c2687SKazutaka Miyasaka            }
706865c2687SKazutaka Miyasaka        }
707865c2687SKazutaka Miyasaka        $parsed_query .= $parsed;
708f5eb7cf0SAndreas Gohr    }
709f5eb7cf0SAndreas Gohr
710865c2687SKazutaka Miyasaka    // cleanup (very sensitive)
711865c2687SKazutaka Miyasaka    $parsed_query .= str_repeat(')', $parens_level);
712865c2687SKazutaka Miyasaka    do {
713865c2687SKazutaka Miyasaka        $parsed_query_old = $parsed_query;
714865c2687SKazutaka Miyasaka        $parsed_query = preg_replace('/(NOT)?\(\)/u', '', $parsed_query);
715865c2687SKazutaka Miyasaka    } while ($parsed_query !== $parsed_query_old);
716865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/(NOT|OR)+\)/u', ')'      , $parsed_query);
717865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/(OR)+/u'      , 'OR'     , $parsed_query);
718865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/\(OR/u'       , '('      , $parsed_query);
719865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/^OR|OR$/u'    , ''       , $parsed_query);
720865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/\)(NOT)?\(/u' , ')AND$1(', $parsed_query);
721865c2687SKazutaka Miyasaka
7222f502d70SKazutaka Miyasaka    // adjustment: make highlightings right
7232f502d70SKazutaka Miyasaka    $parens_level     = 0;
7242f502d70SKazutaka Miyasaka    $notgrp_levels    = array();
7252f502d70SKazutaka Miyasaka    $parsed_query_new = '';
7262f502d70SKazutaka Miyasaka    $tokens = preg_split('/(NOT\(|[()])/u', $parsed_query, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
7272f502d70SKazutaka Miyasaka    foreach ($tokens as $token) {
7282f502d70SKazutaka Miyasaka        if ($token === 'NOT(') {
7292f502d70SKazutaka Miyasaka            $notgrp_levels[] = ++$parens_level;
7302f502d70SKazutaka Miyasaka        } elseif ($token === '(') {
7312f502d70SKazutaka Miyasaka            ++$parens_level;
7322f502d70SKazutaka Miyasaka        } elseif ($token === ')') {
7332f502d70SKazutaka Miyasaka            if ($parens_level-- === end($notgrp_levels)) array_pop($notgrp_levels);
7342f502d70SKazutaka Miyasaka        } elseif (count($notgrp_levels) % 2 === 1) {
7352f502d70SKazutaka Miyasaka            // turn highlight-flag off if terms are logically in "NOT" group
7362f502d70SKazutaka Miyasaka            $token = preg_replace('/([WPN])\+\:/u', '$1-:', $token);
7372f502d70SKazutaka Miyasaka        }
7382f502d70SKazutaka Miyasaka        $parsed_query_new .= $token;
7392f502d70SKazutaka Miyasaka    }
7402f502d70SKazutaka Miyasaka    $parsed_query = $parsed_query_new;
7412f502d70SKazutaka Miyasaka
742865c2687SKazutaka Miyasaka    /**
743865c2687SKazutaka Miyasaka     * convert infix notation string into postfix (Reverse Polish notation) array
744865c2687SKazutaka Miyasaka     * by Shunting-yard algorithm
745865c2687SKazutaka Miyasaka     *
746865c2687SKazutaka Miyasaka     * see: http://en.wikipedia.org/wiki/Reverse_Polish_notation
747865c2687SKazutaka Miyasaka     * see: http://en.wikipedia.org/wiki/Shunting-yard_algorithm
748865c2687SKazutaka Miyasaka     */
749865c2687SKazutaka Miyasaka    $parsed_ary     = array();
750865c2687SKazutaka Miyasaka    $ope_stack      = array();
751865c2687SKazutaka Miyasaka    $ope_precedence = array(')' => 1, 'OR' => 2, 'AND' => 3, 'NOT' => 4, '(' => 5);
752865c2687SKazutaka Miyasaka    $ope_regex      = '/([()]|OR|AND|NOT)/u';
753865c2687SKazutaka Miyasaka
754865c2687SKazutaka Miyasaka    $tokens = preg_split($ope_regex, $parsed_query, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
755865c2687SKazutaka Miyasaka    foreach ($tokens as $token) {
756865c2687SKazutaka Miyasaka        if (preg_match($ope_regex, $token)) {
757865c2687SKazutaka Miyasaka            // operator
758865c2687SKazutaka Miyasaka            $last_ope = end($ope_stack);
75967d812e0SMarius van Witzenburg            while ($last_ope !== false && $ope_precedence[$token] <= $ope_precedence[$last_ope] && $last_ope != '(') {
760865c2687SKazutaka Miyasaka                $parsed_ary[] = array_pop($ope_stack);
761865c2687SKazutaka Miyasaka                $last_ope = end($ope_stack);
762865c2687SKazutaka Miyasaka            }
763865c2687SKazutaka Miyasaka            if ($token == ')') {
764865c2687SKazutaka Miyasaka                array_pop($ope_stack); // this array_pop always deletes '('
765865c2687SKazutaka Miyasaka            } else {
766865c2687SKazutaka Miyasaka                $ope_stack[] = $token;
767865c2687SKazutaka Miyasaka            }
768865c2687SKazutaka Miyasaka        } else {
769865c2687SKazutaka Miyasaka            // operand
770865c2687SKazutaka Miyasaka            $token_decoded = str_replace(array('OP', 'CP'), array('(', ')'), $token);
771865c2687SKazutaka Miyasaka            $parsed_ary[] = $token_decoded;
772865c2687SKazutaka Miyasaka        }
773865c2687SKazutaka Miyasaka    }
774865c2687SKazutaka Miyasaka    $parsed_ary = array_values(array_merge($parsed_ary, array_reverse($ope_stack)));
775865c2687SKazutaka Miyasaka
776865c2687SKazutaka Miyasaka    // cleanup: each double "NOT" in RPN array actually does nothing
777865c2687SKazutaka Miyasaka    $parsed_ary_count = count($parsed_ary);
778865c2687SKazutaka Miyasaka    for ($i = 1; $i < $parsed_ary_count; ++$i) {
779865c2687SKazutaka Miyasaka        if ($parsed_ary[$i] === 'NOT' && $parsed_ary[$i - 1] === 'NOT') {
780865c2687SKazutaka Miyasaka            unset($parsed_ary[$i], $parsed_ary[$i - 1]);
781865c2687SKazutaka Miyasaka        }
782865c2687SKazutaka Miyasaka    }
783865c2687SKazutaka Miyasaka    $parsed_ary = array_values($parsed_ary);
784865c2687SKazutaka Miyasaka
785865c2687SKazutaka Miyasaka    // build return value
786f5eb7cf0SAndreas Gohr    $q = array();
787f5eb7cf0SAndreas Gohr    $q['query']      = $query;
788865c2687SKazutaka Miyasaka    $q['parsed_str'] = $parsed_query;
789865c2687SKazutaka Miyasaka    $q['parsed_ary'] = $parsed_ary;
790f5eb7cf0SAndreas Gohr
791865c2687SKazutaka Miyasaka    foreach ($q['parsed_ary'] as $token) {
792865c2687SKazutaka Miyasaka        if ($token[2] !== ':') continue;
793865c2687SKazutaka Miyasaka        $body = substr($token, 3);
794865c2687SKazutaka Miyasaka
795865c2687SKazutaka Miyasaka        switch (substr($token, 0, 3)) {
7962f502d70SKazutaka Miyasaka            case 'N+:':
797865c2687SKazutaka Miyasaka                     $q['ns'][]        = $body; // for backward compatibility
798865c2687SKazutaka Miyasaka                     break;
7992f502d70SKazutaka Miyasaka            case 'N-:':
8002f502d70SKazutaka Miyasaka                     $q['notns'][]     = $body; // for backward compatibility
8012f502d70SKazutaka Miyasaka                     break;
8022f502d70SKazutaka Miyasaka            case 'W_:':
8032f502d70SKazutaka Miyasaka                     $q['words'][]     = $body;
8042f502d70SKazutaka Miyasaka                     break;
805865c2687SKazutaka Miyasaka            case 'W-:':
806865c2687SKazutaka Miyasaka                     $q['words'][]     = $body;
8072f502d70SKazutaka Miyasaka                     $q['not'][]       = $body; // for backward compatibility
808865c2687SKazutaka Miyasaka                     break;
809865c2687SKazutaka Miyasaka            case 'W+:':
810865c2687SKazutaka Miyasaka                     $q['words'][]     = $body;
8112237b4faSAndreas Gohr                     $q['highlight'][] = $body;
8122f502d70SKazutaka Miyasaka                     $q['and'][]       = $body; // for backward compatibility
813865c2687SKazutaka Miyasaka                     break;
8142f502d70SKazutaka Miyasaka            case 'P-:':
8152f502d70SKazutaka Miyasaka                     $q['phrases'][]   = $body;
8162f502d70SKazutaka Miyasaka                     break;
8172f502d70SKazutaka Miyasaka            case 'P+:':
818865c2687SKazutaka Miyasaka                     $q['phrases'][]   = $body;
8192237b4faSAndreas Gohr                     $q['highlight'][] = $body;
820865c2687SKazutaka Miyasaka                     break;
821865c2687SKazutaka Miyasaka        }
822865c2687SKazutaka Miyasaka    }
8232f502d70SKazutaka Miyasaka    foreach (array('words', 'phrases', 'highlight', 'ns', 'notns', 'and', 'not') as $key) {
824865c2687SKazutaka Miyasaka        $q[$key] = empty($q[$key]) ? array() : array_values(array_unique($q[$key]));
825f5eb7cf0SAndreas Gohr    }
826f5eb7cf0SAndreas Gohr
827f5eb7cf0SAndreas Gohr    return $q;
828f5eb7cf0SAndreas Gohr}
829f5eb7cf0SAndreas Gohr
830865c2687SKazutaka Miyasaka/**
831865c2687SKazutaka Miyasaka * Transforms given search term into intermediate representation
832865c2687SKazutaka Miyasaka *
833865c2687SKazutaka Miyasaka * This function is used in ft_queryParser() and not for general purpose use.
834865c2687SKazutaka Miyasaka *
835865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
83642ea7f44SGerrit Uitslag *
83742ea7f44SGerrit Uitslag * @param Doku_Indexer $Indexer
83842ea7f44SGerrit Uitslag * @param string       $term
83942ea7f44SGerrit Uitslag * @param bool         $consider_asian
84042ea7f44SGerrit Uitslag * @param bool         $phrase_mode
84142ea7f44SGerrit Uitslag * @return string
842865c2687SKazutaka Miyasaka */
8439b41be24STom N Harrisfunction ft_termParser($Indexer, $term, $consider_asian = true, $phrase_mode = false) {
844865c2687SKazutaka Miyasaka    $parsed = '';
845865c2687SKazutaka Miyasaka    if ($consider_asian) {
846865c2687SKazutaka Miyasaka        // successive asian characters need to be searched as a phrase
847865c2687SKazutaka Miyasaka        $words = preg_split('/('.IDX_ASIAN.'+)/u', $term, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
848865c2687SKazutaka Miyasaka        foreach ($words as $word) {
8496ac2077aSKazutaka Miyasaka            $phrase_mode = $phrase_mode ? true : preg_match('/'.IDX_ASIAN.'/u', $word);
8509b41be24STom N Harris            $parsed .= ft_termParser($Indexer, $word, false, $phrase_mode);
851865c2687SKazutaka Miyasaka        }
852865c2687SKazutaka Miyasaka    } else {
853865c2687SKazutaka Miyasaka        $term_noparen = str_replace(array('(', ')'), ' ', $term);
8549b41be24STom N Harris        $words = $Indexer->tokenizer($term_noparen, true);
855865c2687SKazutaka Miyasaka
8562f502d70SKazutaka Miyasaka        // W_: no need to highlight
857865c2687SKazutaka Miyasaka        if (empty($words)) {
858865c2687SKazutaka Miyasaka            $parsed = '()'; // important: do not remove
859865c2687SKazutaka Miyasaka        } elseif ($words[0] === $term) {
860865c2687SKazutaka Miyasaka            $parsed = '(W+:'.$words[0].')';
861865c2687SKazutaka Miyasaka        } elseif ($phrase_mode) {
862865c2687SKazutaka Miyasaka            $term_encoded = str_replace(array('(', ')'), array('OP', 'CP'), $term);
8632f502d70SKazutaka Miyasaka            $parsed = '((W_:'.implode(')(W_:', $words).')(P+:'.$term_encoded.'))';
864865c2687SKazutaka Miyasaka        } else {
865865c2687SKazutaka Miyasaka            $parsed = '((W+:'.implode(')(W+:', $words).'))';
866865c2687SKazutaka Miyasaka        }
867865c2687SKazutaka Miyasaka    }
868865c2687SKazutaka Miyasaka    return $parsed;
869865c2687SKazutaka Miyasaka}
870865c2687SKazutaka Miyasaka
87144156e11SMichael Große/**
87244156e11SMichael Große * Recreate a search query string based on parsed parts, doesn't support negated phrases and `OR` searches
87344156e11SMichael Große *
87444156e11SMichael Große * @param array $and
87544156e11SMichael Große * @param array $not
87644156e11SMichael Große * @param array $phrases
87744156e11SMichael Große * @param array $ns
87844156e11SMichael Große * @param array $notns
87944156e11SMichael Große *
88044156e11SMichael Große * @return string
88144156e11SMichael Große */
88244156e11SMichael Großefunction ft_queryUnparser_simple(array $and, array $not, array $phrases, array $ns, array $notns) {
88344156e11SMichael Große    $query = implode(' ', $and);
88444156e11SMichael Große    if (!empty($not)) {
88544156e11SMichael Große        $query .= ' -' . implode(' -', $not);
88644156e11SMichael Große    }
88744156e11SMichael Große
88844156e11SMichael Große    if (!empty($phrases)) {
88944156e11SMichael Große        $query .= ' "' . implode('" "', $phrases) . '"';
89044156e11SMichael Große    }
89144156e11SMichael Große
89244156e11SMichael Große    if (!empty($ns)) {
89344156e11SMichael Große        $query .= ' @' . implode(' @', $ns);
89444156e11SMichael Große    }
89544156e11SMichael Große
89644156e11SMichael Große    if (!empty($notns)) {
89744156e11SMichael Große        $query .= ' ^' . implode(' ^', $notns);
89844156e11SMichael Große    }
89944156e11SMichael Große
90044156e11SMichael Große    return $query;
90144156e11SMichael Große}
90244156e11SMichael Große
903e3776c06SMichael Hamann//Setup VIM: ex: et ts=4 :
904