yanglib/mdsal-yanglib-rfc7895/src/main/java/org/opendaylight/mdsal/yanglib/rfc7895/YangModuleLibrarySupport.java

   1 /*
   2  * Copyright (c) 2019 PANTHEON.tech s.r.o. 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/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.yanglib.rfc7895;
   9
  10 import static com.google.common.base.Verify.verifyNotNull;
  11
  12 import com.google.common.annotations.Beta;
  13 import com.google.common.collect.Collections2;
  14 import java.io.IOException;
  15 import javax.inject.Inject;
  16 import javax.inject.Singleton;
  17 import org.apache.aries.blueprint.annotation.service.Reference;
  18 import org.eclipse.jdt.annotation.NonNullByDefault;
  19 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
  20 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  21 import org.opendaylight.mdsal.binding.dom.codec.impl.BindingNormalizedNodeCodecRegistry;
  22 import org.opendaylight.mdsal.binding.generator.api.BindingRuntimeContext;
  23 import org.opendaylight.mdsal.binding.generator.api.BindingRuntimeGenerator;
  24 import org.opendaylight.mdsal.yanglib.api.SchemaContextResolver;
  25 import org.opendaylight.mdsal.yanglib.api.YangLibSupport;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.$YangModuleInfoImpl;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.ModulesState;
  28 import org.opendaylight.yangtools.rfc8528.data.api.MountPointContextFactory;
  29 import org.opendaylight.yangtools.rfc8528.data.api.MountPointIdentifier;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  31 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
  32 import org.opendaylight.yangtools.yang.common.QName;
  33 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  34 import org.opendaylight.yangtools.yang.model.parser.api.YangParserException;
  35 import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
  36 import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
  37 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
  38
  39 @Beta
  40 @NonNullByDefault
  41 @Singleton
  42 public final class YangModuleLibrarySupport implements YangLibSupport {
  43     private final BindingDataObjectCodecTreeNode<ModulesState> codec;
  44     private final EffectiveModelContext context;
  45
  46     @Inject
  47     public YangModuleLibrarySupport(final @Reference YangParserFactory parserFactory,
  48             final @Reference BindingRuntimeGenerator generator) throws YangParserException, IOException {
  49         final YangModuleInfo yangLibModule = $YangModuleInfoImpl.getInstance();
  50
  51         context = parserFactory.createParser()
  52                 .addLibSources(Collections2.transform(yangLibModule.getImportedModules(),
  53                     YangModuleLibrarySupport::createSource))
  54                 .addSource(createSource(yangLibModule))
  55                 .buildEffectiveModel();
  56         final BindingCodecTree codecTree = new BindingNormalizedNodeCodecRegistry(BindingRuntimeContext.create(
  57             generator.generateTypeMapping(context), SimpleStrategy.INSTANCE)).getCodecContext();
  58
  59         this.codec = verifyNotNull(codecTree.getSubtreeCodec(InstanceIdentifier.create(ModulesState.class)));
  60     }
  61
  62     @Override
  63     public MountPointContextFactory createMountPointContextFactory(final MountPointIdentifier mountId,
  64             final SchemaContextResolver resolver) {
  65         return new MountPointContextFactoryImpl(mountId, resolver, context, codec);
  66     }
  67
  68     private static YangTextSchemaSource createSource(final YangModuleInfo info) {
  69         final QName name = info.getName();
  70         return YangTextSchemaSource.delegateForByteSource(
  71             RevisionSourceIdentifier.create(name.getLocalName(), name.getRevision()), info.getYangTextByteSource());
  72     }
  73 }