com.github.leeonky.dal.extensions.basic.text.TextExtension Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of DAL-extension-basic Show documentation
DAL-extension-basic
The newest version!
package com.github.leeonky.dal.extensions.basic.text;

import com.github.leeonky.dal.DAL;
import com.github.leeonky.dal.runtime.Extension;
import com.github.leeonky.dal.runtime.RuntimeContextBuilder;
import com.github.leeonky.dal.runtime.RuntimeContextBuilder.DALRuntimeContext;
import com.github.leeonky.dal.runtime.TextAttribute;
import com.github.leeonky.dal.runtime.TextFormatter;

import static com.github.leeonky.dal.extensions.basic.binary.BinaryExtension.readAll;
import static com.github.leeonky.dal.extensions.basic.file.util.FileGroup.register;
import static com.github.leeonky.dal.extensions.basic.text.Methods.json;
import static com.github.leeonky.dal.extensions.basic.text.Methods.yaml;

@SuppressWarnings("unused")
public class TextExtension implements Extension {

    @Override
    public void extend(DAL dal) {
        RuntimeContextBuilder builder = dal.getRuntimeContextBuilder();
        builder.registerStaticMethodExtension(Methods.class);

        builder.registerTextFormatter("JSON", jsonTextFormatter("JSON"));
        builder.registerTextFormatter("json", jsonTextFormatter("json"));
        register("json", inputStream -> json(readAll(inputStream)));
        register("JSON", inputStream -> json(readAll(inputStream)));

        builder.registerTextFormatter("YAML", yamlTextFormatter("YAML"));
        builder.registerTextFormatter("yaml", yamlTextFormatter("yaml"));
        register("yaml", inputStream -> yaml(readAll(inputStream)));
        register("YAML", inputStream -> yaml(readAll(inputStream)));
    }

    private TextFormatter jsonTextFormatter(String label) {
        return new TextFormatter() {
            @Override
            protected Object format(String content, TextAttribute attribute, DALRuntimeContext context) {
                return json(content);
            }

            @Override
            public String description() {
                return "use " + label + " as json object";
            }
        };
    }

    private TextFormatter yamlTextFormatter(String label) {
        return new TextFormatter() {
            @Override
            protected Object format(String content, TextAttribute attribute, DALRuntimeContext context) {
                return yaml(content);
            }

            @Override
            public String description() {
                return "use " + label + " to as yaml object";
            }
        };
    }
}