fix #774 infinite loop in net.sourceforge.phpeclipse.builder.IdentifierIndexManager...
[phpeclipse.git] / net.sourceforge.phpeclipse / src / net / sourceforge / phpdt / internal / corext / template / php / CodeTemplateContext.java
1 /*******************************************************************************
2  * Copyright (c) 2000, 2004 IBM Corporation and others.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Common Public License v1.0
5  * which accompanies this distribution, and is available at
6  * http://www.eclipse.org/legal/cpl-v10.html
7  *
8  * Contributors:
9  *     IBM Corporation - initial API and implementation
10  *******************************************************************************/
11 package net.sourceforge.phpdt.internal.corext.template.php;
12
13 import java.util.Iterator;
14
15 import net.sourceforge.phpdt.core.ICompilationUnit;
16 import net.sourceforge.phpdt.core.IJavaProject;
17 import net.sourceforge.phpeclipse.PHPeclipsePlugin;
18
19 import org.eclipse.jface.text.BadLocationException;
20 import org.eclipse.jface.text.DefaultLineTracker;
21 import org.eclipse.jface.text.ILineTracker;
22 import org.eclipse.jface.text.IRegion;
23 import org.eclipse.jface.text.templates.Template;
24 import org.eclipse.jface.text.templates.TemplateBuffer;
25 import org.eclipse.jface.text.templates.TemplateContext;
26 import org.eclipse.jface.text.templates.TemplateException;
27 import org.eclipse.jface.text.templates.TemplateTranslator;
28 import org.eclipse.jface.text.templates.TemplateVariableResolver;
29
30 public class CodeTemplateContext extends TemplateContext {
31
32         private String fLineDelimiter;
33
34         private IJavaProject fProject;
35
36         public CodeTemplateContext(String contextTypeName, IJavaProject project,
37                         String lineDelim) {
38                 super(PHPeclipsePlugin.getDefault().getCodeTemplateContextRegistry()
39                                 .getContextType(contextTypeName));
40                 fLineDelimiter = lineDelim;
41                 fProject = project;
42         }
43
44         public IJavaProject getJavaProject() {
45                 return fProject;
46         }
47
48         /*
49          * @see net.sourceforge.phpdt.internal.corext.template.TemplateContext#evaluate(net.sourceforge.phpdt.internal.corext.template.Template)
50          */
51         public TemplateBuffer evaluate(Template template)
52                         throws BadLocationException, TemplateException {
53                 // test that all variables are defined
54                 Iterator iterator = getContextType().resolvers();
55                 while (iterator.hasNext()) {
56                         TemplateVariableResolver var = (TemplateVariableResolver) iterator
57                                         .next();
58                         if (var instanceof CodeTemplateContextType.CodeTemplateVariableResolver) {
59                                 // Assert.isNotNull(getVariable(var.getType()), "Variable " +
60                                 // var.getType() + "not defined"); //$NON-NLS-1$ //$NON-NLS-2$
61                         }
62                 }
63
64                 if (!canEvaluate(template))
65                         return null;
66
67                 String pattern = changeLineDelimiter(template.getPattern(),
68                                 fLineDelimiter);
69
70                 TemplateTranslator translator = new TemplateTranslator();
71                 TemplateBuffer buffer = translator.translate(pattern);
72                 getContextType().resolve(buffer, this);
73
74                 return buffer;
75         }
76
77         private static String changeLineDelimiter(String code, String lineDelim) {
78                 try {
79                         ILineTracker tracker = new DefaultLineTracker();
80                         tracker.set(code);
81                         int nLines = tracker.getNumberOfLines();
82                         if (nLines == 1) {
83                                 return code;
84                         }
85
86                         StringBuffer buf = new StringBuffer();
87                         for (int i = 0; i < nLines; i++) {
88                                 if (i != 0) {
89                                         buf.append(lineDelim);
90                                 }
91                                 IRegion region = tracker.getLineInformation(i);
92                                 String line = code.substring(region.getOffset(), region
93                                                 .getOffset()
94                                                 + region.getLength());
95                                 buf.append(line);
96                         }
97                         return buf.toString();
98                 } catch (BadLocationException e) {
99                         // can not happen
100                         return code;
101                 }
102         }
103
104         /*
105          * (non-Javadoc)
106          * 
107          * @see net.sourceforge.phpdt.internal.corext.template.TemplateContext#canEvaluate(net.sourceforge.phpdt.internal.corext.template.Template)
108          */
109         public boolean canEvaluate(Template template) {
110                 return true;
111         }
112
113         public void setCompilationUnitVariables(ICompilationUnit cu) {
114                 setVariable(CodeTemplateContextType.FILENAME, cu.getElementName());
115                 setVariable(CodeTemplateContextType.PACKAGENAME, cu.getParent()
116                                 .getElementName());
117                 setVariable(CodeTemplateContextType.PROJECTNAME, cu.getJavaProject()
118                                 .getElementName());
119         }
120
121         public void setFileNameVariable(String filename, String projectname) {
122                 setVariable(CodeTemplateContextType.FILENAME, filename);
123                 setVariable(CodeTemplateContextType.PROJECTNAME, projectname);
124         }
125 }