2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.yangtools.yang.validator;
11 import com.google.common.base.Preconditions;
13 import java.io.FileFilter;
14 import java.io.FileNotFoundException;
15 import java.io.IOException;
16 import java.util.ArrayList;
17 import java.util.Arrays;
18 import java.util.Collection;
19 import java.util.List;
21 import org.opendaylight.yangtools.yang.common.QName;
22 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
23 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
24 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
25 import org.opendaylight.yangtools.yang.parser.rfc6020.repo.YangStatementStreamSource;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
27 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
28 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
29 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
31 class SystemTestUtils {
33 static final FileFilter YANG_FILE_FILTER = file -> {
34 final String name = file.getName().toLowerCase();
35 return name.endsWith(".yang") && file.isFile();
38 static SchemaContext parseYangSources(final List<String> yangLibDirs, final List<String> yangTestFiles,
39 final Set<QName> supportedFeatures, final boolean recursiveSearch) throws ReactorException, IOException,
40 YangSyntaxErrorException {
42 * Current dir "." should be always present implicitly in the list of
43 * directories where dependencies are searched for
45 if (!yangLibDirs.contains(".")) {
49 final List<File> libFiles = new ArrayList<>();
50 for (final String yangLibDir : yangLibDirs) {
51 libFiles.addAll(getYangFiles(yangLibDir, recursiveSearch));
54 final List<File> testFiles = new ArrayList<>();
55 for (final String yangTestFile : yangTestFiles) {
56 testFiles.add(new File(yangTestFile));
59 return parseYangSources(supportedFeatures, testFiles, libFiles);
62 static SchemaContext parseYangSources(final Set<QName> supportedFeatures, final List<File> testFiles,
63 final List<File> libFiles) throws ReactorException, IOException, YangSyntaxErrorException {
64 final List<StatementStreamSource> testSources = getYangStatementSources(testFiles);
65 final List<StatementStreamSource> libSources = getYangStatementSources(libFiles);
66 return parseYangSources(testSources, libSources, supportedFeatures);
69 static SchemaContext parseYangSources(final List<StatementStreamSource> testSources,
70 final List<StatementStreamSource> libSources, final Set<QName> supportedFeatures) throws ReactorException {
71 Preconditions.checkArgument(testSources != null && !testSources.isEmpty(), "No yang sources");
73 final CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR.newBuild();
74 reactor.addSources(testSources);
75 reactor.addLibSources(libSources);
77 if (supportedFeatures != null) {
78 reactor.setSupportedFeatures(supportedFeatures);
81 return reactor.buildEffective();
84 private static List<StatementStreamSource> getYangStatementSources(final List<File> yangFiles)
85 throws IOException, YangSyntaxErrorException {
86 final List<StatementStreamSource> yangSources = new ArrayList<>(yangFiles.size());
87 for (File file : yangFiles) {
88 yangSources.add(YangStatementStreamSource.create(YangTextSchemaSource.forFile(file)));
93 private static Collection<File> getYangFiles(final String yangSourcesDirectoryPath, final boolean recursiveSearch)
94 throws FileNotFoundException {
95 final File testSourcesDir = new File(yangSourcesDirectoryPath);
96 if (!testSourcesDir.isDirectory()) {
97 throw new FileNotFoundException(String.format("%s no such directory", yangSourcesDirectoryPath));
100 return recursiveSearch ? searchYangFiles(testSourcesDir)
101 : Arrays.asList(testSourcesDir.listFiles(YANG_FILE_FILTER));
104 private static List<File> searchYangFiles(final File dir) {
105 Preconditions.checkNotNull(dir);
106 Preconditions.checkArgument(dir.isDirectory(), "File %s is not a directory", dir.getPath());
108 final List<File> yangFiles = new ArrayList<>();
109 for (final File file : dir.listFiles()) {
110 if (file.isDirectory()) {
111 yangFiles.addAll(searchYangFiles(file));
112 } else if (file.isFile() && file.getName().toLowerCase().endsWith(".yang")) {