def log(self, msg, color=None, bold=False, end='\n'): msg = str(msg).replace("\n", "\n ") + end doc = self._log.getStyledDocument() aset = SimpleAttributeSet() if color: StyleConstants.setForeground(aset, color) if bold: StyleConstants.setBold(aset, True) doc.insertString(self._log.getDocument().getLength(), msg, aset) self._logButton.setEnabled(True)
def new_style(name, color=None, bold=None, italic=None, underline=None): style = self.doc.addStyle(name, self.parent_style) if color is not None: if isinstance(color, str): color = awtColor( int(color[0:2], 16), int(color[2:4], 16), int(color[4:6], 16) ) StyleConstants.setForeground(style, color) if bold is not None: StyleConstants.setBold(style, bold) if italic is not None: StyleConstants.setItalic(style, italic) if underline is not None: StyleConstants.setUnderline(style, underline) return style
def getUiComponent(self): """Burp uses this method to obtain the component that should be used as the contents of the custom tab when it is displayed. Returns a awt.Component. """ # GUI happens here from javax.swing import (JPanel, JSplitPane, JList, JTextPane, JScrollPane, ListSelectionModel, JLabel, JTabbedPane, JEditorPane) from java.awt import BorderLayout panel = JPanel(BorderLayout()) # create a list and then JList out of it. colors = [ "red", "orange", "yellow", "green", "cyan", "blue", "pink", "magenta", "gray", "zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz" ] # create a list - the list is not used in this example list1 = JList(colors) list1.selectionMode = ListSelectionModel.SINGLE_SELECTION # create a StyledDocument for tab 1 from javax.swing.text import DefaultStyledDocument doc = DefaultStyledDocument() # create a JTextPane from doc tab1 = JTextPane(doc) tab1.editable = False # we can add more styles # new styles can be a child of previous styles # our first style is a child of the default style from javax.swing.text import StyleContext, StyleConstants defaultStyle = StyleContext.getDefaultStyleContext().getStyle( StyleContext.DEFAULT_STYLE) # returns a Style regular = doc.addStyle("regular", defaultStyle) StyleConstants.setFontFamily(defaultStyle, "Times New Roman") # make different styles from regular style1 = doc.addStyle("italic", regular) StyleConstants.setItalic(style1, True) style1 = doc.addStyle("bold", regular) StyleConstants.setBold(style1, True) style1 = doc.addStyle("small", regular) StyleConstants.setFontSize(style1, 10) style1 = doc.addStyle("large", regular) StyleConstants.setFontSize(style1, 16) # insert text doc.insertString(doc.length, "This is regular\n", doc.getStyle("regular")) doc.insertString(doc.length, "This is italic\n", doc.getStyle("italic")) doc.insertString(doc.length, "This is bold\n", doc.getStyle("bold")) doc.insertString(doc.length, "This is small\n", doc.getStyle("small")) doc.insertString(doc.length, "This is large\n", doc.getStyle("large")) # create the tabbedpane tabs = JTabbedPane() tabs.addTab("Tab 1", tab1) # create splitpane - horizontal split spl = JSplitPane(JSplitPane.HORIZONTAL_SPLIT, JScrollPane(list1), tabs) panel.add(spl) return panel