migsql 1.0.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6dce65cec7f13004fde63a68b7b963d0378ef6ce
+  data.tar.gz: 314daeaf69aaeafc08238b5253efa6213ee0aba3
+SHA512:
+  metadata.gz: 0470c77ea747a8aa646afec4ff314ecdc2c9f859bccc1c89821276a8101c65f73993dfb0ba1ebfc033f98abef24848b057319a0d02713243d9d9bae77b0e91d3
+  data.tar.gz: 5e441b168abd9ad87ea51124dfe3bd51fb5a6a573acee0c2a866b127507c264e7405219f35ea5e974e63196b314ef4044f4ce7b11572381208ef1aef436ed0e8

data/bin/migsql

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative '../lib/migsql'
+@app = MigSql.new 
+@app.handle_argv(ARGV)

data/lib/migsql/app.rb

@@ -0,0 +1,84 @@
+require 'colorize'
+class MigSql
+
+  def initialize
+    @migration = Migration.new './db/config.yml'
+    @migration.load
+  end
+
+  def handle_argv(argv)
+    if argv.length == 0
+      puts 'Usage: migsql {init|create-migration|migrate}'.white
+    else 
+      case argv[0]
+        when 'init'
+          handle_init
+        when 'create-migration'
+          handle_create_migration argv
+        when 'migrate'
+          handle_migrate argv
+      end  
+    end
+  end
+
+  def handle_init
+    if File.directory?('./db')
+      puts 'Error: the ./db directory already exists'.red
+    else
+      @migration.create_server 'example_database', '127.0.0.1', 'dbname', 'username', 'password'
+      @migration.save
+      puts 'Default configuration created in ./db/config.yml'.green
+    end
+  end 
+
+  def handle_create_migration(argv)
+    migration_name = argv[1]
+    migration_server = get_migration_server(argv[2])
+    if !migration_server.nil?
+      @migration.create_migration migration_server, migration_name
+    end
+  end
+
+  def handle_migrate(argv)
+    if argv[1] == 'to' 
+      migration_server = @migration.get_first_server_name
+      migration_target = argv[2]
+    else
+      migration_server = get_migration_server(argv[1])
+      migration_target = argv[3]
+    end
+    o_target = migration_target
+    if !migration_server.nil? 
+      if migration_target.nil? 
+        migration_target = @migration.get_latest_migration(migration_server)
+      else
+        migration_target = @migration.get_migration_by_name(migration_server, migration_target)
+      end
+      if !migration_target.nil? 
+        from = @migration.get_migration_status(migration_server)
+        plan = @migration.get_migration_plan(migration_server, migration_target, from) 
+        if !plan.nil?
+          @migration.apply_migration_plan(migration_server, plan, migration_target)
+        end
+      else
+        puts "Error:  No migration found with name: #{o_target}".red
+      end
+    end
+  end
+
+  def get_migration_server(server_name)
+    if @migration.count_servers == 0
+      puts 'Error: Please run migsql init first'.red
+    elsif @migration.count_servers > 1 && server_name.nil?
+      puts 'Error: Your config has multiple servers, please specify which server to create the migration on'.red
+    elsif !server_name.nil? && @migration.get_server(server_name).nil?
+      puts "Error: No server named #{server_name} found in your config".red
+      nil
+    elsif @migration.count_servers == 1
+      @migration.get_first_server_name
+    else
+      server_name
+    end
+  end
+
+end

data/lib/migsql/migration.rb

