path: root/web/server/h2o/libh2o/misc/oktavia/tool/web/oktavia-search.jsx

import "oktavia.jsx";
import "binary-util.jsx";
import "query.jsx";
import "query-string-parser.jsx";
import "search-result.jsx";
import "style.jsx";
import "stemmer/stemmer.jsx";
import "console.jsx";


class _Result
{
    var title : string;
    var url : string;
    var content : string;
    var score : int;
    function constructor (title : string, url : string, content : string, score : int)
    {
        this.title = title;
        this.url = url;
        this.content = content;
        this.score = score;
    }
}

class _Proposal
{
    var options : string;
    var label : string;
    var count : int;
    function constructor (options : string, label : string, count : int)
    {
        this.options = options;
        this.label = label;
        this.count = count;
    }
}

class OktaviaSearch
{
    var _oktavia : Oktavia;
    static var _stemmer : Nullable.<Stemmer> = null;
    static var _instance : Nullable.<OktaviaSearch> = null;
    var _queryString : Nullable.<string>;
    var _queries : Query[];
    var _highlight : string;
    var _callback : Nullable.<function(:int, :int):void>;
    var _entriesPerPage : int;
    var _currentPage : int;
    var _result : SearchUnit[];
    var _proposals : Proposal[];
    var _currentFolderDepth : int;

    function constructor (entriesPerPage : int)
    {
        this._oktavia = new Oktavia();
        this._entriesPerPage = entriesPerPage;
        this._currentPage = 1;
        this._queryString = null;
        this._callback = null;
        OktaviaSearch._instance = this;
    }

    static function setStemmer(stemmer : Stemmer) : void
    {
        if (OktaviaSearch._instance)
        {
            OktaviaSearch._instance._oktavia.setStemmer(stemmer);
        }
        else
        {
            OktaviaSearch._stemmer = stemmer;
        }
    }

    function loadIndex (index : string) : void
    {
        if (OktaviaSearch._stemmer)
        {
            this._oktavia.setStemmer(OktaviaSearch._stemmer);
        }
        this._oktavia.load(Binary.base64decode(index));
        if (this._queryString)
        {
            this.search(this._queryString, this._callback);
            this._queryString = null;
            this._callback = null;
        }
    }

    function search (queryString : string, callback : function(:int, :int):void) : void
    {
        if (this._oktavia)
        {
            var queryParser = new QueryStringParser();
            this._queries = queryParser.parse(queryString);
            this._highlight = queryParser.highlight();
            var summary = this._oktavia.search(this._queries);
            if (summary.size() > 0)
            {
                this._result = this._sortResult(summary);
                this._proposals = [] : Proposal[];
                this._currentPage = 1;
            }
            else
            {
                this._result = [] : SearchUnit[];
                if (this._queries.length > 1)
                {
                    this._proposals = summary.getProposal();
                }
                else
                {
                    this._proposals = [] : Proposal[];
                }
                this._currentPage = 1;
            }
            callback(this.resultSize(), this.totalPages());
        }
        else
        {
            this._queryString = queryString;
            this._callback = callback;
        }
    }

    function resultSize () : int
    {
        return this._result.length;
    }

    function totalPages () : int
    {
        return Math.ceil(this._result.length / this._entriesPerPage);
    }

    function currentPage () : int
    {
        return this._currentPage;
    }

    function setCurrentPage (page : int) : void
    {
        this._currentPage = page;
    }

    function hasPrevPage () : boolean
    {
        return (this._currentPage != 1);
    }

    function hasNextPage () : boolean
    {
        return (this._currentPage != this.totalPages());
    }

