org.yangcentral.yangkit.model.impl.codec.IdentityRefStringValueCodecImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yangkit-model-impl Show documentation
There is a newer version: 1.4.5
package org.yangcentral.yangkit.model.impl.codec;

import org.yangcentral.yangkit.base.ErrorCode;
import org.yangcentral.yangkit.common.api.FName;
import org.yangcentral.yangkit.common.api.QName;
import org.yangcentral.yangkit.model.api.codec.IdentityRefStringValueCodec;
import org.yangcentral.yangkit.model.api.codec.YangCodecException;
import org.yangcentral.yangkit.model.api.restriction.Restriction;
import org.yangcentral.yangkit.model.api.schema.ModuleId;
import org.yangcentral.yangkit.model.api.stmt.MainModule;
import org.yangcentral.yangkit.model.api.stmt.Module;
import org.yangcentral.yangkit.model.api.stmt.SubModule;
import org.yangcentral.yangkit.model.api.stmt.TypedDataNode;

import java.net.URI;
import java.util.Iterator;
import java.util.List;
import java.util.Optional;

public class IdentityRefStringValueCodecImpl extends ComplexStringValueCodecImpl implements IdentityRefStringValueCodec {
   public IdentityRefStringValueCodecImpl(TypedDataNode schemaNode) {
      super(schemaNode);
   }

   public QName deserialize(Restriction restriction, String input) throws YangCodecException {
      FName fName = new FName(input);
      Module curModule = this.getSchemaNode().getContext().getCurModule();
      String prefix = null;
      URI namespace = null;
      List mainModuleList;
      if (fName.getPrefix() == null) {
         if (curModule instanceof MainModule) {
            namespace = ((MainModule)curModule).getNamespace().getUri();
            prefix = ((MainModule)curModule).getPrefix().getArgStr();
         } else {
            SubModule sb = (SubModule)curModule;
            mainModuleList = sb.getBelongsto().getMainModules();
            if (mainModuleList.size() == 0) {
               throw new IllegalArgumentException(ErrorCode.INVALID_VALUE.getFieldName());
            }

            namespace = ((MainModule)mainModuleList.get(0)).getNamespace().getUri();
            prefix = sb.getBelongsto().getPrefix().getArgStr();
         }
      } else {
         prefix = fName.getPrefix();
         Optional moduleIdOp = curModule.findModuleByPrefix(fName.getPrefix());
         Module module;
         if (moduleIdOp.isPresent()) {
            ModuleId moduleId = (ModuleId)moduleIdOp.get();
            Optional moduleOp = this.getSchemaNode().getContext().getSchemaContext().getModule(moduleId);
            if (!moduleOp.isPresent()) {
               throw new IllegalArgumentException(ErrorCode.INVALID_VALUE.getFieldName());
            }

            module = (Module)moduleOp.get();

            assert module instanceof MainModule;

            namespace = ((MainModule)module).getNamespace().getUri();
         } else {
            mainModuleList = curModule.getContext().getSchemaContext().getModules();
            Iterator iterator = mainModuleList.iterator();

            while(iterator.hasNext()) {
               module = (Module)iterator.next();
               if (module.getSelfPrefix().equals(prefix)) {
                  namespace = module.getMainModule().getNamespace().getUri();
               }
            }

            if (namespace == null) {
               throw new IllegalArgumentException(ErrorCode.INVALID_VALUE.getFieldName());
            }
         }
      }

      QName qName = new QName(namespace, prefix, fName.getLocalName());
      if (!restriction.evaluated(qName)) {
         throw new YangCodecException(ErrorCode.INVALID_VALUE.getFieldName());
      } else {
         return qName;
      }
   }

   public String serialize(Restriction restriction, QName output) throws YangCodecException {
      if (!restriction.evaluated(output)) {
         throw new YangCodecException(ErrorCode.INVALID_VALUE.getFieldName());
      } else {
         return output.getQualifiedName();
      }
   }
}