acts_as_doc 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12a796bab9820e42c34cc94ad2029dff43521113e20a8686aed667c72a7f1c86
-  data.tar.gz: 7924b82d0461dd99417ed2757e653de560e2a884430586436d931e34f8c85c7d
+  metadata.gz: 84a468753835af1c1f73d8edf468234bd164283c67b18a0f1a4281bea905fc86
+  data.tar.gz: 498222082962d5a7376994cdde490ea64d03440a1a1487b568a9fc9cad6c0d38
 SHA512:
-  metadata.gz: cbc1030672261ae66675f1ebe0d6c7dfa6780f39adc32088a4375e3b51b07539ffbc09473b11a65f66749a8b38b692734637db6f5e3c8dfc5a403bddcb41a2f8
-  data.tar.gz: 3c615324dcbe8ef60ded241251633f40a92a86f4ce523a0eb08967f51e34b1725a86ff6074c0a5c9fecef5285a2a301ee6147f79b0dcfeb01f81d434faeab53d
+  metadata.gz: 7965992ef4dc0b7464b1742a261a668d1793343bee96d876d3b9771328d54dc97740db4df2fafe9f2b8848c2eed7e45d931503e6c19a896a846cbfc5cb2e4e16
+  data.tar.gz: 951e8fc0ac3c327979e8bf2d53c32dcaef6fda75aafb3b7e4e2815695ff72261feda7234ec64f40ee99234156194f5a1973cb8cd141dcdc4879c69cec0d06821

data/acts_as_doc.gemspec

@@ -1,7 +1,7 @@
 # rubocop:disable all
 Gem::Specification.new do |s|
   s.name        = "acts_as_doc"
-  s.version     = "1.0.1"
+  s.version     = "1.1.0"
   s.summary     = "Generate swagger response doc"
   s.description = "Add swagger comment to any ruby file for generating swagger response doc struct"
   s.authors     = ["alex.zang"]

data/lib/acts_as_doc/response_parser.rb

@@ -16,6 +16,17 @@ module ActsAsDoc
 
     REF_FLAG = '$'
 
+    TYPE_MAPPER = {
+      integer: 'integer',
+      string: 'string',
+      decimal: 'number',
+      text: 'string',
+      geography: 'string',
+      json: 'object',
+      jsonb: 'object',
+      array: 'array'
+    }.freeze
+
     # 处理注释
     #
     # @param token [String] one line of comments
@@ -25,24 +36,36 @@ module ActsAsDoc
     end
 
     # Make props recursion
+    #
     # @example
     #   >> ActsAsDoc::ResponseParser.props_recursion!({}, '$a.$b.c', 'c', 'd')
     #   # => {'a' => }
     #
     # @return [Hash] 处理过的hash
     # rubocop:disable all
-    def self.props_recursion!(hash, name, type, desc)
+    def self.props_recursion!(hash, name, type, desc = '', klass = nil)
       arr = name.split('.')
       name = arr.shift.sub(/^\$/, '')
 
       hash[name] = {} unless hash.key?(name)
 
       if arr.empty?
-        hash[name].merge!(type: type, description: desc)
+        if klass && (matches = desc.match(/^\(([a-zA-Z]+)\)$/))
+          columns = Object.const_get(klass).columns.map do |column|
+            [ column.name.to_s, [TYPE_MAPPER[column.type], column.comment] ]
+          end.to_h
+
+          matches[1].split(',').each do |attr_name|
+            c_type, c_desc = columns[attr_name]
+            self.props_recursion!(hash, "$#{name}.#{attr_name}", c_type, c_desc)
+          end
+        else
+          hash[name].merge!(type: type, description: desc)
+        end
       else
         nest_hash = hash[name]
         nest_hash[:properties] = {} unless nest_hash.key?(:properties)
-        self.props_recursion!(nest_hash[:properties], arr.join('.'), type, desc)
+        self.props_recursion!(nest_hash[:properties], arr.join('.'), type, desc, klass)
       end
 
       hash
@@ -62,16 +85,24 @@ module ActsAsDoc
 
         if tag == '@prop'
           desc = arr[3..] ? arr[3..].join(' ') : ''
-          matches = type.match(/\[(?<type>#{SUPPORT_TYPES.join('|')})\]/)
+
+          matches = type.match(/\[(?<type>[a-zA-Z<>]+)\]/)
           type = matches ? matches[:type] : 'string'
 
+          klass_matches = type.match(/(?<type>[a-zA-Z]+)<(?<klass>[a-zA-Z]+)>/)
+          klass = nil
+          if klass_matches
+            type = SUPPORT_TYPES.include?(klass_matches[:type]) ? klass_matches[:type] : 'string'
+            klass = klass_matches[:klass]
+          end
+
           if name.start_with?(':')
             name = name.sub(/^:/, '')
             schema[name] = { type: type, description: desc }
           end
 
           if name.start_with?(REF_FLAG)
-            self.class.props_recursion!(schema, name, type, desc)
+            self.class.props_recursion!(schema, name, type, desc, klass)
           end
         end
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: acts_as_doc
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - alex.zang
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-12-09 00:00:00.000000000 Z
+date: 2022-12-12 00:00:00.000000000 Z
 dependencies: []
 description: Add swagger comment to any ruby file for generating swagger response
   doc struct