File Coverage

blib/lib/Pg/SQL/PrettyPrinter/Node/RangeFunction.pm

Criterion	Covered	Total	%
statement	58	61	95.0
branch	6	12	50.0
condition			n/a
subroutine	16	16	100.0
pod	2	2	100.0
total	82	91	90.1

package Pg::SQL::PrettyPrinter::Node::RangeFunction;

# UTF8 boilerplace, per http://stackoverflow.com/questions/6162484/why-does-modern-perl-avoid-utf-8-by-default/
use v5.26;
use strict;
use warnings;
use warnings qw( FATAL utf8 );
use utf8;
use open qw( :std :utf8 );
use Unicode::Normalize qw( NFC );
use Unicode::Collate;
use Encode qw( decode );

if ( grep /\P{ASCII}/ => @ARGV ) {
    @ARGV = map { decode( 'UTF-8', $_ ) } @ARGV;
}

# If there is __DATA__,then uncomment next line:
# binmode( DATA, ':encoding(UTF-8)' );
# UTF8 boilerplace, per http://stackoverflow.com/questions/6162484/why-does-modern-perl-avoid-utf-8-by-default/

# Useful common code
use autodie;
use Carp         qw( carp croak confess cluck );
use English      qw( -no_match_vars );
use Data::Dumper qw( Dumper );

# give a full stack dump on any untrapped exceptions
local $SIG{ __DIE__ } = sub {
    confess "Uncaught exception: @_" unless $^S;
};

# now promote run-time warnings into stackdumped exceptions
#   *unless* we're in an try block, in which
#   case just generate a clucking stackdump instead
local $SIG{ __WARN__ } = sub {
    if   ( $^S ) { cluck "Trapped warning: @_" }
    else         { confess "Deadly warning: @_" }
};

# Useful common code

use parent qw( Pg::SQL::PrettyPrinter::Node );

sub new {
    my $class = shift;
    my $self  = $class->SUPER::new( @_ );
    bless $self, $class;

    croak( 'Multiple RangeFunctions not yet handled' ) if 1 != scalar @{ $self->{ 'functions' } };
    $self->objectify( 'functions', [ 'alias', 'colnames' ] );
    croak( 'Invalid object inside RangeFunction' ) unless $self->{ 'functions' }->[ 0 ]->isa( 'Pg::SQL::PrettyPrinter::Node::List' );
    croak( 'Multiple RangeFunctions not yet handled (#2)' ) if 1 != scalar @{ $self->{ 'functions' }->[ 0 ]->{ 'items' } };

    return $self;
}

sub as_text {
    my $self      = shift;
    my $full_name = $self->{ 'functions' }->[ 0 ]->{ 'items' }->[ 0 ]->as_text;

    # Add optional WITH ORDINALITY
    $full_name .= ' WITH ORDINALITY' if $self->{ 'ordinality' };

    # Shortcut
    my $A = $self->{ 'alias' };
    return $full_name unless $A;

    my $base_with_alias = sprintf( '%s AS %s', $full_name, $self->quote_ident( $A->{ 'aliasname' } ) );
    return $base_with_alias unless $A->{ 'colnames' };

    return sprintf( '%s ( %s )', $base_with_alias, join( ', ', map { $_->as_ident } @{ $A->{ 'colnames' } } ) );
}

1;