File Coverage

blib/lib/SQL/Translator/Producer/Teng.pm

Criterion	Covered	Total	%
statement	43	43	100.0
branch	6	6	100.0
condition	2	3	66.6
subroutine	10	10	100.0
pod	0	1	0.0
total	61	63	96.8

package SQL::Translator::Producer::Teng;
use 5.008001;
use strict;
use warnings;

our $VERSION = "0.03";

use Text::Xslate;
use Data::Section::Simple;
use DBI;
use SQL::Translator::Schema::Field;

my $_tx;
sub _tx {
    $_tx ||= Text::Xslate->new(
        type   => 'text',
        module => ['Text::Xslate::Bridge::Star'],
        path   => [Data::Section::Simple::get_data_section]
    );
}

sub produce {
    my $translator = shift;
    my $schema = $translator->schema;
    my $args = $translator->producer_args;

    my $package        = $args->{package};
    my $base_row_class = $args->{base_row_class};

    my @tables;
    for my $table ($schema->get_tables) {
        my @pks;
        my @columns;
        for my $field ($table->get_fields) {
            push @columns, {
                name      => $field->name,
                type_name => $field->data_type,
                type      => _get_dbi_const($field->sql_data_type),
            };
            push @pks, $field->name if $field->is_primary_key;
        }

        push @tables, {
            name    => $table->name,
            pks     => \@pks,
            columns => \@columns,
        };
    }

    _tx->render('schema.tx', {
        package        => $package,
        base_row_class => $base_row_class,
        tables         => \@tables,
    });
}

my %CONST_HASH;
sub _get_dbi_const {
    my $val = shift;

    unless (%CONST_HASH) {
        for my $const_key (@{ $DBI::EXPORT_TAGS{sql_types} }) {
            my $const_val = DBI->can($const_key)->();

            unless (exists $CONST_HASH{$const_val}) {
                $CONST_HASH{$const_val} = $const_key;
            }
        }
    }

    $CONST_HASH{$val};
}

1;
__DATA__