lib/SQL/Translator/Parser/DDL/YAML.pm

   1 use MooseX::Declare;
   2 role SQL::Translator::Parser::DDL::YAML {
   3     use MooseX::Types::Moose qw(Str);
   4     use SQL::Translator::Types qw(Schema);
   5     use aliased 'SQL::Translator::Object::Column';
   6     use aliased 'SQL::Translator::Object::Constraint';
   7     use aliased 'SQL::Translator::Object::Index';
   8     use aliased 'SQL::Translator::Object::Procedure';
   9     use aliased 'SQL::Translator::Object::Table';
  10     use aliased 'SQL::Translator::Object::Trigger';
  11     use aliased 'SQL::Translator::Object::View';
  12     use YAML qw(Load);
  13     use MooseX::MultiMethods;
  14
  15     multi method parse(Schema $data) { $data }
  16
  17     multi method parse(Str $data) {
  18         $data = Load($data);
  19         $data = $data->{schema};
  20
  21         my $schema = $self->schema;
  22         my $translator = $self->translator;
  23
  24         #
  25         # Tables
  26         #
  27         my @tables =
  28             map   { $data->{'tables'}{ $_->[1] } }
  29             map   { [ $data->{'tables'}{ $_ }{'order'} || 0, $_ ] }
  30             keys %{ $data->{'tables'} };
  31
  32         for my $tdata ( @tables ) {
  33             my $table = Table->new({ map { $tdata->{$_} ? ($_ => $tdata->{$_}) : () } qw/name extra options/ });
  34             $schema->add_table($table);
  35
  36             my @fields =
  37                 map   { $tdata->{'fields'}{ $_->[1] } }
  38                 map   { [ $tdata->{'fields'}{ $_ }{'order'}, $_ ] }
  39                 keys %{ $tdata->{'fields'} };
  40
  41             for my $fdata ( @fields ) {
  42                 $fdata->{sql_data_type} = $self->data_type_mapping->{$fdata->{data_type}} || -99999;
  43                 $fdata->{table} = $table;
  44
  45                 my $column = Column->new($fdata);
  46                 $table->add_column($column);
  47                 $table->primary_key($column->name) if $fdata->{is_primary_key};
  48             }
  49
  50             for my $idata ( @{ $tdata->{'indices'} || [] } ) {
  51                  $idata->{table} = $table;
  52                  my $index = Index->new($idata);
  53                  $table->add_index($index);
  54             }
  55
  56             for my $cdata ( @{ $tdata->{'constraints'} || [] } ) {
  57                  $cdata->{table} = $table;
  58                  my $constraint = Constraint->new($cdata);
  59                  $table->add_constraint($constraint);
  60             }
  61         }
  62
  63         #
  64         # Views
  65         #
  66         my @views =
  67             map   { $data->{'views'}{ $_->[1] } }
  68             map   { [ $data->{'views'}{ $_ }{'order'}, $_ ] }
  69             keys %{ $data->{'views'} };
  70
  71         for my $vdata ( @views ) {
  72             my $view = View->new($vdata);
  73             $schema->add_view($view);
  74         }
  75
  76         #
  77         # Triggers
  78         #
  79         my @triggers =
  80             map   { $data->{'triggers'}{ $_->[1] } }
  81             map   { [ $data->{'triggers'}{ $_ }{'order'}, $_ ] }
  82             keys %{ $data->{'triggers'} };
  83
  84         for my $tdata ( @triggers ) {
  85             my @columns = delete $tdata->{fields} || ();
  86             my $trigger = Trigger->new($tdata);
  87             $trigger->add_column($schema->get_table($tdata->{on_table})->get_column($_)) for @columns;
  88             $schema->add_trigger($trigger);
  89         }
  90
  91         #
  92         # Procedures
  93         #
  94         my @procedures =
  95             map   { $data->{'procedures'}{ $_->[1] } }
  96             map   { [ $data->{'procedures'}{ $_ }{'order'}, $_ ] }
  97             keys %{ $data->{'procedures'} };
  98
  99         for my $tdata ( @procedures ) {
 100              my $procedure = Procedure->new($tdata);
 101              $schema->add_procedure($procedure);
 102         }
 103
 104         if ( my $tr_data = $data->{'translator'} ) {
 105             $translator->add_drop_table( $tr_data->{'add_drop_table'} );
 106             $translator->filename( $tr_data->{'filename'} );
 107             $translator->no_comments( $tr_data->{'no_comments'} );
 108             $translator->parser_args( $tr_data->{'parser_args'} );
 109             $translator->producer_args( $tr_data->{'producer_args'} );
 110             $translator->parser_type( $tr_data->{'parser_type'} );
 111             $translator->producer_type( $tr_data->{'producer_type'} );
 112             $translator->show_warnings( $tr_data->{'show_warnings'} );
 113             $translator->trace( $tr_data->{'trace'} );
 114         }
 115
 116         return $schema;
 117     }
 118 }