From e7d7286437974b846ec872248cffea158209ada4 Mon Sep 17 00:00:00 2001
From: Thiru Njuguna <uriht@outlook.com>
Date: Sun, 28 Oct 2018 08:14:46 +0300
Subject: [PATCH] migration setup update

---
 lib/natra/generators/app/app_generator.rb          | 14 +++++++++++++-
 lib/natra/generators/app/templates/Gemfile         |  1 -
 .../app/templates/db/migrate/add_extensions.rb     |  7 +++++++
 lib/natra/generators/model/migration.rb.erb        |  2 +-
 4 files changed, 21 insertions(+), 3 deletions(-)
 create mode 100644 lib/natra/generators/app/templates/db/migrate/add_extensions.rb

diff --git a/lib/natra/generators/app/app_generator.rb b/lib/natra/generators/app/app_generator.rb
index b685151..f741e5d 100755
--- a/lib/natra/generators/app/app_generator.rb
+++ b/lib/natra/generators/app/app_generator.rb
@@ -36,10 +36,22 @@ def create_empty_directories
 
         empty_directory File.join(@app_path, 'db/migrate')
 
-        create_file File.join(@app_path, "lib", ".gitkeep")
+        create_file File.join(@app_path, "lib", ".keep")
         template "config/environment.rb", File.join(@app_path, "config/environment.rb")
       end
+      
+      def create_seeds_file
+        create_file File.join(@app_path, "db", "seeds.rb")
+      end
 
+      def initialize_db
+          copy_file('bin/setup', File.join(@app_path, "bin/setup"))
+      end
+      
+      def uuid_setup
+          template 'db/migrate/add_extensions.rb', File.join(@app_path,"db/migrate/#{Time.now.strftime('%Y%m%d')}0000_add_extensions.rb")
+      end
+  
       def create_public_directory
         template "public/favicon.ico", File.join(@app_path, "public/favicon.ico")
       end
diff --git a/lib/natra/generators/app/templates/Gemfile b/lib/natra/generators/app/templates/Gemfile
index 7b588c6..dd3295e 100755
--- a/lib/natra/generators/app/templates/Gemfile
+++ b/lib/natra/generators/app/templates/Gemfile
@@ -6,7 +6,6 @@ gem 'sinatra-activerecord', :require => 'sinatra/activerecord'
 gem 'rake'
 gem 'rack-timeout'
 gem 'require_all'
-gem 'sqlite3'
 gem 'thin'
 gem 'shotgun'
 gem 'scout_apm'
diff --git a/lib/natra/generators/app/templates/db/migrate/add_extensions.rb b/lib/natra/generators/app/templates/db/migrate/add_extensions.rb
new file mode 100644
index 0000000..2d1b42e
--- /dev/null
+++ b/lib/natra/generators/app/templates/db/migrate/add_extensions.rb
@@ -0,0 +1,7 @@
+class AddExtensions < ActiveRecord::Migration
+    def change
+      enable_extension 'hstore'
+      enable_extension 'uuid-ossp'
+      enable_extension 'pgcrypto'
+    end
+end
\ No newline at end of file
diff --git a/lib/natra/generators/model/migration.rb.erb b/lib/natra/generators/model/migration.rb.erb
index 17668b9..e884761 100755
--- a/lib/natra/generators/model/migration.rb.erb
+++ b/lib/natra/generators/model/migration.rb.erb
@@ -1,6 +1,6 @@
 class <%= migration_class_name %> < ActiveRecord::Migration
   def change
-    create_table :<%= table_name %> do |t|
+    create_table :<%= table_name %> ,id: :uuid do |t|
 <% attributes.each do |attribute| -%>
       t.<%= attribute[:type] %> :<%= attribute[:name] %>
 <% end %>