turn data_type_mapping into an attribute

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Producer / SQL / SQLite.pm

diff --git a/lib/SQL/Translator/Producer/SQL/SQLite.pm b/lib/SQL/Translator/Producer/SQL/SQLite.pm
index b014344..f4f093a 100644 (file)
--- a/lib/SQL/Translator/Producer/SQL/SQLite.pm
+++ b/lib/SQL/Translator/Producer/SQL/SQLite.pm
@@ -1,54 +1,46 @@
-package SQL::Translator::Producer::SQL::SQLite;
-use namespace::autoclean;
-use Moose::Role;
-use SQL::Translator::Constants qw(:sqlt_types);
-
-my %data_type_mapping = (
-    SQL_LONGVARCHAR() => 'text',
-    SQL_TIMESTAMP()   => 'timestamp',
-    SQL_INTEGER()     => 'integer',
-    SQL_CHAR()        => 'character',
-    SQL_VARCHAR()     => 'varchar',
-    SQL_BIGINT()      => 'integer',
-);
-
-sub _create_table {
-    my $self = shift;
-    my $table = shift;
-
-    my $sqlite_version = 0;
-
-    my $create_table;
-    my (@create, @column_defs, @index_defs, @constraint_defs);
-
-    $create_table .= 'DROP TABLE '   . $table->name . ";\n" if $self->drop_table;
-    $create_table .= 'CREATE TABLE ' . $table->name . " (\n";
-
-    push @column_defs, $self->_create_column($_) for values %{$table->columns};
-    $create_table .= join(",\n", map { '  ' . $_ } @column_defs ) . "\n)";
-
-    print $create_table . ";\n";
-    return (@create, $create_table, @index_defs, @constraint_defs );
-}
-
-sub _create_column {
-    my $self = shift;
-    my $column = shift;
-
-    my $size = $column->data_type == SQL_TIMESTAMP() ? undef : $column->size;
-    my $default_value = $column->default_value;
-    $default_value =~ s/^now[()]*/CURRENT_TIMESTAMP/i if $default_value;
-
-    my $column_def;
-    $column_def  = $column->name . ' ';
-    $column_def .= defined $data_type_mapping{$column->data_type}
-                   ? $data_type_mapping{$column->data_type}
-                   : $column->data_type;
-    #$column_def .= '(' . $column->size . ')' if $size;
-    $column_def .= ' NOT NULL' unless $column->is_nullable;
-    $column_def .= ' PRIMARY KEY' if $column->is_auto_increment;
-    $column_def .= ' DEFAULT ' . $default_value if $column->default_value && !$column->is_auto_increment;
-    $column_def;
+use MooseX::Declare;
+role SQL::Translator::Producer::SQL::SQLite {
+    use SQL::Translator::Constants qw(:sqlt_types);
+    use SQL::Translator::Types qw(Column Table);
+
+    around _build_data_type_mapping {
+        my $data_type_mapping = $self->$orig;
+        $data_type_mapping->{SQL_FLOAT()} = 'real';
+        $data_type_mapping->{SQL_BIGINT()} = 'integer';
+
+        return $data_type_mapping;
+    };
+    
+    method _create_table(Table $table) {
+        my $sqlite_version = 0;
+    
+        my $create_table;
+        my (@create, @column_defs, @index_defs, @constraint_defs);
+    
+        $create_table .= 'DROP TABLE '   . $table->name . ";\n" if $self->drop_table;
+        $create_table .= 'CREATE TABLE ' . $table->name . " (\n";
+    
+        push @column_defs, $self->_create_column($_) for values %{$table->columns};
+        $create_table .= join(",\n", map { '  ' . $_ } @column_defs ) . "\n)";
+    
+        print $create_table . ";\n";
+        return (@create, $create_table, @index_defs, @constraint_defs );
+    }
+    
+    method _create_column(Column $column) {
+        my $size = $column->data_type == SQL_TIMESTAMP() ? undef : $column->size;
+        my $default_value = $column->default_value;
+        $default_value =~ s/^now[()]*/CURRENT_TIMESTAMP/i if $default_value;
+    
+        my $column_def;
+        $column_def  = $column->name . ' ';
+        $column_def .= defined $self->data_type_mapping->{$column->data_type}
+                       ? $self->data_type_mapping->{$column->data_type}
+                       : $column->data_type;
+        #$column_def .= '(' . $column->size . ')' if $size;
+        $column_def .= ' NOT NULL' unless $column->is_nullable;
+        $column_def .= ' PRIMARY KEY' if $column->is_auto_increment;
+        $column_def .= ' DEFAULT ' . $default_value if $column->default_value && !$column->is_auto_increment;
+        $column_def;
+    }
 }
-
-1;