yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/impl/util/YangSourceFromCapabilitiesResolver.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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/eplv10.html
   7  */
   8 package org.opendaylight.yangtools.yang.parser.impl.util;
   9
  10 import java.io.InputStream;
  11
  12 import org.opendaylight.yangtools.yang.common.QName;
  13 import org.opendaylight.yangtools.yang.model.util.repo.SchemaSourceProvider;
  14 import org.opendaylight.yangtools.yang.model.util.repo.SchemaSourceProviders;
  15 import org.opendaylight.yangtools.yang.model.util.repo.SourceIdentifier;
  16
  17 import com.google.common.base.Optional;
  18
  19 public final class YangSourceFromCapabilitiesResolver extends YangSourceContextResolver {
  20
  21     private final Iterable<QName> capabilities;
  22
  23     public YangSourceFromCapabilitiesResolver(final Iterable<QName> capabilities,
  24             final SchemaSourceProvider<InputStream> schemaSourceProvider) {
  25         super(SchemaSourceProviders.toAdvancedSchemaSourceProvider(schemaSourceProvider));
  26         this.capabilities = capabilities;
  27     }
  28
  29     @Override
  30     public YangSourceContext resolveContext() {
  31         for (QName capability : capabilities) {
  32             resolveCapability(capability);
  33         }
  34         return createSourceContext();
  35     }
  36
  37     private void resolveCapability(final QName capability) {
  38         super.resolveSource(capability.getLocalName(), Optional.fromNullable(capability.getFormattedRevision()));
  39     }
  40
  41     @Override
  42     public Optional<YangModelDependencyInfo> getDependencyInfo(final SourceIdentifier identifier) {
  43         Optional<InputStream> source = getSchemaSource(identifier);
  44         if (source.isPresent()) {
  45             return Optional.of(YangModelDependencyInfo.fromInputStream(source.get()));
  46         }
  47         return Optional.absent();
  48     }
  49
  50     private Optional<InputStream> getSchemaSource(final SourceIdentifier identifier) {
  51         return getSourceProvider().getSchemaSource(identifier.getName(),
  52                 Optional.fromNullable(identifier.getRevision()));
  53     }
  54
  55 }