io.sundr.swagger.experimental.SwaggerRepository Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sundr-swagger Show documentation
There is a newer version: 1.14.0
/*
 *      Copyright 2018 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 io.sundr.swagger.experimental;

import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

import io.sundr.model.ClassRef;
import io.sundr.model.TypeDef;
import io.sundr.model.TypeRef;

public class SwaggerRepository {

  private final Map apis = new HashMap();
  private final Map models = new HashMap();

  public TypeDef registerApi(TypeDef api) {
    if (api != null) {
      apis.put(api.getFullyQualifiedName(), api);
    }
    return api;
  }

  public Set getApis() {
    return Collections.unmodifiableSet(new LinkedHashSet(apis.values()));
  }

  public TypeDef getApi(TypeRef type) {
    if (type instanceof ClassRef) {
      return apis.get(((ClassRef) type).getFullyQualifiedName());
    }
    return null;
  }

  public boolean isApi(TypeDef type) {
    return type != null && apis.containsKey(type.getFullyQualifiedName());
  }

  public boolean isApi(ClassRef classRef) {
    return classRef != null && apis.containsKey(classRef.getFullyQualifiedName());
  }

  public boolean isApi(TypeRef type) {
    if (type instanceof ClassRef) {
      return isApi(((ClassRef) type));
    }
    return false;
  }

  public TypeDef registerModel(TypeDef model) {
    if (model != null) {
      models.put(model.getFullyQualifiedName(), model);
    }
    return model;
  }

  public Set getModels() {
    return Collections.unmodifiableSet(new LinkedHashSet(models.values()));
  }

  public TypeDef getModel(TypeRef type) {
    if (type instanceof ClassRef) {
      return models.get(((ClassRef) type).getFullyQualifiedName());
    }
    return null;
  }

  public boolean isModel(TypeDef type) {
    return type != null && models.containsKey(type.getFullyQualifiedName());
  }

  public boolean isModel(ClassRef classRef) {
    return classRef != null && models.containsKey(classRef.getFullyQualifiedName());
  }

  public boolean isModel(TypeRef type) {
    if (type instanceof ClassRef) {
      return isModel(((ClassRef) type));
    }
    return false;
  }

  public void clear() {
    apis.clear();
    models.clear();
  }
}