root/org.axdt.as3/src/org/axdt/as3/templates/AS3TemplateContext.java @ 3d0622cd77233477dd222f1c00c4f35afeb67815

/**
 * 
 */
package org.axdt.as3.templates;

import java.util.StringTokenizer;

import org.axdt.as3.preferences.AS3EditorPreferences;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.jface.text.BadLocationException;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.IRegion;
import org.eclipse.jface.text.templates.DocumentTemplateContext;
import org.eclipse.jface.text.templates.Template;
import org.eclipse.jface.text.templates.TemplateBuffer;
import org.eclipse.jface.text.templates.TemplateContextType;
import org.eclipse.jface.text.templates.TemplateException;

public class AS3TemplateContext extends DocumentTemplateContext {
        public AS3TemplateContext(TemplateContextType type, IDocument document, int offset, int length) {
                super(type, document, offset, length);
        }

        @Override
        public TemplateBuffer evaluate(Template template) throws BadLocationException,
                        TemplateException {
                return super.evaluate(correctIndention(template, getDocument(), getStart()));
        }

        public static Template correctIndention(Template template, IDocument d, int offset) {
                if (offset == -1 || d.getLength() == 0) return template;
                String pattern = template.getPattern();
                String delim = d.getLegalLineDelimiters()[0];
                IPreferenceStore store = AS3EditorPreferences.getInstance().getStore();
                boolean tab2Space = store.getBoolean(AS3EditorPreferences.TAB_TO_SPACE);
                int tabWidth = store.getInt(AS3EditorPreferences.TAB_WIDTH);
                StringBuffer buf = new StringBuffer();
                for (int i = 0; i < tabWidth; i++)
                        buf.append(' ');
                String tabSpaces = buf.toString();
                for (String nl:d.getLegalLineDelimiters())
                        if (nl.equals("\n")) delim = nl;
                if (pattern.indexOf('\n') < 0 && pattern.indexOf('\r') < 0) return template;
                try {
                        int p = (offset == d.getLength() ? offset - 1 : offset);
                        IRegion info = d.getLineInformationOfOffset(p);
                        int start = info.getOffset();
                        // find white spaces
                        int end = findEndOfWhiteSpace(d, start, offset);
                        buf = new StringBuffer(delim);
                        if (end > start) {
                                // append to input
                                buf.append(d.get(start, end - start));
                        }
                        String indent = buf.toString();
                        buf = new StringBuffer();
                        StringTokenizer tokenizer = new StringTokenizer(pattern, "\r\n\t", true);
                        char lastnl = 0;
                        while (tokenizer.hasMoreTokens()) {
                                String token = tokenizer.nextToken();
                                char c = token.charAt(0);
                                if (c == '\n' || c == '\r') {
                                        if (lastnl == 0 || lastnl == c) buf.append(indent);
                                        lastnl = c;
                                        continue;
                                }
                                lastnl = 0;
                                // TODO: tab2space preference
                                if (tab2Space && token.charAt(0) == '\t') {
                                        buf.append(tabSpaces);
                                } else buf.append(token);
                        }
                        return new Template(template.getName(), template.getDescription(), template
                                        .getContextTypeId(), buf.toString(), template.isAutoInsertable());
                } catch (BadLocationException e) {
                }
                return template;
        }

        private static int findEndOfWhiteSpace(IDocument document, int offset, int end)
                        throws BadLocationException {
                while (offset < end) {
                        char c = document.getChar(offset);
                        if (c != ' ' && c != '\t') {
                                return offset;
                        }
                        offset++;
                }
                return end;
        }
}