db-migrate 0.0.1

data/lib/migrate/storage/mysql.rb

@@ -0,0 +1,77 @@
+require 'mysql2'
+
+module Migrate
+  module Storage
+    class Mysql < DB
+      def initialize(*args)
+        super
+        @conn = Mysql2::Client.new(
+          :database => @config.database,
+          :host => @config.host, 
+          :port => @config.port,
+          :username => @config.user,
+          :password => @config.password,
+        )
+      end
+
+      def create_tables
+        Log.info("Creating version table")
+        self.exec_sql <<-eos
+        CREATE TABLE #{@config.version_info}
+        (
+          version INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
+          description TEXT,
+          created_date TIMESTAMP NOT NULL,
+          last_up TIMESTAMP,
+          last_down TIMESTAMP
+        );
+        eos
+
+        self.exec_sql <<-eos
+        CREATE TABLE #{@config.version_number} (
+          version int(11) not null,
+          PRIMARY KEY (version)
+        );
+        eos
+
+        self.exec_sql <<-eos
+        INSERT INTO #{@config.version_number} VALUES(0);
+        eos
+        Log.success("Version table created")
+      end
+
+      def exec_sql(sql)
+        results = []
+        result = @tx.query sql
+        return [] if result == nil
+
+        result.each do |row|
+          results << row
+        end
+      end
+
+      def has_tx
+        @tx != nil
+      end
+
+      def tx
+        if has_tx
+          yield
+        else
+          begin
+            @conn.query "BEGIN;"
+            @tx = @conn
+            yield
+            @conn.query "COMMIT;"
+          rescue Exception => e
+            @conn.query "ROLLBACK;"
+            raise e
+          ensure
+            @tx = nil
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/migrate/storage/postgres.rb

@@ -0,0 +1,74 @@
+require "pg"
+require "pry"
+
+module Migrate
+  module Storage
+    class Postgres < DB
+      def initialize(*args)
+        super
+        @conn = PG.connect({
+          dbname: @config.database,
+          host: @config.host,
+          user: @config.user,
+          password: @config.password,
+        })
+      end
+
+      def create_tables
+        Log.info("Creating version table")
+        self.exec_sql <<-eos
+        CREATE TABLE #{@config.version_info}
+        (
+          version SERIAL PRIMARY KEY NOT NULL,
+          description TEXT,
+          created_date TIMESTAMP WITH TIME ZONE NOT NULL,
+          last_up TIMESTAMP WITH TIME ZONE,
+          last_down TIMESTAMP WITH TIME ZONE
+        );
+        CREATE UNIQUE INDEX #{@config.version_info}_version_uindex ON #{@config.version_info} (version);
+
+        CREATE TABLE #{@config.version_number}
+        (
+          version INT PRIMARY KEY NOT NULL
+        );
+
+        INSERT INTO #{@config.version_number} VALUES(0);
+        eos
+        Log.success("Version table created")
+      end
+
+
+      def extract_version(results)
+        if results && results.count > 0
+          results[0]["version"]
+        else
+          raise VersionNotFound
+        end
+      end
+
+      def exec_sql(sql)
+        @tx.exec sql
+      end
+
+      def has_tx
+        @tx != nil
+      end
+
+      def tx
+        if has_tx
+          yield
+        else
+          begin
+            @conn.transaction do |tx|
+              @tx = tx
+              yield
+            end
+          ensure
+            @tx = nil
+          end
+        end
+      end
+
+    end
+  end
+end

data/migrate.gemspec

@@ -0,0 +1,21 @@
+Gem::Specification.new do |s|
+  s.name        = 'db-migrate'
+  s.version     = '0.0.1'
+  s.licenses    = ['MIT']
+  s.summary     = "Tool for managing and executing your database migrations."
+  s.description = "#{s.summary} It supports multiple databases and multiple languages for writing migration scripts."
+  s.authors     = ["Ivan Pusic"]
+  s.email       = 'pusic007@gmail.com'
+  s.files       = `git ls-files -z`.split("\x0")
+  s.homepage    = 'https://github.com/ivpusic/migrate'
+  s.executables << 'migrate'
+
+  # deps
+  s.add_runtime_dependency 'thor', ['0.19.1']
+  s.add_runtime_dependency 'highline', ['1.7.8']
+  s.add_runtime_dependency 'json', ['1.8.3']
+  s.add_runtime_dependency 'mysql2', ['0.4.2']
+  s.add_runtime_dependency 'pg', ['0.18.4']
+  s.add_runtime_dependency 'parseconfig', ['1.0.6']
+  s.add_runtime_dependency 'colorize', ['0.7.7']
+end

