scheman 0.0.1

data/lib/scheman/views/mysql.rb

@@ -0,0 +1,267 @@
+module Scheman
+  module Views
+    class Mysql < Base
+      def self.transform
+        @transform ||= Transform.new
+      end
+
+      # @param diff [Hash]
+      def initialize(diff)
+        @diff = diff
+      end
+
+      # @return [String]
+      def to_s
+        self.class.transform.apply(
+          root: @diff
+        )
+      end
+
+      class Transform < Parslet::Transform
+        rule(root: subtree(:root)) do
+          [
+            "BEGIN;",
+            "SET foreign_key_checks=0;",
+            root,
+            "SET foreign_key_checks=1;",
+            "COMMIT;",
+          ].join("\n\n") + "\n"
+        end
+
+        rule(
+          alter_tables: sequence(:alter_tables),
+          create_tables: sequence(:create_tables),
+          drop_tables: sequence(:drop_tables),
+        ) do
+          [
+            CreateTables.new(create_tables).to_s.presence,
+            AlterTables.new(alter_tables).to_s.presence,
+            DropTables.new(drop_tables).to_s.presence,
+          ].compact.join("\n\n")
+        end
+
+        rule(drop_table: subtree(:drop_table)) do
+          DropTable.new(drop_table)
+        end
+
+        rule(create_table: subtree(:create_table)) do
+          CreateTable.new(create_table)
+        end
+
+        rule(field: subtree(:field)) do
+          Field.new(field)
+        end
+
+        rule(qualifier: subtree(:qualifier)) do
+          Qualifier.new(qualifier)
+        end
+
+        rule(index: subtree(:index)) do
+          Index.new(index)
+        end
+
+        rule(add_field: subtree(:add_field)) do
+          AddField.new(add_field)
+        end
+
+        rule(drop_field: subtree(:drop_field)) do
+          DropField.new(drop_field)
+        end
+      end
+
+      class Node
+        def initialize(element)
+          @element = element
+        end
+      end
+
+      class Statements < Node
+        def to_s
+          @element.join("\n\n")
+        end
+      end
+
+      class AlterTables < Statements
+      end
+
+      class DropTables < Statements
+      end
+
+      class CreateTables < Statements
+      end
+
+      class DropTable < Node
+        def to_s
+          "DROP TABLE `#{table_name}`;"
+        end
+
+        private
+
+        def table_name
+          @element[:name]
+        end
+      end
+
+      class AlterTable < Node
+        def to_s
+          "TODO"
+        end
+      end
+
+      class AlterField < Node
+        private
+
+        def table_name
+          @element[:table_name]
+        end
+      end
+
+      class AddField < AlterField
+        def to_s
+          "ALTER TABLE `#{table_name}` ADD COLUMN #{field};"
+        end
+
+        private
+
+        def field
+          Field.new(@element)
+        end
+      end
+
+      class DropField < AlterField
+        def to_s
+          "ALTER TABLE `#{table_name}` DROP COLUMN `#{field_name}`;"
+        end
+
+        private
+
+        def field_name
+          @element[:name]
+        end
+      end
+
+      class CreateTable < Node
+        def to_s
+          str = ""
+          str << "CREATE TABLE `#{table_name}` (\n"
+          str << definitions.join(",\n").indent(2) + "\n"
+          str << ");"
+        end
+
+        private
+
+        def table_name
+          @element[:name]
+        end
+
+        def definitions
+          @element[:fields] + @element[:indices]
+        end
+      end
+
+      class Field < Node
+        def to_s
+          str = "`#{name}` #{type}"
+          str << "(#{values})" if has_values?
+          str << " #{qualifiers}" if has_qualifiers?
+          str
+        end
+
+        private
+
+        # @example
+        #   "id"
+        def name
+          @element[:name]
+        end
+
+        # @example
+        #   "INTEGER"
+        def type
+          @element[:type].upcase
+        end
+
+        def qualifiers
+          @element[:qualifiers].map(&:to_s).join(" ")
+        end
+
+        def values
+          @element[:values].join(", ")
+        end
+
+        def has_qualifiers?
+          !@element[:qualifiers].empty?
+        end
+
+        def has_values?
+          !@element[:values].empty?
+        end
+      end
+
+      class Qualifier < Node
+        def to_s
+          str = type
+          str << " #{value}" if has_value?
+          str
+        end
+
+        private
+
+        # @example
+        #   "NOT NULL"
+        def type
+          @element[:type].upcase.gsub("_", " ")
+        end
+
+        # @example
+        #   "utf8"
+        def value
+          @element[:value]
+        end
+
+        def has_value?
+          @element[:value]
+        end
+      end
+
+      class Index < Node
+        def to_s
+          "#{prefix} (`#{column}`)"
+        end
+
+        def primary_key?
+          !!@element[:primary]
+        end
+
+        def fulltext?
+          @element[:type] == "fulltext"
+        end
+
+        def spatial?
+          @element[:type] == "spatial"
+        end
+
+        # @example
+        #   "id"
+        def column
+          @element[:column]
+        end
+
+        # @example
+        #   "PRIMARY KEY"
+        def prefix
+          case
+          when primary_key?
+            "PRIMARY KEY"
+          when fulltext?
+            "FULLTEXT"
+          when spatial?
+            "SPATIAL"
+          else
+            "KEY"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/scheman.rb

