com.viaoa.util.OAFindFile Maven / Gradle / Ivy

Go to download
package com.viaoa.util;

import java.io.*;
import java.util.*;
import java.util.zip.*;


/**
 * This is used to search for a file from a specific directory and all files below it.
 * If the file is an archive file (zip|jar|war|ear), then it will search within the archive file.
 *  
 * @author vvia
 */
public class OAFindFile {
	static final String[] ZIP_EXTENSIONS = { ".zip", ".jar", ".war", ".ear" };

	private String findFileName;
	private ArrayList list;

	public String[] findAll(String rootFile, String findFileName) throws IOException {
		if (rootFile == null || rootFile.trim().length() == 0) rootFile = ".";
		return findAll(new File(rootFile), findFileName);
	}
	public String[] findAll(File rootFile, String findFileName) throws IOException {
		if (findFileName == null || findFileName.trim().length() == 0) return new String[0];
		this.findFileName = findFileName;
		this.list = new ArrayList();
		findFile(rootFile);
		String[] ss = new String[list.size()];
		this.list.toArray(ss);
		this.list = null;
		return ss;
	}
	
	protected void findFile(File file) throws IOException {
		if (file.isDirectory()) {
	        System.out.println("checking "+file);         
			File[] files = file.listFiles();
			if (files != null) {
    			for (int i = 0; i < files.length; i++) {
    				findFile(files[i]);
    			}
			}
		} 
		else {
			// if ((cnt%100) == 0) System.out.println("Status: " + cnt+") "+file.getAbsolutePath());			
			String fileName = file.getName();
			if (fileName.equalsIgnoreCase(findFileName)) {
				list.add(file.getAbsolutePath());
				//System.out.println("Found #"+(list.size()) + " = " + file);		
			}
			else {
				for (int i = 0; i < ZIP_EXTENSIONS.length; i++) {
					if (fileName.toLowerCase().endsWith(ZIP_EXTENSIONS[i])) {
						//System.out.println("Compressed= "+file.getAbsolutePath());			
						try {
						    findZip(file);
						}
						catch (Exception e) {
						    System.out.println("Error with zip file:" +fileName+", "+e);
						}
						break;
					}
				}
			}
		}
	}

	protected void findZip(File file) throws IOException {
        System.out.println("checking zip "+file);         
		InputStream in = new FileInputStream(file);
		ZipInputStream zin = new ZipInputStream(in);

		ZipEntry en;
		while ((en = zin.getNextEntry()) != null) {
			if (en.isDirectory()) continue;
			String fn = en.getName();
			int pos = fn.lastIndexOf('/');
			if (pos < 0) pos = fn.lastIndexOf('\\');
			if (pos >= 0) fn = fn.substring(pos+1);
			if (fn.equalsIgnoreCase(findFileName)) {
				String s = file.getAbsolutePath() + "!" + en.getName();
				list.add(s);
				// System.out.println("Found #"+(list.size()) + " = " + file + "!" + en);		
			}
		}
	}
	
	
	public static void main(String[] args) throws Exception {
		if (args == null || args.length == 0) {
			System.out.println("Usage: FindFile [fromDirectory|File] SearchFileName");
		}
		else {
			String s1;
			String s2;
			if (args.length == 1) {
				s1 = ".";
				s2 = args[0];
			}
			else {
				s1 = args[0];
				s2 = args[1];
			}
			
			OAFindFile ff = new OAFindFile();
			String[] fileNames = ff.findAll(s1, s2);
			
			for (int i=0; fileNames != null && i < fileNames.length; i++) {
				System.out.println((i+1)+") " + fileNames[i]);
			}
			System.out.println("FindFile done for " + s2+", " + (fileNames.length) + " found");
		}
	}
	
}