@@ -0,0 +1,140 @@
+require 'rake'
+require 'fileutils'
+require 'yaml'
+require 'tiny_tds'
+require 'colorize'
+require_relative 'sqlserver'
+
+class Migration
+  def initialize(path) 
+    @path = path
+    @root = File.dirname(@path)
+    @servers = Hash.new
+  end  
+
+  def create_server(name, address, database, username, password)
+     @servers[name] = SqlServer.new name, address, database, username, password
+  end
+
+  def get_server(name)
+    @servers[name]
+  end
+
+  def get_first_server_name
+    @servers.keys.first
+  end
+   
+  def save
+    FileUtils::mkdir_p @root
+    File.open(@path, 'w') { |f| f.write @servers.to_yaml }
+  end
+
+  def load
+    if File.file?(@path) 
+      @servers = YAML.load_file(@path)
+    end
+  end
+
+  def count_servers 
+    @servers.length
+  end
+
+  def create_migration(server_name, migration_name)
+    if get_migration_by_name(server_name, migration_name).nil?
+      migration_name = "#{DateTime.now.strftime('%Q')}_#{migration_name}"
+      server_root    = "#{@root}/#{server_name}"
+      up 	     = "#{server_root}/#{migration_name}_up.sql"
+      down           = "#{server_root}/#{migration_name}_down.sql"
+      FileUtils::mkdir_p server_root
+      FileUtils::touch up
+      FileUtils::touch down 
+
+      puts "Migration '#{migration_name}' created.".green
+      puts "Up:   #{up}".white
+      puts "Down: #{down}".white
+      return true
+    else
+      puts 'Error: migration name already in use'.red
+      return false
+    end
+  end
+
+  def get_latest_migration(server_name)
+    server_root = "#{@root}/#{server_name}"
+    latest_mig  = File.basename(Dir["#{server_root}/*_up.sql"].sort.reverse[0])
+    /([0-9]+_.*)(_up|_down)\.sql/.match(latest_mig).captures[0]
+  end
+
+  def get_migration_plan(server_name, to, from)
+    to = to || get_latest_migration(server_name)
+    to_i  = /([0-9]+)_?.*/.match(to).captures[0]
+  
+    from_i = /([0-9]+)_?.*/.match(from).captures[0]
+
+    if to_i > from_i 
+      plan = get_up_plan server_name, to_i, from_i
+    elsif to_i < from_i
+      plan = get_down_plan server_name, to_i, from_i
+    else
+      puts 'No migration needed, database already at current level'.green
+    end
+    return plan
+  end
+
+  def get_up_plan(server_name, to, from)
+    plan = Array.new
+    server_root = "#{@root}/#{server_name}"
+    Dir["#{server_root}/*_up.sql"].sort.each {|migration| 
+      current_item = /([0-9]+)_?.*/.match(migration).captures[0]
+      if current_item > from && current_item <= to
+        plan.push(migration)
+      end
+    }
+    plan
+  end
+
+  def get_down_plan(server_name, to, from)
+    plan = Array.new
+    server_root = "#{@root}/#{server_name}"
+    Dir["#{server_root}/*_down.sql"].sort.reverse.each {|migration| 
+      current_item = /([0-9]+)_?.*/.match(migration).captures[0]
+      if current_item > to && current_item <= from
+        plan.push(migration)
+      end
+    }
+    plan
+  end
+
+  def get_migration_by_name(server_name, name)
+    server_root = "#{@root}/#{server_name}"  
+    migration = Dir["#{server_root}/*#{name}_up.sql"][0]
+    if !migration.nil?
+      migration = /([0-9]+_.*)(_up|_down)\.sql/.match(migration).captures[0]
+    end
+    return migration
+  end
+
+  def apply_migration_plan(server_name, migration_plan, final_state)
+    puts "Applying migration to: #{server_name}".yellow
+    migration_plan.each {|migration| 
+      apply_migration server_name, migration
+    }
+    server = get_server(server_name)
+    server.set_migration_status(final_state)
+    puts "Migration Complete, current state: #{final_state}".green
+  end
+
+  def apply_migration(server_name, migration)
+    migration_name, updown = /([0-9]+.*)_(.*)\.sql/.match(migration).captures
+    puts " - #{migration_name} #{updown}".white
+    server = get_server(server_name)
+    server.apply_migration(migration)
+  end
+ 
+  def get_migration_status(server_name) 
+    get_server(server_name).get_migration_status
+  end
+
+end
+
+

