io.github.swagger2markup.extensions.DynamicContentExtension.java Source code

Introduction

Here is the source code for io.github.swagger2markup.extensions.DynamicContentExtension.java

Source

/*
 * Copyright 2016 Robert Winkler
 *
 * 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.github.swagger2markup.extensions;

import com.google.common.collect.Lists;
import com.google.common.collect.Ordering;
import io.github.swagger2markup.Swagger2MarkupConverter;
import io.github.swagger2markup.markup.builder.MarkupLanguage;
import io.github.swagger2markup.spi.ContentContext;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.nio.file.DirectoryStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

public class DynamicContentExtension extends ContentExtension {

    private static final Logger logger = LoggerFactory.getLogger(DynamicContentExtension.class);

    DynamicContentExtension(Swagger2MarkupConverter.Context globalContext, ContentContext contentContext) {
        super(globalContext, contentContext);
    }

    /**
     * Builds extension sections
     *
     * @param extensionMarkupLanguage the MarkupLanguage of the snippets content
     * @param contentPath the path where the content files reside
     * @param prefix      extension file prefix
     * @param levelOffset import markup level offset
     */
    public void extensionsSection(MarkupLanguage extensionMarkupLanguage, Path contentPath, final String prefix,
            int levelOffset) {
        final Collection<String> filenameExtensions = globalContext.getConfig().getMarkupLanguage()
                .getFileNameExtensions().stream().map(fileExtension -> StringUtils.stripStart(fileExtension, "."))
                .collect(Collectors.toList());

        DirectoryStream.Filter<Path> filter = entry -> {
            String fileName = entry.getFileName().toString();
            return fileName.startsWith(prefix) && FilenameUtils.isExtension(fileName, filenameExtensions);
        };

        try (DirectoryStream<Path> extensionFiles = Files.newDirectoryStream(contentPath, filter)) {

            if (extensionFiles != null) {
                List<Path> extensions = Lists.newArrayList(extensionFiles);
                Collections.sort(extensions, Ordering.natural());

                for (Path extension : extensions) {
                    importContent(extension, (reader) -> contentContext.getMarkupDocBuilder().importMarkup(reader,
                            extensionMarkupLanguage, levelOffset));
                }
            }
        } catch (IOException e) {
            if (logger.isDebugEnabled())
                logger.debug("Failed to read extension files from directory {}", contentPath);
        }
    }

}