pg-migrator 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*.swp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'http://rubygems.org'
+
+# Specify your gem's dependencies in pg-migrator.gemspec
+gemspec

data/README.md

@@ -0,0 +1,43 @@
+PG Migrator
+===========
+
+A simple migration, which runs pure Postgres SQL files. 
+
+Drop ```<timestamp>_desc.sql``` files in ```db/migrations``` (and corresponding ```<timestamp>_desc_down.sql``` files if you want to be able to downgrade). 
+
+Gemfile:
+
+    require 'pg-migrator/tasks'
+    # Make sure to have a DATABASE_URL environment variable
+    ENV['DATABASE_URL'] = "postgres://user:passwd@host/db"
+    # or if using sockets and default user
+    ENV['DATABASE_URL'] = "postgres:db"
+
+Then as usually:
+
+    rake db:reset # drops the "public" schema in your database (not the database it self or any other schemas)
+    rake db:migrate:up # runs all migrations in order
+    rake db:migrate:down # runs the down migration for the latest applied migration
+
+License
+=======
+
+Copyright (C) 2011 by Carl Hörberg
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,7 @@
+#!/usr/bin/env rake
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+
+Rake::TestTask.new(:spec) do |t|
+    t.pattern = "spec/*_spec.rb"
+end

data/lib/pg-migrator/version.rb

@@ -0,0 +1,5 @@
+module PG
+  class Migrator
+    VERSION = "0.0.1"
+  end
+end

data/lib/pg-migrator.rb

@@ -0,0 +1,61 @@
+#encoding: utf-8
+require_relative 'pg-migrator/version'
+require 'pg'
+
+module PG
+  class Migrator
+    def initialize(database_url, migrations_dir = './db/migrations')
+      @pg = PGconn.connect database_url
+      @pg.exec 'SET client_min_messages = warning'
+      @migrations_dir = migrations_dir
+    end
+
+    def close
+      @pg.finish
+    end
+
+    def reset
+      @pg.transaction do |conn|
+        conn.exec 'DROP SCHEMA public CASCADE'
+        conn.exec 'CREATE SCHEMA public'
+      end
+      migrate_up
+    end
+
+    def migrate_up
+      @pg.transaction do |conn|
+        conn.exec "CREATE TABLE IF NOT EXISTS migration (
+          id bigint PRIMARY KEY,
+          name varchar(255),
+          applied timestamp DEFAULT current_timestamp)
+        "
+        applied = conn.exec 'SELECT id, name FROM migration ORDER BY id'
+        Dir["#{@migrations_dir}/*.sql"].sort.each do |m|
+          next if m.end_with? '_down.sql'
+          id = m.sub /.*\/(\d+)_.*/, '\1'
+          name = m.sub /.*\/\d+_([^\.]*).sql$/, '\1'
+          next if applied.any? { |a| a['id'] == id }
+          puts "Applying #{File.basename m}"
+          conn.exec File.read m
+          conn.exec 'INSERT INTO migration (id, name) VALUES ($1, $2)', [id, name]
+        end
+      end
+    end
+
+    def migrate_down
+      @pg.transaction do |conn|
+        applied = conn.exec 'SELECT id, name FROM migration ORDER BY applied DESC LIMIT 1'
+        id = applied.first['id']
+        name = applied.first['name']
+        down_sql_file = "#{@migrations_dir}/#{id}_#{name}_down.sql"
+        unless File.exists? down_sql_file
+          puts "No down migration found for #{id} #{name} at #{down_sql_file}"
+          return
+        end
+        puts "Applying #{File.basename down_sql_file}"
+        conn.exec File.read down_sql_file
+        conn.exec "DELETE FROM migration WHERE id = $1", [id]
+      end
+    end
+  end
+end

data/lib/tasks/pg-migrator.rake

