restconf/restconf-openapi/src/main/java/org/opendaylight/restconf/openapi/model/NodeSchemaEntity.java

   1 /*
   2  * Copyright (c) 2024 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.restconf.openapi.model;
   9
  10 import static org.opendaylight.restconf.openapi.model.PropertyEntity.isSchemaNodeMandatory;
  11 import static org.opendaylight.restconf.openapi.util.RestDocgenUtil.widthList;
  12
  13 import com.fasterxml.jackson.core.JsonGenerator;
  14 import java.io.IOException;
  15 import java.util.HashMap;
  16 import java.util.List;
  17 import org.eclipse.jdt.annotation.NonNull;
  18 import org.eclipse.jdt.annotation.Nullable;
  19 import org.opendaylight.restconf.openapi.impl.DefinitionNames;
  20 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  21 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  22 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  23 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  24 import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
  25
  26 public final class NodeSchemaEntity extends SchemaEntity {
  27
  28     public NodeSchemaEntity(final @NonNull SchemaNode value, final @NonNull String title,
  29             final @Nullable String discriminator, final @NonNull String type,
  30             final @NonNull SchemaInferenceStack context, final @NonNull String parentName, final boolean isParentConfig,
  31             final @NonNull DefinitionNames definitionNames, final @NonNull Integer width) {
  32         super(value, title, discriminator, type, context, parentName, isParentConfig, definitionNames, width);
  33     }
  34
  35     @Override
  36     void generateProperties(final @NonNull JsonGenerator generator, final @NonNull List<String> required)
  37             throws IOException {
  38         final var childNodes = new HashMap<String, DataSchemaNode>();
  39         final var dataSchemaNodes = widthList((DataNodeContainer) value(), width);
  40         for (final var childNode : dataSchemaNodes) {
  41             childNodes.put(childNode.getQName().getLocalName(), childNode);
  42         }
  43         final boolean isValueConfig = ((DataSchemaNode) value()).isConfiguration();
  44
  45         for (final var childNode : childNodes.values()) {
  46             if (shouldBeAddedAsProperty(childNode, isValueConfig)) {
  47                 new PropertyEntity(childNode, generator, stack(), required, parentName() + "_"
  48                     + value().getQName().getLocalName(), isValueConfig, definitionNames(), width);
  49             }
  50         }
  51     }
  52
  53     private static boolean shouldBeAddedAsProperty(final DataSchemaNode childNode, final boolean isValueConfig) {
  54         final boolean isChildNodeConfig = childNode.isConfiguration();
  55         if (childNode instanceof ContainerSchemaNode) {
  56             return isChildNodeConfig || isSchemaNodeMandatory(childNode) || !isValueConfig;
  57         }
  58         return isChildNodeConfig || !isValueConfig;
  59     }
  60 }