ru.healexxzt.config.YamlRepresenter Maven / Gradle / Ivy

Go to download
package ru.healexxzt.config;

import org.yaml.snakeyaml.representer.*;
import org.yaml.snakeyaml.nodes.*;
import java.util.*;

public class YamlRepresenter extends Representer
{
    public YamlRepresenter() {
        this.multiRepresenters.put(ConfigurationSection.class, new RepresentConfigurationSection());
        this.multiRepresenters.put(ConfigurationSerializable.class, new RepresentConfigurationSerializable());
    }
    
    private class RepresentConfigurationSection extends RepresentMap
    {
        @Override
        public Node representData(final Object data) {
            return super.representData(((ConfigurationSection)data).getValues(false));
        }
    }
    
    private class RepresentConfigurationSerializable extends RepresentMap
    {
        @Override
        public Node representData(final Object data) {
            final ConfigurationSerializable serializable = (ConfigurationSerializable)data;
            final Map values = new LinkedHashMap();
            values.put("==", ConfigurationSerialization.getAlias(serializable.getClass()));
            values.putAll(serializable.serialize());
            return super.representData(values);
        }
    }
}