@@ -0,0 +1,34 @@
+require_relative '../pg-migrator'
+require 'uri'
+
+namespace :db do
+  task :setup do
+    uri = URI.parse ENV['DATABASE_URL']
+    if uri.opaque
+      @migrator = PG::Migrator.new {:dbname => uri.opaque}
+    else
+      @migrator = PG::Migrator.new(uri.host, uri.port, nil, nil, uri.path.sub(/\//, ''), uri.user, uri.password)
+    end
+  end
+
+  desc 'Drops the "public" schema and runs all the migrations'
+  task :reset => :setup do
+    @migrator.reset 
+  end
+
+  desc 'Migrate up'
+  task :migrate => 'migrate:up'
+
+  namespace :migrate do
+    desc 'Migrate up'
+    task :up => :setup do
+      @migrator.migrate_up
+    end
+
+    desc 'Migrate down the latest applied migration'
+    task :down => :setup do
+      @migrator.migrate_down
+    end
+  end
+end
+

data/pg-migrator.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/pg-migrator/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Carl Hörberg"]
+  gem.email         = ["carl.hoerberg@gmail.com"]
+  gem.description   = %q{Simple PG Migrator}
+  gem.summary       = %q{Drop raw sql files in db/migrations and add require 'pg-migrator/tasks' in your Rakefile}
+  gem.homepage      = ""
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  gem.name          = "pg-migrator"
+  gem.require_paths = ["lib"]
+  gem.version       = PG::Migrator::VERSION
+
+  gem.add_dependency 'pg'
+  gem.add_development_dependency 'minitest'
+
+end

data/spec/migrator_spec.rb

@@ -0,0 +1,80 @@
+require 'minitest/autorun'
+require './lib/pg-migrator'
+
+describe PG::Migrator do
+  before do
+    @pg = PGconn.connect :dbname => 'pg-migrator'
+    @migrator = PG::Migrator.new({:dbname => 'pg-migrator'}, '/tmp')
+  end
+
+  describe 'reset' do
+    it 'deletes previous tables' do
+      @pg.exec 'create table bar (id int)'
+      @migrator.reset
+      tables = @pg.exec "select tablename from pg_tables where schemaname = 'public'"
+      refute_includes tables.values.flatten, 'bar'
+    end
+  end
+
+  describe 'migrate' do
+    before do
+      @migrator.reset
+      up = "create table foo (id int)" 
+      down = "drop table foo" 
+      File.open('/tmp/201109102230_init.sql', 'w') {|f| f.write up }
+      File.open('/tmp/201109102230_init_down.sql', 'w') {|f| f.write down }
+    end
+
+    after do
+      File.delete '/tmp/201109102230_init.sql'
+      File.delete '/tmp/201109102230_init_down.sql'
+    end
+
+    describe 'up' do
+      it 'adds migration table' do
+        @migrator.migrate_up
+        tables = @pg.exec "select tablename from pg_tables where schemaname = 'public'"
+        assert_includes tables.values.flatten, 'migration'
+      end
+
+      it 'execute sql files' do
+        @migrator.migrate_up
+        tables = @pg.exec "select tablename from pg_tables where schemaname = 'public'"
+        assert_includes tables.values.flatten, 'foo'
+      end
+
+      it 'executes sql files in order' do
+      end
+
+      it 'adds records to the migration table' do
+        @migrator.migrate_up
+        migrations = @pg.exec "select * from migration"
+        assert_equal 1, migrations.count
+        assert_equal 201109102230, migrations.first['id'].to_i
+        assert_equal 'init', migrations.first['name']
+      end
+    end
+
+    describe 'down' do 
+
+      it 'executes sql-down files' do
+        @migrator.migrate_up
+        @migrator.migrate_down
+        tables = @pg.exec "select tablename from pg_tables where schemaname = 'public'"
+        refute_includes tables.values.flatten, 'foo'
+      end
+
+      it 'removes records from the migration table' do
+        @migrator.migrate_up
+        @migrator.migrate_down
+        migrations = @pg.exec "select * from migration"
+        assert_equal 0, migrations.count
+      end
+    end
+
+    after do
+      @pg.finish
+      @migrator.close
+    end
+  end
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: pg-migrator
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Carl Hörberg
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-10-24 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: &70158261673600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70158261673600
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: &70158261673180 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70158261673180
+description: Simple PG Migrator
+email:
+- carl.hoerberg@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- lib/pg-migrator.rb
+- lib/pg-migrator/version.rb
+- lib/tasks/pg-migrator.rake
+- pg-migrator.gemspec
+- spec/migrator_spec.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Drop raw sql files in db/migrations and add require 'pg-migrator/tasks' in
+  your Rakefile
+test_files:
+- spec/migrator_spec.rb