BUG-4638: Convert to using new types

[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / reactor / CrossSourceStatementReactor.java

diff --git a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/CrossSourceStatementReactor.java b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/CrossSourceStatementReactor.java
index 06ae43adce10e9ef5cb88bf33dcecdcd16a1889d..c206e0e6375c80b65f6f67062eff1ca23eed1f03 100644 (file)
--- a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/CrossSourceStatementReactor.java
+++ b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/CrossSourceStatementReactor.java
@@ -7,8 +7,23 @@
  */
 package org.opendaylight.yangtools.yang.parser.stmt.reactor;
 
+import java.io.IOException;
+
+import com.google.common.io.ByteSource;
+import java.util.Set;
+import java.io.FileNotFoundException;
+import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
+import java.util.HashMap;
+import java.util.Collections;
+import org.opendaylight.yangtools.yang.model.api.Module;
+import java.io.File;
+import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import java.io.InputStream;
+import java.util.List;
+import java.util.Collection;
+import org.opendaylight.yangtools.yang.parser.spi.validation.ValidationBundlesNamespace.ValidationBundleType;
 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.EffectiveSchemaContext;
-
 import com.google.common.collect.ImmutableMap;
 import java.util.EnumMap;
 import java.util.Map;
@@ -21,9 +36,16 @@ import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
 public class CrossSourceStatementReactor {
 
     private final Map<ModelProcessingPhase,StatementSupportBundle> supportedTerminology;
+    private final Map<ValidationBundleType,Collection<?>> supportedValidation;
 
     CrossSourceStatementReactor(Map<ModelProcessingPhase, StatementSupportBundle> supportedTerminology) {
         this.supportedTerminology = ImmutableMap.copyOf(supportedTerminology);
+        this.supportedValidation = ImmutableMap.of();
+    }
+
+    CrossSourceStatementReactor(Map<ModelProcessingPhase, StatementSupportBundle> supportedTerminology, Map<ValidationBundleType,Collection<?>> supportedValidation) {
+        this.supportedTerminology = ImmutableMap.copyOf(supportedTerminology);
+        this.supportedValidation = ImmutableMap.copyOf(supportedValidation);
     }
 
     public static final Builder builder() {
@@ -37,15 +59,24 @@ public class CrossSourceStatementReactor {
     public static class Builder implements org.opendaylight.yangtools.concepts.Builder<CrossSourceStatementReactor>{
 
         final Map<ModelProcessingPhase,StatementSupportBundle> bundles = new EnumMap<>(ModelProcessingPhase.class);
+        final Map<ValidationBundleType,Collection<?>> validationBundles = new EnumMap<>(ValidationBundleType.class);
 
         public Builder setBundle(ModelProcessingPhase phase,StatementSupportBundle bundle) {
             bundles.put(phase, bundle);
             return this;
         }
 
+
+        public Builder setValidationBundle(
+                ValidationBundleType type,
+                Collection<?> validationBundle) {
+            validationBundles.put(type,validationBundle);
+            return this;
+        }
+
         @Override
         public CrossSourceStatementReactor build() {
-            return new CrossSourceStatementReactor(bundles);
+            return new CrossSourceStatementReactor(bundles, validationBundles);
         }
 
     }
@@ -55,13 +86,19 @@ public class CrossSourceStatementReactor {
         private final BuildGlobalContext context;
 
         public BuildAction() {
-            this.context = new BuildGlobalContext(supportedTerminology);
+            this.context = new BuildGlobalContext(supportedTerminology, supportedValidation);
         }
 
         public void addSource(StatementStreamSource source) {
             context.addSource(source);
         }
 
+        public void addSources(StatementStreamSource... sources) {
+            for (StatementStreamSource source : sources) {
+                context.addSource(source);
+            }
+        }
+
         public EffectiveModelContext build() throws SourceException, ReactorException {
             return context.build();
         }
@@ -70,7 +107,49 @@ public class CrossSourceStatementReactor {
             return context.buildEffective();
         }
 
-    }
+        public SchemaContext buildEffective(Collection<ByteSource> yangByteSources) throws SourceException, ReactorException, IOException {
+
+            for(ByteSource yangByteSource : yangByteSources) {
+                addSource(new YangStatementSourceImpl(yangByteSource.openStream()));
+            }
 
+            return buildEffective();
+        }
+
+        public SchemaContext buildEffective(List<InputStream> yangInputStreams) throws SourceException, ReactorException {
+
+            for(InputStream yangInputStream : yangInputStreams) {
+                addSource(new YangStatementSourceImpl(yangInputStream));
+            }
+
+            return buildEffective();
+        }
+
+        public Map<File, Module> buildEffectiveMappedToSource(
+                List<File> yangFiles) throws SourceException, ReactorException,
+                FileNotFoundException {
 
+            if (yangFiles == null || yangFiles.isEmpty()) {
+                return Collections.emptyMap();
+            }
+
+            Map<String, File> pathToFile = new HashMap<>();
+            Map<File, Module> sourceFileToModule = new HashMap<>();
+
+            for (File yangFile : yangFiles) {
+                addSource(new YangStatementSourceImpl(new NamedFileInputStream(
+                        yangFile, yangFile.getPath())));
+                pathToFile.put(yangFile.getPath(), yangFile);
+            }
+
+            EffectiveSchemaContext schema = buildEffective();
+            Set<Module> modules = schema.getModules();
+            for (Module module : modules) {
+                sourceFileToModule.put(
+                        pathToFile.get(module.getModuleSourcePath()), module);
+            }
+
+            return sourceFileToModule;
+        }
+    }
 }