[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Parser / DDL / MySQL.pm

use MooseX::Declare;
role SQL::Translator::Parser::DDL::MySQL {
    use MooseX::Types::Moose qw(Str);
    use MooseX::MultiMethods;
    use SQL::Translator::Constants qw(:sqlt_types :sqlt_constants);
    use aliased 'SQL::Translator::Object::Column';
    use aliased 'SQL::Translator::Object::Constraint';
    use aliased 'SQL::Translator::Object::ForeignKey';
    use aliased 'SQL::Translator::Object::Index';
    use aliased 'SQL::Translator::Object::PrimaryKey';
    use aliased 'SQL::Translator::Object::Procedure';
    use aliased 'SQL::Translator::Object::Schema';
    use aliased 'SQL::Translator::Object::Table';
    use aliased 'SQL::Translator::Object::View';

    around _build_data_type_mapping {
        my $data_type_mapping = $self->$orig;
        $data_type_mapping->{date} = SQL_DATE();

        return $data_type_mapping;
    }; 

    multi method parse(Schema $data) { $data }

    multi method parse(Str $data) {
        my $parser = Parse::RecDescent->new($self->grammar);

        unless (defined $parser) {
            return $self->error("Error instantiating Parse::RecDescent ".
                "instance: Bad grammar");
        }

        my $translator = $self->translator;

        my $parser_version = $translator->has_parser_args
                             ? $translator->engine_version($translator->parser_args->{mysql_parser_version}, 'mysql') || DEFAULT_PARSER_VERSION
                             : DEFAULT_PARSER_VERSION;
    
        while ($data =~ s#/\*!(\d{5})?(.*?)\*/#($1 && $1 > $parser_version ? '' : $2)#es) { }

        my $result = $parser->startrule($data);
        die "Parse failed" unless defined $result;
    
        my $schema = $translator->schema;
        $schema->name($result->{'database_name'}) if $result->{'database_name'};
    
        my @tables = sort { $result->{'tables'}{ $a }{'order'} <=> $result->{'tables'}{ $b }{'order'} } keys %{ $result->{'tables'} };
    
        for my $table_name ( @tables ) {
            my $tdata = $result->{tables}{ $table_name };
            my $table = Table->new({ name => $tdata->{table_name}, schema => $schema });
            $schema->add_table($table);
            $table->comments( join "\n", @{$tdata->{comments}} ) if $tdata->{comments};
    
            my @fields = sort { $tdata->{'fields'}->{$a}->{'order'} <=> $tdata->{'fields'}->{$b}->{'order'} } keys %{ $tdata->{'fields'} };
    
            for my $fname ( @fields ) {
                my $fdata = $tdata->{fields}{ $fname };
                my $field = Column->new({
                    name              => $fdata->{name},
                    data_type         => $fdata->{data_type},
                    sql_data_type     => $self->data_type_mapping->{$fdata->{data_type}} || -999999,
                    size              => $fdata->{size},
                    default_value     => $fdata->{default},
                    is_auto_increment => $fdata->{is_auto_inc},
                    is_nullable       => $fdata->{null},
                    is_primary_key    => $fdata->{is_primary_key} ? 1 : 0,
                    table             => $table,
                });
                $field->comments($fdata->{comments});
                $table->add_column($field);
    
                $field->is_primary_key(1) if $fdata->{is_primary_key};

                my %extra;
                for my $qual ( qw[ binary unsigned zerofill list collate ],
                        'character set', 'on update' ) {
                    if ( my $val = $fdata->{ $qual } || $fdata->{ uc $qual } ) {
                        next if ref $val eq 'ARRAY' && !@$val;
                        $extra{$qual} = $val;
                        #$field->extra( $qual, $val );
                    }
                }
                $field->extra(\%extra);

                if ( $fdata->{has_index} ) {
                    my $index = Index->new({ name => '', type => 'NORMAL', table => $table });
                    $index->add_column($table->get_column($fdata->{name}));
                    $table->add_index($index);
                }
    
                if ( $fdata->{is_unique} ) {
                    push @{ $tdata->{constraints} }, { name => '', type => 'UNIQUE', fields => [ $fdata->{name} ] };
                }
    
                for my $cdata ( @{ $fdata->{constraints} } ) {
                    next unless lc $cdata->{type} eq 'foreign_key';
                    $cdata->{fields} ||= [ $field->name ];
                    push @{ $tdata->{constraints} }, $cdata;
                }
            }
    
            for my $idata ( @{ $tdata->{indices} || [] } ) {
                my $index = Index->new({ name => $idata->{name} || '', type => uc($idata->{type}), table => $table });
                map { $index->add_column($table->get_column($_)) } @{$idata->{fields}};
                $table->add_index($index);
            }
            
    
            if ( my @options = @{ $tdata->{'table_options'} || [] } ) {
                my @cleaned_options;
                my @ignore_opts = $translator->has_parser_args && $translator->parser_args->{'ignore_opts'}
                    ? split( /,/, $translator->parser_args->{'ignore_opts'} )
                    : ();
                if (@ignore_opts) {
                    my $ignores = { map { $_ => 1 } @ignore_opts };
                    foreach my $option (@options) {
                        # make sure the option isn't in ignore list
                        my ($option_key) = keys %$option;
                        if ( !exists $ignores->{$option_key} ) {
                            push @cleaned_options, $option;
                        }
                    }
                } else {
                    @cleaned_options = @options;
                }
                $table->options( \@cleaned_options ); # or die $table->error;
            }
        }
        
        for my $proc_name ( keys %{ $result->{procedures} } ) {
            my $procedure = Procedure->new({ name  => $proc_name,
                                             owner => $result->{procedures}->{$proc_name}->{owner},
                                             sql   => $result->{procedures}->{$proc_name}->{sql}
            });
            $schema->add_procedure($procedure);
        }
    
        for my $view_name ( keys %{ $result->{'views'} } ) {
            my $view = View->new({ 
                name => $view_name,
                sql  => $result->{'views'}->{$view_name}->{sql},
            });
            $schema->add_view($view);
        }
        for my $table_name (@tables) {
            my $table = $schema->get_table($table_name);
            my $tdata = $result->{tables}{ $table_name };
            for my $cdata ( @{ $tdata->{constraints} || [] } ) {
                my $constraint;
                if (uc $cdata->{type} eq 'PRIMARY_KEY') {
                    $constraint = PrimaryKey->new({ name => $cdata->{name} || '', table => $table });
                    $table->get_column($_)->is_primary_key(1) for @{$cdata->{fields}};
                } elsif (uc $cdata->{type} eq 'FOREIGN_KEY') {
                    $constraint = ForeignKey->new({ name => $cdata->{name} || '',
                                                    table => $table,
                                                    reference_table => $cdata->{reference_table},
                                                    defined $cdata->{reference_fields} ? (reference_columns => $cdata->{reference_fields}) : (),
                                                    on_delete => $cdata->{on_delete} || $cdata->{on_delete_do},
                                                    on_update => $cdata->{on_update} || $cdata->{on_update_do} });
                    $table->get_column($_)->is_foreign_key(1) for @{$cdata->{fields}};
                    $table->get_column($_)->foreign_key_reference($constraint) for @{$cdata->{fields}};
                } else {
                    $constraint = Constraint->new({ name => $cdata->{name} || '', type => uc $cdata->{type}, table => $table });
                }
                $constraint->add_column($table->get_column($_)) for @{$cdata->{fields}};
                $table->add_constraint($constraint);
            }
        }
        return 1;
    }
}
Commit	Line	Data
4f4fd192	1	use MooseX::Declare;
4f4fd192	2	role SQL::Translator::Parser::DDL::MySQL {
acfd88b5	3	use MooseX::Types::Moose qw(Str);
	4	use MooseX::MultiMethods;
	5	use SQL::Translator::Constants qw(:sqlt_types :sqlt_constants);
	6	use aliased 'SQL::Translator::Object::Column';
	7	use aliased 'SQL::Translator::Object::Constraint';
	8	use aliased 'SQL::Translator::Object::ForeignKey';
	9	use aliased 'SQL::Translator::Object::Index';
	10	use aliased 'SQL::Translator::Object::PrimaryKey';
c6365be1	11	use aliased 'SQL::Translator::Object::Procedure';
acfd88b5	12	use aliased 'SQL::Translator::Object::Schema';
acfd88b5	13	use aliased 'SQL::Translator::Object::Table';
c6365be1	14	use aliased 'SQL::Translator::Object::View';
acfd88b5	15
	16	around _build_data_type_mapping {
	17	my $data_type_mapping = $self->$orig;
	18	$data_type_mapping->{date} = SQL_DATE();
	19
	20	return $data_type_mapping;
	21	};
	22
bf2a7e53	23	multi method parse(Schema $data) { $data }
acfd88b5	24
bf2a7e53	25	multi method parse(Str $data) {
acfd88b5	26	my $parser = Parse::RecDescent->new($self->grammar);
acfd88b5	27
acfd88b5	28	unless (defined $parser) {
	29	return $self->error("Error instantiating Parse::RecDescent ".
	30	"instance: Bad grammar");
	31	}
	32
53e0573b	33	my $translator = $self->translator;
	34
	35	my $parser_version = $translator->has_parser_args
	36	? $translator->engine_version($translator->parser_args->{mysql_parser_version}, 'mysql') \|\| DEFAULT_PARSER_VERSION
	37	: DEFAULT_PARSER_VERSION;
acfd88b5	38
acfd88b5	39	while ($data =~ s#/\!(\d{5})?(.?)\*/#($1 && $1 > $parser_version ? '' : $2)#es) { }
bf2a7e53	40
acfd88b5	41	my $result = $parser->startrule($data);
acfd88b5	42	die "Parse failed" unless defined $result;
acfd88b5	43
bf2a7e53	44	my $schema = $translator->schema;
acfd88b5	45	$schema->name($result->{'database_name'}) if $result->{'database_name'};
	46
	47	my @tables = sort { $result->{'tables'}{ $a }{'order'} <=> $result->{'tables'}{ $b }{'order'} } keys %{ $result->{'tables'} };
	48
	49	for my $table_name ( @tables ) {
	50	my $tdata = $result->{tables}{ $table_name };
bf2a7e53	51	my $table = Table->new({ name => $tdata->{table_name}, schema => $schema });
acfd88b5	52	$schema->add_table($table);
	53	$table->comments( join "\n", @{$tdata->{comments}} ) if $tdata->{comments};
	54
	55	my @fields = sort { $tdata->{'fields'}->{$a}->{'order'} <=> $tdata->{'fields'}->{$b}->{'order'} } keys %{ $tdata->{'fields'} };
	56
	57	for my $fname ( @fields ) {
	58	my $fdata = $tdata->{fields}{ $fname };
	59	my $field = Column->new({
	60	name => $fdata->{name},
	61	data_type => $fdata->{data_type},
	62	sql_data_type => $self->data_type_mapping->{$fdata->{data_type}} \|\| -999999,
	63	size => $fdata->{size},
	64	default_value => $fdata->{default},
	65	is_auto_increment => $fdata->{is_auto_inc},
	66	is_nullable => $fdata->{null},
	67	is_primary_key => $fdata->{is_primary_key} ? 1 : 0,
02989ef4	68	table => $table,
acfd88b5	69	});
869bbdf2	70	$field->comments($fdata->{comments});
acfd88b5	71	$table->add_column($field);
acfd88b5	72
49eb21a3	73	$field->is_primary_key(1) if $fdata->{is_primary_key};
acfd88b5	74
	75	my %extra;
	76	for my $qual ( qw[ binary unsigned zerofill list collate ],
	77	'character set', 'on update' ) {
	78	if ( my $val = $fdata->{ $qual } \|\| $fdata->{ uc $qual } ) {
	79	next if ref $val eq 'ARRAY' && !@$val;
	80	$extra{$qual} = $val;
	81	#$field->extra( $qual, $val );
	82	}
	83	}
	84	$field->extra(\%extra);
	85
	86	if ( $fdata->{has_index} ) {
ec2351fe	87	my $index = Index->new({ name => '', type => 'NORMAL', table => $table });
acfd88b5	88	$index->add_column($table->get_column($fdata->{name}));
	89	$table->add_index($index);
	90	}
	91
	92	if ( $fdata->{is_unique} ) {
	93	push @{ $tdata->{constraints} }, { name => '', type => 'UNIQUE', fields => [ $fdata->{name} ] };
acfd88b5	94	}
	95
	96	for my $cdata ( @{ $fdata->{constraints} } ) {
	97	next unless lc $cdata->{type} eq 'foreign_key';
	98	$cdata->{fields} \|\|= [ $field->name ];
	99	push @{ $tdata->{constraints} }, $cdata;
	100	}
	101	}
	102
	103	for my $idata ( @{ $tdata->{indices} \|\| [] } ) {
02989ef4	104	my $index = Index->new({ name => $idata->{name} \|\| '', type => uc($idata->{type}), table => $table });
acfd88b5	105	map { $index->add_column($table->get_column($_)) } @{$idata->{fields}};
	106	$table->add_index($index);
	107	}
	108
	109
af822e89	110	if ( my @options = @{ $tdata->{'table_options'} \|\| [] } ) {
	111	my @cleaned_options;
	112	my @ignore_opts = $translator->has_parser_args && $translator->parser_args->{'ignore_opts'}
	113	? split( /,/, $translator->parser_args->{'ignore_opts'} )
	114	: ();
	115	if (@ignore_opts) {
	116	my $ignores = { map { $_ => 1 } @ignore_opts };
	117	foreach my $option (@options) {
	118	# make sure the option isn't in ignore list
	119	my ($option_key) = keys %$option;
	120	if ( !exists $ignores->{$option_key} ) {
	121	push @cleaned_options, $option;
	122	}
	123	}
	124	} else {
	125	@cleaned_options = @options;
	126	}
	127	$table->options( \@cleaned_options ); # or die $table->error;
	128	}
d69ea17e	129	}
	130
	131	for my $proc_name ( keys %{ $result->{procedures} } ) {
	132	my $procedure = Procedure->new({ name => $proc_name,
	133	owner => $result->{procedures}->{$proc_name}->{owner},
	134	sql => $result->{procedures}->{$proc_name}->{sql}
	135	});
	136	$schema->add_procedure($procedure);
	137	}
acfd88b5	138
d69ea17e	139	for my $view_name ( keys %{ $result->{'views'} } ) {
	140	my $view = View->new({
	141	name => $view_name,
	142	sql => $result->{'views'}->{$view_name}->{sql},
	143	});
	144	$schema->add_view($view);
	145	}
	146	for my $table_name (@tables) {
	147	my $table = $schema->get_table($table_name);
	148	my $tdata = $result->{tables}{ $table_name };
acfd88b5	149	for my $cdata ( @{ $tdata->{constraints} \|\| [] } ) {
	150	my $constraint;
	151	if (uc $cdata->{type} eq 'PRIMARY_KEY') {
acf110dc	152	$constraint = PrimaryKey->new({ name => $cdata->{name} \|\| '', table => $table });
acf110dc	153	$table->get_column($_)->is_primary_key(1) for @{$cdata->{fields}};
acfd88b5	154	} elsif (uc $cdata->{type} eq 'FOREIGN_KEY') {
acf110dc	155	$constraint = ForeignKey->new({ name => $cdata->{name} \|\| '',
02989ef4	156	table => $table,
acfd88b5	157	reference_table => $cdata->{reference_table},
2913dc45	158	defined $cdata->{reference_fields} ? (reference_columns => $cdata->{reference_fields}) : (),
acfd88b5	159	on_delete => $cdata->{on_delete} \|\| $cdata->{on_delete_do},
	160	on_update => $cdata->{on_update} \|\| $cdata->{on_update_do} });
	161	$table->get_column($_)->is_foreign_key(1) for @{$cdata->{fields}};
	162	$table->get_column($_)->foreign_key_reference($constraint) for @{$cdata->{fields}};
	163	} else {
acf110dc	164	$constraint = Constraint->new({ name => $cdata->{name} \|\| '', type => uc $cdata->{type}, table => $table });
acfd88b5	165	}
acf110dc	166	$constraint->add_column($table->get_column($_)) for @{$cdata->{fields}};
acfd88b5	167	$table->add_constraint($constraint);
acfd88b5	168	}
acfd88b5	169	}
84f95c6e	170	return 1;
acfd88b5	171	}
4f4fd192	172	}