File Coverage

blib/lib/FalkorDB/Graph.pm

Criterion	Covered	Total	%
statement	12	91	13.1
branch	0	30	0.0
condition	0	12	0.0
subroutine	4	17	23.5
pod	7	10	70.0
total	23	160	14.3

package FalkorDB::Graph;

use strict;
use warnings;
use Scalar::Util qw(looks_like_number);
use FalkorDB::QueryResult;

sub new {
    my ( $class, $db, $name ) = @_;
    my $self = {
        db   => $db,
        name => $name,
    };
    return bless $self, $class;
}

sub db {
    my ($self) = @_;
    return $self->{db};
}

sub name {
    my ($self) = @_;
    return $self->{name};
}

sub delete {
    my ($self) = @_;
    return $self->db->delete_graph( $self->name );
}

sub query {
    my ( $self, $cypher, $params ) = @_;
    my $query_str = $self->_build_query_string( $cypher, $params );
    my ( $raw_res, $error ) = $self->db->redis->__std_cmd( "GRAPH.QUERY", $self->name, $query_str );
    if ( defined $error ) {
        require Carp;
        Carp::croak("[GRAPH.QUERY] $error");
    }
    return FalkorDB::QueryResult->new_from_raw($raw_res);
}

sub ro_query {
    my ( $self, $cypher, $params ) = @_;
    my $query_str = $self->_build_query_string( $cypher, $params );
    my ( $raw_res, $error ) =
      $self->db->redis->__std_cmd( "GRAPH.RO_QUERY", $self->name, $query_str );
    if ( defined $error ) {
        require Carp;
        Carp::croak("[GRAPH.RO_QUERY] $error");
    }
    return FalkorDB::QueryResult->new_from_raw($raw_res);
}

sub explain {
    my ( $self, $cypher, $params ) = @_;
    my $query_str = $self->_build_query_string( $cypher, $params );
    my ( $raw_res, $error ) =
      $self->db->redis->__std_cmd( "GRAPH.EXPLAIN", $self->name, $query_str );
    if ( defined $error ) {
        require Carp;
        Carp::croak("[GRAPH.EXPLAIN] $error");
    }
    return $raw_res;
}

sub profile {
    my ( $self, $cypher, $params ) = @_;
    my $query_str = $self->_build_query_string( $cypher, $params );
    my ( $raw_res, $error ) =
      $self->db->redis->__std_cmd( "GRAPH.PROFILE", $self->name, $query_str );
    if ( defined $error ) {
        require Carp;
        Carp::croak("[GRAPH.PROFILE] $error");
    }
    return $raw_res;
}

sub create_index {
    my ( $self, $label, $property ) = @_;
    return $self->query("CREATE INDEX FOR (n:$label) ON (n.$property)");
}

sub drop_index {
    my ( $self, $label, $property ) = @_;
    return $self->query("DROP INDEX FOR (n:$label) ON (n.$property)");
}

# --- Parameter Serialization ---

sub _build_query_string {
    my ( $self, $cypher, $params ) = @_;

    return $cypher unless $params && ref $params eq 'HASH';

    my @serialized;
    while ( my ( $k, $v ) = each %$params ) {
        push @serialized, "$k=" . _serialize_param($v);
    }

    if (@serialized) {
        return "CYPHER " . join( " ", @serialized ) . " " . $cypher;
    }

    return $cypher;
}

sub _serialize_param {
    my ($val) = @_;

    if ( !defined $val ) {
        return 'null';
    }

    if ( ref $val ) {
        my $ref_type = ref $val;

        # Handle boolean wrappers (e.g. JSON::PP::Boolean, Types::Serialiser::Boolean)
        if (   $ref_type eq 'JSON::PP::Boolean'
            || $ref_type eq 'Types::Serialiser::Boolean' )
        {
            return $$val ? 'true' : 'false';
        }
        elsif ( $ref_type eq 'ARRAY' ) {
            my @elems = map { _serialize_param($_) } @$val;
            return '[' . join( ', ', @elems ) . ']';
        }
        elsif ( $ref_type eq 'HASH' ) {
            my @pairs;
            while ( my ( $k, $v ) = each %$val ) {
                my $safe_key = $k;
                if ( $k =~ /[^a-zA-Z0-9_]/ ) {
                    $safe_key = "`$k`";
                }
                push @pairs, "$safe_key: " . _serialize_param($v);
            }
            return '{' . join( ', ', @pairs ) . '}';
        }
        else {
            # Stringify fallback for unrecognized references
            return '"' . _escape_string("$val") . '"';
        }
    }

    # Check for boolean keywords passed as strings
    if ( $val eq 'true' || $val eq 'false' ) {
        return $val;
    }

    # Check if value is numeric (so we avoid quoting it)
    if ( looks_like_number($val) && $val !~ /^0\d+/ ) {
        return $val;
    }

    # Default: quote as string
    return '"' . _escape_string($val) . '"';
}

sub _escape_string {
    my ($str) = @_;
    $str =~ s/\\/\\\\/g;
    $str =~ s/"/\\"/g;
    $str =~ s/\n/\\n/g;
    $str =~ s/\r/\\r/g;
    $str =~ s/\t/\\t/g;
    return $str;
}

1;
__END__