ru.cwcode.tkach.config.jackson.yaml.YmlConfigManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of Share Show documentation

CWConfig library share

The newest version!

package ru.cwcode.tkach.config.jackson.yaml;

import ru.cwcode.cwutils.collections.CollectionUtils;
import ru.cwcode.tkach.config.annotation.Description;
import ru.cwcode.tkach.config.base.ConfigPlatform;
import ru.cwcode.tkach.config.base.manager.ConfigPersister;
import ru.cwcode.tkach.config.jackson.JacksonConfigManager;

import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.stream.Collectors;

public class YmlConfigManager extends JacksonConfigManager {
  
  public YmlConfigManager(ConfigPlatform configPlatform) {
    super(configPlatform, new ConfigPersister<>(), new YmlConfigMapper());
    
    persister.setPreprocessor(new ConfigPersister.Preprocessor<>() {
      @Override
      public String preprocess(YmlConfig config, String data) {
        String[] header = config.header();
        if (header != null) {
          data = CollectionUtils.toString(header, "#", "\n", false) + data;
        }
        
        for (var descriptionEntry : Arrays.stream(config.getClass().getDeclaredFields())
                                          .filter(x -> x.isAnnotationPresent(Description.class))
                                          .collect(Collectors.toMap(Field::getName, o -> o.getAnnotation(Description.class).value()))
                                          .entrySet()) {
          
          data = data.replaceFirst("(\n" + descriptionEntry.getKey() + ")",
                                   CollectionUtils.toString(descriptionEntry.getValue(), "\n#", "", false) + "$1");
        }
        
        return data;
      }
    });
  }
}