OSDN Git Service

original
[gb-231r1-is01/GB_2.3_IS01.git] / sdk / eclipse / plugins / com.android.ide.eclipse.adt / src / com / android / ide / eclipse / adt / internal / refactoring / core / FixImportsJob.java
diff --git a/sdk/eclipse/plugins/com.android.ide.eclipse.adt/src/com/android/ide/eclipse/adt/internal/refactoring/core/FixImportsJob.java b/sdk/eclipse/plugins/com.android.ide.eclipse.adt/src/com/android/ide/eclipse/adt/internal/refactoring/core/FixImportsJob.java
new file mode 100644 (file)
index 0000000..ae40790
--- /dev/null
@@ -0,0 +1,145 @@
+/*\r
+ * Copyright (C) 2010 The Android Open Source Project\r
+ *\r
+ * Licensed under the Eclipse Public License, Version 1.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ *      http://www.eclipse.org/org/documents/epl-v10.php\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ */\r
+\r
+package com.android.ide.eclipse.adt.internal.refactoring.core;\r
+\r
+import org.eclipse.core.resources.IFile;\r
+import org.eclipse.core.resources.IMarker;\r
+import org.eclipse.core.resources.IProject;\r
+import org.eclipse.core.resources.IResource;\r
+import org.eclipse.core.resources.IncrementalProjectBuilder;\r
+import org.eclipse.core.resources.WorkspaceJob;\r
+import org.eclipse.core.runtime.CoreException;\r
+import org.eclipse.core.runtime.IProgressMonitor;\r
+import org.eclipse.core.runtime.IStatus;\r
+import org.eclipse.core.runtime.Status;\r
+import org.eclipse.jdt.core.ICompilationUnit;\r
+import org.eclipse.jdt.core.IJavaElement;\r
+import org.eclipse.jdt.core.IJavaProject;\r
+import org.eclipse.jdt.core.IPackageFragment;\r
+import org.eclipse.jdt.core.ISourceRange;\r
+import org.eclipse.jdt.core.JavaCore;\r
+import org.eclipse.jdt.core.dom.CompilationUnit;\r
+import org.eclipse.jdt.core.search.TypeNameMatch;\r
+import org.eclipse.jdt.internal.corext.codemanipulation.CodeGenerationSettings;\r
+import org.eclipse.jdt.internal.corext.codemanipulation.OrganizeImportsOperation;\r
+import org.eclipse.jdt.internal.corext.codemanipulation.OrganizeImportsOperation.IChooseImportQuery;\r
+import org.eclipse.jdt.internal.ui.actions.WorkbenchRunnableAdapter;\r
+import org.eclipse.jdt.internal.ui.javaeditor.EditorUtility;\r
+import org.eclipse.jdt.internal.ui.preferences.JavaPreferencesSettings;\r
+import org.eclipse.jdt.ui.SharedASTProvider;\r
+import org.eclipse.swt.widgets.Display;\r
+import org.eclipse.ui.IEditorPart;\r
+import org.eclipse.ui.PlatformUI;\r
+import org.eclipse.ui.progress.IProgressService;\r
+\r
+/**\r
+ * The helper class which fixes the import errors after refactoring\r
+ *\r
+ */\r
+@SuppressWarnings("restriction")\r
+public class FixImportsJob extends WorkspaceJob {\r
+\r
+    private IFile mAndroidManifest;\r
+\r
+    private String mJavaPackage;\r
+\r
+    /**\r
+     * Creates a new <code>FixImportsJob</code>\r
+     *\r
+     * @param name the job name\r
+     * @param androidManifest the android manifest file\r
+     * @param javaPackage the android java package\r
+     */\r
+    public FixImportsJob(String name, IFile androidManifest, String javaPackage) {\r
+        super(name);\r
+        this.mAndroidManifest = androidManifest;\r
+        this.mJavaPackage = javaPackage;\r
+    }\r
+\r
+    @Override\r
+    public IStatus runInWorkspace(final IProgressMonitor monitor) throws CoreException {\r
+        if (mJavaPackage == null || mAndroidManifest == null || !mAndroidManifest.exists()) {\r
+            return Status.CANCEL_STATUS;\r
+        }\r
+        IProject project = mAndroidManifest.getProject();\r
+        IJavaProject javaProject = JavaCore.create(project);\r
+        if (javaProject == null || !javaProject.isOpen()) {\r
+            return Status.CANCEL_STATUS;\r
+        }\r
+        project.build(IncrementalProjectBuilder.INCREMENTAL_BUILD, monitor);\r
+        IMarker[] markers = project.findMarkers(IMarker.PROBLEM, true, IResource.DEPTH_INFINITE);\r
+        for (int i = 0; i < markers.length; i++) {\r
+            IMarker marker = markers[i];\r
+            IResource resource = marker.getResource();\r
+            try {\r
+                IJavaElement element = JavaCore.create(resource);\r
+                if (element != null && (element instanceof ICompilationUnit)) {\r
+                    final ICompilationUnit cu = (ICompilationUnit) element;\r
+                    IPackageFragment packageFragment = (IPackageFragment) cu\r
+                            .getAncestor(IJavaElement.PACKAGE_FRAGMENT);\r
+                    if (packageFragment != null && packageFragment.exists()) {\r
+                        String packageName = packageFragment.getElementName();\r
+                        if (packageName != null && packageName.startsWith(mJavaPackage)) {\r
+                            CompilationUnit astRoot = SharedASTProvider.getAST(cu,\r
+                                    SharedASTProvider.WAIT_ACTIVE_ONLY, null);\r
+                            CodeGenerationSettings settings = JavaPreferencesSettings\r
+                                    .getCodeGenerationSettings(cu.getJavaProject());\r
+                            final boolean hasAmbiguity[] = new boolean[] {\r
+                                false\r
+                            };\r
+                            IChooseImportQuery query = new IChooseImportQuery() {\r
+                                public TypeNameMatch[] chooseImports(TypeNameMatch[][] openChoices,\r
+                                        ISourceRange[] ranges) {\r
+                                    hasAmbiguity[0] = true;\r
+                                    return new TypeNameMatch[0];\r
+                                }\r
+                            };\r
+                            final OrganizeImportsOperation op = new OrganizeImportsOperation(cu,\r
+                                    astRoot, settings.importIgnoreLowercase, !cu.isWorkingCopy(),\r
+                                    true, query);\r
+                            Display.getDefault().asyncExec(new Runnable() {\r
+\r
+                                public void run() {\r
+                                    try {\r
+                                        IProgressService progressService = PlatformUI\r
+                                                .getWorkbench().getProgressService();\r
+                                        progressService.run(\r
+                                                true,\r
+                                                true,\r
+                                                new WorkbenchRunnableAdapter(op, op\r
+                                                        .getScheduleRule()));\r
+                                        IEditorPart openEditor = EditorUtility.isOpenInEditor(cu);\r
+                                        if (openEditor != null) {\r
+                                            openEditor.doSave(monitor);\r
+                                        }\r
+                                    } catch (Throwable e) {\r
+                                        RefactoringUtil.log(e);\r
+                                    }\r
+                                }\r
+                            });\r
+\r
+                        }\r
+                    }\r
+                }\r
+            } catch (Throwable e) {\r
+                RefactoringUtil.log(e);\r
+            }\r
+        }\r
+        return Status.OK_STATUS;\r
+    }\r
+\r
+}\r