public ProjectModelFetcher(NbGradleProject project, GradleTarget gradleTarget) {
this.settingsFile =
NbGenericModelInfo.findSettingsGradle(project.getProjectDirectoryAsFile());
List<NbGradleExtensionRef> extensions = project.getExtensionRefs();
this.toolingModelNeeds = CollectionUtils.newHashMap(extensions.size());
Map<Object, List<GradleBuildInfoQuery<?>>> buildInfoRequests = Collections.emptyMap();
Map<Object, List<GradleProjectInfoQuery<?>>> projectInfoRequests = new HashMap<>();
List<Class<?>> models = new LinkedList<>();
for (NbGradleExtensionRef extensionRef : extensions) {
String extensionName = extensionRef.getName();
GradleModelDefQuery2 modelQuery = extensionRef.getModelNeeds().getQuery2();
GradleModelDef modelDef = modelQuery.getModelDef(gradleTarget);
models.addAll(modelDef.getToolingModels());
addAllToMultiMap(extensionName, modelDef.getProjectInfoQueries(), projectInfoRequests);
addAllToMultiMap(extensionName, modelDef.getToolingModels(), toolingModelNeeds);
}
modelFetcher = new GenericModelFetcher(buildInfoRequests, projectInfoRequests, models);
}
private NbGradleModel parseModel(
FetchedProjectModels projectModels, Map<File, ProjectModelsOfExtensions> extensionModels) {
Throwable issue = projectModels.getIssue();
if (issue != null) {
issues.add(ModelLoadIssues.projectModelLoadError(mainProject, projectModels, null, issue));
}
NbGradleMultiProjectDef projectDef =
new NbGradleMultiProjectDef(projectModels.getProjectDef());
NbGenericModelInfo genericInfo =
new NbGenericModelInfo(projectDef, modelFetcher.getSettingsFile());
NbGradleModel.Builder result = new NbGradleModel.Builder(genericInfo);
File projectDir = genericInfo.getProjectDir();
ProjectExtensionModelCache projectCache = cache.tryGetProjectCache(projectDir);
for (NbGradleExtensionRef extension : extensions) {
String extensionName = extension.getName();
CachedModel cachedModel =
projectCache != null ? projectCache.tryGetModel(extensionName) : null;
Object extensionModel;
if (cachedModel != null) {
extensionModel = cachedModel.model;
} else {
ModelLoadResult modelLoadResult = modelLoadResultCache.get(extensionName);
if (modelLoadResult == null) {
modelLoadResult = getModelLoadResult(extension, projectDir, extensionModels);
modelLoadResultCache.put(extensionName, modelLoadResult);
} else {
modelLoadResult = modelLoadResult.withMainProject(projectDir);
}
ParsedModel<?> parsedModels = extension.parseModel(modelLoadResult);
extensionModel = parsedModels.getMainModel();
for (Map.Entry<File, ?> entry : parsedModels.getOtherProjectsModel().entrySet()) {
cache.getProjectCache(entry.getKey()).addModel(extensionName, entry.getValue());
}
}
result.setModelForExtension(extension, extensionModel);
}
return result.create();
}
public List<Object> getToolingModelsForExtension(
NbGradleExtensionRef extension, FetchedProjectModels projectModels) {
List<Class<?>> needs = toolingModelNeeds.get(extension.getName());
if (needs == null || needs.isEmpty()) {
return Collections.emptyList();
}
Map<Class<?>, Object> allModels = projectModels.getToolingModels();
List<Object> result = new ArrayList<>(needs.size());
for (Class<?> need : needs) {
Object model = allModels.get(need);
if (model != null) {
result.add(model);
}
}
return result;
}
private ModelLoadResult getModelLoadResult(
NbGradleExtensionRef extension,
File defaultProjectDir,
Map<File, ProjectModelsOfExtensions> extensionModels) {
Map<File, Lookup> lookups = CollectionUtils.newHashMap(extensionModels.size());
String extensionName = extension.getName();
for (Map.Entry<File, ProjectModelsOfExtensions> entry : extensionModels.entrySet()) {
Lookup lookup = entry.getValue().getExtensionLookups().get(extensionName);
if (lookup == null) {
lookup = Lookup.EMPTY;
}
lookups.put(entry.getKey(), lookup);
}
return new ModelLoadResult(defaultProjectDir, lookups);
}
private Map<String, Lookup> createLookups(FetchedProjectModels projectModels) {
GenericProjectProperties genericProperties =
projectModels.getProjectDef().getMainProject().getGenericProperties();
Map<String, Lookup> result = CollectionUtils.newHashMap(extensions.size());
for (NbGradleExtensionRef extension : extensions) {
String extensionName = extension.getName();
List<Object> models = new ArrayList<>();
addProjectInfoResults(projectModels, extension, models);
addAllNullSafe(models, modelFetcher.getToolingModelsForExtension(extension, projectModels));
models.add(genericProperties);
result.put(extensionName, Lookups.fixed(models.toArray()));
}
return result;
}
private void addProjectInfoResults(
FetchedProjectModels projectModels, NbGradleExtensionRef extension, List<Object> results) {
List<BuilderResult> builderResults =
projectModels.getProjectInfoResults().get(extension.getName());
if (builderResults == null) {
return;
}
for (BuilderResult builderResult : builderResults) {
BuilderIssue issue = builderResult.getIssue();
if (issue != null) {
issues.add(ModelLoadIssues.builderError(mainProject, projectModels, extension, issue));
}
Object resultObject = builderResult.getResultObject();
if (resultObject != null) {
results.add(resultObject);
}
}
}