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