Skip to main content
summaryrefslogtreecommitdiffstats
blob: b6a2455b46afb70d2db538b28cd099f85bc25c3a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
/*******************************************************************************
 * Copyright (c) 2005, 2007 committers of openArchitectureWare and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     committers of openArchitectureWare - initial API and implementation
 *******************************************************************************/

package org.eclipse.xtend.shared.ui.test;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileWriter;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IMarker;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.IWorkspaceDescription;
import org.eclipse.core.resources.IncrementalProjectBuilder;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.MultiStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.internal.xpand2.pr.util.FSIO;
import org.eclipse.jdt.core.IAccessRule;
import org.eclipse.jdt.core.IClasspathAttribute;
import org.eclipse.jdt.core.IClasspathEntry;
import org.eclipse.jdt.core.IJavaModelMarker;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.internal.core.ClasspathEntry;
import org.eclipse.jdt.internal.core.JavaProject;
import org.eclipse.ui.dialogs.IOverwriteQuery;
import org.eclipse.ui.wizards.datatransfer.ImportOperation;
import org.eclipse.ui.wizards.datatransfer.ZipFileStructureProvider;
import org.eclipse.xtend.shared.ui.Activator;
import org.eclipse.xtend.shared.ui.core.IModelMarker;


public class TestEnvironment {
    private boolean fIsOpen = false;

    private boolean fWasBuilt = false;

    private IWorkspace fWorkspace = null;

    private Hashtable fProjects = null;

    private void addBuilderSpecs(final String projectName) {
        try {
            final IProject project = getProject(projectName);
            final IProjectDescription description = project.getDescription();
            description.setNatureIds(new String[] { JavaCore.NATURE_ID, Activator.getNatureId() });

            project.setDescription(description, null);
        } catch (final CoreException e) {
            handleCoreException(e);
        }
    }

