/*
* Copyright (C) 2013 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.builder.internal.compiler;
import com.android.builder.internal.util.concurrent.WaitableExecutor;
import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
/**
* Class to search for source files (by extension) in a set of source folders.
*/
public class SourceSearcher {
private final List<File> mSourceFolders;
private final String[] mExtensions;
private WaitableExecutor mExecutor;
public interface SourceFileProcessor {
void processFile(File sourceFile) throws IOException, InterruptedException;
}
public SourceSearcher(List<File> sourceFolders, String... extensions) {
mSourceFolders = sourceFolders;
mExtensions = extensions;
}
public void setUseExecutor(boolean useExecutor) {
if (useExecutor) {
mExecutor = new WaitableExecutor();
} else {
mExecutor = null;
}
}
public void search(SourceFileProcessor processor)
throws IOException, InterruptedException, ExecutionException {
for (File file : mSourceFolders) {
processFile(file, processor);
}
if (mExecutor != null) {
mExecutor.waitForTasks();
}
}
private void processFile(final File file, final SourceFileProcessor processor)
throws IOException, InterruptedException {
if (file.isFile()) {
// get the extension of the file.
if (checkExtension(file)) {
if (mExecutor != null) {
mExecutor.execute(new Callable() {
@Override
public Object call() throws Exception {
processor.processFile(file);
return null;
}
});
} else {
processor.processFile(file);
}
}
} else if (file.isDirectory()) {
File[] children = file.listFiles();
if (children != null) {
for (File child : children) {
processFile(child, processor);
}
}
}
}
private boolean checkExtension(File file) {
if (mExtensions.length == 0) {
return true;
}
String filename = file.getName();
int pos = filename.indexOf('.');
if (pos != -1) {
String extension = filename.substring(pos + 1);
for (String ext : mExtensions) {
if (ext.equalsIgnoreCase(extension)) {
return true;
}
}
}
return false;
}
}