Change styles of existing HTMLDocument content in JTextPane

I am trying to build a chat client that uses JTextPane to display the conversations. I am having problems with highlighting the past sentences of a chat-participant chosen by the user. When implementing this I need to stick to using HTMLDocument as some of the content has to be HTML.

The best idea seemed to be using a different named styles for each user participating in the conversation. The idea is that when the text of a particular person needs to be highlighted, I just update his personal style and everything he has said should get highlighted as by magic. Unfortunately this does not work.

So to add text I use:

public void addMessage(String from, String message){
    HTMLDocument doc = (HTMLDocument) textPane.getStyledDocument();
    if(doc != null){
        try {
            String stylename = "from_" + from;
            if(textPane.getStyle(stylename) == null){
                LOG.debug("Did not find style. Adding new: " + stylename);
                Style nameStyle = textPane.addStyle(stylename, null);
     开发者_开发问答           StyleConstants.setForeground(nameStyle, Color.black);
                StyleConstants.setBold(nameStyle, true);
            }else{
                LOG.debug("Found existing style: " + textPane.getStyle(stylename));
            }
            doc.insertString(doc.getLength(), from + ": ", textPane.getStyle(stylename));
            doc.insertString(doc.getLength(), message + "\n", null);
        } catch (BadLocationException ble) {
            LOG.error("Could not insert text to tab");
        }
    }
}

So far so good... The text gets displayed in the textPane as I wished. However when I try to update the stylesheet at some future point and call:

public void highlight(String name, boolean highlight){
    Style fromStyle = textPane.getStyle("from_" + name);
    if(fromStyle != null){
        LOG.debug("found style, changing color");
        StyleConstants.setForeground(fromStyle, Color.red);
    }else{
        LOG.debug("fromStyle was NULL");
    }
}

..I can see that the style is found and my code gets executed - yet nothing changes on the screen.

I would like to ask if you have any suggestions on how I might try to resolve this issue. Is there a way to make this work with named styles or should I take some completely different approach?

Thank you

You just modified the Style object, you still have to apply the modified Style using setCharacterAttributes.

I have since created a solution which works for me. I will post here a functioning testcase for anyone interested. I ended up with a solution where I updated the style class in the StyleSheet and then called htmlDocument.setParagraphAttributes for every paragraph element.

import java.awt.BorderLayout;
import java.awt.Container;
import java.awt.EventQueue;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;

import javax.swing.JFrame;
import javax.swing.JTextPane;
import javax.swing.SwingUtilities;
import javax.swing.text.BadLocationException;
import javax.swing.text.Element;
import javax.swing.text.html.HTMLDocument;
import javax.swing.text.html.HTMLEditorKit;
import javax.swing.text.html.StyleSheet;

public class StyleChangeTest extends JFrame {

    public final static int APP_WIDTH = 640;
    public final static int APP_HEIGHT = 400;
    public JTextPane jTextPane;
    public StyleSheet styleSheet;
    public HTMLDocument htmlDocument;
    public HTMLEditorKit htmlEditorKit;
    public Element bodyElement;

    public static StyleChangeTest jTextPaneApp;

    public static void main(String[] args) throws InterruptedException, InvocationTargetException{
        EventQueue.invokeAndWait(new Runnable() {
            public void run() {
                try {
                    jTextPaneApp = new StyleChangeTest();
                    jTextPaneApp.setVisible(true);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        System.out.println("Started");
        Thread.currentThread().sleep(1000);
        SwingUtilities.invokeLater(new Runnable() {
            public void run() {
                try {
                    jTextPaneApp.change();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        System.out.println("Finished");
    }

    public StyleChangeTest(){
        setSize(APP_WIDTH, APP_HEIGHT);
        setResizable(false);
        setTitle("JTextPane App");

        styleSheet = new StyleSheet();
        styleSheet.addRule(".someclass1 {color: blue;}");
        styleSheet.addRule(".someclass2 {color: green;}");

        htmlEditorKit = new HTMLEditorKit();
        htmlEditorKit.setStyleSheet(styleSheet);
        htmlDocument = (HTMLDocument) htmlEditorKit.createDefaultDocument();
        jTextPane = new JTextPane();
        jTextPane.setEditorKit(htmlEditorKit);
        jTextPane.setDocument(htmlDocument);

        try {
            Element htmlElement = htmlDocument.getRootElements()[0];
            bodyElement = htmlElement.getElement(0);

            Container contentPane = getContentPane();
            contentPane.setLayout(new BorderLayout());
            contentPane.add(jTextPane, BorderLayout.CENTER);
            addWindowListener(new WindowAdapter(){
                public void windowClosing(WindowEvent e){
                    System.exit(0);
                }
            });

            addContent("<font class=someclass1>test 1</font><br>");
            addContent("<font class=someclass2>test 2</font><br>");

        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    public void reapplyStyles() {
        System.out.println("Reformating...");
        Element sectionElem = bodyElement.getElement(bodyElement.getElementCount() - 1);
        int paraCount = sectionElem.getElementCount();
        for (int i=0; i<paraCount; i++) {
            Element paraElem = sectionElem.getElement(i);
            //System.out.println("\tParagraph: " + (i+1) + " - " + paraElem);
            int rangeStart = paraElem.getStartOffset();
            int rangeEnd = paraElem.getEndOffset();
            htmlDocument.setParagraphAttributes(rangeStart, rangeEnd-rangeStart, paraElem.getAttributes(), true);
        }
    }


    public void change() throws BadLocationException, IOException{
        styleSheet = htmlEditorKit.getStyleSheet();
        styleSheet.addRule(".someclass1 {color: red;}");
        reapplyStyles();
        addContent("<font class=someclass1>test 3</font><br>");
    }


    private void addContent(String content) throws BadLocationException, IOException{
        Element contentElement = bodyElement.getElement(bodyElement.getElementCount() - 1);

        StringBuffer sbHtml = new StringBuffer();
        sbHtml.append("<font class=someclass>" + content + "</font><br>");

        htmlDocument.insertBeforeEnd(contentElement, sbHtml.toString());
    }

}