daiku 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5dd1097cb00c2e89d1d4ac2ef0c7d7507951d849
-  data.tar.gz: 65357c4faee3832c917bd05291e95df1c8a731dd
+  metadata.gz: ea12ee95a2500430471e9afbae6a3348d1800a33
+  data.tar.gz: 70440ea6816cf4069e2d3992c973a29f224a107e
 SHA512:
-  metadata.gz: bb8d0624c0f23c8c6ea9b17d1bba949bbefb54092d2f585f68056fbdd3953bc337f99990459abcb8dddce675e91e35fac90a10c2291763b81f3dc06db36a5fb6
-  data.tar.gz: b77e23dd95b9444e40358f34d8065405ccb9a162707efb30ae5685e968424738637e868d2d744505be98875adb88eea6c7bf40cf894d75466499f4035e4dfbcd
+  metadata.gz: 43a594f9f70de2468ab4bc093fb9b08de4f145780572b8db4f29b2a288df3d0d9454b5285a6c801004942958e2b08f7d1df1a4b4220a45ac2aa5aa50071cf8e0
+  data.tar.gz: 2810331d3b90801f397278fd8fc9b27ecee3fadab3bd352499a8a0019631bb6deedf0fb7294187744d26d8433094c6889b5adfb3058b2661b0692f358258c34c

data/lib/daiku.rb

@@ -27,6 +27,9 @@ require 'daiku/plugins/honeybadger/cli'
 require 'daiku/plugins/newrelic'
 require 'daiku/plugins/newrelic/cli'
 
+require 'daiku/plugins/sequel'
+require 'daiku/plugins/sequel/cli'
+
 require 'daiku/plugins/sidekiq'
 require 'daiku/plugins/sidekiq/cli'
 

data/lib/daiku/plugins/datamapper/cli.rb

@@ -17,7 +17,7 @@ module Daiku
           insert_into_file "#{app}/lib/boot.rb", :after => "#models\n" do
             <<-DMREQ.strip_heredoc
               require File.join($app_root, 'config/datamapper')
-              # insert require statements for models here
+              # require models here
               DataMapper.finalize
             DMREQ
           end

data/lib/daiku/plugins/sequel.rb

@@ -0,0 +1,11 @@
+# encoding: utf-8
+
+module Daiku
+  class Plugins
+    class Sequel < ::Daiku::Plugin
+      plugin_name 'sequel'
+      plugin_type 'models'
+    end
+  end
+end
+

data/lib/daiku/plugins/sequel/_templates/001_create_extension_hstore.rb.tt

@@ -0,0 +1,9 @@
+Sequel.migration do
+  up do
+    run 'CREATE EXTENSION hstore;'
+  end
+
+  down do
+    run 'DROP EXTENSION IF EXISTS hstore;'
+  end
+end

data/lib/daiku/plugins/sequel/_templates/002_create_extension_postgis.rb.tt

@@ -0,0 +1,15 @@
+Sequel.migration do
+  up do
+    run 'CREATE EXTENSION postgis;'
+    run 'CREATE EXTENSION postgis_topology;'
+    run 'CREATE EXTENSION fuzzystrmatch;'
+    run 'CREATE EXTENSION postgis_tiger_geocoder;'
+  end
+
+  down do
+    run 'DROP EXTENSION IF EXISTS postgis;'
+    run 'DROP EXTENSION IF EXISTS postgis_topology;'
+    run 'DROP EXTENSION IF EXISTS fuzzystrmatch;'
+    run 'DROP EXTENSION IF EXISTS postgis_tiger_geocoder;'
+  end
+end

data/lib/daiku/plugins/sequel/_templates/db.rake.tt

@@ -0,0 +1,14 @@
+namespace :db do
+  desc "Run Sequel migrations"
+  task :migrate => :environment, [:version] do |t, args|
+    Sequel.extension :migration
+    db = Sequel.connect(ENV.fetch("DATABASE_URL"))
+    if args[:version]
+      puts "Migrating to version #{args[:version]}"
+      Sequel::Migrator.run(db, "#{$app_root}/db/migrations", target: args[:version].to_i)
+    else
+      puts "Migrating to latest"
+      Sequel::Migrator.run(db, "#{$app_root}/db/migrations")
+    end
+  end
+end

data/lib/daiku/plugins/sequel/_templates/sequel.rb.tt

