mkxms-mssql 1.0.0 → 1.1.0

data/lib/mkxms/mssql/utils.rb

@@ -50,6 +50,92 @@ module Mkxms::Mssql::Utils
       item.send(@children_message).each(&blk)
     end
   end
+  
+  module StringHelpers
+    def expand_tabs(tabstops_every = 8)
+      self.lines.map do |l|
+        if l.include?("\t")
+          segs = l.split("\t")
+          segs[0...-1].map do |seg|
+            # seg length must _increase_ to a multiple of 8
+            spaces_needed = tabstops_every - (seg.length + 1) % tabstops_every + 1
+            seg + ' ' * spaces_needed
+          end.join('') + segs[-1]
+        else
+          l
+        end
+      end.join('')
+    end
+    
+    def sql_quoted
+      %Q{N'#{gsub("'", "''")}'}
+    end
+  end
+  
+  # Primes in the interval [100, 255].  This enumerator can be queried by
+  # classes that generate RAISERROR statements to provide unique-ish context
+  # by passing the next multiple of one of the values taken from this
+  # enumerator for each RAISERROR statement output (as a literal number in
+  # the generated SQL).  This will assist
+  RAISERROR_STATE_BASE = [
+    101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163 ,167, 173,
+    179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251
+  ].each unless defined? RAISERROR_STATE_BASE
+  
+  # Create one instance of this class to write a sequence of similar
+  # RAISERROR messages.  The state of each message will be unique within the
+  # sequence until the 256th message.  The particular order is unique to
+  # all other instances of this class.
+  class RaiserrorWriter
+    # Severity:
+    #   11 is the minimum to transfer into a CATCH
+    #   19 or higher can only be raised by sysadmin
+    #   20 or higher is fatal to the connection
+    
+    SYMBOLIC_SEVERITIES = {
+      :warning => 1,
+      :error => 11,
+    }
+    
+    def initialize(message, severity: 11)
+      # Get a unique prime to use as an ideal to generate the 0-255 state-value
+      # space.  With luck, the number is fairly unique to the message.
+      severity = map_severity(severity)
+      @state_base = RAISERROR_STATE_BASE.next
+      @index = 1 # Start at 1 because 0 is the kernel -- every @state_base * 0 == 0
+      @message = message
+      @severity = severity
+    end
+    
+    attr_reader :state_base
+    attr_accessor :message, :severity
+    
+    def map_severity(value)
+      SYMBOLIC_SEVERITIES.fetch(value, value)
+    end
+    
+    def current_statement(*args, severity: nil)
+      severity = map_severity(severity || self.severity)
+      state_str = current_error_marker
+      full_message = %Q{N'#{message.gsub("'", "''")} (search for "#{state_str}")'}
+      trailing_args = ([state_str] + args.map(&:to_s)).join(', ')
+      %Q{RAISERROR (#{full_message}, #{severity}, #{trailing_args})}.tap do |stmt|
+        stmt.define_singleton_method(:error_marker) {state_str}
+      end
+    end
+    
+    def current_error_marker
+      "/*ERR*/ #{current_state} /*ERR*/"
+    end
+    
+    def current_state
+      (state_base * @index) % 256
+    end
+    
+    def next_statement(*args, **kwargs)
+      current_statement(*args, **kwargs).tap {@index += 1}
+    end
+  end
 end
 
 class << Mkxms::Mssql::Utils
@@ -80,4 +166,14 @@ class << Mkxms::Mssql::Utils
       l
     end.join('')
   end
+  
+  def dry_run?
+    @dry_run.tap do |v|
+      (break @dry_run = !!(ENV.fetch('DRY_RUN', '') =~ /^(y(es)?|t(rue)?|1)$/i)) if v.nil?
+    end
+  end
+end
+
+class String
+  include Mkxms::Mssql::Utils::StringHelpers
 end

data/lib/mkxms/mssql/version.rb

@@ -1,5 +1,5 @@
 module Mkxms
   module Mssql
-    VERSION = "1.0.0"
+    VERSION = "1.1.0"
   end
 end

data/lib/mkxms/mssql/view_handler.rb

@@ -34,14 +34,11 @@ module Mkxms::Mssql
       a = node.attributes
       
       @view = View.new(a).tap do |v|
+        store_properties_on v
         views << v
       end
     end
     
-    def extended_properties
-      @view.extended_properties
-    end
-    
     def handle_definition_element(parse); end
     
     def handle_references_element(parse)

data/spec/utils/indented_string_builder_spec.rb

@@ -213,6 +213,27 @@ describe Mkxms::Mssql::IndentedStringBuilder do
         test_instance = test_class.new
         expect(test_instance.to_s).to eql("BEGIN\n  command\nEND\n")
       end
+      
+      it "handles blank lines properly" do
+        test_class = Class.new(described_class) do
+          def initialize
+            super
+            
+            add_level_one
+          end
+          
+          def add_level_one
+            dsl {
+              puts "foo"
+              puts
+            }
+            puts "bar"
+          end
+        end
+        
+        test_instance = test_class.new
+        expect(test_instance.to_s).to eql("foo\n\nbar\n")
+      end
     end
   end
 end

data/spec/utils/query_cursor_spec.rb

@@ -16,7 +16,7 @@ describe Mkxms::Mssql::QueryCursor do
       sql.puts "  -- Handle a row"
     end
     
-    expect(sql.string).to eql("DECLARE @schema_id INT, @schema_name SYSNAME;\nDECLARE test_cursor CURSOR LOCAL FOR\nSELECT schema_id, name FROM sys.schemas;\nFETCH NEXT FROM test_cursor INTO @schema_id, @schema_name;\nWHILE @@FETCH_STATUS = 0\nBEGIN\n  -- Handle a row\n  FETCH NEXT FROM test_cursor INTO @schema_id, @schema_name;\nEND;\n")
+    expect(sql.string).to eql("DECLARE @schema_id INT, @schema_name SYSNAME;\nDECLARE test_cursor CURSOR LOCAL FOR\nSELECT schema_id, name FROM sys.schemas;\nOPEN test_cursor;\nFETCH NEXT FROM test_cursor INTO @schema_id, @schema_name;\nWHILE @@FETCH_STATUS = 0\nBEGIN\n  -- Handle a row\n  FETCH NEXT FROM test_cursor INTO @schema_id, @schema_name;\nEND;\n")
   end
   
   it "provides a expectation loop" do
@@ -52,6 +52,6 @@ describe Mkxms::Mssql::QueryCursor do
       end
     end
     
-    expect(sql.string).to eql("DECLARE @column_name SYSNAME, @is_sorted_descending BIT;\nDECLARE test_cursor CURSOR LOCAL FOR\nSELECT c.name, ic.is_descending_key FROM sys.index_columns ic JOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id WHERE ic.object_id = @relation_id AND ic.index_id = @index_id AND ic.key_ordinal >= 1 ORDER BY ic.key_ordinal;\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is foo.\n-- Test is ascending.\nEND;\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is bar.\n-- Test is ascending.\nEND;\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is baz.\n-- Test is descending.\nEND;\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS = 0\nBEGIN\n-- Handle extra entry/ies\nEND;\nCLOSE  test_cursor; DEALLOCATE  test_cursor;\n")
+    expect(sql.string).to eql("DECLARE @column_name SYSNAME, @is_sorted_descending BIT;\nDECLARE test_cursor CURSOR LOCAL FOR\nSELECT c.name, ic.is_descending_key FROM sys.index_columns ic JOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id WHERE ic.object_id = @relation_id AND ic.index_id = @index_id AND ic.key_ordinal >= 1 ORDER BY ic.key_ordinal;\nOPEN test_cursor;\n\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is foo.\n-- Test is ascending.\nEND;\n\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is bar.\n-- Test is ascending.\nEND;\n\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS <> 0\nBEGIN\n-- Handle missing entry\nEND ELSE BEGIN\n-- Test column name is baz.\n-- Test is descending.\nEND;\nFETCH NEXT FROM test_cursor INTO @column_name, @is_sorted_descending;\nIF @@FETCH_STATUS = 0\nBEGIN\n-- Handle extra entry/ies\nEND;\nCLOSE  test_cursor; DEALLOCATE  test_cursor;\n")
   end
 end

data/spec/utils/sql_string_manipulators_spec.rb

@@ -0,0 +1,59 @@
+require 'mkxms/mssql/sql_string_manipulators'
+
+describe Mkxms::Mssql::SqlStringManipulators do
+  module T
+    extend Mkxms::Mssql::SqlStringManipulators
+  end
+  
+  context "dedenting" do
+    it "handles a string with no newlines" do
+      expect(T.dedent("foo")).to eq("foo")
+    end
+    
+    it "handles a string with a all lines at a constant indent" do
+      s = T.dedent %Q{
+        Jack and Jill went up the hill
+        to fetch a pail of water
+      }
+      expect(s.lines.size).to eq(2)
+      expect(s.lines[0]).not_to start_with(' ')
+    end
+    
+    it "handles nested indentation" do
+      s = T.dedent %Q{
+        one
+          two
+        one
+          two
+      }
+      expect(s.lines.size).to eq(4)
+      expect(s.lines[0]).not_to start_with(' ')
+      expect(s.lines[1]).to start_with('  t')
+    end
+    
+    it "allows leading blank lines" do
+      s = T.dedent %Q{
+        
+        after the break
+      }
+      expect(s.lines[0]).to match(/^\s*$/)
+    end
+    
+    it "allows trailing blank lines" do
+      s = T.dedent %Q{
+        before the break
+        
+      }
+      expect(s.lines[1]).to match(/^\s*$/)
+    end
+    
+    it "allows internal blank lines" do
+      s = T.dedent %Q{
+        before
+        
+        after
+      }
+      expect(s.lines[1]).to eq("\n")
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mkxms-mssql
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Richard Weeks
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-28 00:00:00.000000000 Z
+date: 2018-02-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: xmigra
@@ -85,8 +85,16 @@ files:
 - lib/mkxms/mssql/access_object_definition.rb
 - lib/mkxms/mssql/adoption_script_writer.rb
 - lib/mkxms/mssql/check_constraint_handler.rb
+- lib/mkxms/mssql/clr_aggregate_handler.rb
+- lib/mkxms/mssql/clr_assembly_handler.rb
+- lib/mkxms/mssql/clr_function_handler.rb
+- lib/mkxms/mssql/clr_impl.rb
+- lib/mkxms/mssql/clr_stored_procedure_handler.rb
+- lib/mkxms/mssql/clr_type_handler.rb
 - lib/mkxms/mssql/database_handler.rb
+- lib/mkxms/mssql/declaratives_creator.rb
 - lib/mkxms/mssql/default_constraint_handler.rb
+- lib/mkxms/mssql/dml_trigger_handler.rb
 - lib/mkxms/mssql/engine.rb
 - lib/mkxms/mssql/exceptions.rb
 - lib/mkxms/mssql/filegroup_handler.rb
@@ -96,16 +104,21 @@ files:
 - lib/mkxms/mssql/index_column.rb
 - lib/mkxms/mssql/index_handler.rb
 - lib/mkxms/mssql/keylike_constraint_helper.rb
+- lib/mkxms/mssql/keywords.rb
 - lib/mkxms/mssql/permission_handler.rb
 - lib/mkxms/mssql/primary_key_handler.rb
 - lib/mkxms/mssql/property_handler.rb
 - lib/mkxms/mssql/query_cursor.rb
+- lib/mkxms/mssql/references_handler.rb
 - lib/mkxms/mssql/role_handler.rb
+- lib/mkxms/mssql/scalar_type_handler.rb
 - lib/mkxms/mssql/schema_handler.rb
 - lib/mkxms/mssql/sql_string_manipulators.rb
 - lib/mkxms/mssql/statistics_handler.rb
 - lib/mkxms/mssql/stored_procedure_handler.rb
+- lib/mkxms/mssql/synonym_handler.rb
 - lib/mkxms/mssql/table_handler.rb
+- lib/mkxms/mssql/table_type_handler.rb
 - lib/mkxms/mssql/unique_constraint_handler.rb
 - lib/mkxms/mssql/utils.rb
 - lib/mkxms/mssql/version.rb
@@ -113,6 +126,7 @@ files:
 - mkxms-mssql.gemspec
 - spec/utils/indented_string_builder_spec.rb
 - spec/utils/query_cursor_spec.rb
+- spec/utils/sql_string_manipulators_spec.rb
 homepage: ''
 licenses:
 - MIT
@@ -133,10 +147,11 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.1
+rubygems_version: 2.4.8
 signing_key: 
 specification_version: 4
 summary: XMigra source files from MS-SQL database description.
 test_files:
 - spec/utils/indented_string_builder_spec.rb
 - spec/utils/query_cursor_spec.rb
+- spec/utils/sql_string_manipulators_spec.rb