File Coverage

blib/lib/DBIx/Skinny/ProxyTable.pm

Criterion	Covered	Total	%
statement	15	47	31.9
branch	0	10	0.0
condition	0	6	0.0
subroutine	5	11	45.4
pod	3	5	60.0
total	23	79	29.1

package DBIx::Skinny::ProxyTable;
use strict;
use warnings;
our $VERSION = '0.10';
use DBIx::Skinny::ProxyTable::Rule;
use Carp;
use DBIx::Skinny::Util qw();

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

sub skinny {
    my $self = $_[0];
    $self->{skinny};
}

sub set {
    my ($self, $from, $to) = @_;

    $self->_validate($from);
    $self->_validate($to);

    my $skinny = $self->skinny;
    my $schema = $skinny->schema;
    my $_schema_info = $schema->schema_info;
    $_schema_info->{$to} = $_schema_info->{$from};

    $self;
}

# This method is safety net for creating wrong table name or executing SQL injection.
sub _validate {
    my ($self, $str) = @_;
    if ( $str !~ /^[a-zA-Z0-9_]+$/ ) {
        Carp::croak("$str should be normal character");
    }
}

sub copy_table {
    my ($self, $from, $to) = @_;

    $self->_validate($from);
    $self->_validate($to);
    my $dbd = $self->skinny->dbd && ref $self->skinny->dbd;
    if ( $dbd && $dbd =~ /^DBIx::Skinny::DBD::(.+)$/ ) {
        $dbd = $1;
        if ( $dbd eq 'mysql' ) {
            $self->skinny->dbh->do(sprintf(q{ CREATE TABLE IF NOT EXISTS %s LIKE %s }, $to, $from));
        } elsif ( $dbd eq 'SQLite' ) {
            my $record = $self->skinny->dbh->selectcol_arrayref(q{
                SELECT sql FROM
                    ( SELECT * FROM sqlite_master UNION ALL
                    SELECT * FROM sqlite_temp_master)
                WHERE type != 'meta' and tbl_name = ?
            }, {}, $from)->[0]
                or Carp::croak("Can't find table $from in sqlite_master or sqlite_temp_master");
            my $sql = $record;
            $sql =~ s/TABLE $from \(/TABLE IF NOT EXISTS $to (/;
            $self->skinny->dbh->do($sql);
        } else {
            die "DBIx::Skinny::DBD::$dbd is not supported";
        }
    }
}

sub rule {
    my ($self, $base, @args) = @_;
    return DBIx::Skinny::ProxyTable::Rule->new($self, $base, @args);
}

1;
__END__