-package SQL::Translator::Parser::DDL::MySQL;
-use Moose::Role;
+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';
-no Moose::Role;
+ around _build_data_type_mapping {
+ my $data_type_mapping = $self->$orig;
+ $data_type_mapping->{date} = SQL_DATE();
-1;
+ 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 $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);
+ }
+ }
+
+ 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);
+ }
+ return 1;
+ }
+}
projects / dbsrgits/SQL-Translator-2.0-ish.git / blobdiff