xref: /dokuwiki/inc/fulltext.php (revision 3850270ca84449d731fbd46587b055beb83221dd)
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
25*3850270cSMichael Große * @param string     $sort
26*3850270cSMichael Große * @param int|string $after  only show results with an modified time after this date, accepts timestap or strtotime arguments
27*3850270cSMichael Große * @param int|string $before only show results with an modified time before this date, accepts timestap or strtotime arguments
28*3850270cSMichael Große *
2942ea7f44SGerrit Uitslag * @return array
30f5eb7cf0SAndreas Gohr */
31*3850270cSMichael Großefunction ft_pageSearch($query,&$highlight, $sort = null, $after = null, $before = null){
326840140fSChris Smith
33*3850270cSMichael Große    if ($sort === null) {
34*3850270cSMichael Große        $sort = 'hits';
35*3850270cSMichael Große    }
36*3850270cSMichael Große    $data = [
37*3850270cSMichael Große        'query' => $query,
38*3850270cSMichael Große        'sort' => $sort,
39*3850270cSMichael Große        'after' => $after,
40*3850270cSMichael Große        'before' => $before
41*3850270cSMichael Große    ];
426840140fSChris Smith    $data['highlight'] =& $highlight;
436840140fSChris Smith
446840140fSChris Smith    return trigger_event('SEARCH_QUERY_FULLPAGE', $data, '_ft_pageSearch');
456840140fSChris Smith}
46865c2687SKazutaka Miyasaka
47865c2687SKazutaka Miyasaka/**
48865c2687SKazutaka Miyasaka * Returns a list of matching documents for the given query
49865c2687SKazutaka Miyasaka *
50865c2687SKazutaka Miyasaka * @author Andreas Gohr <andi@splitbrain.org>
51865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
5242ea7f44SGerrit Uitslag *
5342ea7f44SGerrit Uitslag * @param array $data event data
5442ea7f44SGerrit Uitslag * @return array matching documents
55865c2687SKazutaka Miyasaka */
566840140fSChris Smithfunction _ft_pageSearch(&$data) {
579b41be24STom N Harris    $Indexer = idx_get_indexer();
589b41be24STom N Harris
59865c2687SKazutaka Miyasaka    // parse the given query
609b41be24STom N Harris    $q = ft_queryParser($Indexer, $data['query']);
61865c2687SKazutaka Miyasaka    $data['highlight'] = $q['highlight'];
626840140fSChris Smith
63865c2687SKazutaka Miyasaka    if (empty($q['parsed_ary'])) return array();
64506fa893SAndreas Gohr
65f5eb7cf0SAndreas Gohr    // lookup all words found in the query
669b41be24STom N Harris    $lookup = $Indexer->lookup($q['words']);
67f5eb7cf0SAndreas Gohr
68865c2687SKazutaka Miyasaka    // get all pages in this dokuwiki site (!: includes nonexistent pages)
69865c2687SKazutaka Miyasaka    $pages_all = array();
709b41be24STom N Harris    foreach ($Indexer->getPages() as $id) {
719b41be24STom N Harris        $pages_all[$id] = 0; // base: 0 hit
72f5eb7cf0SAndreas Gohr    }
73f5eb7cf0SAndreas Gohr
74865c2687SKazutaka Miyasaka    // process the query
75865c2687SKazutaka Miyasaka    $stack = array();
76865c2687SKazutaka Miyasaka    foreach ($q['parsed_ary'] as $token) {
77865c2687SKazutaka Miyasaka        switch (substr($token, 0, 3)) {
78865c2687SKazutaka Miyasaka            case 'W+:':
792f502d70SKazutaka Miyasaka            case 'W-:':
802f502d70SKazutaka Miyasaka            case 'W_:': // word
81865c2687SKazutaka Miyasaka                $word    = substr($token, 3);
82865c2687SKazutaka Miyasaka                $stack[] = (array) $lookup[$word];
83865c2687SKazutaka Miyasaka                break;
842f502d70SKazutaka Miyasaka            case 'P+:':
852f502d70SKazutaka Miyasaka            case 'P-:': // phrase
86865c2687SKazutaka Miyasaka                $phrase = substr($token, 3);
87865c2687SKazutaka Miyasaka                // since phrases are always parsed as ((W1)(W2)...(P)),
88865c2687SKazutaka Miyasaka                // the end($stack) always points the pages that contain
89865c2687SKazutaka Miyasaka                // all words in this phrase
90865c2687SKazutaka Miyasaka                $pages  = end($stack);
91865c2687SKazutaka Miyasaka                $pages_matched = array();
92865c2687SKazutaka Miyasaka                foreach(array_keys($pages) as $id){
93a7e8b43eSMichael Hamann                    $evdata = array(
94a7e8b43eSMichael Hamann                        'id' => $id,
95a7e8b43eSMichael Hamann                        'phrase' => $phrase,
96a7e8b43eSMichael Hamann                        'text' => rawWiki($id)
97a7e8b43eSMichael Hamann                    );
98a7e8b43eSMichael Hamann                    $evt = new Doku_Event('FULLTEXT_PHRASE_MATCH',$evdata);
99a7e8b43eSMichael Hamann                    if ($evt->advise_before() && $evt->result !== true) {
100a7e8b43eSMichael Hamann                        $text = utf8_strtolower($evdata['text']);
101865c2687SKazutaka Miyasaka                        if (strpos($text, $phrase) !== false) {
102a7e8b43eSMichael Hamann                            $evt->result = true;
103a7e8b43eSMichael Hamann                        }
104a7e8b43eSMichael Hamann                    }
105a7e8b43eSMichael Hamann                    $evt->advise_after();
106a7e8b43eSMichael Hamann                    if ($evt->result === true) {
107865c2687SKazutaka Miyasaka                        $pages_matched[$id] = 0; // phrase: always 0 hit
108865c2687SKazutaka Miyasaka                    }
109865c2687SKazutaka Miyasaka                }
110865c2687SKazutaka Miyasaka                $stack[] = $pages_matched;
111865c2687SKazutaka Miyasaka                break;
1122f502d70SKazutaka Miyasaka            case 'N+:':
1132f502d70SKazutaka Miyasaka            case 'N-:': // namespace
114de3383c6SMichael Große                $ns = cleanID(substr($token, 3)) . ':';
115865c2687SKazutaka Miyasaka                $pages_matched = array();
116865c2687SKazutaka Miyasaka                foreach (array_keys($pages_all) as $id) {
117865c2687SKazutaka Miyasaka                    if (strpos($id, $ns) === 0) {
118865c2687SKazutaka Miyasaka                        $pages_matched[$id] = 0; // namespace: always 0 hit
119865c2687SKazutaka Miyasaka                    }
120865c2687SKazutaka Miyasaka                }
121865c2687SKazutaka Miyasaka                $stack[] = $pages_matched;
122865c2687SKazutaka Miyasaka                break;
123865c2687SKazutaka Miyasaka            case 'AND': // and operation
124865c2687SKazutaka Miyasaka                list($pages1, $pages2) = array_splice($stack, -2);
125865c2687SKazutaka Miyasaka                $stack[] = ft_resultCombine(array($pages1, $pages2));
126865c2687SKazutaka Miyasaka                break;
127865c2687SKazutaka Miyasaka            case 'OR':  // or operation
128865c2687SKazutaka Miyasaka                list($pages1, $pages2) = array_splice($stack, -2);
129865c2687SKazutaka Miyasaka                $stack[] = ft_resultUnite(array($pages1, $pages2));
130865c2687SKazutaka Miyasaka                break;
131865c2687SKazutaka Miyasaka            case 'NOT': // not operation (unary)
132865c2687SKazutaka Miyasaka                $pages   = array_pop($stack);
133865c2687SKazutaka Miyasaka                $stack[] = ft_resultComplement(array($pages_all, $pages));
134a21136cdSAndreas Gohr                break;
135a21136cdSAndreas Gohr        }
136f5eb7cf0SAndreas Gohr    }
137865c2687SKazutaka Miyasaka    $docs = array_pop($stack);
138865c2687SKazutaka Miyasaka
139865c2687SKazutaka Miyasaka    if (empty($docs)) return array();
140865c2687SKazutaka Miyasaka
141865c2687SKazutaka Miyasaka    // check: settings, acls, existence
142865c2687SKazutaka Miyasaka    foreach (array_keys($docs) as $id) {
143865c2687SKazutaka Miyasaka        if (isHiddenPage($id) || auth_quickaclcheck($id) < AUTH_READ || !page_exists($id, '', false)) {
144865c2687SKazutaka Miyasaka            unset($docs[$id]);
145f5eb7cf0SAndreas Gohr        }
146f5eb7cf0SAndreas Gohr    }
147f5eb7cf0SAndreas Gohr
148*3850270cSMichael Große    $docs = _ft_filterResultsByTime($docs, $data['after'], $data['before']);
1498d0e286aSMichael Große
1508d0e286aSMichael Große    if ($data['sort'] === 'mtime') {
1518d0e286aSMichael Große        uksort($docs, 'ft_pagemtimesorter');
1528d0e286aSMichael Große    } else {
153865c2687SKazutaka Miyasaka        // sort docs by count
154f5eb7cf0SAndreas Gohr        arsort($docs);
1558d0e286aSMichael Große    }
156f5eb7cf0SAndreas Gohr
157f5eb7cf0SAndreas Gohr    return $docs;
158f5eb7cf0SAndreas Gohr}
159f5eb7cf0SAndreas Gohr
160f5eb7cf0SAndreas Gohr/**
16154f4c056SAndreas Gohr * Returns the backlinks for a given page
16254f4c056SAndreas Gohr *
163320f489aSMichael Hamann * Uses the metadata index.
16407ff0babSMichael Hamann *
16507ff0babSMichael Hamann * @param string $id           The id for which links shall be returned
16607ff0babSMichael Hamann * @param bool   $ignore_perms Ignore the fact that pages are hidden or read-protected
16707ff0babSMichael Hamann * @return array The pages that contain links to the given page
16854f4c056SAndreas Gohr */
16907ff0babSMichael Hamannfunction ft_backlinks($id, $ignore_perms = false){
170320f489aSMichael Hamann    $result = idx_get_indexer()->lookupKey('relation_references', $id);
17154f4c056SAndreas Gohr
17263773904SAndreas Gohr    if(!count($result)) return $result;
17363773904SAndreas Gohr
17463773904SAndreas Gohr    // check ACL permissions
17563773904SAndreas Gohr    foreach(array_keys($result) as $idx){
17607ff0babSMichael Hamann        if(($ignore_perms !== true && (
17707ff0babSMichael Hamann                isHiddenPage($result[$idx]) || auth_quickaclcheck($result[$idx]) < AUTH_READ
17807ff0babSMichael Hamann            )) || !page_exists($result[$idx], '', false)){
17963773904SAndreas Gohr            unset($result[$idx]);
18063773904SAndreas Gohr        }
18163773904SAndreas Gohr    }
18263773904SAndreas Gohr
18354f4c056SAndreas Gohr    sort($result);
18454f4c056SAndreas Gohr    return $result;
18554f4c056SAndreas Gohr}
18654f4c056SAndreas Gohr
18754f4c056SAndreas Gohr/**
188a05e297aSAndreas Gohr * Returns the pages that use a given media file
189a05e297aSAndreas Gohr *
190ffec1009SMichael Hamann * Uses the relation media metadata property and the metadata index.
191a05e297aSAndreas Gohr *
192ffec1009SMichael Hamann * Note that before 2013-07-31 the second parameter was the maximum number of results and
193ffec1009SMichael Hamann * permissions were ignored. That's why the parameter is now checked to be explicitely set
194ffec1009SMichael Hamann * to true (with type bool) in order to be compatible with older uses of the function.
195ffec1009SMichael Hamann *
196ffec1009SMichael Hamann * @param string $id           The media id to look for
197ffec1009SMichael Hamann * @param bool   $ignore_perms Ignore hidden pages and acls (optional, default: false)
198ffec1009SMichael Hamann * @return array A list of pages that use the given media file
199a05e297aSAndreas Gohr */
200ffec1009SMichael Hamannfunction ft_mediause($id, $ignore_perms = false){
201ffec1009SMichael Hamann    $result = idx_get_indexer()->lookupKey('relation_media', $id);
202a05e297aSAndreas Gohr
203ffec1009SMichael Hamann    if(!count($result)) return $result;
204a05e297aSAndreas Gohr
205ffec1009SMichael Hamann    // check ACL permissions
206ffec1009SMichael Hamann    foreach(array_keys($result) as $idx){
207ffec1009SMichael Hamann        if(($ignore_perms !== true && (
208ffec1009SMichael Hamann                    isHiddenPage($result[$idx]) || auth_quickaclcheck($result[$idx]) < AUTH_READ
209ffec1009SMichael Hamann                )) || !page_exists($result[$idx], '', false)){
210ffec1009SMichael Hamann            unset($result[$idx]);
211a05e297aSAndreas Gohr        }
212a05e297aSAndreas Gohr    }
213a05e297aSAndreas Gohr
214a05e297aSAndreas Gohr    sort($result);
215a05e297aSAndreas Gohr    return $result;
216a05e297aSAndreas Gohr}
217a05e297aSAndreas Gohr
218a05e297aSAndreas Gohr
219a05e297aSAndreas Gohr/**
220506fa893SAndreas Gohr * Quicksearch for pagenames
221506fa893SAndreas Gohr *
222506fa893SAndreas Gohr * By default it only matches the pagename and ignores the
22380423ab6SAdrian Lang * namespace. This can be changed with the second parameter.
22480423ab6SAdrian Lang * The third parameter allows to search in titles as well.
225506fa893SAndreas Gohr *
2268d22f1e9SAndreas Gohr * The function always returns titles as well
2276840140fSChris Smith *
2288d22f1e9SAndreas Gohr * @triggers SEARCH_QUERY_PAGELOOKUP
229506fa893SAndreas Gohr * @author   Andreas Gohr <andi@splitbrain.org>
2308d22f1e9SAndreas Gohr * @author   Adrian Lang <lang@cosmocode.de>
23142ea7f44SGerrit Uitslag *
23242ea7f44SGerrit Uitslag * @param string     $id       page id
23342ea7f44SGerrit Uitslag * @param bool       $in_ns    match against namespace as well?
23442ea7f44SGerrit Uitslag * @param bool       $in_title search in title?
235*3850270cSMichael Große * @param int|string $after    only show results with an modified time after this date, accepts timestap or strtotime arguments
236*3850270cSMichael Große * @param int|string $before   only show results with an modified time before this date, accepts timestap or strtotime arguments
237*3850270cSMichael Große *
23842ea7f44SGerrit Uitslag * @return string[]
239506fa893SAndreas Gohr */
240*3850270cSMichael Großefunction ft_pageLookup($id, $in_ns=false, $in_title=false, $after = null, $before = null){
241*3850270cSMichael Große    $data = [
242*3850270cSMichael Große        'id' => $id,
243*3850270cSMichael Große        'in_ns' => $in_ns,
244*3850270cSMichael Große        'in_title' => $in_title,
245*3850270cSMichael Große        'after' => $after,
246*3850270cSMichael Große        'before' => $before
247*3850270cSMichael Große    ];
2488d22f1e9SAndreas Gohr    $data['has_titles'] = true; // for plugin backward compatibility check
2496840140fSChris Smith    return trigger_event('SEARCH_QUERY_PAGELOOKUP', $data, '_ft_pageLookup');
2506840140fSChris Smith}
2516840140fSChris Smith
25242ea7f44SGerrit Uitslag/**
25342ea7f44SGerrit Uitslag * Returns list of pages as array(pageid => First Heading)
25442ea7f44SGerrit Uitslag *
25542ea7f44SGerrit Uitslag * @param array &$data event data
25642ea7f44SGerrit Uitslag * @return string[]
25742ea7f44SGerrit Uitslag */
2586840140fSChris Smithfunction _ft_pageLookup(&$data){
25980423ab6SAdrian Lang    // split out original parameters
2606840140fSChris Smith    $id = $data['id'];
261940f24fcSMichael Große    $Indexer = idx_get_indexer();
262940f24fcSMichael Große    $parsedQuery = ft_queryParser($Indexer, $id);
263940f24fcSMichael Große    if (count($parsedQuery['ns']) > 0) {
264940f24fcSMichael Große        $ns = cleanID($parsedQuery['ns'][0]) . ':';
265940f24fcSMichael Große        $id = implode(' ', $parsedQuery['highlight']);
266b0f6db0cSAdrian Lang    }
267b0f6db0cSAdrian Lang
2688d22f1e9SAndreas Gohr    $in_ns    = $data['in_ns'];
2698d22f1e9SAndreas Gohr    $in_title = $data['in_title'];
27080423ab6SAdrian Lang    $cleaned = cleanID($id);
2719b41be24STom N Harris
2729b41be24STom N Harris    $Indexer = idx_get_indexer();
2739b41be24STom N Harris    $page_idx = $Indexer->getPages();
2749b41be24STom N Harris
2759b41be24STom N Harris    $pages = array();
2765479a8c3SAndreas Gohr    if ($id !== '' && $cleaned !== '') {
2779b41be24STom N Harris        foreach ($page_idx as $p_id) {
2789b41be24STom N Harris            if ((strpos($in_ns ? $p_id : noNSorNS($p_id), $cleaned) !== false)) {
2799b41be24STom N Harris                if (!isset($pages[$p_id]))
28067c15eceSMichael Hamann                    $pages[$p_id] = p_get_first_heading($p_id, METADATA_DONT_RENDER);
281506fa893SAndreas Gohr            }
282506fa893SAndreas Gohr        }
283f078bb00STom N Harris        if ($in_title) {
284c66f16a3SMichael Hamann            foreach ($Indexer->lookupKey('title', $id, '_ft_pageLookupTitleCompare') as $p_id) {
285f078bb00STom N Harris                if (!isset($pages[$p_id]))
28667c15eceSMichael Hamann                    $pages[$p_id] = p_get_first_heading($p_id, METADATA_DONT_RENDER);
287f078bb00STom N Harris            }
288f078bb00STom N Harris        }
289d0bdf765SAdrian Lang    }
2900c074a52SMichael Hamann
291d0bdf765SAdrian Lang    if (isset($ns)) {
2920c074a52SMichael Hamann        foreach (array_keys($pages) as $p_id) {
2930c074a52SMichael Hamann            if (strpos($p_id, $ns) !== 0) {
2940c074a52SMichael Hamann                unset($pages[$p_id]);
295d0bdf765SAdrian Lang            }
296d0bdf765SAdrian Lang        }
297506fa893SAndreas Gohr    }
29863773904SAndreas Gohr
29980423ab6SAdrian Lang    // discard hidden pages
30080423ab6SAdrian Lang    // discard nonexistent pages
30163773904SAndreas Gohr    // check ACL permissions
30263773904SAndreas Gohr    foreach(array_keys($pages) as $idx){
30380423ab6SAdrian Lang        if(!isVisiblePage($idx) || !page_exists($idx) ||
30480423ab6SAdrian Lang           auth_quickaclcheck($idx) < AUTH_READ) {
30563773904SAndreas Gohr            unset($pages[$idx]);
30663773904SAndreas Gohr        }
30763773904SAndreas Gohr    }
30863773904SAndreas Gohr
309*3850270cSMichael Große    $pages = _ft_filterResultsByTime($pages, $data['after'], $data['before']);
3101b48999cSMichael Große
3113d2017d9SAdrian Lang    uksort($pages,'ft_pagesorter');
3128d22f1e9SAndreas Gohr    return $pages;
313506fa893SAndreas Gohr}
314506fa893SAndreas Gohr
3151b48999cSMichael Große
3161b48999cSMichael Große/**
3171b48999cSMichael Große * @param array      $results search results in the form pageid => value
318*3850270cSMichael Große * @param int|string $after   only returns results with an modified time after this date, accepts timestap or strtotime arguments
319*3850270cSMichael Große * @param int|string $before  only returns results with an modified time after this date, accepts timestap or strtotime arguments
3201b48999cSMichael Große *
3211b48999cSMichael Große * @return array
3221b48999cSMichael Große */
323*3850270cSMichael Großefunction _ft_filterResultsByTime(array $results, $after, $before) {
324*3850270cSMichael Große    if ($after || $before) {
3251b48999cSMichael Große        $after = is_int($after) ? $after : strtotime($after);
3261b48999cSMichael Große        $before = is_int($before) ? $before : strtotime($before);
3271b48999cSMichael Große
3281b48999cSMichael Große        foreach ($results as $id => $value) {
3291b48999cSMichael Große            $mTime = filemtime(wikiFN($id));
3301b48999cSMichael Große            if ($after && $after > $mTime) {
3311b48999cSMichael Große                unset($results[$id]);
3321b48999cSMichael Große                continue;
3331b48999cSMichael Große            }
3341b48999cSMichael Große            if ($before && $before < $mTime) {
3351b48999cSMichael Große                unset($results[$id]);
3361b48999cSMichael Große            }
3371b48999cSMichael Große        }
3381b48999cSMichael Große    }
3391b48999cSMichael Große
3401b48999cSMichael Große    return $results;
3411b48999cSMichael Große}
3421b48999cSMichael Große
343506fa893SAndreas Gohr/**
344c66f16a3SMichael Hamann * Tiny helper function for comparing the searched title with the title
345c66f16a3SMichael Hamann * from the search index. This function is a wrapper around stripos with
346c66f16a3SMichael Hamann * adapted argument order and return value.
34742ea7f44SGerrit Uitslag *
34842ea7f44SGerrit Uitslag * @param string $search searched title
34942ea7f44SGerrit Uitslag * @param string $title  title from index
35042ea7f44SGerrit Uitslag * @return bool
351c66f16a3SMichael Hamann */
352c66f16a3SMichael Hamannfunction _ft_pageLookupTitleCompare($search, $title) {
353c66f16a3SMichael Hamann    return stripos($title, $search) !== false;
354c66f16a3SMichael Hamann}
355c66f16a3SMichael Hamann
356c66f16a3SMichael Hamann/**
357f31eb72bSAndreas Gohr * Sort pages based on their namespace level first, then on their string
358f31eb72bSAndreas Gohr * values. This makes higher hierarchy pages rank higher than lower hierarchy
359f31eb72bSAndreas Gohr * pages.
36042ea7f44SGerrit Uitslag *
36142ea7f44SGerrit Uitslag * @param string $a
36242ea7f44SGerrit Uitslag * @param string $b
36342ea7f44SGerrit Uitslag * @return int Returns < 0 if $a is less than $b; > 0 if $a is greater than $b, and 0 if they are equal.
364f31eb72bSAndreas Gohr */
365f31eb72bSAndreas Gohrfunction ft_pagesorter($a, $b){
366f31eb72bSAndreas Gohr    $ac = count(explode(':',$a));
367f31eb72bSAndreas Gohr    $bc = count(explode(':',$b));
368f31eb72bSAndreas Gohr    if($ac < $bc){
369f31eb72bSAndreas Gohr        return -1;
370f31eb72bSAndreas Gohr    }elseif($ac > $bc){
371f31eb72bSAndreas Gohr        return 1;
372f31eb72bSAndreas Gohr    }
373f31eb72bSAndreas Gohr    return strcmp ($a,$b);
374f31eb72bSAndreas Gohr}
375f31eb72bSAndreas Gohr
376f31eb72bSAndreas Gohr/**
3778d0e286aSMichael Große * Sort pages by their mtime, from newest to oldest
3788d0e286aSMichael Große *
3798d0e286aSMichael Große * @param string $a
3808d0e286aSMichael Große * @param string $b
3818d0e286aSMichael Große *
3828d0e286aSMichael 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
3838d0e286aSMichael Große */
3848d0e286aSMichael Großefunction ft_pagemtimesorter($a, $b) {
3858d0e286aSMichael Große    $mtimeA = filemtime(wikiFN($a));
3868d0e286aSMichael Große    $mtimeB = filemtime(wikiFN($b));
3878d0e286aSMichael Große    return $mtimeB - $mtimeA;
3888d0e286aSMichael Große}
3898d0e286aSMichael Große
3908d0e286aSMichael Große/**
391506fa893SAndreas Gohr * Creates a snippet extract
392506fa893SAndreas Gohr *
393506fa893SAndreas Gohr * @author Andreas Gohr <andi@splitbrain.org>
39460e91a17SAndreas Gohr * @triggers FULLTEXT_SNIPPET_CREATE
39542ea7f44SGerrit Uitslag *
39642ea7f44SGerrit Uitslag * @param string $id page id
39742ea7f44SGerrit Uitslag * @param array $highlight
39842ea7f44SGerrit Uitslag * @return mixed
399506fa893SAndreas Gohr */
400546d3a99SAndreas Gohrfunction ft_snippet($id,$highlight){
401506fa893SAndreas Gohr    $text = rawWiki($id);
4024f0030ddSAndreas Gohr    $text = str_replace("\xC2\xAD",'',$text); // remove soft-hyphens
40360e91a17SAndreas Gohr    $evdata = array(
40460e91a17SAndreas Gohr            'id'        => $id,
40560e91a17SAndreas Gohr            'text'      => &$text,
40660e91a17SAndreas Gohr            'highlight' => &$highlight,
40760e91a17SAndreas Gohr            'snippet'   => '',
40860e91a17SAndreas Gohr            );
40960e91a17SAndreas Gohr
41060e91a17SAndreas Gohr    $evt = new Doku_Event('FULLTEXT_SNIPPET_CREATE',$evdata);
41160e91a17SAndreas Gohr    if ($evt->advise_before()) {
412ced0762eSchris        $match = array();
413ced0762eSchris        $snippets = array();
4149ee93076Schris        $utf8_offset = $offset = $end = 0;
415ced0762eSchris        $len = utf8_strlen($text);
4169ee93076Schris
417546d3a99SAndreas Gohr        // build a regexp from the phrases to highlight
4182237b4faSAndreas Gohr        $re1 = '('.join('|',array_map('ft_snippet_re_preprocess', array_map('preg_quote_cb',array_filter((array) $highlight)))).')';
419b571ff2dSChuck Kollars        $re2 = "$re1.{0,75}(?!\\1)$re1";
420b571ff2dSChuck Kollars        $re3 = "$re1.{0,45}(?!\\1)$re1.{0,45}(?!\\1)(?!\\2)$re1";
421546d3a99SAndreas Gohr
422b571ff2dSChuck Kollars        for ($cnt=4; $cnt--;) {
423b571ff2dSChuck Kollars            if (0) {
424b571ff2dSChuck Kollars            } else if (preg_match('/'.$re3.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
425b571ff2dSChuck Kollars            } else if (preg_match('/'.$re2.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
426b571ff2dSChuck Kollars            } else if (preg_match('/'.$re1.'/iu',$text,$match,PREG_OFFSET_CAPTURE,$offset)) {
427b571ff2dSChuck Kollars            } else {
428b571ff2dSChuck Kollars                break;
429b571ff2dSChuck Kollars            }
430ced0762eSchris
431ced0762eSchris            list($str,$idx) = $match[0];
432ced0762eSchris
433ced0762eSchris            // convert $idx (a byte offset) into a utf8 character offset
434ced0762eSchris            $utf8_idx = utf8_strlen(substr($text,0,$idx));
435ced0762eSchris            $utf8_len = utf8_strlen($str);
436ced0762eSchris
437ced0762eSchris            // establish context, 100 bytes surrounding the match string
438ced0762eSchris            // first look to see if we can go 100 either side,
439ced0762eSchris            // then drop to 50 adding any excess if the other side can't go to 50,
440ced0762eSchris            $pre = min($utf8_idx-$utf8_offset,100);
441ced0762eSchris            $post = min($len-$utf8_idx-$utf8_len,100);
442ced0762eSchris
443ced0762eSchris            if ($pre>50 && $post>50) {
444ced0762eSchris                $pre = $post = 50;
445ced0762eSchris            } else if ($pre>50) {
446ced0762eSchris                $pre = min($pre,100-$post);
447ced0762eSchris            } else if ($post>50) {
448ced0762eSchris                $post = min($post, 100-$pre);
449ef3e3cddSMichael Hamann            } else if ($offset == 0) {
450ced0762eSchris                // both are less than 50, means the context is the whole string
45110ffc9ddSAndreas Gohr                // make it so and break out of this loop - there is no need for the
45210ffc9ddSAndreas Gohr                // complex snippet calculations
453ced0762eSchris                $snippets = array($text);
454ced0762eSchris                break;
455ced0762eSchris            }
456ced0762eSchris
45710ffc9ddSAndreas Gohr            // establish context start and end points, try to append to previous
45810ffc9ddSAndreas Gohr            // context if possible
4599ee93076Schris            $start = $utf8_idx - $pre;
460ced0762eSchris            $append = ($start < $end) ? $end : false;  // still the end of the previous context snippet
4619ee93076Schris            $end = $utf8_idx + $utf8_len + $post;      // now set it to the end of this context
462ced0762eSchris
463ced0762eSchris            if ($append) {
464ced0762eSchris                $snippets[count($snippets)-1] .= utf8_substr($text,$append,$end-$append);
465ced0762eSchris            } else {
466ced0762eSchris                $snippets[] = utf8_substr($text,$start,$end-$start);
467ced0762eSchris            }
468ced0762eSchris
469ced0762eSchris            // set $offset for next match attempt
47043d58b76SMichael Hamann            // continue matching after the current match
47143d58b76SMichael Hamann            // if the current match is not the longest possible match starting at the current offset
47243d58b76SMichael Hamann            // this prevents further matching of this snippet but for possible matches of length
47343d58b76SMichael Hamann            // smaller than match length + context (at least 50 characters) this match is part of the context
47443d58b76SMichael Hamann            $utf8_offset = $utf8_idx + $utf8_len;
47543d58b76SMichael Hamann            $offset = $idx + strlen(utf8_substr($text,$utf8_idx,$utf8_len));
4769ee93076Schris            $offset = utf8_correctIdx($text,$offset);
4779ee93076Schris        }
4789ee93076Schris
479ced0762eSchris        $m = "\1";
480b571ff2dSChuck Kollars        $snippets = preg_replace('/'.$re1.'/iu',$m.'$1'.$m,$snippets);
481b571ff2dSChuck Kollars        $snippet = preg_replace('/'.$m.'([^'.$m.']*?)'.$m.'/iu','<strong class="search_hit">$1</strong>',hsc(join('... ',$snippets)));
482bd2cb6fcSchris
48360e91a17SAndreas Gohr        $evdata['snippet'] = $snippet;
48460e91a17SAndreas Gohr    }
48560e91a17SAndreas Gohr    $evt->advise_after();
48660e91a17SAndreas Gohr    unset($evt);
48760e91a17SAndreas Gohr
48860e91a17SAndreas Gohr    return $evdata['snippet'];
489506fa893SAndreas Gohr}
490506fa893SAndreas Gohr
491506fa893SAndreas Gohr/**
49226eb848cSGina Haeussge * Wraps a search term in regex boundary checks.
49342ea7f44SGerrit Uitslag *
49442ea7f44SGerrit Uitslag * @param string $term
49542ea7f44SGerrit Uitslag * @return string
49626eb848cSGina Haeussge */
4972237b4faSAndreas Gohrfunction ft_snippet_re_preprocess($term) {
49835594613SKazutaka Miyasaka    // do not process asian terms where word boundaries are not explicit
49935594613SKazutaka Miyasaka    if(preg_match('/'.IDX_ASIAN.'/u',$term)){
50035594613SKazutaka Miyasaka        return $term;
50135594613SKazutaka Miyasaka    }
50235594613SKazutaka Miyasaka
5033161005dSAndreas Gohr    if (UTF8_PROPERTYSUPPORT) {
50484e581a6SAndreas Gohr        // unicode word boundaries
50584e581a6SAndreas Gohr        // see http://stackoverflow.com/a/2449017/172068
50684e581a6SAndreas Gohr        $BL = '(?<!\pL)';
50784e581a6SAndreas Gohr        $BR = '(?!\pL)';
5083161005dSAndreas Gohr    } else {
5093161005dSAndreas Gohr        // not as correct as above, but at least won't break
5103161005dSAndreas Gohr        $BL = '\b';
5113161005dSAndreas Gohr        $BR = '\b';
5123161005dSAndreas Gohr    }
5133161005dSAndreas Gohr
5142237b4faSAndreas Gohr    if(substr($term,0,2) == '\\*'){
5152237b4faSAndreas Gohr        $term = substr($term,2);
5162237b4faSAndreas Gohr    }else{
51784e581a6SAndreas Gohr        $term = $BL.$term;
5182237b4faSAndreas Gohr    }
5192237b4faSAndreas Gohr
5202237b4faSAndreas Gohr    if(substr($term,-2,2) == '\\*'){
5212237b4faSAndreas Gohr        $term = substr($term,0,-2);
5222237b4faSAndreas Gohr    }else{
52384e581a6SAndreas Gohr        $term = $term.$BR;
5242237b4faSAndreas Gohr    }
5258a803caeSAndreas Gohr
52684e581a6SAndreas Gohr    if($term == $BL || $term == $BR || $term == $BL.$BR) $term = '';
5272237b4faSAndreas Gohr    return $term;
52826eb848cSGina Haeussge}
52926eb848cSGina Haeussge
53026eb848cSGina Haeussge/**
531f5eb7cf0SAndreas Gohr * Combine found documents and sum up their scores
532f5eb7cf0SAndreas Gohr *
533f5eb7cf0SAndreas Gohr * This function is used to combine searched words with a logical
534f5eb7cf0SAndreas Gohr * AND. Only documents available in all arrays are returned.
535f5eb7cf0SAndreas Gohr *
536f5eb7cf0SAndreas Gohr * based upon PEAR's PHP_Compat function for array_intersect_key()
537f5eb7cf0SAndreas Gohr *
538f5eb7cf0SAndreas Gohr * @param array $args An array of page arrays
53942ea7f44SGerrit Uitslag * @return array
540f5eb7cf0SAndreas Gohr */
541f5eb7cf0SAndreas Gohrfunction ft_resultCombine($args){
542f5eb7cf0SAndreas Gohr    $array_count = count($args);
543134f4ab2SAndreas Gohr    if($array_count == 1){
544134f4ab2SAndreas Gohr        return $args[0];
545134f4ab2SAndreas Gohr    }
546134f4ab2SAndreas Gohr
547f5eb7cf0SAndreas Gohr    $result = array();
54809c27a6dSGuy Brand    if ($array_count > 1) {
549a21136cdSAndreas Gohr        foreach ($args[0] as $key => $value) {
550a21136cdSAndreas Gohr            $result[$key] = $value;
551f5eb7cf0SAndreas Gohr            for ($i = 1; $i !== $array_count; $i++) {
552a21136cdSAndreas Gohr                if (!isset($args[$i][$key])) {
553a21136cdSAndreas Gohr                    unset($result[$key]);
554a21136cdSAndreas Gohr                    break;
555f5eb7cf0SAndreas Gohr                }
556a21136cdSAndreas Gohr                $result[$key] += $args[$i][$key];
557f5eb7cf0SAndreas Gohr            }
558f5eb7cf0SAndreas Gohr        }
55909c27a6dSGuy Brand    }
560f5eb7cf0SAndreas Gohr    return $result;
561f5eb7cf0SAndreas Gohr}
562f5eb7cf0SAndreas Gohr
563f5eb7cf0SAndreas Gohr/**
564865c2687SKazutaka Miyasaka * Unites found documents and sum up their scores
565f5eb7cf0SAndreas Gohr *
566865c2687SKazutaka Miyasaka * based upon ft_resultCombine() function
567865c2687SKazutaka Miyasaka *
568865c2687SKazutaka Miyasaka * @param array $args An array of page arrays
56942ea7f44SGerrit Uitslag * @return array
57042ea7f44SGerrit Uitslag *
571865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
572865c2687SKazutaka Miyasaka */
573865c2687SKazutaka Miyasakafunction ft_resultUnite($args) {
574865c2687SKazutaka Miyasaka    $array_count = count($args);
575865c2687SKazutaka Miyasaka    if ($array_count === 1) {
576865c2687SKazutaka Miyasaka        return $args[0];
577865c2687SKazutaka Miyasaka    }
578865c2687SKazutaka Miyasaka
579865c2687SKazutaka Miyasaka    $result = $args[0];
580865c2687SKazutaka Miyasaka    for ($i = 1; $i !== $array_count; $i++) {
581865c2687SKazutaka Miyasaka        foreach (array_keys($args[$i]) as $id) {
582865c2687SKazutaka Miyasaka            $result[$id] += $args[$i][$id];
583865c2687SKazutaka Miyasaka        }
584865c2687SKazutaka Miyasaka    }
585865c2687SKazutaka Miyasaka    return $result;
586865c2687SKazutaka Miyasaka}
587865c2687SKazutaka Miyasaka
588865c2687SKazutaka Miyasaka/**
589865c2687SKazutaka Miyasaka * Computes the difference of documents using page id for comparison
590865c2687SKazutaka Miyasaka *
591865c2687SKazutaka Miyasaka * nearly identical to PHP5's array_diff_key()
592865c2687SKazutaka Miyasaka *
593865c2687SKazutaka Miyasaka * @param array $args An array of page arrays
59442ea7f44SGerrit Uitslag * @return array
59542ea7f44SGerrit Uitslag *
596865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
597865c2687SKazutaka Miyasaka */
598865c2687SKazutaka Miyasakafunction ft_resultComplement($args) {
599865c2687SKazutaka Miyasaka    $array_count = count($args);
600865c2687SKazutaka Miyasaka    if ($array_count === 1) {
601865c2687SKazutaka Miyasaka        return $args[0];
602865c2687SKazutaka Miyasaka    }
603865c2687SKazutaka Miyasaka
604865c2687SKazutaka Miyasaka    $result = $args[0];
605865c2687SKazutaka Miyasaka    foreach (array_keys($result) as $id) {
606865c2687SKazutaka Miyasaka        for ($i = 1; $i !== $array_count; $i++) {
607865c2687SKazutaka Miyasaka            if (isset($args[$i][$id])) unset($result[$id]);
608865c2687SKazutaka Miyasaka        }
609865c2687SKazutaka Miyasaka    }
610865c2687SKazutaka Miyasaka    return $result;
611865c2687SKazutaka Miyasaka}
612865c2687SKazutaka Miyasaka
613865c2687SKazutaka Miyasaka/**
614865c2687SKazutaka Miyasaka * Parses a search query and builds an array of search formulas
615865c2687SKazutaka Miyasaka *
616865c2687SKazutaka Miyasaka * @author Andreas Gohr <andi@splitbrain.org>
617865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
61842ea7f44SGerrit Uitslag *
61942ea7f44SGerrit Uitslag * @param Doku_Indexer $Indexer
62042ea7f44SGerrit Uitslag * @param string $query search query
62142ea7f44SGerrit Uitslag * @return array of search formulas
622f5eb7cf0SAndreas Gohr */
6239b41be24STom N Harrisfunction ft_queryParser($Indexer, $query){
624865c2687SKazutaka Miyasaka    /**
625865c2687SKazutaka Miyasaka     * parse a search query and transform it into intermediate representation
626865c2687SKazutaka Miyasaka     *
627865c2687SKazutaka Miyasaka     * in a search query, you can use the following expressions:
628865c2687SKazutaka Miyasaka     *
629865c2687SKazutaka Miyasaka     *   words:
630865c2687SKazutaka Miyasaka     *     include
631865c2687SKazutaka Miyasaka     *     -exclude
632865c2687SKazutaka Miyasaka     *   phrases:
633865c2687SKazutaka Miyasaka     *     "phrase to be included"
634865c2687SKazutaka Miyasaka     *     -"phrase you want to exclude"
635865c2687SKazutaka Miyasaka     *   namespaces:
636865c2687SKazutaka Miyasaka     *     @include:namespace (or ns:include:namespace)
637865c2687SKazutaka Miyasaka     *     ^exclude:namespace (or -ns:exclude:namespace)
638865c2687SKazutaka Miyasaka     *   groups:
639865c2687SKazutaka Miyasaka     *     ()
640865c2687SKazutaka Miyasaka     *     -()
641865c2687SKazutaka Miyasaka     *   operators:
642865c2687SKazutaka Miyasaka     *     and ('and' is the default operator: you can always omit this)
6437871d415SKazutaka Miyasaka     *     or  (or pipe symbol '|', lower precedence than 'and')
644865c2687SKazutaka Miyasaka     *
645865c2687SKazutaka Miyasaka     * e.g. a query [ aa "bb cc" @dd:ee ] means "search pages which contain
646865c2687SKazutaka Miyasaka     *      a word 'aa', a phrase 'bb cc' and are within a namespace 'dd:ee'".
647865c2687SKazutaka Miyasaka     *      this query is equivalent to [ -(-aa or -"bb cc" or -ns:dd:ee) ]
648865c2687SKazutaka Miyasaka     *      as long as you don't mind hit counts.
649865c2687SKazutaka Miyasaka     *
650865c2687SKazutaka Miyasaka     * intermediate representation consists of the following parts:
651865c2687SKazutaka Miyasaka     *
652865c2687SKazutaka Miyasaka     *   ( )           - group
653865c2687SKazutaka Miyasaka     *   AND           - logical and
654865c2687SKazutaka Miyasaka     *   OR            - logical or
655865c2687SKazutaka Miyasaka     *   NOT           - logical not
6562f502d70SKazutaka Miyasaka     *   W+:, W-:, W_: - word      (underscore: no need to highlight)
6572f502d70SKazutaka Miyasaka     *   P+:, P-:      - phrase    (minus sign: logically in NOT group)
6582f502d70SKazutaka Miyasaka     *   N+:, N-:      - namespace
659865c2687SKazutaka Miyasaka     */
660865c2687SKazutaka Miyasaka    $parsed_query = '';
661865c2687SKazutaka Miyasaka    $parens_level = 0;
662865c2687SKazutaka Miyasaka    $terms = preg_split('/(-?".*?")/u', utf8_strtolower($query), -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
663865c2687SKazutaka Miyasaka
664865c2687SKazutaka Miyasaka    foreach ($terms as $term) {
665865c2687SKazutaka Miyasaka        $parsed = '';
666865c2687SKazutaka Miyasaka        if (preg_match('/^(-?)"(.+)"$/u', $term, $matches)) {
667865c2687SKazutaka Miyasaka            // phrase-include and phrase-exclude
668865c2687SKazutaka Miyasaka            $not = $matches[1] ? 'NOT' : '';
6699b41be24STom N Harris            $parsed = $not.ft_termParser($Indexer, $matches[2], false, true);
670f5eb7cf0SAndreas Gohr        } else {
671865c2687SKazutaka Miyasaka            // fix incomplete phrase
672865c2687SKazutaka Miyasaka            $term = str_replace('"', ' ', $term);
673865c2687SKazutaka Miyasaka
674865c2687SKazutaka Miyasaka            // fix parentheses
675865c2687SKazutaka Miyasaka            $term = str_replace(')'  , ' ) ', $term);
676865c2687SKazutaka Miyasaka            $term = str_replace('('  , ' ( ', $term);
677865c2687SKazutaka Miyasaka            $term = str_replace('- (', ' -(', $term);
678865c2687SKazutaka Miyasaka
6797871d415SKazutaka Miyasaka            // treat pipe symbols as 'OR' operators
6807871d415SKazutaka Miyasaka            $term = str_replace('|', ' or ', $term);
6817871d415SKazutaka Miyasaka
682865c2687SKazutaka Miyasaka            // treat ideographic spaces (U+3000) as search term separators
683865c2687SKazutaka Miyasaka            // FIXME: some more separators?
684865c2687SKazutaka Miyasaka            $term = preg_replace('/[ \x{3000}]+/u', ' ',  $term);
685865c2687SKazutaka Miyasaka            $term = trim($term);
686865c2687SKazutaka Miyasaka            if ($term === '') continue;
687865c2687SKazutaka Miyasaka
688865c2687SKazutaka Miyasaka            $tokens = explode(' ', $term);
689865c2687SKazutaka Miyasaka            foreach ($tokens as $token) {
690865c2687SKazutaka Miyasaka                if ($token === '(') {
691865c2687SKazutaka Miyasaka                    // parenthesis-include-open
692865c2687SKazutaka Miyasaka                    $parsed .= '(';
693865c2687SKazutaka Miyasaka                    ++$parens_level;
694865c2687SKazutaka Miyasaka                } elseif ($token === '-(') {
695865c2687SKazutaka Miyasaka                    // parenthesis-exclude-open
696865c2687SKazutaka Miyasaka                    $parsed .= 'NOT(';
697865c2687SKazutaka Miyasaka                    ++$parens_level;
698865c2687SKazutaka Miyasaka                } elseif ($token === ')') {
699865c2687SKazutaka Miyasaka                    // parenthesis-any-close
700865c2687SKazutaka Miyasaka                    if ($parens_level === 0) continue;
701865c2687SKazutaka Miyasaka                    $parsed .= ')';
702865c2687SKazutaka Miyasaka                    $parens_level--;
703865c2687SKazutaka Miyasaka                } elseif ($token === 'and') {
704865c2687SKazutaka Miyasaka                    // logical-and (do nothing)
705865c2687SKazutaka Miyasaka                } elseif ($token === 'or') {
706865c2687SKazutaka Miyasaka                    // logical-or
707865c2687SKazutaka Miyasaka                    $parsed .= 'OR';
708865c2687SKazutaka Miyasaka                } elseif (preg_match('/^(?:\^|-ns:)(.+)$/u', $token, $matches)) {
709865c2687SKazutaka Miyasaka                    // namespace-exclude
7102f502d70SKazutaka Miyasaka                    $parsed .= 'NOT(N+:'.$matches[1].')';
711865c2687SKazutaka Miyasaka                } elseif (preg_match('/^(?:@|ns:)(.+)$/u', $token, $matches)) {
712865c2687SKazutaka Miyasaka                    // namespace-include
7132f502d70SKazutaka Miyasaka                    $parsed .= '(N+:'.$matches[1].')';
714865c2687SKazutaka Miyasaka                } elseif (preg_match('/^-(.+)$/', $token, $matches)) {
715865c2687SKazutaka Miyasaka                    // word-exclude
7169b41be24STom N Harris                    $parsed .= 'NOT('.ft_termParser($Indexer, $matches[1]).')';
717865c2687SKazutaka Miyasaka                } else {
718865c2687SKazutaka Miyasaka                    // word-include
7199b41be24STom N Harris                    $parsed .= ft_termParser($Indexer, $token);
720865c2687SKazutaka Miyasaka                }
721865c2687SKazutaka Miyasaka            }
722865c2687SKazutaka Miyasaka        }
723865c2687SKazutaka Miyasaka        $parsed_query .= $parsed;
724f5eb7cf0SAndreas Gohr    }
725f5eb7cf0SAndreas Gohr
726865c2687SKazutaka Miyasaka    // cleanup (very sensitive)
727865c2687SKazutaka Miyasaka    $parsed_query .= str_repeat(')', $parens_level);
728865c2687SKazutaka Miyasaka    do {
729865c2687SKazutaka Miyasaka        $parsed_query_old = $parsed_query;
730865c2687SKazutaka Miyasaka        $parsed_query = preg_replace('/(NOT)?\(\)/u', '', $parsed_query);
731865c2687SKazutaka Miyasaka    } while ($parsed_query !== $parsed_query_old);
732865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/(NOT|OR)+\)/u', ')'      , $parsed_query);
733865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/(OR)+/u'      , 'OR'     , $parsed_query);
734865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/\(OR/u'       , '('      , $parsed_query);
735865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/^OR|OR$/u'    , ''       , $parsed_query);
736865c2687SKazutaka Miyasaka    $parsed_query = preg_replace('/\)(NOT)?\(/u' , ')AND$1(', $parsed_query);
737865c2687SKazutaka Miyasaka
7382f502d70SKazutaka Miyasaka    // adjustment: make highlightings right
7392f502d70SKazutaka Miyasaka    $parens_level     = 0;
7402f502d70SKazutaka Miyasaka    $notgrp_levels    = array();
7412f502d70SKazutaka Miyasaka    $parsed_query_new = '';
7422f502d70SKazutaka Miyasaka    $tokens = preg_split('/(NOT\(|[()])/u', $parsed_query, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
7432f502d70SKazutaka Miyasaka    foreach ($tokens as $token) {
7442f502d70SKazutaka Miyasaka        if ($token === 'NOT(') {
7452f502d70SKazutaka Miyasaka            $notgrp_levels[] = ++$parens_level;
7462f502d70SKazutaka Miyasaka        } elseif ($token === '(') {
7472f502d70SKazutaka Miyasaka            ++$parens_level;
7482f502d70SKazutaka Miyasaka        } elseif ($token === ')') {
7492f502d70SKazutaka Miyasaka            if ($parens_level-- === end($notgrp_levels)) array_pop($notgrp_levels);
7502f502d70SKazutaka Miyasaka        } elseif (count($notgrp_levels) % 2 === 1) {
7512f502d70SKazutaka Miyasaka            // turn highlight-flag off if terms are logically in "NOT" group
7522f502d70SKazutaka Miyasaka            $token = preg_replace('/([WPN])\+\:/u', '$1-:', $token);
7532f502d70SKazutaka Miyasaka        }
7542f502d70SKazutaka Miyasaka        $parsed_query_new .= $token;
7552f502d70SKazutaka Miyasaka    }
7562f502d70SKazutaka Miyasaka    $parsed_query = $parsed_query_new;
7572f502d70SKazutaka Miyasaka
758865c2687SKazutaka Miyasaka    /**
759865c2687SKazutaka Miyasaka     * convert infix notation string into postfix (Reverse Polish notation) array
760865c2687SKazutaka Miyasaka     * by Shunting-yard algorithm
761865c2687SKazutaka Miyasaka     *
762865c2687SKazutaka Miyasaka     * see: http://en.wikipedia.org/wiki/Reverse_Polish_notation
763865c2687SKazutaka Miyasaka     * see: http://en.wikipedia.org/wiki/Shunting-yard_algorithm
764865c2687SKazutaka Miyasaka     */
765865c2687SKazutaka Miyasaka    $parsed_ary     = array();
766865c2687SKazutaka Miyasaka    $ope_stack      = array();
767865c2687SKazutaka Miyasaka    $ope_precedence = array(')' => 1, 'OR' => 2, 'AND' => 3, 'NOT' => 4, '(' => 5);
768865c2687SKazutaka Miyasaka    $ope_regex      = '/([()]|OR|AND|NOT)/u';
769865c2687SKazutaka Miyasaka
770865c2687SKazutaka Miyasaka    $tokens = preg_split($ope_regex, $parsed_query, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
771865c2687SKazutaka Miyasaka    foreach ($tokens as $token) {
772865c2687SKazutaka Miyasaka        if (preg_match($ope_regex, $token)) {
773865c2687SKazutaka Miyasaka            // operator
774865c2687SKazutaka Miyasaka            $last_ope = end($ope_stack);
77567d812e0SMarius van Witzenburg            while ($last_ope !== false && $ope_precedence[$token] <= $ope_precedence[$last_ope] && $last_ope != '(') {
776865c2687SKazutaka Miyasaka                $parsed_ary[] = array_pop($ope_stack);
777865c2687SKazutaka Miyasaka                $last_ope = end($ope_stack);
778865c2687SKazutaka Miyasaka            }
779865c2687SKazutaka Miyasaka            if ($token == ')') {
780865c2687SKazutaka Miyasaka                array_pop($ope_stack); // this array_pop always deletes '('
781865c2687SKazutaka Miyasaka            } else {
782865c2687SKazutaka Miyasaka                $ope_stack[] = $token;
783865c2687SKazutaka Miyasaka            }
784865c2687SKazutaka Miyasaka        } else {
785865c2687SKazutaka Miyasaka            // operand
786865c2687SKazutaka Miyasaka            $token_decoded = str_replace(array('OP', 'CP'), array('(', ')'), $token);
787865c2687SKazutaka Miyasaka            $parsed_ary[] = $token_decoded;
788865c2687SKazutaka Miyasaka        }
789865c2687SKazutaka Miyasaka    }
790865c2687SKazutaka Miyasaka    $parsed_ary = array_values(array_merge($parsed_ary, array_reverse($ope_stack)));
791865c2687SKazutaka Miyasaka
792865c2687SKazutaka Miyasaka    // cleanup: each double "NOT" in RPN array actually does nothing
793865c2687SKazutaka Miyasaka    $parsed_ary_count = count($parsed_ary);
794865c2687SKazutaka Miyasaka    for ($i = 1; $i < $parsed_ary_count; ++$i) {
795865c2687SKazutaka Miyasaka        if ($parsed_ary[$i] === 'NOT' && $parsed_ary[$i - 1] === 'NOT') {
796865c2687SKazutaka Miyasaka            unset($parsed_ary[$i], $parsed_ary[$i - 1]);
797865c2687SKazutaka Miyasaka        }
798865c2687SKazutaka Miyasaka    }
799865c2687SKazutaka Miyasaka    $parsed_ary = array_values($parsed_ary);
800865c2687SKazutaka Miyasaka
801865c2687SKazutaka Miyasaka    // build return value
802f5eb7cf0SAndreas Gohr    $q = array();
803f5eb7cf0SAndreas Gohr    $q['query']      = $query;
804865c2687SKazutaka Miyasaka    $q['parsed_str'] = $parsed_query;
805865c2687SKazutaka Miyasaka    $q['parsed_ary'] = $parsed_ary;
806f5eb7cf0SAndreas Gohr
807865c2687SKazutaka Miyasaka    foreach ($q['parsed_ary'] as $token) {
808865c2687SKazutaka Miyasaka        if ($token[2] !== ':') continue;
809865c2687SKazutaka Miyasaka        $body = substr($token, 3);
810865c2687SKazutaka Miyasaka
811865c2687SKazutaka Miyasaka        switch (substr($token, 0, 3)) {
8122f502d70SKazutaka Miyasaka            case 'N+:':
813865c2687SKazutaka Miyasaka                     $q['ns'][]        = $body; // for backward compatibility
814865c2687SKazutaka Miyasaka                     break;
8152f502d70SKazutaka Miyasaka            case 'N-:':
8162f502d70SKazutaka Miyasaka                     $q['notns'][]     = $body; // for backward compatibility
8172f502d70SKazutaka Miyasaka                     break;
8182f502d70SKazutaka Miyasaka            case 'W_:':
8192f502d70SKazutaka Miyasaka                     $q['words'][]     = $body;
8202f502d70SKazutaka Miyasaka                     break;
821865c2687SKazutaka Miyasaka            case 'W-:':
822865c2687SKazutaka Miyasaka                     $q['words'][]     = $body;
8232f502d70SKazutaka Miyasaka                     $q['not'][]       = $body; // for backward compatibility
824865c2687SKazutaka Miyasaka                     break;
825865c2687SKazutaka Miyasaka            case 'W+:':
826865c2687SKazutaka Miyasaka                     $q['words'][]     = $body;
8272237b4faSAndreas Gohr                     $q['highlight'][] = $body;
8282f502d70SKazutaka Miyasaka                     $q['and'][]       = $body; // for backward compatibility
829865c2687SKazutaka Miyasaka                     break;
8302f502d70SKazutaka Miyasaka            case 'P-:':
8312f502d70SKazutaka Miyasaka                     $q['phrases'][]   = $body;
8322f502d70SKazutaka Miyasaka                     break;
8332f502d70SKazutaka Miyasaka            case 'P+:':
834865c2687SKazutaka Miyasaka                     $q['phrases'][]   = $body;
8352237b4faSAndreas Gohr                     $q['highlight'][] = $body;
836865c2687SKazutaka Miyasaka                     break;
837865c2687SKazutaka Miyasaka        }
838865c2687SKazutaka Miyasaka    }
8392f502d70SKazutaka Miyasaka    foreach (array('words', 'phrases', 'highlight', 'ns', 'notns', 'and', 'not') as $key) {
840865c2687SKazutaka Miyasaka        $q[$key] = empty($q[$key]) ? array() : array_values(array_unique($q[$key]));
841f5eb7cf0SAndreas Gohr    }
842f5eb7cf0SAndreas Gohr
843f5eb7cf0SAndreas Gohr    return $q;
844f5eb7cf0SAndreas Gohr}
845f5eb7cf0SAndreas Gohr
846865c2687SKazutaka Miyasaka/**
847865c2687SKazutaka Miyasaka * Transforms given search term into intermediate representation
848865c2687SKazutaka Miyasaka *
849865c2687SKazutaka Miyasaka * This function is used in ft_queryParser() and not for general purpose use.
850865c2687SKazutaka Miyasaka *
851865c2687SKazutaka Miyasaka * @author Kazutaka Miyasaka <kazmiya@gmail.com>
85242ea7f44SGerrit Uitslag *
85342ea7f44SGerrit Uitslag * @param Doku_Indexer $Indexer
85442ea7f44SGerrit Uitslag * @param string       $term
85542ea7f44SGerrit Uitslag * @param bool         $consider_asian
85642ea7f44SGerrit Uitslag * @param bool         $phrase_mode
85742ea7f44SGerrit Uitslag * @return string
858865c2687SKazutaka Miyasaka */
8599b41be24STom N Harrisfunction ft_termParser($Indexer, $term, $consider_asian = true, $phrase_mode = false) {
860865c2687SKazutaka Miyasaka    $parsed = '';
861865c2687SKazutaka Miyasaka    if ($consider_asian) {
862865c2687SKazutaka Miyasaka        // successive asian characters need to be searched as a phrase
863865c2687SKazutaka Miyasaka        $words = preg_split('/('.IDX_ASIAN.'+)/u', $term, -1, PREG_SPLIT_DELIM_CAPTURE | PREG_SPLIT_NO_EMPTY);
864865c2687SKazutaka Miyasaka        foreach ($words as $word) {
8656ac2077aSKazutaka Miyasaka            $phrase_mode = $phrase_mode ? true : preg_match('/'.IDX_ASIAN.'/u', $word);
8669b41be24STom N Harris            $parsed .= ft_termParser($Indexer, $word, false, $phrase_mode);
867865c2687SKazutaka Miyasaka        }
868865c2687SKazutaka Miyasaka    } else {
869865c2687SKazutaka Miyasaka        $term_noparen = str_replace(array('(', ')'), ' ', $term);
8709b41be24STom N Harris        $words = $Indexer->tokenizer($term_noparen, true);
871865c2687SKazutaka Miyasaka
8722f502d70SKazutaka Miyasaka        // W_: no need to highlight
873865c2687SKazutaka Miyasaka        if (empty($words)) {
874865c2687SKazutaka Miyasaka            $parsed = '()'; // important: do not remove
875865c2687SKazutaka Miyasaka        } elseif ($words[0] === $term) {
876865c2687SKazutaka Miyasaka            $parsed = '(W+:'.$words[0].')';
877865c2687SKazutaka Miyasaka        } elseif ($phrase_mode) {
878865c2687SKazutaka Miyasaka            $term_encoded = str_replace(array('(', ')'), array('OP', 'CP'), $term);
8792f502d70SKazutaka Miyasaka            $parsed = '((W_:'.implode(')(W_:', $words).')(P+:'.$term_encoded.'))';
880865c2687SKazutaka Miyasaka        } else {
881865c2687SKazutaka Miyasaka            $parsed = '((W+:'.implode(')(W+:', $words).'))';
882865c2687SKazutaka Miyasaka        }
883865c2687SKazutaka Miyasaka    }
884865c2687SKazutaka Miyasaka    return $parsed;
885865c2687SKazutaka Miyasaka}
886865c2687SKazutaka Miyasaka
88744156e11SMichael Große/**
88844156e11SMichael Große * Recreate a search query string based on parsed parts, doesn't support negated phrases and `OR` searches
88944156e11SMichael Große *
89044156e11SMichael Große * @param array $and
89144156e11SMichael Große * @param array $not
89244156e11SMichael Große * @param array $phrases
89344156e11SMichael Große * @param array $ns
89444156e11SMichael Große * @param array $notns
89544156e11SMichael Große *
89644156e11SMichael Große * @return string
89744156e11SMichael Große */
89844156e11SMichael Großefunction ft_queryUnparser_simple(array $and, array $not, array $phrases, array $ns, array $notns) {
89944156e11SMichael Große    $query = implode(' ', $and);
90044156e11SMichael Große    if (!empty($not)) {
90144156e11SMichael Große        $query .= ' -' . implode(' -', $not);
90244156e11SMichael Große    }
90344156e11SMichael Große
90444156e11SMichael Große    if (!empty($phrases)) {
90544156e11SMichael Große        $query .= ' "' . implode('" "', $phrases) . '"';
90644156e11SMichael Große    }
90744156e11SMichael Große
90844156e11SMichael Große    if (!empty($ns)) {
90944156e11SMichael Große        $query .= ' @' . implode(' @', $ns);
91044156e11SMichael Große    }
91144156e11SMichael Große
91244156e11SMichael Große    if (!empty($notns)) {
91344156e11SMichael Große        $query .= ' ^' . implode(' ^', $notns);
91444156e11SMichael Große    }
91544156e11SMichael Große
91644156e11SMichael Große    return $query;
91744156e11SMichael Große}
91844156e11SMichael Große
919e3776c06SMichael Hamann//Setup VIM: ex: et ts=4 :
920