mytextpane.java

SqlPlus(1.01) 这是一个纯JAVA的数据操作软件

package com.jimw.mysqlplus;

import java.util.StringTokenizer;

import java.awt.Color;
import java.awt.event.*;

import javax.swing.*;
import javax.swing.text.*;
import javax.swing.event.*;
import javax.swing.text.rtf.RTFEditorKit;

public class MyTextPane extends JTextPane {

    protected StyleContext m_context;
    protected DefaultStyledDocument m_doc;
    private   MutableAttributeSet keyAttr,functionAttr,normalAttr;
    private   MutableAttributeSet inputAttributes =
                new RTFEditorKit().getInputAttributes();

    private final int KEYWORD_TYPE = 0,FUNCTION_TYPE = 1;

    public MyTextPane() {
        super();
        m_context = new StyleContext();
        m_doc = new DefaultStyledDocument(m_context);
        this.setDocument(m_doc);

        this.addKeyListener(new KeyAdapter() {
            public void keyReleased(KeyEvent ke) {
                syntaxParse();
            }
        });
        keyAttr = new SimpleAttributeSet();
        StyleConstants.setForeground(keyAttr, Color.blue);

        functionAttr = new SimpleAttributeSet();
        StyleConstants.setForeground(functionAttr, Color.red);

        normalAttr = new SimpleAttributeSet();
        StyleConstants.setForeground(normalAttr, Color.black);
    }

    public void syntaxParse() {
        try {
            String s = null;
            Element root = m_doc.getDefaultRootElement();
//取当前行
            int cursorPos = this.getCaretPosition();//当前光标的位置
            int line = root.getElementIndex(cursorPos);

            Element para = root.getElement(line);
            int start = para.getStartOffset();
            int end = para.getEndOffset() - 1; //删除\r字符
            s = m_doc.getText(start, end - start).toUpperCase();

            int i = 0;
            int xStart = 0;
            int type = KEYWORD_TYPE;
//分析关键字---
            m_doc.setCharacterAttributes(start, s.length(),normalAttr, false);
            MyStringTokenizer st = new MyStringTokenizer(s);
            while( st.hasMoreTokens()) {
                type = KEYWORD_TYPE;
                s = st.nextToken();
                if ( s == null) return;
                for (i = 0; i < MySqlPlusFrame.command.length; i++ ) {
                    if (s.equals(MySqlPlusFrame.command[i])) break;
                }
                if ( i >= MySqlPlusFrame.command.length ) {
                  for (i = 0; i < MySqlPlusFrame.keyWords.length; i++ ) {
                      if (s.equals(MySqlPlusFrame.keyWords[i])) break;
                  }
                  if ( i >= MySqlPlusFrame.keyWords.length ) {
                    for (i = 0; i < MySqlPlusFrame.function.length; i++ ) {
                        if ( s.equals(MySqlPlusFrame.function[i]) ||
                             (s.indexOf(MySqlPlusFrame.function[i]+"(") == 0) ) { //函数必须包括"("
                            type = FUNCTION_TYPE;
                            break;
                        }
                    }
                    if ( i >= MySqlPlusFrame.function.length ) continue;
                  }
                }

                xStart = st.getCurrPosition();

                if (type == FUNCTION_TYPE)
                    m_doc.setCharacterAttributes(start+xStart, MySqlPlusFrame.function[i].length(),
                                             functionAttr, false);
                else
                    m_doc.setCharacterAttributes(start+xStart, s.length() + 1,
                                             keyAttr, false);
            }
            inputAttributes.addAttributes(normalAttr);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    public void insertString(String str) {
        try {
            m_doc.insertString(m_doc.getLength(),"\r\n"+str,normalAttr);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    public String getSelectedText() {
        String query = super.getSelectedText();
        if (query == null) {
            Element root = m_doc.getDefaultRootElement();
//取当前行
            int cursorPos = this.getCaretPosition();//当前光标的位置
            int line = root.getElementIndex(cursorPos);

            Element para = root.getElement(line);
            int start = para.getStartOffset();
            int end = para.getEndOffset();
            try {
                super.setSelectionStart(start);
                super.setSelectionEnd(end);
                query = m_doc.getText(start, end - start).toUpperCase();
            } catch (Exception ex) {
                ex.printStackTrace();
            }

            if (query == null)
                return null;
        }
        query = query.trim();
        if (query.equals("")) return null;
        return query;
    }

    public static void main(String[] args) {
        JFrame frame = new JFrame("test text pane");
        frame.getContentPane().add(new MyTextPane());
        WindowListener wndCloser = new WindowAdapter() {
            public void windowClosing(WindowEvent e) {
                System.exit(0);
            }
        };
        frame.addWindowListener(wndCloser);
        frame.pack();
        frame.show();
    }
}

/*返回字符串所在的位置
*
*/
class MyStringTokenizer extends StringTokenizer{
    String sval = " ";
    String oldStr,str;
    int m_currPosition = 0,m_beginPosition=0;
    MyStringTokenizer(String str) {
        super(str," ");
        this.oldStr = str;
        this.str = str;
    }

    public String nextToken() {
       try {
           String s = super.nextToken();
           int pos = -1;

           if (oldStr.equals(s)) {
               return s;
           }

           pos = str.indexOf(s + sval);
           if ( pos == -1) {
               pos = str.indexOf(sval + s);
               if ( pos == -1)
                   return null;
               else pos += 1;
           }

           int xBegin = pos + s.length();
           str = str.substring(xBegin);

           m_currPosition = m_beginPosition + pos;
           m_beginPosition = m_beginPosition + xBegin;
           return s;
       } catch (java.util.NoSuchElementException ex) {
           ex.printStackTrace();
           return null;
       }
    }

   public int getCurrPosition() {
       return m_currPosition;
   }
}