migrations 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 45ef69a67e679944490603eaa27ea10f1581e248
+  data.tar.gz: 4a7c63fe40651c10b01e2c4e5c978f6477b83084
+SHA512:
+  metadata.gz: dad6e96c85c347d66b3ed58f62a144acc0be6437f89962ecf7ca218bf0ac1d24d407ae5c6e9f51219b1a001c8f6601f1508d3fc6a585018445b3c57312c0ab42
+  data.tar.gz: 8e7e2b8c33af112c4fa7902a193701c67dac37fc136b6ed08b0349ba5502606e18dd3d1b19b3798e9437feca1c37fa1f194dbd16d34d284a2e26f07ab4d44779

data/bin/migrate

@@ -0,0 +1,319 @@
+#!/usr/bin/ruby
+
+require 'migrations'
+
+class MigrationCommand
+  def initialize(args)
+    if args.empty?
+      usage
+    end
+
+    @prompt = true
+
+    if respond_to? args[0]
+      method = args.shift
+
+      if args[args.length-1] == '-y'
+        @prompt = false
+        args.pop
+      end
+
+      if args.empty?
+        send method
+      else
+        send method, *args
+      end
+    else
+      usage
+    end
+  end
+
+  def usage
+    puts "\nMigrations for ruby by m4rkw - https://github.com/m4rkw/migration\n\n"
+    puts "Usage:\n\n"
+
+    format(
+      'migrate list' => 'list all migrations',
+      'migrate up [-y]' => 'run all pending migrations',
+      'migrate up <n> [-y]' => 'run <n> pending migrations',
+      'migrate down [-y]' => 'revert previous migration',
+      'migrate down <n> [-y]' => 'revert previous <n> migrations',
+      'migrate down all [-y]' => 'revert all previous migrations',
+      'migrate to <version> [-y]' => 'migrate up or down to a specific version',
+      'migrate add <name> [-y]' => 'generate a new migration class',
+      'migrate set <version || "none"> [-y]' => 'set the migrated version without running any migrations',
+      'migrate apply <version> [-y]' => 'apply a specific migration without updating the version',
+      'migrate revert <version> [-y]' => 'revert a specific migration without updating the version'
+    )
+    puts "\n"
+    exit
+  end
+
+  def format(keys)
+    width = (keys.keys.max do |a,b| a.length <=> b.length end).length + 2
+
+    keys.each do |key, value|
+      puts "#{key.ljust(width,' ')}: #{value}"
+    end
+  end
+
+  def format_table(list)
+    widths = {}
+    total = 0
+
+    list[0].keys.each do |key|
+      widths[key] = key.length
+      list.each do |item|
+        if table_value(item[key]).length > widths[key]
+          widths[key] = table_value(item[key]).length
+        end
+      end
+      total += widths[key]
+    end
+
+    table_spacer(widths)
+
+    list[0].keys.each do |key|
+      print "| #{key.to_s.rjust(widths[key],' ')} "
+    end
+
+    puts "|"
+
+    table_spacer(widths)
+
+    list.each do |item|
+      item.each do |key,value|
+        print "| #{table_value(value).rjust(widths[key],' ')} "
+      end
+      puts "|"
+    end
+
+    table_spacer(widths)
+  end
+
+  def table_spacer(widths)
+    widths.each do |key, width|
+      print "+-" + ("-" * width) + "-"
+    end
+    puts "+"
+  end
+
+  def table_value(value)
+    if !!value == value
+      return value ? 'yes' : 'no'
+    end
+    if value.is_a?(Time)
+      return value.strftime('%Y-%m-%d %H:%M:%S')
+    end
+
+    value
+  end
+
+  def list
+    m = Migrations.new
+
+    list = m.list
+
+    if list.empty?
+      puts "\nNo migrations found.\n\n"
+    else
+      puts "\n"
+      format_table(list)
+      puts "\n"
+    end
+  end
+
+  def prompt_continue
+    print "\nContinue? [y/N] "
+    input = STDIN.gets.strip
+
+    if input[0].downcase != 'y'
+      puts "Aborting.\n"
+      exit
+    end
+  end
+
+  def list_migrations(title, versions)
+    puts "\n#{title}\n\n"
+
+    versions.each do |version|
+      puts version
+    end
+  end
+
+  def up(n=nil)
+    m = Migrations.new
+
+    pending = m.pending(n)
+
+    if pending.empty?
+      puts "\nNo migrations pending.\n\n"
+    else
+      if @prompt
+        list_migrations "Migrations pending:",pending
+        prompt_continue
+      end
+
+      puts "\n"
+      m.up(n)
+      puts "\n"
+    end
+  end
+
+  def down(n=nil)
+    if n.nil?
+      n = 1
+    elsif n == 'all'
+      n = nil
+    end
+
+    m = Migrations.new
+
+    applied = m.applied(n)
+
+    if applied.empty?
+      puts "\nNo migrations applied.\n\n"
+    else
+      if @prompt
+        list_migrations "Migrations to roll back:",applied
+        prompt_continue
+      end
+
+      puts "\n"
+      m.down(n)
+      puts "\n"
+    end
+  end
+
+  def add(name)
+    name = name.gsub(/[^a-zA-Z0-9_]+/,'')
+
+    m = Migrations.new
+
+    puts "\n"
+    m.generate(name)
+    puts "\n"
+  end
+
+  def to(version)
+    m = Migrations.new
+
+    pending = m.pending
+
+    if pending.include? version
+      to_apply = []
+
+      pending.each do |pending_version|
+        to_apply.push pending_version
+
+        if pending_version == version
+          break
+        end
+      end
+
+      if to_apply.empty?
+        puts "\nNothing to do, we are already at that version.\n\n"
+        return
+      end
+
+      if @prompt
+        list_migrations "Migrations to apply:", to_apply
+        prompt_continue
+      end
+
+      puts "\n"
+      to_apply.each do |version|
+        m.apply version
+      end
+      puts "\n"
+      return
+
+    else
+      applied = m.applied
+
+      if applied.include? version
+        if applied[0] == version
+          puts "\nNothing to do, we are already at that version.\n\n"
+          return
+        end
+        to_revert = []
+        applied.each do |applied_version|
+          if applied_version == version
+            break
+          end
+          to_revert.push applied_version
+        end
+
+        if @prompt
+          list_migrations "Migrations to roll back:", to_revert
+          prompt_continue
+        end
+
+        puts "\n"
+        to_revert.each do |version|
+          m.revert version
+        end
+        puts "\n"
+        return
+      end
+    end
+
+   puts "\nVersion not found.\n\n"
+  end
+
+  def set(version)
+    m = Migrations.new
+
+    pending = m.pending
+
+    if pending.include? version
+      pending.each do |pending_version|
+        m.insert pending_version
+
+        if pending_version == version
+          break
+        end
+      end
+
+      puts "\nVersion has been set to: #{version}\n\n"
+      return
+    else
+      applied = m.applied
+
+      if applied.first == version
+        puts "\nNothing to do, we are already at version: #{version}\n\n"
+        return
+      end
+
+      if (applied.include?(version) || version == "none")
+        applied.each do |applied_version|
+          if applied_version == version
+            break
+          end
+          m.remove applied_version
+        end
+
+        if version == "none"
+          puts "\nVersion has been set to (none)\n\n"
+        else
+          puts "\nVersion has been set to: #{version}\n\n"
+        end
+        return
+      end
+    end
+
+    puts "\nVersion #{version} was not found.\n\n"
+  end
+
+  def apply(version)
+    m = Migrations.new
+    m.apply(version,false)
+  end
+
+  def revert(version)
+    m = Migrations.new
+    m.revert(version,false)
+  end
+end
+
+MigrationCommand.new(ARGV)