data/spec/lib/config_spec.rb

@@ -0,0 +1,103 @@
+describe "Config" do
+  let(:fixtures) { "spec/lib/fixtures"  }
+  let(:config) { Config.new(fixtures, "example.config") }
+  let(:config_hash) {
+    {
+      host: "localhost",
+      port: ("5432").to_i,
+      database: "mydb",
+      user: "postgres",
+      password: "password",
+      version_info: "version_info",
+      version_number: "version_number"
+    }
+  }
+
+  it "should create new instance" do
+    expect(config).not_to eq(nil)
+  end
+
+  it "should save root" do
+    expect(config.root).to eq("spec/lib/fixtures")
+  end
+
+  it "should not find file" do
+    config = Config.new(".", "custom_file.conf")
+    expect(config.exists?).to eq(false)
+  end
+
+  it "should find file" do
+    expect(config.exists?).to eq(true)
+  end
+
+  it "should create config file" do
+    config_path = "#{fixtures}/test.config"
+    begin
+      expect(File.exist? config_path).to be false
+
+      config = Config.new(fixtures, "test.config")
+      config.init(config_hash)
+      expect(File.exist? config_path). to be true
+    ensure
+      if File.exist? config_path
+        File.delete config_path
+      end
+    end
+  end
+
+  it "should load configuration" do
+    config.init(config_hash)
+    config.load!
+    expect(config.host).to eq("localhost")
+    expect(config.port).to eq("5432")
+    expect(config.database).to eq("mydb")
+    expect(config.user).to eq("postgres")
+    expect(config.password).to eq("password")
+    expect(config.version_info).to eq("version_info")
+    expect(config.version_number).to eq("version_number")
+  end
+
+  it "should remove configuration" do
+    config_path = "#{fixtures}/test.config"
+    begin
+      expect(File.exist? config_path).to be false
+
+      config = Config.new(fixtures, "test.config")
+      config.init(config_hash)
+      expect(File.exist? config_path).to be true
+      config.delete
+      expect(File.exist? config_path).to be false
+    ensure
+      if File.exist? config_path
+        File.delete config_path
+      end
+    end
+  end
+
+  [{type: "pg", cls: Storage::Postgres, conf: $pg_config_hash}, 
+   {type: "mysql", cls: Storage::Mysql, conf: $mysql_config_hash}].each do |storage|
+    context storage[:type] do
+      it "should be able to get database instance" do
+        config.init(storage[:conf])
+        config.load!
+        expect(config.get_db).to be_kind_of(storage[:cls])
+      end
+    end
+  end
+
+  [{type: "go", cls: Lang::Go}, {type: "sql", cls: Lang::Sql},
+   {type: "ruby", cls: Lang::Ruby}, {type: "javascript", cls: Lang::Javascript}, 
+   {type: "python", cls: Lang::Python}].each do |lang|
+    context lang[:type] do
+      it "should be able to get language instance" do
+        config_hash = $pg_config_hash.merge({
+          lang: lang[:type]
+        })
+
+        config.init(config_hash)
+        config.load!
+        expect(config.get_lang).to be_kind_of(lang[:cls])
+      end
+    end
+  end
+end

data/spec/lib/fixtures/go/down.go

@@ -0,0 +1,7 @@
+package main
+
+import "fmt"
+
+func main() {
+	fmt.Println("works")
+}

data/spec/lib/fixtures/go/up.go

@@ -0,0 +1,7 @@
+package main
+
+import "fmt"
+
+func main() {
+	fmt.Println("works")
+}

data/spec/lib/fixtures/js/down.js

@@ -0,0 +1 @@
+console.log("works");

data/spec/lib/fixtures/js/up.js

@@ -0,0 +1 @@
+console.log("works");

data/spec/lib/fixtures/py/down.py

@@ -0,0 +1 @@
+print "works"

data/spec/lib/fixtures/py/up.py

@@ -0,0 +1 @@
+print "works"

data/spec/lib/fixtures/rb/down.rb

@@ -0,0 +1 @@
+puts "works"

data/spec/lib/fixtures/rb/up.rb

@@ -0,0 +1 @@
+puts "works"

data/spec/lib/fixtures/sql/down.sql

@@ -0,0 +1 @@
+-- todo;

data/spec/lib/fixtures/sql/up.sql

@@ -0,0 +1 @@
+-- todo;

