Add support for formatting EffectiveModelContext
[mdsal.git] / yanglib / mdsal-yanglib-rfc8525 / src / main / java / org / opendaylight / mdsal / yanglib / rfc8525 / YangLibrarySupport.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.rfc8525;
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.eclipse.jdt.annotation.NonNullByDefault;
18 import org.opendaylight.binding.runtime.api.BindingRuntimeGenerator;
19 import org.opendaylight.binding.runtime.api.DefaultBindingRuntimeContext;
20 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
21 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTreeFactory;
22 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
23 import org.opendaylight.mdsal.binding.dom.codec.api.BindingIdentityCodec;
24 import org.opendaylight.mdsal.yanglib.api.SchemaContextResolver;
25 import org.opendaylight.mdsal.yanglib.api.YangLibSupport;
26 import org.opendaylight.mdsal.yanglib.api.YangLibraryContentBuilder;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.$YangModuleInfoImpl;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.ModulesState;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibrary;
30 import org.opendaylight.yangtools.rfc8528.data.api.MountPointContextFactory;
31 import org.opendaylight.yangtools.rfc8528.data.api.MountPointIdentifier;
32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
33 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
34 import org.opendaylight.yangtools.yang.common.QName;
35 import org.opendaylight.yangtools.yang.common.Revision;
36 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
37 import org.opendaylight.yangtools.yang.model.parser.api.YangParserException;
38 import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
39 import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
40 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
41
42 @Beta
43 @NonNullByDefault
44 @Singleton
45 public final class YangLibrarySupport implements YangLibSupport {
46     private static final Revision REVISION = YangLibrary.QNAME.getRevision().orElseThrow();
47
48     private final BindingDataObjectCodecTreeNode<YangLibrary> codec;
49     @SuppressWarnings("deprecation")
50     private final BindingDataObjectCodecTreeNode<ModulesState> legacyCodec;
51     private final BindingIdentityCodec identityCodec;
52     private final EffectiveModelContext context;
53     private final BindingCodecTree codecTree;
54
55     @Inject
56     public YangLibrarySupport(final YangParserFactory parserFactory, final BindingRuntimeGenerator generator,
57             final BindingCodecTreeFactory codecFactory) throws YangParserException, IOException {
58         final YangModuleInfo yangLibModule = $YangModuleInfoImpl.getInstance();
59
60         context = parserFactory.createParser()
61                 .addLibSources(Collections2.transform(yangLibModule.getImportedModules(),
62                     YangLibrarySupport::createSource))
63                 .addSource(createSource(yangLibModule))
64                 .buildEffectiveModel();
65         codecTree = codecFactory.create(DefaultBindingRuntimeContext.create(
66             generator.generateTypeMapping(context), SimpleStrategy.INSTANCE));
67
68         this.identityCodec = codecTree.getIdentityCodec();
69         this.codec = verifyNotNull(codecTree.getSubtreeCodec(InstanceIdentifier.create(YangLibrary.class)));
70         this.legacyCodec = verifyNotNull(codecTree.getSubtreeCodec(InstanceIdentifier.create(ModulesState.class)));
71     }
72
73     @Override
74     public MountPointContextFactory createMountPointContextFactory(final MountPointIdentifier mountId,
75             final SchemaContextResolver resolver) {
76         return new MountPointContextFactoryImpl(mountId, resolver, context, identityCodec, codec, legacyCodec);
77     }
78
79     @Override
80     public Revision implementedRevision() {
81         return REVISION;
82     }
83
84     @Override
85     public YangLibraryContentBuilder newContentBuilder() {
86         return new YangLibraryContentBuilderImpl(codecTree);
87     }
88
89     private static YangTextSchemaSource createSource(final YangModuleInfo info) {
90         final QName name = info.getName();
91         return YangTextSchemaSource.delegateForByteSource(
92             RevisionSourceIdentifier.create(name.getLocalName(), name.getRevision()), info.getYangTextByteSource());
93     }
94 }