cn.msuno.swagger.spring.boot.autoconfigure.mappers.VendorExtensionsMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of swagger-spring-boot-starter Show documentation

Project for Spring Boot

There is a newer version: 2.5.2

Show newest version

/*
 *
 *  Copyright 2015-2018 the original author or 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 cn.msuno.swagger.spring.boot.autoconfigure.mappers;

import static com.google.common.base.Strings.isNullOrEmpty;
import static com.google.common.collect.FluentIterable.from;
import static com.google.common.collect.Maps.newHashMap;
import static com.google.common.collect.Maps.newTreeMap;

import java.util.List;
import java.util.Map;

import org.mapstruct.Mapper;

import com.google.common.base.Function;

import springfox.documentation.service.ListVendorExtension;
import springfox.documentation.service.ObjectVendorExtension;
import springfox.documentation.service.StringVendorExtension;
import springfox.documentation.service.VendorExtension;

@Mapper
public class VendorExtensionsMapper {

  public Map mapExtensions(List from) {
    Map extensions = newTreeMap();
    Iterable listExtensions = from(from)
        .filter(ListVendorExtension.class);
    for (ListVendorExtension each : listExtensions) {
      extensions.put(each.getName(), each.getValue());
    }
    Iterable> objectExtensions = from(from)
        .filter(ObjectVendorExtension.class)
        .transform(toExtensionMap());
    for (Map each : objectExtensions) {
      extensions.putAll(each);
    }
    Iterable propertyExtensions = from(from)
        .filter(StringVendorExtension.class);
    for (StringVendorExtension each : propertyExtensions) {
      extensions.put(each.getName(), each.getValue());
    }
    return extensions;
  }

  private Function> toExtensionMap() {
    return new Function>() {
      @Override
      public Map apply(ObjectVendorExtension input) {
        if (!isNullOrEmpty(input.getName())) {
          Map map = newHashMap();
          map.put(input.getName(), mapExtensions(input.getValue()));
          return map;
        }
        return propertiesAsMap(input);
      }
    };
  }

  private Map propertiesAsMap(ObjectVendorExtension input) {
    Map properties = newHashMap();
    Iterable stringExtensions = from(input.getValue()).filter(StringVendorExtension.class);
    for (StringVendorExtension property : stringExtensions) {
      properties.put(property.getName(), property.getValue());
    }
    Iterable objectExtensions = from(input.getValue()).filter(ObjectVendorExtension.class);
    for (ObjectVendorExtension property : objectExtensions) {
      properties.put(property.getName(), mapExtensions(property.getValue()));
    }
    return properties;
  }
}