mkxms-mssql 1.0.0 → 1.1.0

data/lib/mkxms/mssql/dml_trigger_handler.rb

@@ -0,0 +1,107 @@
+require 'mkxms/mssql/clr_impl'
+require 'mkxms/mssql/references_handler'
+require 'mkxms/mssql/utils'
+
+module Mkxms; end
+
+module Mkxms::Mssql
+  class Trigger
+    extend Utils::InitializedAttributes
+    include ExtendedProperties, Property::Hosting, Property::SchemaScoped
+    include Dependencies
+    include Utils::SchemaQualifiedName
+    include XMigra::MSSQLSpecifics
+    
+    def initialize(schema, name, timing, execute_as: nil, disabled: false, not_replicable: false)
+      @schema = schema
+      @name = name
+      @timing = timing
+      @execute_as = execute_as
+      @disabled = disabled
+      @not_replicable = not_replicable
+    end
+    
+    attr_accessor :schema, :name, :table, :timing, :execute_as, :disabled, :not_replicable, :clr_impl
+    attr_init(:events) {[]}
+    attr_init(:definition) {""}
+    
+    def to_sql
+      def_sql = clr_impl ? clr_definition : definition
+      [def_sql.expand_tabs.gsub(/ +\n/, "\n")].tap do |result|
+        unless (ep_sql = extended_properties_sql).empty?
+          result << ep_sql.joined_on_new_lines
+        end
+        if disabled
+          result << "DISABLE TRIGGER #{qualified_name} ON #{table.qualified_name};"
+        end
+      end.join(ddl_block_separator)
+    end
+    
+    def clr_definition
+      [].tap do |lines|
+        lines << "CREATE TRIGGER #{schema}.#{name}"
+        lines << "ON #{table.qualified_name}"
+        case execute_as
+        when 'OWNER'
+          lines << "WITH EXECUTE AS OWNER"
+        when String
+          lines << "WITH EXECUTE AS #{execute_as.sql_quoted}"
+        end
+        lines << "#{timing} #{events.join(', ')}"
+        lines << "NOT FOR REPLICATION" if not_replicable
+        lines << "AS EXTERNAL NAME #{clr_impl.full_specifier};"
+      end.join("\n")
+    end
+  end
+  
+  class DmlTriggerHandler
+    extend XMigra::MSSQLSpecifics
+    include PropertyHandler::ElementHandler
+    include ReferencesHandler::ElementHandler
+    
+    def initialize(triggers, node)
+      a = node.attributes
+      
+      @trigger = Trigger.new(
+        a['schema'],
+        a['name'],
+        a['timing'],
+        execute_as: a['execute_as'],
+        disabled: a['disabled'],
+        not_replicable: a['not-for-replication'],
+      ).tap do |t|
+        store_properties_on t
+        triggers << t
+      end
+    end
+    
+    def dependencies
+      @trigger.dependencies
+    end
+    
+    def handle_table_element(parse)
+      a = parse.node.attributes
+      @trigger.table = Reference.new(a['schema'], a['name'])
+    end
+    
+    def handle_event_element(parse)
+      a = parse.node.attributes
+      @trigger.events << a['type']
+    end
+    
+    def handle_do_element(parse); end
+    
+    def handle_text(text, parent_element)
+      case [parent_element.namespace, parent_element.name]
+      when ['', 'do']
+        @trigger.definition << text
+      end
+    end
+    
+    # This function handles a CLR implementation
+    def handle_implementation_element(parse)
+      a = parse.node.attributes
+      @trigger.clr_impl = ClrMethod.new(a['assembly'], a['class'], a['method'])
+    end
+  end
+end

data/lib/mkxms/mssql/filegroup_handler.rb

@@ -31,15 +31,12 @@ module Mkxms::Mssql
         end
       ]
       @filegroup = Filegroup.new(**group_options).tap do |fg|
+        store_properties_on fg
         filegroups << fg
       end
       @files = []
     end
     
-    def extended_properties
-      @filegroup.extended_properties
-    end
-    
     def handle_file_element(parse)
       parse.context = DatabaseFile.new(@files, parse.node)
     end

data/lib/mkxms/mssql/function_handler.rb