data/lib/migsql/sqlserver.rb

@@ -0,0 +1,81 @@
+require 'rake'
+require 'fileutils'
+require 'yaml'
+require 'tiny_tds'
+require 'colorize'
+
+class SqlServer
+  def initialize(name, address, database, username, password) 
+    @name     = name
+    @address  = address
+    @database = database
+    @username = username
+    @password = password
+  end
+
+  def name 
+    return @name
+  end
+  def address
+    return @address
+  end
+  def database
+    return @database
+  end
+  def username 
+    return @username
+  end
+  def password
+    return @password
+  end
+
+  def get_sql(name)
+    File.read("#{File.dirname(__FILE__)}/../../sql/#{name}.sql")
+  end
+
+  def get_client
+    TinyTds::Client.new(
+      :username => username, 
+      :password => password, 
+      :host     => address,
+      :database => database
+    )
+  end
+
+  def get_migration_status
+    client = get_client
+    begin
+      results = client.execute("SELECT migration FROM _migration")
+      result = results.each(:first => true)[0]['migration']
+      if result.length == 0 
+        return '0'
+      else
+        return result
+      end
+    rescue Exception => ex
+      return '0'
+    end
+  end
+ 
+  def set_migration_status(to)
+    client = get_client
+    sql = [
+      get_sql('create_migration_table'),
+      "UPDATE _migration SET migration = '#{to}'"
+    ].join(' ')  
+    client.execute(sql).each
+  end
+ 
+  def remove_migration
+    client = get_client
+    client.execute('DROP TABLE _migration')
+  end
+
+  def apply_migration(path) 
+    client = get_client
+    sql    = File.read(path)
+    client.execute(sql)
+  end
+
+end
+

data/lib/migsql.rb

@@ -0,0 +1,3 @@
+require_relative 'migsql/sqlserver.rb'
+require_relative 'migsql/migration.rb'
+require_relative 'migsql/app.rb'

data/sql/create_migration_table.sql

@@ -0,0 +1,14 @@
+IF NOT EXISTS(SELECT * FROM sys.objects WHERE object_id = OBJECT_ID(N'[dbo].[_migration]') AND TYPE IN (N'U'))
+BEGIN
+  CREATE TABLE [dbo].[_migration](
+    [migration] [varchar](100) NOT NULL,
+    [date_applied] [datetime] NOT NULL
+  ) ON [PRIMARY];
+
+  INSERT INTO [dbo].[_migration](
+    migration, 
+    date_applied
+  ) VALUES(
+    '0', '1990-01-01'
+  )
+END

data/sql/create_test_table.sql

@@ -0,0 +1,5 @@
+CREATE TABLE [dbo].[MigrationTest](
+  [col1] [int] NOT NULL,
+  [col2] [int] NOT NULL
+) ON [PRIMARY];
+

data/sql/populate_test_table.sql

@@ -0,0 +1,2 @@
+INSERT INTO MigrationTest(col1, col2) 
+VALUES('12345', '67890')

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: migsql
+version: !ruby/object:Gem::Version
+  version: 1.0.4
+platform: ruby
+authors:
+- Karl Stoney
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-06-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tiny_tds
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Simple lightweight up/down MSSQL migrations
+email: karl@jambr.co.uk
+executables:
+- migsql
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/migsql/migration.rb
+- lib/migsql/app.rb
+- lib/migsql/sqlserver.rb
+- lib/migsql.rb
+- bin/migsql
+- sql/create_migration_table.sql
+- sql/populate_test_table.sql
+- sql/create_test_table.sql
+homepage: https://github.com/Stono/migsql
+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.1.11
+signing_key: 
+specification_version: 4
+summary: Simple lightweight up/down MSSQL migrations
+test_files: []