@@ -0,0 +1,27 @@
+# encoding: utf-8
+
+require 'bundler'
+Bundler.require(:sequel)
+
+# environment setup
+ENV['DATABASE_LOG_LEVEL'] ||= 'info'
+ENV['DATABASE_MAX_CONNECTIONS'] ||= 4
+ENV['DATABASE_SSLMODE'] ||= 'prefer'
+ENV['DATABASE_TIMEOUT'] ||= 10
+
+# database instance options
+sequel_opts = { 
+  loggers: [Logger.new($stdout, ENV['DATABASE_LOG_LEVEL'])],
+  max_connections: ENV['DATABASE_MAX_CONNECTIONS'].to_i,
+  sslmode: ENV['DATABASE_SSLMODE'],
+  connect_timeout: ENV['DATABASE_TIMEOUT']
+}
+
+# database connection
+DB = Sequel.connect(ENV['DATABASE_URL'], sequel_opts)
+
+# load postgres specific datatype support
+DB.extension :pg_array
+DB.extension :pg_hstore
+DB.extension :pg_json
+DB.extension :pg_streaming

data/lib/daiku/plugins/sequel/cli.rb

@@ -0,0 +1,52 @@
+# encoding: utf-8
+
+require 'thor'
+
+module Daiku
+  class Plugins
+    class Sequel
+      class CLI < ::Thor::Group
+        include ::Thor::Actions
+        argument :app
+        class_option :'sequel-postgis', type: :boolean, desc: 'Enable postgis extension via Sequel migration', default: true
+
+        def self.source_root
+          File.expand_path('../', __FILE__)
+        end
+
+        def bootrb
+          insert_into_file "#{app}/lib/boot.rb", :after => "#models\n" do
+            <<-DMREQ.strip_heredoc
+              require File.join($app_root, 'config/sequel')
+              # require models here
+            DMREQ
+          end
+        end
+
+        def config
+          template('_templates/sequel.rb.tt', "#{app}/config/sequel.rb")
+        end
+
+        def db_migrations
+          empty_directory("#{app}/db")
+          empty_directory("#{app}/db/migrations")
+          create_file("#{app}/db/migrations/.gitkeep", "")
+          template('_templates/001_create_extension_hstore.rb.tt', "#{app}/app/db/migrations/001_create_extension_hstore.rb")
+          if options[:'sequel-postgis']
+            template('_templates/002_create_extension_postgis.rb.tt', "#{app}/app/db/migrations/002_create_extension_postgis.rb")
+          end
+        end
+
+        def gemfile
+          content = <<-SGEMS.strip_heredoc
+            group :sequel do
+              gem 'sequel'
+              gem 'sequel_pg'
+            end
+          SGEMS
+          append_to_file "#{app}/Gemfile", content
+        end
+      end
+    end
+  end
+end

data/lib/daiku/version.rb

@@ -1,3 +1,3 @@
 module Daiku
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/spec/daiku/plugins_spec.rb

@@ -11,6 +11,7 @@ describe Daiku::Plugins do
       Daiku::Plugins::Grunt,
       Daiku::Plugins::Honeybadger,
       Daiku::Plugins::Newrelic,
+      Daiku::Plugins::Sequel,
       Daiku::Plugins::Sidekiq,
       Daiku::Plugins::Vcr
     ]
@@ -47,7 +48,7 @@ describe Daiku::Plugins do
   end
 
   it "can filter plugins by type" do
-    subject.filter_plugins('models', :type).must_equal [subject::Datamapper.meta]
+    subject.filter_plugins('models', :type).must_equal [subject::Datamapper.meta, subject::Sequel.meta]
   end
 
   it "returns names of all plugins" do
@@ -55,6 +56,6 @@ describe Daiku::Plugins do
   end
 
   it "returns names of plugins of type: models" do
-    subject.plugin_names('models').must_equal ['datamapper']
+    subject.plugin_names('models').must_equal ['datamapper', 'sequel']
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: daiku
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Eric Marden
@@ -133,6 +133,12 @@ files:
 - lib/daiku/plugins/honeybadger/cli.rb
 - lib/daiku/plugins/newrelic.rb
 - lib/daiku/plugins/newrelic/cli.rb
+- lib/daiku/plugins/sequel.rb
+- lib/daiku/plugins/sequel/_templates/001_create_extension_hstore.rb.tt
+- lib/daiku/plugins/sequel/_templates/002_create_extension_postgis.rb.tt
+- lib/daiku/plugins/sequel/_templates/db.rake.tt
+- lib/daiku/plugins/sequel/_templates/sequel.rb.tt
+- lib/daiku/plugins/sequel/cli.rb
 - lib/daiku/plugins/sidekiq.rb
 - lib/daiku/plugins/sidekiq/_templates/sidekiq.rb.tt
 - lib/daiku/plugins/sidekiq/cli.rb