@@ -45,14 +45,11 @@ module Mkxms::Mssql
       a = node.attributes
       
       @function = Function.new(a).tap do |f|
+        store_properties_on f
         functions << f
       end
     end
     
-    def extended_properties
-      @function.extended_properties
-    end
-    
     def handle_definition_element(parse); end
     
     def handle_references_element(parse)

data/lib/mkxms/mssql/indented_string_builder.rb

@@ -48,7 +48,13 @@ module Mkxms::Mssql
       
       def puts(*args, &blk)
         if blk
-          @builder.puts(*args) {IndentedStringBuilder.new.tap {|i| i.dsl(&blk)}}
+          if args[0].kind_of? Range
+            @builder.puts(args[0].begin, *args[1..-1])
+            indented(&blk)
+            @builder.puts(args[0].end, *args[1..-1])
+          else
+            @builder.puts(*args) {IndentedStringBuilder.new.tap {|i| i.dsl(&blk)}}
+          end
         else
           @builder.puts(*args)
         end
@@ -123,7 +129,7 @@ module Mkxms::Mssql
         subbed_multiline = false
         s.scan(scan_pattern) do |m|
           chunk_range, completed = completed...$~.begin(0), $~.end(0)
-          chunk_empty = !chunk_range.cover?(chunk_range.begin)
+          chunk_empty = !chunk_range.cover?(chunk_range.begin) && s != ''
           
           case
           when m == "\n"

data/lib/mkxms/mssql/index_handler.rb

@@ -75,14 +75,11 @@ module Mkxms::Mssql
     
     def initialize(indexes, node)
       @index = Index.new(node.attributes).tap do |i|
+        store_properties_on i
         indexes << i
       end
     end
     
-    def extended_properties
-      @index.extended_properties
-    end
-    
     def handle_column_element(parse)
       a = parse.node.attributes
       

data/lib/mkxms/mssql/keywords.rb

