java文件过滤器

挑战任务参加“绿盟杯”竞赛的小明想要开发一个图片查看器，他想只显示文件夹下所有图片类型的文件。
你来帮小明实现这个功能吧。
编程要求基本功能与第二题类似，编写代码实现对给定文件夹目录结构的展示，如果是文件夹则在其名字之前加上±-若是文件则加上–，上级目录与下级目录、下级文件用两个空格作为间隔，依照文件夹、文件的首字母顺序依次打印；另外需要对文件进行过滤，只显示图片类型的文件，本关需要过滤的图片文件类型有：“jpg，png，bmp”，请补充完善右侧代码区中的showDirStructure(File dir)函数实现本关要求的功能，其中函数参数含义如下：

dir：指定要显示的文件夹
测试说明样例1输入：src/step3/root
输出：

样例2输入：src/step3/dir
输出：

提示：
你可以通过如下链接下载本关涉及到的目录文件：
https://www.educoder.net/attachments/download/202620/step3Dir.zip

开始挑战吧，祝你成功！
代码：

package step3;
import java.io.File;import java.util.*;import java.util.Arrays;public class Task {        /********** BEGIN **********/    public static void showDirStructure(File file) {
        if (file.isDirectory()) {
            System.out.println("+--" + file.getName());
        }
        int Blank = 2;
        haha(file, Blank);
    }
 
    public static void haha(File file, int Blank ) {
        File[] files = file.listFiles();        Arrays.sort(files);        for (int i = 0; i < files.length; i++) {
            if (files[i].isDirectory()) {
                for (int k = 0; k < Blank; k++) {
                    System.out.print(" ");
                }
                System.out.println("+--" + files[i].getName());
                
                haha(files[i], Blank+2);
            } else {
                int end =  files[i].toString().indexOf(".");
                String suffix = files[i].toString().substring(end+1);
            
                if(suffix.equals("jpg") || suffix.equals("png") || suffix.equals("bmp")) {
                    for (int k = 0; k < Blank; k++) {
                        System.out.print(" ");
                    }
                    System.out.println("--" + files[i].getName());
                }
            }
        }
    }    
    /********** END **********/}