    function pageIndexes () : string[]
    {
        var result = [] : string[];
        var total = this.totalPages();
        if (total < 10)
        {
            for (var i = 1; i <= total; i++)
            {
                result.push(i as string);
            }
        }
        else if (this._currentPage <= 5)
        {
            for (var i = 1; i <= 7; i++)
            {
                result.push(i as string);
            }
            result.push('...', total as string);
        }
        else if (total - 5 <= this._currentPage)
        {
            result.push('1', '...');
            for (var i = total - 8; i <= total; i++)
            {
                result.push(i as string);
            }
        }
        else
        {
            result.push('1', '...');
            for (var i = this._currentPage - 3; i <= this._currentPage + 3; i++)
            {
                result.push(i as string);
            }
            result.push('...', total as string);
        }
        return result;
    }

    function getResult () : _Result[]
    {
        var style = new Style('html');
        var start = (this._currentPage - 1) * this._entriesPerPage;
        var last = Math.min(this._currentPage * this._entriesPerPage, this._result.length);
        var metadata = this._oktavia.getPrimaryMetadata();
        var num = 250;

        var results = [] : _Result[];

        for (var i = start; i < last; i++)
        {
            var unit = this._result[i];
            var info = metadata.getInformation(unit.id).split(Oktavia.eob);

            var offset = info[0].length + 1;
            var content = metadata.getContent(unit.id);
            var start = 0;
            var positions = unit.getPositions();
            if (content.indexOf(info[0]) == 1)
            {
                content = content.slice(info[0].length + 2, content.length);
                start += (info[0].length + 2);
            }
            var end = start + num;
            var split = false;
            if (positions[0].position > end - positions[0].word.length)
            {
                end = positions[0].position + Math.floor(num / 2);
                split = true;
            }
            for (var j = positions.length - 1; j > -1; j--)
            {
                var pos = positions[j];
                if (pos.position + pos.word.length < end)
                {
                    content = [
                        content.slice(0, pos.position - start),
                        style.convert('<hit>*</hit>').replace('*', content.slice(pos.position - start, pos.position + pos.word.length - start)),
                        content.slice(pos.position + pos.word.length - start, content.length)
                    ].join('');
                }
            }
            var text : string;
            if (split)
            {
                text = [
                    content.slice(0, Math.floor(num / 2)) + ' ...',
                    content.slice(-Math.floor(num / 2), end - start)].join('<br/>');
            }
            else
            {
                text = content.slice(0, end - start) + ' ...<br/>';
            }
            text = text.replace(Oktavia.eob, ' ').replace(/(<br\/>)(<br\/>)+/, '<br/><br/>');
            results.push(new _Result(info[0], info[1], text, unit.score));
        }
        return results;
    }

    function getHighlight () : string
    {
        return this._highlight;
    }

    function getProposals () : _Proposal[]
    {
        var style = new Style('html');
        var results = [] : _Proposal[];

        if (this._queries.length > 1)
        {
            for (var i = 0; i < this._proposals.length; i++)
            {
                var proposal = this._proposals[i];
                if (proposal.expect > 0)
                {
                    var label = [] : string[];
                    var option = [] : string[];
                    for (var j = 0; j < this._queries.length; j++)
                    {
                        if (j != proposal.omit)
                        {
                            label.push(style.convert('<hit>' + this._queries[j].toString() + '</hit>'));
                            option.push(this._queries[j].toString());
                        }
                        else
                        {
                            label.push(style.convert('<del>' + this._queries[j].toString() + '</del>'));
                        }
                    }
                    results.push(new _Proposal(option.join(' '), label.join('&nbsp;'), proposal.expect));
                }
            }
        }
        return results;
    }

    function _sortResult (summary : SearchSummary) : SearchUnit[]
    {
        for (var i = 0; i < summary.result.units.length; i++)
        {
            var score = 0;
            var unit = summary.result.units[i];
            for (var pos in unit.positions)
            {
                var position = unit.positions[pos];
                if (this._oktavia.wordPositionType(position.position))
                {
                    score += 10;
                }
                else
                {
                    score += 1;
                }
                if (!position.stemmed)
                {
                    score += 2;
                }
            }
            unit.score = score;
        }
        return summary.getSortedResult();
    }
}

class _Main
{
    static function main(args : string[]) : void
    {
    }
}