Сортировка файлов по типу

public enum FileTypes {
    AUDIO,
    VIDEO,
    IMAGE;
}

import java.io.File;
 
public class Audio implements FileExtensionFilter {
    private static final String MP3 = ".mp3";
    private static final String WAV = ".wav";
    private static final String WMA = ".wma";
 
    public boolean accept(File pathname) {
        if (pathname.isFile()) {
            if (pathname.getName().endsWith(MP3)
                    || pathname.getName().endsWith(WAV)
                    || pathname.getName().endsWith(WMA)) {
                return true;
            }
        }
 
        return false;
    }

Map<FileTypes, List<File>> sortFilesByType(List<File> files) { }

Map< List<File>, FileTypes>

sortFilesByType(List<File> files) { }

    Map<FileTypes, List<File>> sortFilesByType(List<File> files){
        Audio audio = new Audio();
        Image image = new Image();
        Video video = new Video();
        Function<File, FileTypes> compl_file = i -> audio.accept(i) ? FileTypes.AUDIO 
                                                  : image.accept(i) ? FileTypes.IMAGE 
                                                  : video.accept(i) ? FileTypes.VIDEO
                                                  : null;
        return files.stream()
                .filter(i -> compl_file.apply(i) != null)
                .collect(Collectors.groupingBy(compl_file::apply));
    }