Contents

require 'pg_query'

module PgObjects
  ##
  # Reads directives from SQL-comments
  #
  #    --!depends_on [name_of_dependency]
  #
  # name_of_dependency: short or full name of object as well as object_name
  #
  #    --!multistatement
  #
  # use when there are more than one SQL command in object file
  class Parser
    ROUTES = [
      ['DefineStmt', 'defnames', 0, 'String', 'str'],
      ['CreateFunctionStmt', 'funcname', 0, 'String', 'str'],
      %w[CreateTrigStmt trigname],
      %w[CreateEventTrigStmt trigname],
      %w[CompositeTypeStmt typevar RangeVar relname],
      %w[ViewStmt view RangeVar relname],
      ['CreateConversionStmt', 'conversion_name', 0, 'String', 'str'],
      %w[CreateTableAsStmt into IntoClause rel RangeVar relname],
      ['CreateOpClassStmt', 'opclassname', 0, 'String', 'str']
    ].freeze

    class << self
      def fetch_directives(text)
        {
          depends_on: fetch_dependencies(text),
          multistatement: fetch_multistatement(text)
        }
      end

      def fetch_object_name(text)
        parsed = PgQuery.parse(text).tree.dig(0, 'RawStmt', 'stmt')
        ROUTES.map { |route| parsed.dig(*route) }.compact[0]
      rescue PgQuery::ParseError, NoMethodError
        nil
      end

      private

      def fetch_dependencies(text)
        text.split("\n").select { |ln| ln =~ /^(--|#)!/ }.map { |ln| ln.split(' ')[1] if ln =~ /!depends_on/ }.compact
      end

      def fetch_multistatement(text)
        text.split("\n").select { |ln| ln =~ /^(--|#)!/ }.select { |ln| ln =~ /!multistatement/ }.present?
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
pg_objects-0.4.2	lib/pg_objects/parser.rb