data/lib/migrations/migration.rb

@@ -0,0 +1,6 @@
+
+class Migration
+  def initialize(db)
+    @db = db
+  end
+end

data/lib/migrations.rb

@@ -0,0 +1,207 @@
+
+require 'fileutils'
+require 'sequel'
+
+class Migrations
+  attr_accessor :path
+  attr_accessor :verbose
+
+  def initialize(db=nil)
+    config_file = "config/config.rb"
+
+    if db
+      @db = db
+    elsif File.exist? config_file
+      config = {}
+      instance_eval(File.read(config_file)).each do |key, value|
+        config[key] = value
+      end
+
+      @db = Sequel.connect("#{config[:db][:type]}://#{config[:db][:user]}:#{config[:db][:pass]}@#{config[:db][:host]}/#{config[:db][:name]}")
+    else
+      raise "#{config_file} was not found and no database handle was passed to us."
+    end
+
+    @path = 'migrations'
+    @verbose = true
+  end
+
+  def get_migrations
+    ensure_migration_table
+
+    filesystem = []
+    database = {}
+    all = []
+
+    Dir.glob(@path + '/*.rb').map do |migration|
+      class_name = migration.match(/(M[0-9]{8}_[0-9]{6}_[a-zA-Z0-9_]+)\.rb\z/)[1]
+
+      filesystem.push class_name
+      all.push class_name
+    end
+
+    @db[:migration].each do |migration|
+      database[migration[:version]] = migration[:applied_at]
+      if !all.include? migration[:version]
+        all.push migration
+      end
+    end
+
+    [filesystem,database,all]
+  end
+
+  def list
+    filesystem, database, all = get_migrations
+
+    list = []
+
+    all.sort.each do |migration|
+      list.push({
+        :version => migration,
+        :applied => database[migration] ? true : false,
+        :applied_at => database[migration] ? database[migration] : ''
+      })
+    end
+
+    list
+  end
+
+  def pending(n=nil)
+    filesystem, database, all = get_migrations
+
+    pending = []
+
+    filesystem.sort.each do |migration|
+      if !database.keys.include?(migration) and (n.nil? or pending.length < n)
+        pending.push migration
+      end
+    end
+
+    pending
+  end
+
+  def applied(n=nil)
+    filesystem, database, all = get_migrations
+
+    applied = []
+
+    database.keys.sort.reverse.each do |migration|
+      if (n.nil? or applied.length < n)
+        applied.push migration
+      end
+    end
+
+    applied
+  end
+
+  def generate(name)
+    if !File.exist? @path
+      FileUtils.mkdir_p @path
+    end
+
+    t = Time.now
+
+    while 1
+      t = Time.at(t.to_i + 1)
+      class_name = 'M' + t.strftime('%Y%m%d_%H%M%S') + '_' + name.gsub(/[^a-zA-Z0-9_]/i, '')
+      migration = "#{@path}/#{class_name}.rb"
+
+      if !File.exist? migration
+        break
+      end
+    end
+
+    File.open(migration,'w') do |f|
+      f.write("
+require 'migrations/migration'
+
+class #{class_name} < Migration
+  def up
+  end
+
+  def down
+  end
+end
+")
+      @verbose and puts "Create new migration: #{migration}"
+
+      true
+    end
+  end
+
+  def ensure_migration_table
+    begin
+      table = @db[:migration].map(:version)
+    rescue Sequel::DatabaseError => e
+      if e.message.match /Table .*? doesn\'t exist/
+        @db.create_table :migration do
+          primary_key :id
+          String :version, :unique => true, :null => false
+          DateTime :applied_at
+        end
+      else
+        raise Sequel::DatabaseError e
+      end
+    end
+  end
+
+  def up(n=nil)
+    ensure_migration_table
+
+    pending(n).each do |version|
+      apply(version)
+    end
+  end
+
+  def down(n=nil)
+    ensure_migration_table
+
+    applied(n).each do |version|
+      revert(version)
+    end
+  end
+
+  def apply(version, update_table=true)
+    migration = @path + '/' + version + '.rb'
+
+    load migration
+
+    m = Object::const_get(version).new(@db)
+
+    @verbose and print "Applying: #{version} ... "
+
+    m.up
+
+    @verbose and puts "ok"
+
+    update_table and insert(version)
+  end
+
+  def insert(version)
+    @db[:migration].insert(:version => version, :applied_at => Time.now)
+  end
+
+  def revert(version, update_table=true)
+    migration = @path + '/' + version + '.rb'
+
+    if !File.exist? migration
+      raise "Migration file #{migration} not found."
+    end
+
+    load migration
+
+    m = Object::const_get(version).new(@db)
+
+    @verbose and print "Rolling back: #{version} ... "
+
+    m.down
+
+    @verbose and puts "ok"
+
+    update_table and remove(version)
+  end
+
+  def remove(version)
+    @db[:migration].where('version = ?',version).delete
+  end
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: migrations
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- m4rkw
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.25'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.25'
+description: Ruby database migration library
+email: m@rkw.io
+executables:
+- migrate
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/migrate
+- lib/migrations.rb
+- lib/migrations/migration.rb
+homepage: https://github.com/m4rkw/migrations
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Migrations
+test_files: []