data/spec/lib/lang/lang_spec.rb

@@ -0,0 +1,46 @@
+describe "Lang" do
+  langs = [
+    Lang::Go.new,
+    Lang::Javascript.new,
+    Lang::Ruby.new,
+    Lang::Sql.new($dbs[0]),
+    Lang::Python.new
+  ]
+
+  langs.each do |lang|
+    dir = "spec/lib/fixtures/v01"
+
+    after(:each) do
+      if Dir.exist? dir
+        Dir.glob("#{dir}/{up.*,down.*}").each do |file|
+          File.delete(file)
+        end
+        Dir.rmdir dir
+      end
+    end
+
+    it "#{lang.ext} should be able to create new migration" do
+      Dir.mkdir dir
+      lang.create_migration(dir)
+
+      up = "#{dir}/up.#{lang.ext}"
+      down = "#{dir}/down.#{lang.ext}"
+
+      expect(File.exist? up).to be true
+      expect(File.exist? down).to be true
+    end
+
+    context "#{lang.ext} when migrating up" do
+      [true, false].each do |is_up|
+        it "should be able to execute migration script" do
+          lang_dir = "spec/lib/fixtures/#{lang.ext}"
+
+          if lang.ext != "sql"
+            out = lang.exec_migration(lang_dir, is_up)
+            expect(out.strip).to eq("works")
+          end
+        end
+      end
+    end
+  end
+end

data/spec/lib/migrator_spec.rb

@@ -0,0 +1,103 @@
+describe "Migrator" do
+  $dbs.each do |db|
+    config = db.config
+    migrator = Migrator.new(config)
+
+    around(:each) do |test|
+      db.tx do
+        test.run
+      end
+    end
+
+    context db.type do
+      it "should be able to initialize" do
+        expect{Migrator.new(config)}.not_to raise_error(Exception)
+      end
+
+      it "should make new migration" do
+        migration_dir = nil
+
+        begin
+          result = db.exec_sql("SELECT * FROM #{config.version_info} ORDER BY version DESC")
+          version = db.extract_version result
+
+          migration_dir = migrator.new("description")
+
+          result = db.exec_sql("SELECT * FROM #{config.version_info} ORDER BY version DESC")
+          new_version = db.extract_version result
+          expect(new_version.to_i).to eq(version.to_i + 1)
+          expect(Dir.exist? migration_dir).to be true
+        ensure
+          if migration_dir != nil
+            Dir.glob("#{migration_dir}/{up.*,down.*}").each do |file|
+              File.delete file
+            end
+
+            Dir.rmdir migration_dir
+          end
+        end
+      end
+
+      create_migration_dir = lambda do |version|
+        migration = db.get_migration(version)
+        migration_dir = migrator.migration_dir(migration)
+
+        if not Dir.exist? migration_dir
+          Dir.mkdir migration_dir
+          config.get_lang().create_migration(migration_dir)
+        end
+      end
+
+      it "should execute one up migration" do
+        current = db.current_version().to_i
+        create_migration_dir.call(current + 1)
+        migrator.up
+
+        expect(current.to_i + 1).to eq(db.current_version().to_i)
+      end
+
+      it "should execute multiple up migration" do
+        current = db.current_version().to_i
+        create_migration_dir.call(current + 1)
+        create_migration_dir.call(current + 2)
+
+        migrator.up(current + 2)
+        expect(current.to_i + 2).to eq(db.current_version().to_i)
+      end
+
+      it "should execute one down migration" do
+        current = db.current_version().to_i
+        create_migration_dir.call(current)
+
+        migrator.down
+        expect(current.to_i - 1).to eq(db.current_version().to_i)
+      end
+
+      it "should execute multiple up migration" do
+        current = db.current_version().to_i
+        create_migration_dir.call(current)
+        create_migration_dir.call(current - 1)
+
+        migrator.down(current - 2)
+        expect(current.to_i - 2).to eq(db.current_version().to_i)
+      end
+
+      it "should get current version" do
+        current = db.current_version().to_i
+        expect(migrator.current_version().to_i).to eq(current)
+      end
+
+      it "should delete one version" do
+        delete = db.current_version().to_i + 1
+        migrator.delete(delete)
+        expect{db.get_migration(delete)}.to raise_error(VersionNotFound)
+      end
+
+      it "should not delete current version" do
+        current = db.current_version().to_i
+        migrator.delete(current)
+        expect(db.get_migration(current)["version"].to_i).to eq(current)
+      end
+    end
+  end
+end