File Coverage

blib/lib/Otogiri/Plugin/TableInfo.pm

Criterion	Covered	Total	%
statement	51	61	83.6
branch	11	20	55.0
condition			n/a
subroutine	12	12	100.0
pod	5	5	100.0
total	79	98	80.6

package Otogiri::Plugin::TableInfo;
use 5.008005;
use strict;
use warnings;

use Otogiri;
use Otogiri::Plugin;
use DBIx::Inspector;
use Otogiri::Plugin::TableInfo::Pg;

our $VERSION = "0.03";

our @EXPORT = qw(show_tables show_views show_create_table show_create_view desc);

sub show_tables {
    my ($self, $like_regex) = @_;
    my $inspector = DBIx::Inspector->new(dbh => $self->dbh);
    my @result = map { $_->name } $inspector->tables;
    @result = grep { $_ =~ /$like_regex/ } @result if ( defined $like_regex );
    return @result;
}

sub show_views {
    my ($self, $like_regex) = @_;
    my $inspector = DBIx::Inspector->new(dbh => $self->dbh);
    my @result = map { $_->name } $inspector->views;
    @result = grep { $_ =~ /$like_regex/ } @result if ( defined $like_regex );
    return @result;
}


sub show_create_table {
    my ($self, $table_name) = @_;
    my $inspector = DBIx::Inspector->new(dbh => $self->dbh);
    my $table = $inspector->table($table_name);

    return if ( !defined $table );

    my $driver_name = $self->{dsn}->{driver};

    if ( $driver_name eq 'mysql' ) {
        my ($row) = $self->search_by_sql("SHOW CREATE TABLE $table_name");
        return $row->{'Create Table'};
    }
    elsif ( $driver_name eq 'SQLite' ) {
        return $table->{SQLITE_SQL};
    }
    elsif ( $driver_name eq 'Pg' ) {
        my $pg = Otogiri::Plugin::TableInfo::Pg->new($self);
        return $pg->show_create_table($table_name);
    }
    return;
}

sub show_create_view {
    my ($self, $view_name) = @_;
    my $inspector = DBIx::Inspector->new(dbh => $self->dbh);
    my $view = $inspector->view($view_name);

    return if ( !defined $view );

    my $driver_name = $self->{dsn}->{driver};

    if ( $driver_name eq 'mysql' ) {
        my ($row) = $self->search_by_sql("SHOW CREATE VIEW $view_name");
        return $row->{'Create View'};
    }
    elsif ( $driver_name eq 'SQLite' ) {
        return $view->{SQLITE_SQL};
    }
    elsif ( $driver_name eq 'Pg' ) {
        my $pg = Otogiri::Plugin::TableInfo::Pg->new($self);
        return $pg->show_create_view($view_name);
    }
    return;
}

sub desc {
    my ($self, $table_name) = @_;
    $self->show_create_table($table_name);
}



1;
__END__