lib/SQL/Translator/Producer/YAML.pm

   1 use MooseX::Declare;
   2 role SQL::Translator::Producer::YAML {
   3     use YAML qw(Dump);
   4     use SQL::Translator::Types qw(Column Constraint Index Procedure Table Trigger View);
   5
   6     method produce {
   7         my $schema = $self->schema;
   8         my $translator = $self->translator;
   9
  10         return Dump({
  11             schema => {
  12                 tables =>     { map { ($_->name => $self->_create_table($_)) } $schema->get_tables, },
  13                 views =>      { map { ($_->name => $self->_create_view($_)) } $schema->get_views, },
  14                 triggers =>   { map { ($_->name => $self->_create_trigger($_)) } $schema->get_triggers, },
  15                 procedures => { map { ($_->name => $self->_create_procedure($_)) } $schema->get_procedures, },
  16             },
  17             translator => {
  18                 add_drop_table => $translator->add_drop_table,
  19                 filename       => $translator->filename,
  20                 no_comments    => $translator->no_comments,
  21                 parser_args    => $translator->parser_args || {},
  22                 producer_args  => $translator->producer_args || {},
  23                 parser_type    => $translator->parser_type,
  24                 producer_type  => $translator->producer_type,
  25                 show_warnings  => $translator->show_warnings,
  26                 trace          => $translator->trace,
  27                 version        => $translator->version,
  28             },
  29             keys %{$schema->extra} ? ('extra' => { $schema->extra } ) : (),
  30         });
  31     }
  32
  33     method _create_table(Table $table) {
  34         return {
  35             'name'        => $table->name,
  36             'options'     => $table->options || [],
  37             $table->comments ? ('comments'    => $table->comments ) : (),
  38             'constraints' => [ map { $self->_create_constraint($_) } $table->get_constraints ],
  39             'indices'     => [ map { $self->_create_index($_) } $table->get_indices ],
  40             'fields'      => { map { ($_->name => $self->_create_field($_)) } $table->get_fields, },
  41             'order'       => $table->order,
  42             keys %{$table->extra} ? ('extra' => { $table->extra } ) : (),
  43         };
  44     }
  45
  46     method _create_constraint(Constraint $constraint) {
  47         return {
  48             'deferrable'       => scalar $constraint->deferrable,
  49             'expression'       => scalar $constraint->expression || '',
  50             'fields'           => [ $constraint->fields ],
  51             'match_type'       => scalar $constraint->match_type,
  52             'name'             => scalar $constraint->name,
  53             'options'          => $constraint->options || [],
  54             'on_delete'        => scalar $constraint->on_delete || '',
  55             'on_update'        => scalar $constraint->on_update || '',
  56             'reference_fields' => [ map { ref $_ ? $_->name : $_ } $constraint->reference_fields ],
  57             'reference_table'  => $constraint->reference_table || '',
  58             'type'             => scalar $constraint->type,
  59             keys %{$constraint->extra} ? ('extra' => { $constraint->extra } ) : (),
  60         };
  61     }
  62
  63     method _create_field(Column $field) {
  64         return {
  65             'name'              => $field->name,
  66             'data_type'         => scalar $field->data_type,
  67             'size'              => [ $field->size ],
  68             'default_value'     => scalar $field->default_value,
  69             'is_nullable'       => $field->is_nullable,
  70             'is_primary_key'    => scalar $field->is_primary_key,
  71             'is_unique'         => $field->is_unique,
  72             'order'             => scalar $field->order,
  73             $field->is_auto_increment ? ('is_auto_increment' => 1) : (),
  74             $field->comments ? ('comments' => $field->comments) : (),
  75             keys %{$field->extra} ? ('extra' => { $field->extra } ) : (),
  76         };
  77     }
  78
  79     method _create_procedure(Procedure $procedure) {
  80         return {
  81             'name'       => scalar $procedure->name,
  82             'sql'        => scalar $procedure->sql,
  83             'parameters' => scalar $procedure->parameters,
  84             'owner'      => scalar $procedure->owner,
  85             'comments'   => scalar $procedure->comments,
  86             keys %{$procedure->extra} ? ('extra' => { $procedure->extra } ) : (),
  87         };
  88     }
  89
  90     method _create_trigger(Trigger $trigger) {
  91         return {
  92             'name'                => scalar $trigger->name,
  93             'perform_action_when' => scalar $trigger->perform_action_when,
  94             'database_events'     => [ $trigger->database_events ],
  95             'fields'              => $trigger->fields ? [ $trigger->fields ] : undef,
  96             'on_table'            => scalar $trigger->on_table,
  97             'action'              => scalar $trigger->action,
  98             'order'               => $trigger->order,
  99             keys %{$trigger->extra} ? ('extra' => { $trigger->extra } ) : (),
 100         };
 101     }
 102
 103     method _create_view(View $view) {
 104         return {
 105             'name'   => scalar $view->name,
 106             'sql'    => scalar $view->sql,
 107             'fields' => $view->fields ? [ $view->fields ] : '',
 108             'order'  => $view->order,
 109             keys %{$view->extra} ? ('extra' => { $view->extra } ) : (),
 110         };
 111     }
 112
 113     method _create_index(Index $index) {
 114         return {
 115             'name'      => scalar $index->name,
 116             'type'      => scalar $index->type,
 117             'fields'    => [ $index->fields ],
 118             'options'   => scalar $index->options,
 119             keys %{$index->extra} ? ('extra' => { $index->extra } ) : (),
 120         };
 121     }
 122 }