gerrit.o-ran-sc Code Review - smo/teiv.git/blob

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2024 Ericsson
   4  *  Modifications Copyright (C) 2024 OpenInfra Foundation Europe
   5  *  ================================================================================
   6  *  Licensed under the Apache License, Version 2.0 (the "License");
   7  *  you may not use this file except in compliance with the License.
   8  *  You may obtain a copy of the License at
   9  *
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *
  18  *  SPDX-License-Identifier: Apache-2.0
  19  *  ============LICENSE_END=========================================================
  20  */
  21 package org.oran.smo.yangtools.parser.model.statements.ietf;
  22
  23 import java.util.List;
  24 import java.util.Map;
  25
  26 import org.oran.smo.yangtools.parser.model.statements.AbstractStatement;
  27 import org.oran.smo.yangtools.parser.model.statements.AbstractStatementClassSupplier;
  28
  29 /**
  30  * Class supplier for extensions defined in various IETF RFCs.
  31  *
  32  * @author Mark Hollmann
  33  */
  34 public class IetfExtensionsClassSupplier extends AbstractStatementClassSupplier {
  35
  36     @Override
  37     public Map<String, List<String>> getHandledStatements() {
  38         return CIETF.HANDLED_STATEMENTS;
  39     }
  40
  41     @Override
  42     public <T extends AbstractStatement> Class<T> getJavaClazzForStatement(final String moduleName,
  43             final String extStatementName) {
  44
  45         if (CIETF.IETF_YANG_SCHEMA_MOUNT_MODULE_NAME.equals(moduleName)) {
  46             return handleIetfYangSchemaMountExtensions(extStatementName);
  47         }
  48
  49         if (CIETF.IETF_YANG_METADATA_MODULE_NAME.equals(moduleName)) {
  50             return handleIetfYangMetadataExtensions(extStatementName);
  51         }
  52
  53         if (CIETF.IETF_NETCONF_ACM_MODULE_NAME.equals(moduleName)) {
  54             return handleIetfNetconfAcmExtensions(extStatementName);
  55         }
  56
  57         return null;
  58     }
  59
  60     private static final String IETF_EXTENSIONS_JAVA_PACKAGE = YIetfMountPoint.class.getPackage().getName();
  61     private static final String IETF_EXTENSIONS_JAVA_CLASS_PREFIX = "YIetf";
  62
  63     private <T extends AbstractStatement> Class<T> handleIetfYangSchemaMountExtensions(final String extStatementName) {
  64         return getJavaClazzForStatement(IETF_EXTENSIONS_JAVA_PACKAGE, IETF_EXTENSIONS_JAVA_CLASS_PREFIX, extStatementName);
  65     }
  66
  67     private <T extends AbstractStatement> Class<T> handleIetfYangMetadataExtensions(final String extStatementName) {
  68         return getJavaClazzForStatement(IETF_EXTENSIONS_JAVA_PACKAGE, IETF_EXTENSIONS_JAVA_CLASS_PREFIX, extStatementName);
  69     }
  70
  71     private <T extends AbstractStatement> Class<T> handleIetfNetconfAcmExtensions(final String extStatementName) {
  72         return getJavaClazzForStatement(IETF_EXTENSIONS_JAVA_PACKAGE, IETF_EXTENSIONS_JAVA_CLASS_PREFIX, extStatementName);
  73     }
  74 }