/*
 * Copyright 2016-2018 shardingsphere.io.
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * </p>
 */

package io.shardingsphere.orchestration.internal.yaml.representer;

import org.yaml.snakeyaml.introspector.Property;
import org.yaml.snakeyaml.nodes.CollectionNode;
import org.yaml.snakeyaml.nodes.MappingNode;
import org.yaml.snakeyaml.nodes.Node;
import org.yaml.snakeyaml.nodes.NodeTuple;
import org.yaml.snakeyaml.nodes.ScalarNode;
import org.yaml.snakeyaml.nodes.SequenceNode;
import org.yaml.snakeyaml.nodes.Tag;
import org.yaml.snakeyaml.representer.Represent;
import org.yaml.snakeyaml.representer.Representer;

import java.util.Collection;
import java.util.HashSet;

/**
 * Datasource parameter representer.
 *
 * @author panjuan
 */
public class DataSourceParameterRepresenter extends Representer {
    
    private static Collection<String> reservedNodeNames = new HashSet<>();
    
    static {
        reservedNodeNames.add("password");
    }
    
    public DataSourceParameterRepresenter() {
        super();
        nullRepresenter = new DataSourceParameterRepresenter.NullRepresent();
    }
    
    @Override
    protected NodeTuple representJavaBeanProperty(final Object javaBean, final Property property, final Object propertyValue, final Tag customTag) {
        NodeTuple tuple = super.representJavaBeanProperty(javaBean, property, propertyValue, customTag);
        Node valueNode = tuple.getValueNode();
        Node keyNode = tuple.getKeyNode();
        
        if (isWantedNodeTuple(tuple.getKeyNode())) {
            return tuple;
        }
        
        if (isUnwantedNodeTuple(tuple.getValueNode())) {
            return null;
        }
        
        if (keyNode instanceof ScalarNode && ((ScalarNode) keyNode).getValue().equals("password")) {
            return tuple;
        }
        if (Tag.NULL.equals(valueNode.getTag())) {
            return null;
        }
        if (valueNode instanceof CollectionNode) {
            if (Tag.SEQ.equals(valueNode.getTag()) && ((SequenceNode) valueNode).getValue().isEmpty()) {
                return null;
            }
            if (Tag.MAP.equals(valueNode.getTag()) && ((MappingNode) valueNode).getValue().isEmpty()) {
                return null;
            }
        }
        return tuple;
    }
    
    private boolean isWantedNodeTuple(final Node keyNode) {
        return keyNode instanceof ScalarNode && reservedNodeNames.contains(((ScalarNode) keyNode).getValue());
    }
    
    private boolean isUnwantedNodeTuple(final Node valueNode) {
        return isNullNode(valueNode) || isEmptyCollectionNode(valueNode);
    }
    
    private boolean isNullNode(final Node valueNode) {
        return Tag.NULL.equals(valueNode.getTag());
    }
    
    private boolean isEmptyCollectionNode(final Node valueNode) {
        return valueNode instanceof CollectionNode && (isEmptySequenceNode(valueNode) || isEmptyMappingNode(valueNode));
    }
    
    private boolean isEmptySequenceNode(final Node valueNode) {
        return Tag.SEQ.equals(valueNode.getTag()) && ((SequenceNode) valueNode).getValue().isEmpty();
    }
    
    private boolean isEmptyMappingNode(final Node valueNode) {
        return Tag.MAP.equals(valueNode.getTag()) && ((MappingNode) valueNode).getValue().isEmpty();
    }
    
    private class NullRepresent implements Represent {
        public Node representData(final Object data) {
            return representScalar(Tag.NULL, "");
        }
    }
}

@@ -1897,107 +1897,8 @@
 g);%0A
-        Node valueNode = tuple.getValueNode();%0A        Node keyNode = tuple.getKeyNode();%0A        %0A
     
@@ -1938,32 +1938,32 @@
 etKeyNode())) %7B%0A
+
             retu
@@ -1978,33 +1978,24 @@
 ;%0A        %7D%0A
-        %0A
         if (
@@ -2031,32 +2031,32 @@
 ValueNode())) %7B%0A
+
             retu
@@ -2078,596 +2078,8 @@
   %7D%0A
-        %0A        if (keyNode instanceof ScalarNode && ((ScalarNode) keyNode).getValue().equals(%22password%22)) %7B%0A            return tuple;%0A        %7D%0A        if (Tag.NULL.equals(valueNode.getTag())) %7B%0A            return null;%0A        %7D%0A        if (valueNode instanceof CollectionNode) %7B%0A            if (Tag.SEQ.equals(valueNode.getTag()) && ((SequenceNode) valueNode).getValue().isEmpty()) %7B%0A                return null;%0A            %7D%0A            if (Tag.MAP.equals(valueNode.getTag()) && ((MappingNode) valueNode).getValue().isEmpty()) %7B%0A                return null;%0A            %7D%0A        %7D%0A
     

modify representJavaBeanProperty