yaml mapping reader

Author: hauner

src/main/groovy/com/github/hauner/openapi/spring/generatr/TypeMappingReader.groovy

@@ -0,0 +1,128 @@
+/*
+ * Copyright 2019 the original authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.github.hauner.openapi.spring.generatr
+
+import com.github.hauner.openapi.spring.generatr.mapping.EndpointTypeMapping
+import com.github.hauner.openapi.spring.generatr.mapping.ParameterTypeMapping
+import com.github.hauner.openapi.spring.generatr.mapping.ResponseTypeMapping
+import com.github.hauner.openapi.spring.generatr.mapping.TypeMapping
+import org.yaml.snakeyaml.Yaml
+
+import java.util.regex.Matcher
+import java.util.regex.Pattern
+
+/**
+ *  Reader for mapping yaml.
+ *
+ *  @author Martin Hauner
+ */
+class TypeMappingReader {
+    private Pattern GENERIC_INLINE = ~/(.+?)<(.+?)>/
+
+    List<?> read (String typeMappings) {
+        if (typeMappings == null) {
+            return []
+        }
+
+        String mapping = typeMappings
+        if (isFileName (typeMappings)) {
+            mapping = new File(typeMappings).text
+        }
+
+        Yaml yaml = new Yaml()
+        Map props = yaml.load (mapping)
+        parse (props)
+    }
+
+    private List<?> parse (Map<String, ?> props) {
+        //def version = props.get ('openapi-generatr-spring')
+
+        def root = props.get ('map') as Map<String, ?>
+
+        def mappings = readTypeMappings (root)
+
+        def paths = root.get ('paths') as Map<String, ?>
+        paths.each {
+            def epm = new EndpointTypeMapping(path: it.key)
+            epm.typeMappings = readTypeMappings (it.value as Map<String, ?>)
+            mappings.add (epm)
+        }
+
+        mappings
+    }
+
+    private List<?> readTypeMappings (Map<String, ?> root) {
+        def mappings = []
+
+        def types = root.get ('types') as List<Map<String, ?> >
+        types.each { Map<String, ?> it ->
+            mappings.add (readTypMapping (it))
+        }
+
+        def responses = root.get ('responses') as List<Map<String, ?> >
+        responses.each {
+            mappings.add (readResponseTypeMapping (it))
+        }
+
+        def parameters = root.get ('parameters') as List<Map<String, ?> >
+        parameters.each {
+            mappings.add (readParameterTypeMapping (it))
+        }
+
+        return mappings
+    }
+
+    private ParameterTypeMapping readParameterTypeMapping (Map<String, ?> source) {
+        def name = source.name
+        def mapping = readTypMapping (source)
+        new ParameterTypeMapping (parameterName: name, mapping: mapping)
+    }
+
+    private ResponseTypeMapping readResponseTypeMapping (Map<String, ?> source) {
+        def content = source.content
+        def mapping = readTypMapping (source)
+        new ResponseTypeMapping(contentType: content, mapping: mapping)
+    }
+
+    private TypeMapping readTypMapping (Map<String, ?> source) {
+        Matcher matcher = source.to =~ GENERIC_INLINE
+
+        String from = source.from
+        String to = source.to
+        List<String> generics = []
+
+        // has inline generics
+        if (matcher.find ()) {
+            to = matcher.group (1)
+            generics = matcher
+                .group (2)
+                .split (',')
+                .collect { it.trim () }
+
+        // has explicit generic list
+        } else if (source.containsKey ('generics')) {
+            generics = source.generics as List
+        }
+
+        new TypeMapping (sourceTypeName: from, targetTypeName: to, genericTypeNames: generics)
+    }
+
+    private boolean isFileName (String name) {
+        name.endsWith ('.yaml') || name.endsWith ('.yml')
+    }
+
+}

src/main/groovy/com/github/hauner/openapi/spring/generatr/mapping/ResponseTypeMapping.groovy

@@ -32,12 +32,19 @@ class ResponseTypeMapping {
     /**
      * The OpenAPI schema type that should be mapped to the {@link #targetTypeName} java type.
      */
+    @Deprecated // use mapping
     String sourceTypeName
 
     /**
      * The fully qualified java type name that will be used for {@link #sourceTypeName} in an
      * {@link #contentType} response.
      */
+    @Deprecated // use mapping
     String targetTypeName
 
+    /**
+     * Type mapping valid only for responses with {@link #contentType}.
+     */
+    TypeMapping mapping
+
 }

src/test/groovy/com/github/hauner/openapi/learn/RegexSpec.groovy

@@ -51,4 +51,14 @@ class RegexSpec extends Specification {
         expect:
         "prefix\nXXXX\npostfix" ==~ /(?s)prefix(.+?)postfix/
     }
+
+    void "extract generic types" () {
+        Pattern p = ~/.+?<(.+?)>/
+        Matcher m = "mapping.Bar<java.lang.String, java.lang.Boolean>" =~ p
+
+        expect:
+        m.find ()
+        m.group (1) == "java.lang.String, java.lang.Boolean"
+    }
+
 }