    /**
     * Adds a binary class with the given contents to the given package in the
     * workspace. The package is created if necessary. If a class with the same
     * name already exists, it is replaced. A workspace must be open, and the
     * given class name must not end with ".class". Returns the path of the
     * added class.
     */
    public IPath addBinaryClass(final IPath packagePath, final String className, final byte[] contents) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath classPath = packagePath.append(className + ".class"); //$NON-NLS-1$
        createFile(classPath, contents);
        return classPath;
    }

    /**
     * Adds a binary class with the given contents to the given package in the
     * workspace. The package is created if necessary. If a class with the same
     * name already exists, it is replaced. A workspace must be open, and the
     * given class name must not end with ".class". Returns the path of the
     * added class.
     */
    public IPath addBinaryClass(final IPath packageFragmentRootPath, final String packageName, final String className,
            final byte[] contents) {
        /* make sure the package exists */
        if (packageName != null && packageName.length() > 0) {
            final IPath packagePath = addPackage(packageFragmentRootPath, packageName);

            return addBinaryClass(packagePath, className, contents);
        }
        return addBinaryClass(packageFragmentRootPath, className, contents);

    }

    /**
     * Adds a class with the given contents to the given package in the
     * workspace. The package is created if necessary. If a class with the same
     * name already exists, it is replaced. A workspace must be open, and the
     * given class name must not end with ".java". Returns the path of the added
     * class.
     */
    public IPath addClass(final IPath packagePath, final String className, final String contents) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath classPath = packagePath.append(className + ".java"); //$NON-NLS-1$
        createFile(classPath, contents.getBytes());
        return classPath;
    }

    /**
     * Adds a class with the given contents to the given package in the
     * workspace. The package is created if necessary. If a class with the same
     * name already exists, it is replaced. A workspace must be open, and the
     * given class name must not end with ".java". Returns the path of the added
     * class.
     */
    public IPath addClass(final IPath packageFragmentRootPath, final String packageName, final String className,
            final String contents) {
        /* make sure the package exists */
        if (packageName != null && packageName.length() > 0) {
            final IPath packagePath = addPackage(packageFragmentRootPath, packageName);

            return addClass(packagePath, className, contents);
        }
        return addClass(packageFragmentRootPath, className, contents);
    }

    /**
     * Adds a package to the given package fragment root in the workspace. The
     * package fragment root is created if necessary. If a package with the same
     * name already exists, it is not replaced. A workspace must be open.
     * Returns the path of the added package.
     */
    public IPath addPackage(final IPath packageFragmentRootPath, final String packageName) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath path = packageFragmentRootPath.append(packageName.replace('.', IPath.SEPARATOR));
        createFolder(path);
        return path;
    }

    public IPath addPackageFragmentRoot(final IPath projectPath, final String sourceFolderName)
            throws JavaModelException {
        return addPackageFragmentRoot(projectPath, sourceFolderName, null, null);
    }

    /**
     * Adds a package fragment root to the workspace. If a package fragment root
     * with the same name already exists, it is not replaced. A workspace must
     * be open. Returns the path of the added package fragment root.
     */
    public IPath addPackageFragmentRoot(final IPath projectPath, final String sourceFolderName,
            final IPath[] exclusionPatterns, final String specificOutputLocation) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath path = getPackageFragmentRootPath(projectPath, sourceFolderName);
        createFolder(path);
        IPath outputPath = null;
        if (specificOutputLocation != null) {
            outputPath = getPackageFragmentRootPath(projectPath, specificOutputLocation);
            createFolder(outputPath);
        }
        final IClasspathEntry entry = JavaCore.newSourceEntry(path, exclusionPatterns == null ? new Path[0]
                : exclusionPatterns, outputPath);
        addEntry(projectPath, entry);
        return path;
    }

    public IPath addProject(final String projectName) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IProject project = createProject(projectName);

        return project.getFullPath();
    }

    public void addRequiredProject(final IPath projectPath, final IPath requiredProjectPath) throws JavaModelException {
        addRequiredProject(projectPath, requiredProjectPath, new IPath[] {}/*
                                                                             * include
                                                                             * all
                                                                             */,
                new IPath[] {}/* exclude none */, false);
    }

    /**
     * Adds a project to the classpath of a project.
     */
    @SuppressWarnings("restriction")
	public void addRequiredProject(final IPath projectPath, final IPath requiredProjectPath,
            final IPath[] accessibleFiles, final IPath[] nonAccessibleFiles, final boolean isExported)
            throws JavaModelException {
        checkAssertion("required project must not be in project", !projectPath.isPrefixOf(requiredProjectPath)); //$NON-NLS-1$
        final IAccessRule[] accessRules = ClasspathEntry.getAccessRules(accessibleFiles, nonAccessibleFiles);
        addEntry(projectPath, JavaCore.newProjectEntry(requiredProjectPath, accessRules, true,
                new IClasspathAttribute[0], isExported));
    }

    public void addExternalJars(final IPath projectPath, final String[] jars) throws JavaModelException {
        addExternalJars(projectPath, jars, false);
    }

    public void addExternalJar(final IPath projectPath, final String jar) throws JavaModelException {
        addExternalJar(projectPath, jar, false);
    }

    /**
     * Add the given folder to the list of source folders in the given Java Projekt
     * @param projectPath Project to modify
     * @param folder Folder to add
     */
    public IClasspathEntry addFolderToSourceFolders(final IPath projectPath, final IPath folder) {
    	IClasspathEntry newSourceEntry = null;
    	try {
        	newSourceEntry = JavaCore.newSourceEntry(folder);
        	addEntry(projectPath, newSourceEntry);
        } catch (final JavaModelException e) {
            e.printStackTrace();
            checkAssertion("JavaModelException", false); //$NON-NLS-1$
        }
        return newSourceEntry;
    }

    /**
     * Adds an external jar to the classpath of a project.
     */
    public void addExternalJars(final IPath projectPath, final String[] jars, final boolean isExported)
            throws JavaModelException {
        for (int i = 0, max = jars.length; i < max; i++) {
            final String jar = jars[i];
            checkAssertion("file name must end with .zip or .jar", jar.endsWith(".zip") || jar.endsWith(".jar")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
            addEntry(projectPath, JavaCore.newLibraryEntry(new Path(jar), null, null, isExported));
        }
    }

    /**
     * Adds an external jar to the classpath of a project.
     */
    public void addExternalJar(final IPath projectPath, final String jar, final boolean isExported)
            throws JavaModelException {
        checkAssertion("file name must end with .zip or .jar", jar.endsWith(".zip") || jar.endsWith(".jar")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
        addEntry(projectPath, JavaCore.newLibraryEntry(new Path(jar), null, null, isExported));
    }

    private void addEntry(final IPath projectPath, final IClasspathEntry entryPath) throws JavaModelException {
        final IClasspathEntry[] classpath = getClasspath(projectPath);
        final IClasspathEntry[] newClaspath = new IClasspathEntry[classpath.length + 1];
        System.arraycopy(classpath, 0, newClaspath, 0, classpath.length);
        newClaspath[classpath.length] = entryPath;
        setClasspath(projectPath, newClaspath);
    }

    /**
     * Returns the class path.
     */
	@SuppressWarnings("restriction")
	public IClasspathEntry[] getClasspath(final IPath projectPath) {
        try {
            checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
            final JavaProject javaProject = (JavaProject) JavaCore.create(getProject(projectPath));
            return javaProject.getExpandedClasspath();
        } catch (final JavaModelException e) {
            e.printStackTrace();
            checkAssertion("JavaModelException", false); //$NON-NLS-1$
            return null; // not reachable
        }
    }

    /**
     * Adds a file.
     */
    public IPath addFile(final IPath root, final String fileName, final String contents) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath filePath = root.append(fileName);
        createFile(filePath, contents.getBytes());
        return filePath;
    }

    /**
     * Adds a folder.
     */
    public IPath addFolder(final IPath root, final String folderName) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath folderPath = root.append(folderName);
        createFolder(folderPath);
        return folderPath;
    }

    public IPath addInternalJar(final IPath projectPath, final String zipName, final byte[] contents)
            throws JavaModelException {
        return addInternalJar(projectPath, zipName, contents, false);
    }

    /**
     * Adds a jar with the given contents to the the workspace. If a jar with
     * the same name already exists, it is replaced. A workspace must be open,
     * and the given zip name must end with ".zip" or ".jar". Returns the path
     * of the added jar.
     */
    public IPath addInternalJar(final IPath projectPath, final String zipName, final byte[] contents,
            final boolean isExported) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        checkAssertion("zipName must end with .zip or .jar", zipName.endsWith(".zip") || zipName.endsWith(".jar")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
        final IPath path = projectPath.append(zipName);

        /* remove any existing zip from the java model */
        removeInternalJar(projectPath, zipName);

        createFile(path, contents);
        addEntry(projectPath, JavaCore.newLibraryEntry(path, null, null, isExported));
        return path;
    }

    private void checkAssertion(final String message, boolean b) {
        if (!b)
            throw new RuntimeException(message);
    }

    /**
     * Closes the testing environment and frees up any resources. Once the
     * testing environment is closed, it shouldn't be used any more.
     */
    public void close() {
        try {
            if (fProjects != null) {
                final Enumeration projectNames = fProjects.keys();
                while (projectNames.hasMoreElements()) {
                    final String projectName = (String) projectNames.nextElement();
                    getJavaProject(projectName).getJavaModel().close();
                }
            }
            closeWorkspace();
        } catch (final JavaModelException e) {
            e.printStackTrace();
        } catch (final RuntimeException e) {
            e.printStackTrace();
        }
    }

    /**
     * Close a project from the workspace.
     */
    public void closeProject(final IPath projectPath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            getJavaProject(projectPath).getProject().close(null);
        } catch (final CoreException e) {
            e.printStackTrace();
        }
    }

    private void closeWorkspace() {
        fIsOpen = false;
        fWasBuilt = false;
    }

    private IFile createFile(final IPath path, final byte[] contents) {
        try {
            final IFile file = fWorkspace.getRoot().getFile(path);

            final ByteArrayInputStream is = new ByteArrayInputStream(contents);
            if (file.exists()) {
                file.setContents(is, true, false, null);
            } else {
                file.create(is, true, null);
            }
            return file;
        } catch (final CoreException e) {
            handle(e);
        }
        return null;
    }

    private IFolder createFolder(final IPath path) {
        checkAssertion("root", !path.isRoot()); //$NON-NLS-1$

        /* don't create folders for projects */
        if (path.segmentCount() <= 1)
            return null;

        final IFolder folder = fWorkspace.getRoot().getFolder(path);
        if (!folder.exists()) {
            /* create the parent folder if necessary */
            createFolder(path.removeLastSegments(1));

            try {
                folder.create(true, true, null);
            } catch (final CoreException e) {
                handle(e);
            }
        }
        return folder;
    }

    public IProject createProject(final String projectName) {
        IProject project = null;
        try {
            project = fWorkspace.getRoot().getProject(projectName);
            project.create(null, null);
            project.open(null);
            fProjects.put(projectName, project);
            addBuilderSpecs(projectName);
        } catch (final CoreException e) {
            handle(e);
        }

        return project;
    }

    /**
     * Batch builds the workspace. A workspace must be open.
     */
    public void fullBuild() {
        waitForAutoBuild();
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            getWorkspace().build(IncrementalProjectBuilder.FULL_BUILD, null);
        } catch (final CoreException e) {
            handle(e);
        }
        fWasBuilt = true;
    }

    /**
     * Batch builds a project. A workspace must be open.
     */
    public void fullBuild(final IPath projectPath) {
        waitForAutoBuild();
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            getProject(projectPath).build(IncrementalProjectBuilder.FULL_BUILD, null);
        } catch (final CoreException e) {
            handle(e);
        }
        fWasBuilt = true;
    }

    /**
     * Returns the Java Model element for the project.
     */
    public IJavaProject getJavaProject(final IPath projectPath) {
        final IJavaProject javaProject = JavaCore.create(getProject(projectPath));
        if (javaProject == null)
            throw new NullPointerException("No javaProject for path : " + projectPath.toPortableString());
        return javaProject;
    }

    /**
     * Returns the Java Model element for the project.
     */
    public IJavaProject getJavaProject(final String projectName) {
        final IJavaProject javaProject = JavaCore.create(getProject(projectName));
        if (javaProject == null)
            throw new NullPointerException("No javaProject for name : " + projectName);
        return javaProject;
    }

    /**
     * Return output location for a project.
     */
    public IPath getOutputLocation(final IPath projectPath) {
        try {
            final IJavaProject javaProject = JavaCore.create(getProject(projectPath));
            return javaProject.getOutputLocation();
        } catch (final CoreException e) {
            // ignore
        }
        return null;
    }

    /**
     * Return all problems with workspace.
     */
    public IMarker[] getMarkers() {
        return getMarkersFor(getWorkspaceRootPath());
    }

    /**
     * Return all problems with the specified element.
     */
    public IMarker[] getMarkersFor(final IPath path) {
        return getMarkersFor(path, false);
    }

    /**
     * Return all problems with the specified element.
     */
	public IMarker[] getMarkersFor(final IPath path, final boolean storeRange) {
        IResource resource;
        if (path.equals(getWorkspaceRootPath())) {
            resource = getWorkspace().getRoot();
        } else {
            final IProject p = getProject(path);
            if (p != null && path.equals(p.getFullPath())) {
                resource = getProject(path.lastSegment());
            } else if (path.getFileExtension() == null) {
                resource = getWorkspace().getRoot().getFolder(path);
            } else {
                resource = getWorkspace().getRoot().getFile(path);
            }
        }
        try {
            final ArrayList problems = new ArrayList();
            IMarker[] markers = resource.findMarkers(IModelMarker.PROBLEM, true, IResource.DEPTH_INFINITE);
            problems.addAll(Arrays.asList(markers));
            markers = resource.findMarkers(IModelMarker.WARNING, true, IResource.DEPTH_INFINITE);
            problems.addAll(Arrays.asList(markers));

            return (IMarker[]) problems.toArray(new IMarker[problems.size()]);
        } catch (final CoreException e) {
            // ignore
        }
        return new IMarker[0];
    }

    /**
     * Return all problems with the specified element.
     */
    public IMarker[] getTaskMarkersFor(final IPath path) {
        IResource resource = null;
        if (path.equals(getWorkspaceRootPath())) {
            resource = getWorkspace().getRoot();
        } else {
            final IProject p = getProject(path);
            if (p != null && path.equals(p.getFullPath())) {
                resource = getProject(path.lastSegment());
            } else if (path.getFileExtension() == null) {
                resource = getWorkspace().getRoot().getFolder(path);
            } else {
                resource = getWorkspace().getRoot().getFile(path);
            }
        }
        try {
            if (resource != null)
                return resource.findMarkers(IJavaModelMarker.TASK_MARKER, true, IResource.DEPTH_INFINITE);
        } catch (final CoreException e) {
            // ignore
        }
        return new IMarker[0];
    }

    /**
     * Return the path of the package with the given name. A workspace must be
     * open, and the package must exist.
     */
    public IPath getPackagePath(final IPath root, final String packageName) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        if (packageName.length() == 0)
            return root;
        return root.append(packageName.replace('.', IPath.SEPARATOR));
    }

    /**
     * Return the path of the package fragment root with the given name. A
     * workspace must be open, and the package fragment root must exist.
     */
    public IPath getPackageFragmentRootPath(final IPath projectPath, final String name) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        if (name.length() == 0)
            return projectPath;
        return projectPath.append(name);
    }

    /**
     * Returns the core project.
     */
    public IProject getProject(final String projectName) {
        return (IProject) fProjects.get(projectName);
    }

    /**
     * Returns the core project.
     */
    public IProject getProject(final IPath projectPath) {
        return (IProject) fProjects.get(projectPath.lastSegment());
    }

    /**
     * Returns the workspace.
     */
    public IWorkspace getWorkspace() {
        return fWorkspace;
    }

    /**
     * Returns the path of workspace root.
     */
    public IPath getWorkspaceRootPath() {
        return getWorkspace().getRoot().getLocation();
    }

    private IPath getJarRootPath(final IPath projectPath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        return getProject(projectPath).getFullPath();
    }

    void handle(final Exception e) {
        if (e instanceof CoreException) {
            handleCoreException((CoreException) e);
        } else
            throw new RuntimeException(e);
    }

    /**
     * Handles a core exception thrown during a testing environment operation
     */
    private void handleCoreException(final CoreException e) {
        e.printStackTrace();
        final IStatus status = e.getStatus();
        String message = e.getMessage();
        if (status.isMultiStatus()) {
            final MultiStatus multiStatus = (MultiStatus) status;
            final IStatus[] children = multiStatus.getChildren();
            final StringBuffer buffer = new StringBuffer();
            for (int i = 0, max = children.length; i < max; i++) {
                final IStatus child = children[i];
                if (child != null) {
                    buffer.append(child.getMessage());
                    buffer.append(System.getProperty("line.separator"));//$NON-NLS-1$
                }
            }
            message = String.valueOf(buffer);
        }
        throw new RuntimeException("Core exception in testing environment: " + message, e); //$NON-NLS-1$
    }

    /**
     * Incrementally builds the workspace. A workspace must be open.
     */
    public void incrementalBuild() {
        waitForAutoBuild();
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        checkAssertion("the workspace must have been built", fWasBuilt); //$NON-NLS-1$
        try {
            getWorkspace().build(IncrementalProjectBuilder.INCREMENTAL_BUILD, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Incrementally builds a project. A workspace must be open.
     */
    public void incrementalBuild(final IPath projectPath) {
        waitForAutoBuild();
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        checkAssertion("the workspace must have been built", fWasBuilt); //$NON-NLS-1$
        try {
            getProject(projectPath).build(IncrementalProjectBuilder.INCREMENTAL_BUILD, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    public boolean isAutoBuilding() {
        final IWorkspace w = getWorkspace();
        final IWorkspaceDescription d = w.getDescription();
        return d.isAutoBuilding();
    }

    /**
     * Open an empty workspace.
     */
    public void openEmptyWorkspace() {
        close();
        openWorkspace();
        fProjects = new Hashtable(10);
        setup();
    }

    /**
     * Close a project from the workspace.
     */
    public void openProject(final IPath projectPath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            getJavaProject(projectPath).getProject().open(null);
        } catch (final CoreException e) {
            e.printStackTrace();
        }
    }

    private void openWorkspace() {
        try {
            closeWorkspace();

            fWorkspace = ResourcesPlugin.getWorkspace();

            // turn off auto-build -- the tests determine when builds occur
            final IWorkspaceDescription description = fWorkspace.getDescription();
            description.setAutoBuilding(false);
            fWorkspace.setDescription(description);
        } catch (final Exception e) {
            handle(e);
        }
    }

    /**
     * Renames a compilation unit int the given package in the workspace. A
     * workspace must be open.
     */
    public void renameCU(final IPath packagePath, final String cuName, final String newName) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IFolder packageFolder = fWorkspace.getRoot().getFolder(packagePath);
        try {
            packageFolder.getFile(cuName).move(packageFolder.getFile(newName).getFullPath(), true, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Removes a binary class from the given package in the workspace. A
     * workspace must be open, and the given class name must not end with
     * ".class".
     */
    public void removeBinaryClass(final IPath packagePath, String className) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        className += ".class"; //$NON-NLS-1$
        final IFolder packageFolder = fWorkspace.getRoot().getFolder(packagePath);
        try {
            packageFolder.getFile(className).delete(true, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Removes a class from the given package in the workspace. A workspace must
     * be open, and the given class name must not end with ".java".
     */
    public void removeClass(final IPath packagePath, String className) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        className += ".java"; //$NON-NLS-1$
        final IFolder packageFolder = fWorkspace.getRoot().getFolder(packagePath);
        try {
            packageFolder.getFile(className).delete(true, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Removes a package from the given package fragment root in the workspace.
     * A workspace must be open.
     */
    public void removePackage(final IPath packageFragmentRootPath, final String packageName) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IPath path = packageFragmentRootPath.append(packageName.replace('.', IPath.SEPARATOR));
        final IFolder folder = fWorkspace.getRoot().getFolder(path);
        try {
            folder.delete(false, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Removes the given package fragment root from the the workspace. A
     * workspace must be open.
     */
    public void removePackageFragmentRoot(final IPath projectPath, final String packageFragmentRootName)
            throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        if (packageFragmentRootName.length() > 0) {
            final IFolder folder = getProject(projectPath).getFolder(packageFragmentRootName);
            if (folder.exists()) {
                try {
                    folder.delete(false, null);
                } catch (final CoreException e) {
                    handle(e);
                }
            }
        }
        final IPath rootPath = getPackageFragmentRootPath(projectPath, packageFragmentRootName);
        removeEntry(projectPath, rootPath);
    }

    /**
     * Remove a project from the workspace.
     */
    public void removeProject(final IPath projectPath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            getJavaProject(projectPath).close();
        } catch (final JavaModelException e) {
            e.printStackTrace();
        }
        final IProject project = getProject(projectPath);
        try {
        	project.refreshLocal(IProject.DEPTH_INFINITE, new NullProgressMonitor());
            project.delete(true, true, null);
        } catch (final CoreException ignore) {
//            handle(e);
        }

    }

    /**
     * Search the user hard-drive for a Java class library. Returns null if none
     * could be found.
     * 
     * Example of use: [org.eclipse.jdt.core.tests.util.Util.getJavaClassLib()]
     */
    public String[] getJavaClassLibs() {
        final String jreDir = getJREDirectory();
        if (jreDir == null)
            return new String[] {};
        final String vmName = System.getProperty("java.vm.name");
        if ("J9".equals(vmName)) {
            return new String[] { toNativePath(jreDir + "/lib/jclMax/classes.zip") };
        } else if ("Mac OS X".equals(System.getProperty("os.name"))) {
        	return new String[] { toNativePath(jreDir + "/../Classes/classes.jar") };
        } else {
	        final File file = new File(jreDir + "/lib/rt.jar");
	        if (file.exists())
	            return new String[] { toNativePath(jreDir + "/lib/rt.jar") };
	        return new String[] { toNativePath(jreDir + "/lib/core.jar"), toNativePath(jreDir + "/lib/security.jar"),
	                toNativePath(jreDir + "/lib/graphics.jar") };
        }
    }

    /**
     * Returns the JRE directory this tests are running on. Returns null if none
     * could be found.
     * 
     * Example of use: [org.eclipse.jdt.core.tests.util.Util.getJREDirectory()]
     */
    public String getJREDirectory() {
        return System.getProperty("java.home");
    }

    /**
     * Makes the given path a path using native path separators as returned by
     * File.getPath() and trimming any extra slash.
     */
    public String toNativePath(final String path) {
        final String nativePath = path.replace('\\', File.separatorChar).replace('/', File.separatorChar);
        return nativePath.endsWith("/") || nativePath.endsWith("\\") ? nativePath.substring(0, nativePath.length() - 1)
                : nativePath;
    }

    /**
     * Remove a required project from the classpath
     */
    public void removeRequiredProject(final IPath projectPath, final IPath requiredProject) throws JavaModelException {
        removeEntry(projectPath, requiredProject);
    }

    /**
     * Remove all elements in the workspace.
     */
    public void resetWorkspace() {
        if (fProjects != null) {
            final Enumeration projectNames = fProjects.keys();
            while (projectNames.hasMoreElements()) {
                final String projectName = (String) projectNames.nextElement();
                removeProject(getProject(projectName).getFullPath());
            }
        }
    }
    
    /**
    * Remove a given classpath entry from the projects class path
    * @param projectPath The project to modify
    * @param entry The entry to remove
    * @throws JavaModelException
    */
    public void removeClasspathEntry(final IPath projectPath,
			final IClasspathEntry entry) throws JavaModelException {
		checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
		final IClasspathEntry[] oldEntries = getClasspath(projectPath);
		for (int i = 0; i < oldEntries.length; ++i) {
			if (entry.equals(oldEntries[i])) {
				final IClasspathEntry[] newEntries = new IClasspathEntry[oldEntries.length - 1];
				System.arraycopy(oldEntries, 0, newEntries, 0, i);
				System.arraycopy(oldEntries, i + 1, newEntries, i,
						oldEntries.length - i - 1);
				setClasspath(projectPath, newEntries);
				break;
			}
		}
	}
    

    /**
	 * Removes the given internal jar from the workspace. A workspace must be
	 * open.
	 */
    public void removeInternalJar(final IPath projectPath, final String zipName) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        checkAssertion("zipName must end with .zip or .jar", zipName.endsWith(".zip") || zipName.endsWith(".jar")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$

        /* remove zip from the java model (it caches open zip files) */
        final IPath zipPath = getJarRootPath(projectPath).append(zipName);
        try {
            getJavaProject(projectPath).getPackageFragmentRoot(getWorkspace().getRoot().getFile(zipPath)).close();
        } catch (final JavaModelException e) {
            e.printStackTrace();
        }
        removePackageFragmentRoot(projectPath, zipName);

        final IFile file = getProject(projectPath).getFile(zipName);
        try {
            file.delete(false, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Remove an external jar from the classpath.
     */
    public void removeExternalJar(final IPath projectPath, final IPath jarPath) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        removeEntry(projectPath, jarPath);
    }

    private void removeEntry(final IPath projectPath, final IPath entryPath) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IClasspathEntry[] oldEntries = getClasspath(projectPath);
        for (int i = 0; i < oldEntries.length; ++i) {
            if (oldEntries[i].getPath().equals(entryPath)) {
                final IClasspathEntry[] newEntries = new IClasspathEntry[oldEntries.length - 1];
                System.arraycopy(oldEntries, 0, newEntries, 0, i);
                System.arraycopy(oldEntries, i + 1, newEntries, i, oldEntries.length - i - 1);
                setClasspath(projectPath, newEntries);
            }
        }
    }

    /**
     * Remove a file
     */
    public void removeFile(final IPath filePath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        try {
            fWorkspace.getRoot().getFile(filePath).delete(true, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    /**
     * Remove a folder
     */
    public void removeFolder(final IPath folderPath) {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IFolder folder = fWorkspace.getRoot().getFolder(folderPath);
        try {
            folder.delete(true, null);
        } catch (final CoreException e) {
            handle(e);
        }
    }

    public void setAutoBuilding(final boolean value) {
        try {
            final IWorkspace w = getWorkspace();
            final IWorkspaceDescription d = w.getDescription();
            d.setAutoBuilding(value);
            w.setDescription(d);
        } catch (final CoreException e) {
            e.printStackTrace();
            checkAssertion("CoreException", false); //$NON-NLS-1$
        }
    }

    public void setBuildOrder(final String[] projects) {
        try {
            final IWorkspace w = getWorkspace();
            final IWorkspaceDescription d = w.getDescription();
            d.setBuildOrder(projects);
            w.setDescription(d);
        } catch (final CoreException e) {
            e.printStackTrace();
            checkAssertion("CoreException", false); //$NON-NLS-1$
        }
    }

    public void setClasspath(final IPath projectPath, final IClasspathEntry[] entries) throws JavaModelException {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        final IJavaProject javaProject = JavaCore.create(getProject(projectPath));
        javaProject.setRawClasspath(entries, null);
    }

    public IPath setExternalOutputFolder(final IPath projectPath, final String name, final IPath externalOutputLocation) {
        IPath result = null;
        try {
            checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
            final IProject p = getProject(projectPath);
            final IFolder f = p.getFolder(name);
            f.createLink(externalOutputLocation, IResource.ALLOW_MISSING_LOCAL, null);

            result = f.getFullPath();
            final IJavaProject javaProject = JavaCore.create(p);
            javaProject.setOutputLocation(result, null);
        } catch (final CoreException e) {
            e.printStackTrace();
            checkAssertion("CoreException", false); //$NON-NLS-1$
        }
        return result;
    }

    public IPath setOutputFolder(final IPath projectPath, final String outputFolder) {
        IPath outputPath = null;
        try {
            checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
            final IJavaProject javaProject = JavaCore.create(getProject(projectPath));
            outputPath = projectPath.append(outputFolder);
            javaProject.setOutputLocation(outputPath, null);
        } catch (final JavaModelException e) {
            e.printStackTrace();
            checkAssertion("JavaModelException", false); //$NON-NLS-1$
        }
        return outputPath;
    }

    private void setup() {
        fIsOpen = true;
    }

    /**
     * Wait for autobuild notification to occur
     */
    public void waitForAutoBuild() {
        checkAssertion("a workspace must be open", fIsOpen); //$NON-NLS-1$
        boolean wasInterrupted = false;
        do {
            try {
                Job.getJobManager().join(ResourcesPlugin.FAMILY_AUTO_BUILD, null);
                wasInterrupted = false;
            } catch (final OperationCanceledException e) {
                handle(e);
            } catch (final InterruptedException e) {
                wasInterrupted = true;
            }
        } while (wasInterrupted);
        fWasBuilt = true;
    }

    public void moveFile(final IPath root, final IPath tpl) {
        try {
            fWorkspace.getRoot().getFile(tpl).move(root.append(tpl.lastSegment()), true, new NullProgressMonitor());
        } catch (final CoreException e) {
            throw new RuntimeException(e);
        }

    }

    public void changeFile(final IPath tpl, final InputStream contents) {
        try {
            fWorkspace.getRoot().getFile(tpl).setContents(contents, true, true, new NullProgressMonitor());
        } catch (final CoreException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * Imports one or more zipped projects into the workspace. The workspace must be opened before
     * invoking this method. The Zip file is assumed to have the following structure:
<pre>
/           Zip Root
/Proj1      Project#1 Root
 +-- res1   Resource in Project#1
 +-- res2   "
/Proj2      Project#2 Root
...
</pre>
     * @param resourceStream Stream for reading the Zip File
     * @return The imported projects
     * @since 25.07.2007
     * @author Karsten Thoms
     */
    public List<IProject> importZippedProject (InputStream resourceStream) {
    	List<IProject> result = new ArrayList<IProject>();
    	
        try {
            IPath root = fWorkspace.getRoot().getFullPath();
            File tempFile = File.createTempFile("oawtest", "");
            tempFile.deleteOnExit();
            FSIO.writeSingleFile(new FileWriter(tempFile), new BufferedReader(new InputStreamReader(resourceStream)));
            ZipFile zipFile = new ZipFile(tempFile);
            ZipFileStructureProvider zipFileStructureProvider = new ZipFileStructureProvider(zipFile);
            ZipEntry zipRoot = zipFileStructureProvider.getRoot();
            
            // Assume that the root entry in the ZIP file is named "/"
            if ("/".equals(zipRoot.getName())) {
            	// Iterate over the contained projects
            	for (Iterator<?> itProject= zipFileStructureProvider.getChildren(zipRoot).iterator(); itProject.hasNext(); ) {
                	ZipEntry projectEntry = (ZipEntry) itProject.next();
                	// The project name has a trailing slash that we need to remove
                	String projectName = projectEntry.getName().substring(0, projectEntry.getName().length()-1);

                	// Create this project
                	IProject proj = createProject(projectName);
                	// Iterate over the project's children and import them recursively
            		for (Iterator it2=zipFileStructureProvider.getChildren(projectEntry).iterator(); it2.hasNext(); ) {
            			ZipEntry projectContent = (ZipEntry) it2.next();
            			// We use the ImportOperation to facilitate project import 
	                    ImportOperation op = new ImportOperation(
	                    		root, 
	                    		projectContent, 
	                    		zipFileStructureProvider, 
	                    		new IOverwriteQuery() {
	        						public String queryOverwrite(String pathString) {
	        							return IOverwriteQuery.ALL;
	        						}});
	                    op.run(new NullProgressMonitor());
            		}
            		result.add(proj);
            	}
            }
        } catch (final RuntimeException e) {
            throw e;
	    } catch (final Exception e) {
	        throw new RuntimeException(e);
	    }
	    return result;
    }
}

Back to the top