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

Go to download
/*
 *      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 io.sundr.codegen.model.ClassRef;
import io.sundr.codegen.model.TypeDef;
import io.sundr.codegen.model.TypeRef;

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

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).getDefinition().getFullyQualifiedName());
        }
        return null;
    }

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

    public boolean isApi(TypeRef type) {
        if (type instanceof ClassRef) {
            return isApi(((ClassRef)type).getDefinition());
        }
        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).getDefinition().getFullyQualifiedName());
        }
        return null;
    }

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

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

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