use of java.util.AbstractList in project xtext-eclipse by eclipse.
the class XtextGrammarQuickfixProvider method fixExternalImportedPackage.
@Fix(INVALID_PACKAGE_REFERENCE_EXTERNAL)
public void fixExternalImportedPackage(final Issue issue, IssueResolutionAcceptor acceptor) {
if (issue.getData().length == 1)
acceptor.accept(issue, "Update the imported package '" + issue.getData()[0] + "'", "Fix the bogus package import\n" + "import '" + issue.getData()[0] + "'", NULL_QUICKFIX_IMAGE, new IModification() {
@Override
public void apply(IModificationContext context) throws BadLocationException {
String replaceString = valueConverterService.toString(issue.getData()[0], "STRING");
IXtextDocument document = context.getXtextDocument();
final List<String> importedPackages = document.priorityReadOnly(new IUnitOfWork<List<String>, XtextResource>() {
@Override
public List<String> exec(XtextResource state) throws Exception {
IResourceDescriptions descriptions = resourceDescriptionsProvider.getResourceDescriptions(state);
ResourceSet resourceSet = state.getResourceSet();
final Map<URI, URI> uriMap = Maps.newHashMap();
EPackage ePackage = loadPackageFromIndex(descriptions, resourceSet, uriMap, issue.getData()[0]);
if (ePackage != null) {
final Map<String, EPackage> packagePerNsURI = Maps.newHashMap();
packagePerNsURI.put(ePackage.getNsURI(), ePackage);
final Set<URI> updatedReferences = fixReferencesInPackages(ePackage, packagePerNsURI, uriMap, descriptions, resourceSet);
if (updatedReferences.isEmpty())
return null;
Iterator<EPackage> iterator = packagePerNsURI.values().iterator();
while (iterator.hasNext()) {
EPackage pack = iterator.next();
Resource resource = pack.eResource();
if (!resource.getURI().isPlatformResource()) {
iterator.remove();
}
}
final List<String> result = Lists.newArrayList();
new WorkspaceModifyOperation() {
/* workspace lock */
@Override
protected void execute(IProgressMonitor monitor) throws CoreException, InvocationTargetException, InterruptedException {
try {
for (EPackage pack : packagePerNsURI.values()) {
Resource resource = pack.eResource();
resource.save(Collections.singletonMap(XMLResource.OPTION_URI_HANDLER, new URIHandlerImpl.PlatformSchemeAware() {
@Override
public URI deresolve(URI uri) {
// replace archive uris with platform:/plugin
if (!uri.isArchive() || !updatedReferences.contains(uri)) {
return super.deresolve(uri);
}
URI withoutFragment = uri.trimFragment();
if (uriMap.containsKey(withoutFragment)) {
withoutFragment = uriMap.get(withoutFragment);
}
return super.deresolve(withoutFragment.appendFragment(uri.fragment()));
}
}));
result.add(resource.getURI().toString());
}
} catch (IOException ioe) {
throw new InvocationTargetException(ioe);
}
}
}.run(new NullProgressMonitor());
for (int i = resourceSet.getResources().size() - 1; i >= 0; i--) {
Resource resource = resourceSet.getResources().get(i);
if (!resource.getContents().isEmpty() && resource.getContents().get(0) instanceof GenModel) {
resourceSet.getResources().remove(i);
}
}
return result;
}
return null;
}
private Set<URI> fixReferencesInPackages(EPackage ePackage, Map<String, EPackage> packagePerNsURI, Map<URI, URI> uriMap, IResourceDescriptions descriptions, ResourceSet resourceSet) {
Set<URI> result = Sets.newHashSet();
Map<EObject, Collection<Setting>> allReferences = EcoreUtil.CrossReferencer.find(Collections.singletonList(ePackage));
for (final Setting setting : Iterables.concat(allReferences.values())) {
if (setting.getEStructuralFeature().isChangeable()) {
final Object referenced = setting.get(true);
List<Object> references = null;
if (referenced instanceof EObject) {
references = new AbstractList<Object>() {
@Override
public Object set(int index, Object element) {
setting.set(element);
return referenced;
}
@Override
public Object get(int index) {
return referenced;
}
@Override
public int size() {
return 1;
}
};
} else {
@SuppressWarnings("unchecked") List<Object> casted = (List<Object>) referenced;
references = casted;
}
for (int i = 0; i < references.size(); i++) {
if (references.get(i) instanceof EObject) {
EObject referencedEObject = (EObject) references.get(i);
EPackage transitive = EcoreUtil2.getContainerOfType(referencedEObject, EPackage.class);
if (isRegisteredPackage(transitive)) {
if (referencedEObject instanceof EDataType)
continue;
if (referencedEObject == EcorePackage.Literals.EOBJECT)
continue;
EPackage fromWorkspace = packagePerNsURI.get(transitive.getNsURI());
if (fromWorkspace == null && !packagePerNsURI.containsKey(transitive.getNsURI())) {
fromWorkspace = loadPackageFromIndex(descriptions, resourceSet, uriMap, transitive.getNsURI());
packagePerNsURI.put(transitive.getNsURI(), fromWorkspace);
}
if (fromWorkspace != null) {
String fragment = transitive.eResource().getURIFragment(referencedEObject);
EObject replacement = fromWorkspace.eResource().getEObject(fragment);
if (replacement != null) {
result.add(EcoreUtil.getURI(replacement));
references.set(i, replacement);
}
}
}
}
}
}
}
return result;
}
private boolean isRegisteredPackage(EPackage ePackage) {
return ePackage != null && (ePackage.eResource() == null || ePackage.getNsURI().equals(ePackage.eResource().getURI().toString()));
}
private EPackage loadPackageFromIndex(IResourceDescriptions descriptions, ResourceSet resourceSet, Map<URI, URI> uriMap, String nsURI) {
Iterable<IEObjectDescription> fixUs = descriptions.getExportedObjects(EcorePackage.Literals.EPACKAGE, QualifiedName.create(nsURI), false);
for (IEObjectDescription description : fixUs) {
if (description.getEObjectURI().isPlatformResource()) {
EObject result = resourceSet.getEObject(description.getEObjectURI(), true);
if (result instanceof EPackage) {
return (EPackage) result;
}
}
}
URI genModelURI = EcorePlugin.getEPackageNsURIToGenModelLocationMap(false).get(nsURI);
if (genModelURI != null) {
Resource genmodelResource = resourceSet.getResource(genModelURI, true);
GenModel genModel = (GenModel) genmodelResource.getContents().get(0);
for (GenPackage genPackage : genModel.getGenPackages()) {
Object object = genPackage.eGet(GenModelPackage.Literals.GEN_PACKAGE__ECORE_PACKAGE, false);
if (object instanceof EObject) {
EObject proxy = (EObject) object;
URI proxyURI = EcoreUtil.getURI(proxy);
URI resolvedProxyURI = proxyURI.resolve(genModelURI);
if (nsURI.equals(genPackage.getEcorePackage().getNsURI())) {
EPackage result = genPackage.getEcorePackage();
uriMap.put(result.eResource().getURI(), resolvedProxyURI.trimFragment());
return result;
}
}
}
}
return null;
}
});
String delimiter = document.get(issue.getOffset(), 1);
if (!replaceString.startsWith(delimiter)) {
replaceString = delimiter + replaceString.substring(1, replaceString.length() - 1) + delimiter;
}
document.replace(issue.getOffset(), issue.getLength(), replaceString);
if (importedPackages != null && !importedPackages.isEmpty()) {
final Shell shell = workbench.getActiveWorkbenchWindow().getShell();
shell.getDisplay().asyncExec(new Runnable() {
@Override
public void run() {
String title = "Please update the Ecore2XtextDslProjectContributor that generates the language.";
String message = "Please make sure that the Ecore2XtextDslProjectContributor that generates the language is up-to date.\n" + "Especially important is the registration of the referenced packages.\n" + "Please refer to the reference documentation for details.";
MessageDialog dialog = new MessageDialog(shell, title, null, message, MessageDialog.INFORMATION, new String[] { "Open Documentation", "Close" }, 1);
if (dialog.open() == 0) {
try {
workbench.getBrowserSupport().getExternalBrowser().openURL(new URL(GRAMMAR_LANG_DOC));
} catch (Exception e) {
// ignore
}
}
}
});
}
}
});
}
Aggregations