@@ -0,0 +1,17 @@
+require "active_support/core_ext/array/wrap"
+require "active_support/core_ext/enumerable"
+require "active_support/core_ext/hash/slice"
+require "active_support/core_ext/object/blank"
+require "active_support/core_ext/object/try"
+require "active_support/core_ext/string/indent"
+require "parslet"
+
+require "scheman/diff"
+require "scheman/errors"
+require "scheman/parser_builder"
+require "scheman/parsers/base"
+require "scheman/parsers/mysql"
+require "scheman/schema"
+require "scheman/version"
+require "scheman/views/base"
+require "scheman/views/mysql"

data/scheman.gemspec

@@ -0,0 +1,25 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "scheman/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "scheman"
+  spec.version       = Scheman::VERSION
+  spec.authors       = ["Ryo Nakamura"]
+  spec.email         = ["r7kamura@gmail.com"]
+  spec.summary       = "Manage database schema based on schema definition file."
+  spec.homepage      = "https://github.com/r7kamura/scheman"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "activesupport"
+  spec.add_dependency "parslet"
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec", "2.14.1"
+end

data/spec/scheman/diff_spec.rb

@@ -0,0 +1,76 @@
+require "spec_helper"
+
+describe Scheman::Diff do
+  let(:instance) do
+    described_class.new(args)
+  end
+
+  let(:args) do
+    {
+      before: before_schema,
+      after: after_schema,
+      type: type,
+    }
+  end
+
+  let(:type) do
+    "mysql"
+  end
+
+  let(:before_schema) do
+    <<-EOS.strip_heredoc
+      CREATE TABLE `table1` (
+        `column1` INTEGER NOT NULL AUTO INCREMENT,
+        PRIMARY KEY (`column1`)
+      );
+
+      CREATE TABLE `table2` (
+        `column1` INTEGER NOT NULL AUTO INCREMENT,
+        PRIMARY KEY (`column1`)
+      );
+    EOS
+  end
+
+  let(:after_schema) do
+    <<-EOS.strip_heredoc
+      CREATE TABLE `table1` (
+        `column2` VARCHAR(255) NOT NULL,
+        PRIMARY KEY (`column1`)
+      );
+
+      CREATE TABLE `table3` (
+        `column1` INTEGER NOT NULL AUTO INCREMENT,
+        PRIMARY KEY (`column1`)
+      );
+    EOS
+  end
+
+  describe "#to_s" do
+    subject do
+      instance.to_s
+    end
+
+    it "returns a diff in SQL" do
+      should == <<-EOS.strip_heredoc
+        BEGIN;
+
+        SET foreign_key_checks=0;
+
+        CREATE TABLE `table3` (
+          `column1` INTEGER NOT NULL AUTO INCREMENT,
+          PRIMARY KEY (`column1`)
+        );
+
+        ALTER TABLE `table1` ADD COLUMN `column2` VARCHAR(255) NOT NULL;
+
+        ALTER TABLE `table1` DROP COLUMN `column1`;
+
+        DROP TABLE `table2`;
+
+        SET foreign_key_checks=1;
+
+        COMMIT;
+      EOS
+    end
+  end
+end