package Demo;

import java.io.File;
import java.io.FilenameFilter;
import java.util.Scanner;
import java.lang.Exception;
import java.io.IOException;
public class Demo8_1 {
	public static void main(String[] args) {
		while(true) {
			System.out.println("1、按關鍵字檢索文件");
			System.out.println("2、按后綴名找出文件");
			System.out.println("3、退出");
			System.out.println("請選擇你的操作");
			Scanner sr = new Scanner(System.in);
			int choose = sr.nextInt();
			if(choose==1) {
				searchByKey();	
			}else if(choose==2) {
				searchBySuffix();
			}else if(choose==3) {
				exit();
			}
		}
	}
	
	
	
	
	//退出
	public static void exit() {
		System.out.println("你已經(jīng)退出系統(tǒng)，感謝使用！");
		System.exit(0);
	}
	
	
	//按照關鍵字查找
	public static void searchByKey() {
		Scanner sr = new Scanner(System.in);
		System.out.println("請輸入要檢索文件的位置：");
		String dirPath = sr.next();
		File dir = new File(dirPath);
		File[]files = dir.listFiles();
		String[] fileNames = new String[files.length];
		System.out.println("請輸入檢索文件關鍵字");
		String mainName = sr.next();
		//(dir.getName()+"/"+mainName);
		Filter filter = new Filter();
		for(int i=0;i<files.length;i++) {
			fileNames[i] = files[i].getName();
			if(filter.acceptLikeName(fileNames[i],mainName)) {
				System.out.println(fileNames[i]);
			}
		}
			
	}
	
	
	
	//按照后綴名查找
	public static void searchBySuffix() {
		Scanner sr = new Scanner(System.in);
		System.out.println("請輸入要檢索文件的位置：");
		String dirPath = sr.next();
		File dir = new File(dirPath);
		File[]files = dir.listFiles();
		String[] fileNames = new String[files.length];
		Filter filter = new Filter();
		System.out.println("請輸入后綴名");
		String endName = sr.next();
		
		for(int i=0;i<files.length;i++) {
			fileNames[i] = files[i].getName();
			if(filter.acceptEndName(dir,fileNames[i],endName)) {
				System.out.println("后綴名為"+endName+"的有："+fileNames[i]);
			}
		}
	}
}
	
//繼承類FilenameFileter
	class Filter implements FilenameFilter {
		public boolean accept(File dir,String name) {
			
			return true;
		}
		public boolean acceptLikeName(String name,String mainName) {
			if(name.contains(mainName)) {
				return true;
			}
			return false;
		}
		public boolean acceptEndName(File dir,String name,String endName) {
			File file = new File(dir,name);
			if(file.isFile()&&name.endsWith(endName)) {
				return true;
			}
			return false;
		}
		
	}