com.github.leeonky.dal.extensions.basic.file.PathExtension Maven / Gradle / Ivy

Go to download
package com.github.leeonky.dal.extensions.basic.file;

import com.github.leeonky.dal.DAL;
import com.github.leeonky.dal.extensions.basic.file.util.PathJavaClassPropertyAccessor;
import com.github.leeonky.dal.extensions.basic.file.util.ToString;
import com.github.leeonky.dal.extensions.basic.file.util.Util;
import com.github.leeonky.dal.runtime.*;

import java.io.FileInputStream;
import java.nio.file.Path;

import static com.github.leeonky.util.Suppressor.get;

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

    @Override
    public void extend(DAL dal) {
        RuntimeContextBuilder runtimeContextBuilder = dal.getRuntimeContextBuilder();
        runtimeContextBuilder.registerImplicitData(Path.class, file -> get(() -> new FileInputStream(file.toFile())))
                .registerDALCollectionFactory(Path.class, new DALCollectionFactory() {
                    @Override
                    public boolean isList(Path path) {
                        return path.toFile().isDirectory();
                    }

                    @Override
                    public DALCollection create(Path path) {
                        return new CollectionDALCollection<>(Util.listFile(path));
                    }
                })
                .registerPropertyAccessor(Path.class, new PathJavaClassPropertyAccessor())
                .registerDumper(Path.class, data -> ((Path) data.instance()).toFile().isDirectory()
                        ? Util.PATH_DIR_DUMPER : Util.PATH_FILE_DUMPER)
                .getConverter().addTypeConverter(Path.class, String.class, ToString::name)
        ;
    }
}