package org.eclipse.jgit.api;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.api.errors.JGitInternalException;
import org.eclipse.jgit.errors.ConfigInvalidException;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.submodule.SubmoduleStatus;
import org.eclipse.jgit.submodule.SubmoduleStatusType;
import org.eclipse.jgit.submodule.SubmoduleWalk;
import org.eclipse.jgit.treewalk.filter.PathFilterGroup;
public class SubmoduleStatusCommand extends
GitCommand<Map<String, SubmoduleStatus>> {
private final Collection<String> paths;
public SubmoduleStatusCommand(Repository repo) {
super(repo);
paths = new ArrayList<>();
}
public SubmoduleStatusCommand addPath(String path) {
paths.add(path);
return this;
}
@Override
public Map<String, SubmoduleStatus> call() throws GitAPIException {
checkCallable();
try (SubmoduleWalk generator = SubmoduleWalk.forIndex(repo)) {
if (!paths.isEmpty())
generator.setFilter(PathFilterGroup.createFromStrings(paths));
Map<String, SubmoduleStatus> statuses = new HashMap<>();
while (generator.next()) {
SubmoduleStatus status = getStatus(generator);
statuses.put(status.getPath(), status);
}
return statuses;
} catch (IOException | ConfigInvalidException e) {
throw new JGitInternalException(e.getMessage(), e);
}
}
private SubmoduleStatus getStatus(SubmoduleWalk generator)
throws IOException, ConfigInvalidException {
ObjectId id = generator.getObjectId();
String path = generator.getPath();
if (generator.getModulesPath() == null)
return new SubmoduleStatus(SubmoduleStatusType.MISSING, path, id);
if (generator.getConfigUrl() == null)
return new SubmoduleStatus(SubmoduleStatusType.UNINITIALIZED, path,
id);
ObjectId headId = null;
try (Repository subRepo = generator.getRepository()) {
if (subRepo == null) {
return new SubmoduleStatus(SubmoduleStatusType.UNINITIALIZED,
path, id);
}
headId = subRepo.resolve(Constants.HEAD);
}
if (headId == null)
return new SubmoduleStatus(SubmoduleStatusType.UNINITIALIZED, path,
id, headId);
if (!headId.equals(id))
return new SubmoduleStatus(SubmoduleStatusType.REV_CHECKED_OUT,
path, id, headId);
return new SubmoduleStatus(SubmoduleStatusType.INITIALIZED, path, id,
headId);
}
}