• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• inql-master
  • .github
    • ISSUE_TEMPLATE
      • feature_request.md
      • bug_report.md
  • Makefile
  • inql
    • introspection.py
    • actions
      • executor.py
      • browser.py
      • sendto.py
      • doc.txt
      • __init__.py
      • setcustomheader.py
    • burp_loader.py
    • constants.py
    • __main__.py
    • generators
      • html.py
      • schema.py
      • query.py
      • doc.txt
      • __init__.py
    • __init__.py
    • utils.py
    • burp_ext
      • scanner.py
      • extender.py
      • doc.txt
      • __init__.py
      • tab.py
      • editor.py
    • widgets
      • filetree.py
      • omnibar.py
      • doc.txt
      • __init__.py
      • tab.py
      • fileview.py
      • payloadview.py
      • propertyeditor.py
  • LICENSE
  • setup.py
  • README.md
  • bin
    • inql
  • requirements.txt
  • .gitignore
  • docs

import platform

from inql.utils import nop_evt

if platform.system() != "Java":
    print("Load this file inside jython, if you need the stand-alone tool run: inql")
    exit(-1)

from java.awt.event import FocusListener, KeyAdapter, KeyEvent
from javax.swing import JFrame, JPanel, JTextField, JButton
from java.awt import (BorderLayout, Color)


class _HintTextField(FocusListener, KeyAdapter):
    """
    HintTextField is a class responsible for showing an hint while the textfield is empty
    """

    def __init__(self, hint=None, action=None):
        if not hint: hint = 'hint'
        if not action: action = nop_evt
        self.this = JTextField(hint)
        self._hint = hint
        self._showing_hint = True
        self._enter_listener = None
        self.this.addFocusListener(self)
        self.this.addKeyListener(self)
        self.set_enter_evt_listener(action)

    def set_enter_evt_listener(self, enter_listener):
        """
        Add an evt listener to HintTextField

        :param enter_listener: lambda event listener
        :return:
        """
        self._enter_listener = enter_listener

    def keyPressed(self, e):
        """
        KeyAdapter override

        :param e: event containing the key pressed
        :return: None
        """
        if self._enter_listener and e.getKeyCode() == KeyEvent.VK_ENTER:
            self._enter_listener(e)
      
    def focusGained(self, e):
        """
        FocusListener override

        :param e: unused
        :return: None
        """
        if self.getText() == "":
            self.this.setText("")
            self._showing_hint = False

    def focusLost(self, e):
        """
        FocusListener override

        :param e: unused
        :return: None
        """
        if self.getText() == "":
            self.this.setText(self._hint)
            self._showing_hint = True

    def getText(self):
        """
        :return: the current text or "" if no text is wrote inside the textfield
        """
        if self._showing_hint:
            return ""
        else:
            return self.this.getText()

    def setText(self, txt):
        """
        Set Text

        :param txt: a string
        :return: None
        """
        self.this.setText(txt)
        self._showing_hint = False

    def reset(self):
        """
        Reset the HintBox
        :return: None
        """
        self.this.setText(self._hint)
        self._showing_hint = True

class Omnibar:
    """
    Omnibar represents a chrome alike textbox with behaviour similar to the one of a normal browser
    """

    def __init__(self, hint=None, label=None, action=None):
        if not hint: hint = 'Omnibar hint'
        if not label: label = 'Run'
        if not action: action = nop_evt
        self.this = JPanel()
        self.this.setLayout(BorderLayout())

        # Add an hinttextfield
        self._text = _HintTextField(hint, action)
        self.this.add(BorderLayout.CENTER, self._text.this)

        # Add a run buttpn
        button = JButton(label)
        button.addActionListener(action)
        self.this.add(BorderLayout.EAST, button)

    def getText(self):
        """
        :return: the current text or "" if no text is wrote inside the textfield
        """
        return self._text.getText()

    def setText(self, txt):
        """
        Set Text

        :param txt: a string
        :return: None
        """
        self._text.setText(txt)

    def reset(self):
        """
        Reset the HintBox
        :return: None
        """
        self._text.reset()

if __name__ == "__main__":
    frame = JFrame("Omnibar")
    frame.setForeground(Color.black)
    frame.setBackground(Color.lightGray)
    cp = frame.getContentPane()
    cp.add(Omnibar().this)
    frame.pack()
    frame.setVisible(True)
    frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)