Changeset 2538 for Search-Query/trunk/lib/Search/Query/Dialect/KSx.pm

Timestamp:

02/23/10 09:40:38 (2 years ago)

Author:

karpet

Message:

ok. real KS support.

File:

• Search-Query/trunk/lib/Search/Query/Dialect/KSx.pm (modified) (2 diffs)

Search-Query/trunk/lib/Search/Query/Dialect/KSx.pm

@@ -5 +5 @@
 use Carp;
 use Data::Dump qw( dump );
+use Search::Query::Field::KSx;
 
 our $VERSION = '0.08';
+
+__PACKAGE__->mk_accessors(
+    qw(
+        wildcard
+        fuzzify
+        )
+);
 
 =head1 NAME
@@ -23 +31 @@
 =head1 METHODS
 
-This class is a subclass of Search::Query::Dialect::Native. 
-Only new or overridden methods are documented here.
-
-=cut
+This class is a subclass of Search::Query::Dialect. Only new or overridden
+methods are documented here.
+
+=cut
+
+=head2 init
+
+Overrides base method and sets SWISH-appropriate defaults.
+Can take the following params, also available as standard attribute
+methods.
+
+=over
+
+=item wildcard
+
+Default is '*'.
+
+=item fuzzify
+
+If true, a wildcard is automatically appended to each query term.
+
+=back
+
+=cut
+
+sub init {
+    my $self = shift;
+
+    $self->SUPER::init(@_);
+
+    #carp dump $self;
+    $self->{wildcard} = '*';
+    if ( $self->parser->fields ) {
+        $self->{default_field} ||= $self->parser->default_field
+            || [ sort keys %{ $self->parser->fields } ];
+    }
+    else {
+        $self->{default_field} ||= $self->parser->default_field
+            || 'swishdefault';
+    }
+    if ( $self->{default_field} and !ref( $self->{default_field} ) ) {
+        $self->{default_field} = [ $self->{default_field} ];
+    }
+
+    return $self;
+}
+
+=head2 stringify
+
+Returns the Query object as a normalized string.
+
+=cut
+
+my %op_map = (
+    '+' => 'AND',
+    ''  => 'OR',
+    '-' => 'NOT',
+);
+
+sub stringify {
+    my $self = shift;
+    my $tree = shift || $self;
+
+    my @q;
+    foreach my $prefix ( '+', '', '-' ) {
+        my @clauses;
+        my $joiner = $op_map{$prefix};
+        next unless exists $tree->{$prefix};
+        for my $clause ( @{ $tree->{$prefix} } ) {
+            push( @clauses, $self->stringify_clause( $clause, $prefix ) );
+        }
+        next if !@clauses;
+
+        push @q, join( " $joiner ", grep { defined and length } @clauses );
+    }
+
+    return join " AND ", @q;
+}
+
+sub _doctor_value {
+    my ( $self, $clause ) = @_;
+
+    my $value = $clause->{value};
+
+    if ( $self->fuzzify ) {
+        $value .= '*' unless $value =~ m/[\*\%]/;
+    }
+
+    # normalize wildcard
+    my $wildcard = $self->wildcard;
+    $value =~ s/[\*\%]/$wildcard/g;
+
+    return $value;
+}
+
+=head2 stringify_clause( I<leaf>, I<prefix> )
+
+Called by stringify() to handle each Clause in the Query tree.
+
+=cut
+
+sub stringify_clause {
+    my $self   = shift;
+    my $clause = shift;
+    my $prefix = shift;
+
+    #warn dump $clause;
+    #warn "prefix = '$prefix'";
+
+    if ( $clause->{op} eq '()' ) {
+        if ( $clause->has_children and $clause->has_children == 1 ) {
+            return $self->stringify( $clause->{value} );
+        }
+        else {
+            return
+                ( $prefix eq '-' ? 'NOT ' : '' ) . "("
+                . $self->stringify( $clause->{value} ) . ")";
+        }
+    }
+
+    # make sure we have a field
+    my @fields
+        = $clause->{field}
+        ? ( $clause->{field} )
+        : ( @{ $self->_get_default_field } );
+
+    # what value
+    my $value
+        = ref $clause->{value}
+        ? $clause->{value}
+        : $self->_doctor_value($clause);
+
+    my $wildcard = $self->wildcard;
+
+    # normalize operator
+    my $op = $clause->{op} || ":";
+    if ( $op eq '=' ) {
+        $op = ':';
+    }
+    if ( $prefix eq '-' ) {
+        $op = '!' . $op;
+    }
+    if ( $value =~ m/\%/ ) {
+        $op = $prefix eq '-' ? '!~' : '~';
+    }
+
+    my $quote = $clause->quote || '';
+
+    my @buf;
+NAME: for my $name (@fields) {
+        my $field = $self->_get_field($name);
+
+        if ( defined $field->callback ) {
+            push( @buf, $field->callback->( $field, $op, $value ) );
+            next NAME;
+        }
+
+        #warn dump [ $name, $op, $quote, $value ];
+
+        # invert fuzzy
+        if ( $op eq '!~' ) {
+            $value .= $wildcard unless $value =~ m/\Q$wildcard/;
+            push( @buf,
+                join( '', 'NOT ', $name, '=', qq/$quote$value$quote/ ) );
+        }
+
+        # fuzzy
+        elsif ( $op eq '~' ) {
+            $value .= $wildcard unless $value =~ m/\Q$wildcard/;
+            push( @buf, join( '', $name, '=', qq/$quote$value$quote/ ) );
+        }
+
+        # invert
+        elsif ( $op eq '!:' ) {
+            push( @buf,
+                join( '', 'NOT ', $name, ':', qq/$quote$value$quote/ ) );
+        }
+
+        # range
+        elsif ( $op eq '..' ) {
+            if ( ref $value ne 'ARRAY' or @$value != 2 ) {
+                croak "range of values must be a 2-element ARRAY";
+            }
+
+            # we support only numbers at this point
+            for my $v (@$value) {
+                if ( $v =~ m/\D/ ) {
+                    croak "non-numeric range values are not supported: $v";
+                }
+            }
+
+            my @range = ( $value->[0] .. $value->[1] );
+            push( @buf,
+                join( '', $name, ':', '(', join( ' OR ', @range ), ')' ) );
+
+        }
+
+        # invert range
+        elsif ( $op eq '!..' ) {
+            if ( ref $value ne 'ARRAY' or @$value != 2 ) {
+                croak "range of values must be a 2-element ARRAY";
+            }
+
+            # we support only numbers at this point
+            for my $v (@$value) {
+                if ( $v =~ m/\D/ ) {
+                    croak "non-numeric range values are not supported: $v";
+                }
+            }
+
+            my @range = ( $value->[0] .. $value->[1] );
+            push( @buf,
+                join( '', '-', $name, ':', '( ', join( ' ', @range ), ' )' )
+            );
+        }
+
+        # standard
+        else {
+            push( @buf, join( '', $name, ':', qq/$quote$value$quote/ ) );
+        }
+    }
+    my $joiner = $prefix eq '-' ? ' AND ' : ' OR ';
+    return
+          ( scalar(@buf) > 1 ? '(' : '' )
+        . join( $joiner, @buf )
+        . ( scalar(@buf) > 1 ? ')' : '' );
+}
+
+=head2 field_class
+
+Returns "Search::Query::Field::KSx".
+
+=cut
+
+sub field_class {'Search::Query::Field::KSx'}
 
 1;