Commit | Line | Data |
4f4fd192 |
1 | use MooseX::Declare; |
2 | class SQL::Translator { |
e525194f |
3 | use Try::Tiny; |
631efe08 |
4 | use MooseX::Types::Moose qw(Bool HashRef Int Str Undef); |
c6b7881f |
5 | use MooseX::Aliases; |
ee1a2ac9 |
6 | use SQL::Translator::Types qw(DBIHandle Parser Producer Schema); |
7 | use SQL::Translator::Object::Schema; |
8 | |
2d6021dc |
9 | our $VERSION = '0.001'; |
7aecfb4c |
10 | $VERSION = eval $VERSION; |
2d6021dc |
11 | |
4f4fd192 |
12 | has 'parser' => ( |
13 | isa => Str, |
ee1a2ac9 |
14 | is => 'rw', |
c6b7881f |
15 | alias => 'from', |
4f4fd192 |
16 | ); |
17 | |
18 | has 'producer' => ( |
19 | isa => Str, |
ee1a2ac9 |
20 | is => 'rw', |
c6b7881f |
21 | alias => 'to', |
4f4fd192 |
22 | ); |
23 | |
24 | has '_parser' => ( |
25 | isa => Parser, |
26 | is => 'rw', |
27 | lazy_build => 1, |
28 | handles => [ qw(parse) ], |
c6b7881f |
29 | predicate => 'has_parser', |
4f4fd192 |
30 | ); |
31 | |
32 | has '_producer' => ( |
33 | isa => Producer, |
34 | is => 'rw', |
35 | lazy_build => 1, |
36 | handles => [ qw(produce) ], |
c6b7881f |
37 | predicate => 'has_producer', |
4f4fd192 |
38 | ); |
39 | |
40 | has 'dbh' => ( |
41 | isa => DBIHandle, |
42 | is => 'ro', |
43 | predicate => 'has_dbh', |
44 | ); |
ee1a2ac9 |
45 | |
46 | has 'schema' => ( |
47 | isa => Schema, |
48 | is => 'rw', |
5f60616d |
49 | lazy => 1, |
c062b437 |
50 | clearer => '_clear_schema', |
5f60616d |
51 | default => sub { SQL::Translator::Object::Schema->new }, |
ee1a2ac9 |
52 | ); |
53 | |
54 | has 'parser_args' => ( |
55 | isa => HashRef, |
56 | is => 'rw', |
23043f8d |
57 | predicate => 'has_parser_args', |
ee1a2ac9 |
58 | ); |
59 | |
60 | has 'producer_args' => ( |
61 | isa => HashRef, |
62 | is => 'rw', |
23043f8d |
63 | predicate => 'has_producer_args', |
4f4fd192 |
64 | ); |
4e32b65d |
65 | |
66 | has 'data' => ( |
67 | isa => Str, |
68 | is => 'rw', |
69 | ); |
4f4fd192 |
70 | |
7aecfb4c |
71 | has 'version' => (isa => Str, is => 'ro', default => $VERSION); |
ee1a2ac9 |
72 | has 'add_drop_table' => (isa => Bool, is => 'rw', default => 0); |
73 | has 'no_comments' => (isa => Bool, is => 'rw', default => 0); |
bdf688cd |
74 | has 'show_warnings' => (isa => Bool, is => 'rw', default => 0); |
ee1a2ac9 |
75 | has 'trace' => (isa => Bool, is => 'rw', default => 0); |
6127f0bd |
76 | has 'quote_table_names' => (isa => Bool, is => 'rw', default => 0); |
77 | has 'quote_field_names' => (isa => Bool, is => 'rw', default => 0); |
ee1a2ac9 |
78 | has 'filename' => (isa => Str, is => 'rw'); |
79 | |
2d6021dc |
80 | has '_producer_mapping' => ( |
81 | isa => HashRef, |
82 | is => 'ro', |
83 | default => sub { { MySQL => 'SQL::MySQL', SQLite => 'SQL::SQLite', PostgreSQL => 'SQL::PostgreSQL', XML => 'XML', YAML => 'YAML' } } |
84 | ); |
85 | |
2c64974b |
86 | has 'error' => ( |
87 | is => 'rw', |
88 | isa => Str, |
89 | ); |
90 | |
4f4fd192 |
91 | method _build__parser { |
92 | my $class = 'SQL::Translator::Parser'; |
93 | |
94 | Class::MOP::load_class($class); |
95 | |
15adaf18 |
96 | my $parser; |
97 | if ($self->has_dbh) { |
ee1a2ac9 |
98 | $parser = $class->new({ translator => $self, dbh => $self->dbh }); |
15adaf18 |
99 | } else { |
ee1a2ac9 |
100 | $parser = $class->new({ translator => $self, type => $self->parser || '' }); |
15adaf18 |
101 | } |
4f4fd192 |
102 | |
103 | return $parser; |
104 | } |
105 | |
106 | method _build__producer { |
2d6021dc |
107 | my $mapping = $self->_producer_mapping; |
108 | |
4f4fd192 |
109 | my $class = 'SQL::Translator::Producer'; |
2d6021dc |
110 | my $role = $class . '::' . $mapping->{$self->producer}; |
d1684085 |
111 | |
4f4fd192 |
112 | Class::MOP::load_class($class); |
2d6021dc |
113 | Class::MOP::load_class($role); |
4f4fd192 |
114 | |
ee1a2ac9 |
115 | my $producer = $class->new({ translator => $self }); |
4f4fd192 |
116 | $role->meta->apply($producer); |
117 | |
118 | return $producer; |
c16f2fa9 |
119 | } |
ee1a2ac9 |
120 | |
121 | method translate(:$data, :$producer?, :$producer_args?, :$parser?, :$parser_args?) { |
631efe08 |
122 | my $return; |
123 | |
c6b7881f |
124 | $self->_clear_schema if defined $parser; |
4e32b65d |
125 | $data ||= $self->data; |
c6b7881f |
126 | |
127 | $parser ||= $self->parser unless $self->has_parser; |
631efe08 |
128 | if (defined $parser) { |
ee1a2ac9 |
129 | $self->_clear_parser; |
130 | $self->parser($parser); |
d1684085 |
131 | $self->parse($data); |
631efe08 |
132 | $return = $self->schema; |
133 | } |
134 | |
c6b7881f |
135 | $producer ||= $self->producer unless $self->has_producer; |
631efe08 |
136 | if (defined $producer) { |
ee1a2ac9 |
137 | $self->_clear_producer; |
ee1a2ac9 |
138 | $self->producer($producer); |
631efe08 |
139 | $return = $self->produce; |
ee1a2ac9 |
140 | } |
631efe08 |
141 | |
142 | return $return; |
ee1a2ac9 |
143 | } |
144 | |
145 | method parser_type { return $self->parser } |
146 | method producer_type { return $self->producer } |
ffe3a4b1 |
147 | |
631efe08 |
148 | method engine_version(Int|Str|Undef $v, Str $target = 'perl') { |
149 | return undef unless $v; |
150 | |
ffe3a4b1 |
151 | my @vers; |
152 | |
153 | # X.Y.Z style |
154 | if ( $v =~ / ^ (\d+) \. (\d{1,3}) (?: \. (\d{1,3}) )? $ /x ) { |
155 | push @vers, $1, $2, $3; |
156 | } |
157 | |
158 | # XYYZZ (mysql) style |
159 | elsif ( $v =~ / ^ (\d) (\d{2}) (\d{2}) $ /x ) { |
160 | push @vers, $1, $2, $3; |
161 | } |
162 | |
163 | # XX.YYYZZZ (perl) style or simply X |
164 | elsif ( $v =~ / ^ (\d+) (?: \. (\d{3}) (\d{3}) )? $ /x ) { |
165 | push @vers, $1, $2, $3; |
166 | } |
167 | else { |
168 | #how do I croak sanely here? |
169 | die "Unparseable MySQL version '$v'"; |
170 | } |
171 | |
172 | if ($target eq 'perl') { |
173 | return sprintf ('%d.%03d%03d', map { $_ || 0 } (@vers) ); |
174 | } |
175 | elsif ($target eq 'mysql') { |
176 | return sprintf ('%d%02d%02d', map { $_ || 0 } (@vers) ); |
177 | } |
178 | else { |
179 | #how do I croak sanely here? |
180 | die "Unknown version target '$target'"; |
181 | } |
182 | } |
4f4fd192 |
183 | } |