factor/basis/help/search/search.factor

63 lines
2.0 KiB
Factor

! Copyright (C) 2012 John Benediktsson
! See http://factorcode.org/license.txt for BSD license
USING: arrays assocs combinators combinators.short-circuit fry
help help.apropos help.markup help.stylesheet help.topics io
io.streams.string io.styles kernel math memoize namespaces
sequences sequences.deep sorting splitting strings unicode.case
unicode.categories words ;
IN: help.search
<PRIVATE
: search-words ( str -- seq )
>lower "-" split [ [ blank? ] split-when ] map concat ;
: element-value ( element -- str )
dup array? [
dup ?first {
{ \ $link [ second article-name ] }
{ \ $vocab-link [ second ] }
{ \ $emphasis [ second ] }
{ \ $subsection [ second article-name ] }
{ \ $subsections [ rest [ article-name ] map " " join ] }
{ \ $description [ rest [ element-value ] map " " join ] }
{ \ $notes [ rest [ element-value ] map " " join ] }
{ \ $snippet [ rest [ element-value ] map " " join ] }
[ 2drop f ]
} case
] [ dup string? [ drop f ] unless ] if ;
MEMO: article-words ( name -- words )
article-content [ element-value ] map " " join search-words
[ [ digit? ] all? not ] filter
[ [ { [ letter? ] [ digit? ] } 1|| not ] trim ] map! harvest ;
: (search-articles) ( string -- seq' )
search-words [ { } ] [
[ all-articles ] dip
dup length 1 > [
'[ article-words _ swap subseq? ] filter
] [
first '[ article-words [ _ head? ] any? ] filter
] if
] if-empty [ article-name ] sort-with ;
PRIVATE>
: search-articles ( string -- )
[
last-element off
[
"Search results for “" "”" surround
title-style get [ format ] ($block)
]
[
(search-articles) [ word? ] partition swap
"Articles" "Words"
[ over empty? [ 2drop ] [ $heading $completions ] if ]
bi-curry@ bi*
] bi
] with-default-style nl ;