@@ -0,0 +1,492 @@
+require 'set'
+
+module Mkxms; end
+
+module Mkxms::Mssql
+  KEYWORDS = %w[
+    ABSOLUTE
+    ACTION
+    ADA
+    ADD
+    ADMIN
+    AFTER
+    AGGREGATE
+    ALIAS
+    ALL
+    ALLOCATE
+    ALTER
+    AND
+    ANY
+    ARE
+    ARRAY
+    AS
+    ASC
+    ASENSITIVE
+    ASSERTION
+    ASYMMETRIC
+    AT
+    ATOMIC
+    AUTHORIZATION
+    AVG
+    BACKUP
+    BEFORE
+    BEGIN
+    BETWEEN
+    BINARY
+    BIT
+    BIT_LENGTH
+    BLOB
+    BOOLEAN
+    BOTH
+    BREADTH
+    BREAK
+    BROWSE
+    BULK
+    BY
+    CALL
+    CALLED
+    CARDINALITY
+    CASCADE
+    CASCADED
+    CASE
+    CAST
+    CATALOG
+    CHAR
+    CHARACTER
+    CHARACTER_LENGTH
+    CHAR_LENGTH
+    CHECK
+    CHECKPOINT
+    CLASS
+    CLOB
+    CLOSE
+    CLUSTERED
+    COALESCE
+    COLLATE
+    COLLATION
+    COLLECT
+    COLUMN
+    COMMIT
+    COMPLETION
+    COMPUTE
+    CONDITION
+    CONNECT
+    CONNECTION
+    CONSTRAINT
+    CONSTRAINTS
+    CONSTRUCTOR
+    CONTAINS
+    CONTAINSTABLE
+    CONTINUE
+    CONVERT
+    CORR
+    CORRESPONDING
+    COUNT
+    COVAR_POP
+    COVAR_SAMP
+    CREATE
+    CROSS
+    CUBE
+    CUME_DIST
+    CURRENT
+    CURRENT_CATALOG
+    CURRENT_DATE
+    CURRENT_DEFAULT_TRANSFORM_GROUP
+    CURRENT_PATH
+    CURRENT_ROLE
+    CURRENT_SCHEMA
+    CURRENT_TIME
+    CURRENT_TIMESTAMP
+    CURRENT_TRANSFORM_GROUP_FOR_TYPE
+    CURRENT_USER
+    CURSOR
+    CYCLE
+    DATA
+    DATABASE
+    DATE
+    DAY
+    DBCC
+    DEALLOCATE
+    DEC
+    DECIMAL
+    DECLARE
+    DEFAULT
+    DEFERRABLE
+    DEFERRED
+    DELETE
+    DENY
+    DEPTH
+    DEREF
+    DESC
+    DESCRIBE
+    DESCRIPTOR
+    DESTROY
+    DESTRUCTOR
+    DETERMINISTIC
+    DIAGNOSTICS
+    DICTIONARY
+    DISCONNECT
+    DISK
+    DISTINCT
+    DISTRIBUTED
+    DOMAIN
+    DOUBLE
+    DROP
+    DUMP
+    DYNAMIC
+    EACH
+    ELEMENT
+    ELSE
+    END
+    END-EXEC
+    EQUALS
+    ERRLVL
+    ESCAPE
+    EVERY
+    EXCEPT
+    EXCEPTION
+    EXEC
+    EXECUTE
+    EXISTS
+    EXIT
+    EXTERNAL
+    EXTRACT
+    FALSE
+    FETCH
+    FILE
+    FILLFACTOR
+    FILTER
+    FIRST
+    FLOAT
+    FOR
+    FOREIGN
+    FORTRAN
+    FOUND
+    FREE
+    FREETEXT
+    FREETEXTTABLE
+    FROM
+    FULL
+    FULLTEXTTABLE
+    FUNCTION
+    FUSION
+    GENERAL
+    GET
+    GLOBAL
+    GO
+    GOTO
+    GRANT
+    GROUP
+    GROUPING
+    HAVING
+    HOLD
+    HOLDLOCK
+    HOST
+    HOUR
+    IDENTITY
+    IDENTITYCOL
+    IDENTITY_INSERT
+    IF
+    IGNORE
+    IMMEDIATE
+    IN
+    INCLUDE
+    INDEX
+    INDICATOR
+    INITIALIZE
+    INITIALLY
+    INNER
+    INOUT
+    INPUT
+    INSENSITIVE
+    INSERT
+    INT
+    INTEGER
+    INTERSECT
+    INTERSECTION
+    INTERVAL
+    INTO
+    IS
+    ISOLATION
+    ITERATE
+    JOIN
+    KEY
+    KILL
+    LANGUAGE
+    LARGE
+    LAST
+    LATERAL
+    LEADING
+    LEFT
+    LESS
+    LEVEL
+    LIKE
+    LIKE_REGEX
+    LIMIT
+    LINENO
+    LN
+    LOAD
+    LOCAL
+    LOCALTIME
+    LOCALTIMESTAMP
+    LOCATOR
+    LOWER
+    MAP
+    MATCH
+    MAX
+    MEMBER
+    MERGE
+    METHOD
+    MIN
+    MINUTE
+    MOD
+    MODIFIES
+    MODIFY
+    MODULE
+    MONTH
+    MULTISET
+    NAMES
+    NATIONAL
+    NATURAL
+    NCHAR
+    NCLOB
+    NEW
+    NEXT
+    NO
+    NOCHECK
+    NONCLUSTERED
+    NONE
+    NORMALIZE
+    NOT
+    NULL
+    NULLIF
+    NUMERIC
+    OBJECT
+    OCCURRENCES_REGEX
+    OCTET_LENGTH
+    OF
+    OFF
+    OFFSETS
+    OLD
+    ON
+    ONLY
+    OPEN
+    OPENDATASOURCE
+    OPENQUERY
+    OPENROWSET
+    OPENXML
+    OPERATION
+    OPTION
+    OR
+    ORDER
+    ORDINALITY
+    OUT
+    OUTER
+    OUTPUT
+    OVER
+    OVERLAPS
+    OVERLAY
+    PAD
+    PARAMETER
+    PARAMETERS
+    PARTIAL
+    PARTITION
+    PASCAL
+    PATH
+    PERCENT
+    PERCENTILE_CONT
+    PERCENTILE_DISC
+    PERCENT_RANK
+    PIVOT
+    PLAN
+    POSITION
+    POSITION_REGEX
+    POSTFIX
+    PRECISION
+    PREFIX
+    PREORDER
+    PREPARE
+    PRESERVE
+    PRIMARY
+    PRINT
+    PRIOR
+    PRIVILEGES
+    PROC
+    PROCEDURE
+    PUBLIC
+    RAISERROR
+    RANGE
+    READ
+    READS
+    READTEXT
+    REAL
+    RECONFIGURE
+    RECURSIVE
+    REF
+    REFERENCES
+    REFERENCING
+    REGR_AVGX
+    REGR_AVGY
+    REGR_COUNT
+    REGR_INTERCEPT
+    REGR_R2
+    REGR_SLOPE
+    REGR_SXX
+    REGR_SXY
+    REGR_SYY
+    RELATIVE
+    RELEASE
+    REPLICATION
+    RESTORE
+    RESTRICT
+    RESULT
+    RETURN
+    RETURNS
+    REVERT
+    REVOKE
+    RIGHT
+    ROLE
+    ROLLBACK
+    ROLLUP
+    ROUTINE
+    ROW
+    ROWCOUNT
+    ROWGUIDCOL
+    ROWS
+    RULE
+    SAVE
+    SAVEPOINT
+    SCHEMA
+    SCOPE
+    SCROLL
+    SEARCH
+    SECOND
+    SECTION
+    SECURITYAUDIT
+    SELECT
+    SEMANTICKEYPHRASETABLE
+    SEMANTICSIMILARITYDETAILSTABLE
+    SEMANTICSIMILARITYTABLE
+    SENSITIVE
+    SEQUENCE
+    SESSION
+    SESSION_USER
+    SET
+    SETS
+    SETUSER
+    SHUTDOWN
+    SIMILAR
+    SIZE
+    SMALLINT
+    SOME
+    SPACE
+    SPECIFIC
+    SPECIFICTYPE
+    SQL
+    SQLCA
+    SQLCODE
+    SQLERROR
+    SQLEXCEPTION
+    SQLSTATE
+    SQLWARNING
+    START
+    STATE
+    STATEMENT
+    STATIC
+    STATISTICS
+    STDDEV_POP
+    STDDEV_SAMP
+    STRUCTURE
+    SUBMULTISET
+    SUBSTRING
+    SUBSTRING_REGEX
+    SUM
+    SYMMETRIC
+    SYSTEM
+    SYSTEM_USER
+    TABLE
+    TABLESAMPLE
+    TEMPORARY
+    TERMINATE
+    TEXTSIZE
+    THAN
+    THEN
+    TIME
+    TIMESTAMP
+    TIMEZONE_HOUR
+    TIMEZONE_MINUTE
+    TO
+    TOP
+    TRAILING
+    TRAN
+    TRANSACTION
+    TRANSLATE
+    TRANSLATE_REGEX
+    TRANSLATION
+    TREAT
+    TRIGGER
+    TRIM
+    TRUE
+    TRUNCATE
+    TRY_CONVERT
+    TSEQUAL
+    UESCAPE
+    UNDER
+    UNION
+    UNIQUE
+    UNKNOWN
+    UNNEST
+    UNPIVOT
+    UPDATE
+    UPDATETEXT
+    UPPER
+    USAGE
+    USE
+    USER
+    USING
+    VALUE
+    VALUES
+    VARCHAR
+    VARIABLE
+    VARYING
+    VAR_POP
+    VAR_SAMP
+    VIEW
+    WAITFOR
+    WHEN
+    WHENEVER
+    WHERE
+    WHILE
+    WIDTH_BUCKET
+    WINDOW
+    WITH
+    WITHIN
+    WITHOUT
+    WORK
+    WRITE
+    WRITETEXT
+    XMLAGG
+    XMLATTRIBUTES
+    XMLBINARY
+    XMLCAST
+    XMLCOMMENT
+    XMLCONCAT
+    XMLDOCUMENT
+    XMLELEMENT
+    XMLEXISTS
+    XMLFOREST
+    XMLITERATE
+    XMLNAMESPACES
+    XMLPARSE
+    XMLPI
+    XMLQUERY
+    XMLSERIALIZE
+    XMLTABLE
+    XMLTEXT
+    XMLVALIDATE
+    YEAR
+    ZONE
+  ]
+  
+  KEYWORDS_SET = Set.new KEYWORDS
+end