#!/usr/bin/perl

# This file is part of Koha.
#
# Koha is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# Koha is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Koha; if not, see <https://www.gnu.org/licenses>.

use Modern::Perl;
use Test::NoWarnings;
use Test::More tests => 10;
use Net::Z3950::ZOOM;

use t::lib::Mocks;

use C4::Search qw( buildQuery  );

#FIXME: would it be better to use our real ccl.properties file?
sub _get_ccl_properties {
    my $config = q(
        kw 1=1016
        wrdl 4=6
        rtrn 5=1
        right-Truncation rtrn
        rk 2=102
        ti 1=4
        title ti
        ext 4=1 6=3
        Title-cover 1=36
        r1 9=32
        r2 9=28
        r3 9=26
        r4 9=24
        r5 9=22
        r6 9=20
        r7 9=18
        r8 9=16
        r9 9=14
        phr 4=1
        fuzzy 5=103
        ccode 1=8009
        nb 1=7
        ns 1=8
    );
    return $config;
}

#FIXME: We should add QueryFuzzy and QueryStemming preferences to expand test permutations

subtest "test weighted autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["test"];
    my $indexes   = [""];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error,        undef,                        "Error is correct" );
    is( $query,        '(rk=(kw,wrdl,rtrn=test )) ', 'Query is correct with auto truncation' );
    is( $simple_query, 'test',                       "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=test',              'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: test',              'Query desc is correct' );
    is( $limit,        '',                           "Limit is correct" );
    is( $limit_cgi,    '',                           "Limit cgi is correct" );
    is( $limit_desc,   '',                           "Limit desc is correct" );
    is( $query_type,   undef,                        "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);

};

subtest "test* weighted autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["test*"];
    my $indexes   = [""];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error,        undef,                        "Error is correct" );
    is( $query,        '(rk=(kw,wrdl,rtrn=test )) ', 'Query is correct with manual truncation' );
    is( $simple_query, 'test*',                      "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=test%2A',           'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: test*',             'Query desc is correct' );
    is( $limit,        '',                           "Limit is correct" );
    is( $limit_cgi,    '',                           "Limit cgi is correct" );
    is( $limit_desc,   '',                           "Limit desc is correct" );
    is( $query_type,   undef,                        "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "test weighted not-autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '0' );
    t::lib::Mocks::mock_preference( 'QueryFuzzy',        '1' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["test"];
    my $indexes   = [""];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error, undef, "Error is correct" );
    is(
        $query,
        '(rk=(Title-cover,ext,r1="test" or ti,ext,r2="test" or Title-cover,phr,r3="test" or ti,wrdl,r4="test" or wrdl,fuzzy,r8="test" or wrdl,right-Truncation,r9="test" or wrdl,r9="test")) ',
        'Query is correct with weighted fields'
    );
    is( $simple_query, 'test',          "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=test', 'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: test', 'Query desc is correct' );
    is( $limit,        '',              "Limit is correct" );
    is( $limit_cgi,    '',              "Limit cgi is correct" );
    is( $limit_desc,   '',              "Limit desc is correct" );
    is( $query_type,   undef,           "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "test ccode:REF weighted autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["test"];
    my $indexes   = [""];
    my $limits    = ["ccode:REF"];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error,        undef,                                     "Error is correct" );
    is( $query,        '(rk=(kw,wrdl,rtrn=test )) and ccode=REF', 'Query is correct with auto truncation and limits' );
    is( $simple_query, 'test',                                    "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=test',                           'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: test',                           'Query desc is correct' );
    is( $limit,        'and ccode=REF',                           "Limit is correct" );
    is( $limit_cgi,    '&limit=ccode%3AREF',                      "Limit cgi is correct" );
    is( $limit_desc,   'ccode:REF',                               "Limit desc is correct" );
    is( $query_type,   undef,                                     "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "test ccode:REF weighted not-autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '0' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["test"];
    my $indexes   = [""];
    my $limits    = ["ccode:REF"];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error, undef, "Error is correct" );
    is(
        $query,
        '(rk=(Title-cover,ext,r1="test" or ti,ext,r2="test" or Title-cover,phr,r3="test" or ti,wrdl,r4="test" or wrdl,fuzzy,r8="test" or wrdl,right-Truncation,r9="test" or wrdl,r9="test")) and ccode=REF',
        'Query is correct with weighted fields and limits'
    );
    is( $simple_query, 'test',               "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=test',      'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: test',      'Query desc is correct' );
    is( $limit,        'and ccode=REF',      "Limit is correct" );
    is( $limit_cgi,    '&limit=ccode%3AREF', "Limit cgi is correct" );
    is( $limit_desc,   'ccode:REF',          "Limit desc is correct" );
    is( $query_type,   undef,                "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "kw:one and title:two ccode:REF weighted autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );

    my $config    = _get_ccl_properties();
    my $operators = ["and"];
    my $operands  = [ "one", "two" ];
    my $indexes   = [ "kw",  "title" ];
    my $limits    = ["ccode:REF"];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error, undef, "Error is correct" );
    is(
        $query, '(rk=(kw,wrdl,rtrn=one )) and (rk=(title,wrdl,rtrn=two )) and ccode=REF',
        'Query is correct with auto truncation, limits, and using indexes and operators'
    );
    is( $simple_query, 'one',                                 "Simple query is correct?" );
    is( $query_cgi,    'idx=kw&q=one&op=and&idx=title&q=two', 'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: one and title,wrdl: two',    'Query desc is correct' );
    is( $limit,        'and ccode=REF',                       "Limit is correct" );
    is( $limit_cgi,    '&limit=ccode%3AREF',                  "Limit cgi is correct" );
    is( $limit_desc,   'ccode:REF',                           "Limit desc is correct" );
    is( $query_type,   undef,                                 "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "one and two weighted autotruncated" => sub {
    plan tests => 13;

    t::lib::Mocks::mock_preference( 'QueryWeightFields', '1' );
    t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["one and two"];
    my $indexes   = [""];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
        C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
    is( $error, undef,                               "Error is correct" );
    is( $query, '(rk=(kw,wrdl,rtrn=one and two )) ', 'Query is correct with auto truncation and unstructured query' );
    is( $simple_query, 'one and two',                "Simple query is correct" );
    is( $query_cgi,    'idx=kw&q=one%20and%20two',   'Query cgi is correct' );
    is( $query_desc,   'kw,wrdl: one and two',       'Query desc is correct' );
    is( $limit,        '',                           "Limit is correct" );
    is( $limit_cgi,    '',                           "Limit cgi is correct" );
    is( $limit_desc,   '',                           "Limit desc is correct" );
    is( $query_type,   undef,                        "Query type is correct" );
    my $q = Net::Z3950::ZOOM::query_create();
    my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
    my $res = Net::Z3950::ZOOM::query_ccl2rpn(
        $q,           $query,      $config,
        $ccl_errcode, $ccl_errstr, $ccl_errpos
    );
    is( $res,         0, "created CCL2RPN query" );
    is( $ccl_errcode, 0 );
    is( $ccl_errstr,  "" );
    is( $ccl_errpos,  0 );
    Net::Z3950::ZOOM::query_destroy($q);
};

subtest "test with ISBN variations" => sub {
    plan tests => 12;

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["1565926994"];
    my $indexes   = ["nb"];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    foreach my $sample (
        { state => 0, query => 'nb=(rk=(1565926994)) ' },
        {
            state => 1,
            query =>
                'kw,wrdl=(rk=((nb=1-56592-699-4 OR nb=1-56592-699-4 OR nb=978-1-56592-699-8 OR nb=1565926994 OR nb=9781565926998))) '
        }
        )
    {
        t::lib::Mocks::mock_preference( 'SearchWithISBNVariations', $sample->{state} );

        # Test with disabled variatioms
        my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
            C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
        say 'Q: > ', $query;
        is( $error, undef,            "Error is correct" );
        is( $query, $sample->{query}, 'Search ISBN when variations is ' . $sample->{state} );
        my $q = Net::Z3950::ZOOM::query_create();
        my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
        my $res = Net::Z3950::ZOOM::query_ccl2rpn(
            $q,           $query,      $config,
            $ccl_errcode, $ccl_errstr, $ccl_errpos
        );
        is( $res,         0, "created CCL2RPN query" );
        is( $ccl_errcode, 0 );
        is( $ccl_errstr,  "" );
        is( $ccl_errpos,  0 );
        Net::Z3950::ZOOM::query_destroy($q);
    }

};

subtest "test with ISSN variations" => sub {
    plan tests => 12;

    my $config    = _get_ccl_properties();
    my $operators = [""];
    my $operands  = ["2434561X"];
    my $indexes   = ["ns"];
    my $limits    = [""];
    my $sort_by   = [""];
    my ( $scan, $lang );

    foreach my $sample (
        { state => 0, query => 'ns=(rk=(2434561X)) ' },
        { state => 1, query => 'kw,wrdl=(rk=((ns=2434-561X OR ns=2434561X))) ' }
        )
    {
        t::lib::Mocks::mock_preference( 'SearchWithISSNVariations', $sample->{state} );

        # Test with disabled variatioms
        my ( $error, $query, $simple_query, $query_cgi, $query_desc, $limit, $limit_cgi, $limit_desc, $query_type ) =
            C4::Search::buildQuery( $operators, $operands, $indexes, $limits, $sort_by, $scan, $lang );
        say 'Q: > ', $query;
        is( $error, undef,            "Error is correct" );
        is( $query, $sample->{query}, 'Search ISSN when variations is ' . $sample->{state} );
        my $q = Net::Z3950::ZOOM::query_create();
        my ( $ccl_errcode, $ccl_errstr, $ccl_errpos ) = ( 0, "", 0 );
        my $res = Net::Z3950::ZOOM::query_ccl2rpn(
            $q,           $query,      $config,
            $ccl_errcode, $ccl_errstr, $ccl_errpos
        );
        is( $res,         0, "created CCL2RPN query" );
        is( $ccl_errcode, 0 );
        is( $ccl_errstr,  "" );
        is( $ccl_errpos,  0 );
        Net::Z3950::ZOOM::query_destroy($q);
    }

};