class MainWindow(QMainWindow):
"""
Main window for the player application.
"""
def __init__(self, bind_address: str, port: int, filesToRead: [str]):
"""
:param bind_address: address to bind to when listening for live connections.
:param port: port to bind to when listening for live connections.
:param filesToRead: replay files to open.
"""
super().__init__()
self.liveWindow = LiveWindow(bind_address, port)
self.replayWindow = ReplayWindow()
self.tabManager = QTabWidget()
self.tabManager.addTab(self.replayWindow, "Replays")
self.tabManager.addTab(self.liveWindow, "Live connections")
self.setCentralWidget(self.tabManager)
openAction = QAction("Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip("Open a replay file")
openAction.triggered.connect(self.onOpenFile)
menuBar = self.menuBar()
fileMenu = menuBar.addMenu("File")
fileMenu.addAction(openAction)
for fileName in filesToRead:
self.replayWindow.openFile(fileName)
def onOpenFile(self):
fileName, _ = QFileDialog.getOpenFileName(self, "Open File")
if fileName:
self.replayWindow.openFile(fileName)
class MainWindow(QMainWindow):
"""
Main window for the player application.
"""
def __init__(self, bind_address: str, port: int, filesToRead: [str]):
"""
:param bind_address: address to bind to when listening for live connections.
:param port: port to bind to when listening for live connections.
:param filesToRead: replay files to open.
"""
super().__init__()
self.liveWindow = LiveWindow(bind_address, port)
self.replayWindow = ReplayWindow()
self.tabManager = QTabWidget()
self.tabManager.addTab(self.liveWindow, "Live connections")
self.tabManager.addTab(self.replayWindow, "Replays")
self.setCentralWidget(self.tabManager)
openAction = QAction("Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip("Open a replay file")
openAction.triggered.connect(self.onOpenFile)
windowsRAction = QAction("Windows+R", self)
windowsRAction.setShortcut("Ctrl+Alt+R")
windowsRAction.setStatusTip("Send a Windows+R key sequence")
windowsRAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_R]))
windowsLAction = QAction("Windows+L", self)
windowsLAction.setShortcut("Ctrl+Alt+L")
windowsLAction.setStatusTip("Send a Windows+L key sequence")
windowsLAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_L]))
windowsEAction = QAction("Windows+E", self)
windowsEAction.setShortcut("Ctrl+Alt+E")
windowsEAction.setStatusTip("Send a Windows+E key sequence")
windowsEAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_E]))
typeTextAction = QAction("Type text...", self)
typeTextAction.setShortcut("Ctrl+Alt+T")
typeTextAction.setStatusTip("Simulate typing on the keyboard")
typeTextAction.triggered.connect(self.sendText)
menuBar = self.menuBar()
fileMenu = menuBar.addMenu("File")
fileMenu.addAction(openAction)
commandMenu = menuBar.addMenu("Command")
commandMenu.addAction(windowsRAction)
commandMenu.addAction(windowsLAction)
commandMenu.addAction(windowsEAction)
commandMenu.addAction(typeTextAction)
for fileName in filesToRead:
self.replayWindow.openFile(fileName)
def onOpenFile(self):
fileName, _ = QFileDialog.getOpenFileName(self, "Open File")
if fileName:
self.tabManager.setCurrentWidget(self.replayWindow)
self.replayWindow.openFile(fileName)
def sendKeySequence(self, keys: [Qt.Key]):
if self.tabManager.currentWidget() is self.liveWindow:
self.liveWindow.sendKeySequence(keys)
def sendText(self):
if self.tabManager.currentWidget() is not self.liveWindow:
return
text, success = QInputDialog.getMultiLineText(self, "Type text...",
"Text to type:")
if not success:
return
self.liveWindow.sendText(text)
class MainWindow(QMainWindow):
"""
Main window for the player application.
"""
updateCountSignal = Signal()
def __init__(self, bind_address: str, port: int, filesToRead: [str]):
"""
:param bind_address: address to bind to when listening for live connections.
:param port: port to bind to when listening for live connections.
:param filesToRead: replay files to open.
"""
super().__init__()
# TODO : Rework into a class if we add more options later.
self.options = {
"focusNewTab":
True, # Useful whenever we are getting flooded with connections (or scanned), and we only want to monitor one at a time.
"closeTabOnCtrlW": True # Allow user to toggle Ctrl+W passthrough.
}
self.liveWindow = LiveWindow(bind_address,
port,
self.updateCountSignal,
self.options,
parent=self)
self.replayWindow = ReplayWindow(self.options, parent=self)
self.tabManager = QTabWidget()
self.tabManager.addTab(self.liveWindow, "Live connections")
self.tabManager.addTab(self.replayWindow, "Replays")
self.setCentralWidget(self.tabManager)
self.updateCountSignal.connect(self.updateTabConnectionCount)
self.resizeObservers: List[BaseTab] = []
# File menu
openAction = QAction("Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip("Open a replay file")
openAction.triggered.connect(self.onOpenFile)
# Command menu
windowsRAction = QAction("Windows+R", self)
windowsRAction.setShortcut("Ctrl+Alt+R")
windowsRAction.setStatusTip("Send a Windows+R key sequence")
windowsRAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_R]))
windowsLAction = QAction("Windows+L", self)
windowsLAction.setShortcut("Ctrl+Alt+L")
windowsLAction.setStatusTip("Send a Windows+L key sequence")
windowsLAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_L]))
windowsEAction = QAction("Windows+E", self)
windowsEAction.setShortcut("Ctrl+Alt+E")
windowsEAction.setStatusTip("Send a Windows+E key sequence")
windowsEAction.triggered.connect(
lambda: self.sendKeySequence([Qt.Key.Key_Meta, Qt.Key.Key_E]))
typeTextAction = QAction("Type text...", self)
typeTextAction.setShortcut("Ctrl+Alt+T")
typeTextAction.setStatusTip("Simulate typing on the keyboard")
typeTextAction.triggered.connect(self.sendText)
# Options menu
focusTabAction = QAction("Focus new connections", self)
focusTabAction.setCheckable(True)
focusTabAction.setChecked(self.options.get("focusNewTab"))
focusTabAction.triggered.connect(lambda: self.toggleFocusNewTab())
closeTabOnCtrlW = QAction("Close current tab on Ctrl+W", self)
closeTabOnCtrlW.setCheckable(True)
closeTabOnCtrlW.setChecked(self.options.get("closeTabOnCtrlW"))
closeTabOnCtrlW.triggered.connect(lambda: self.toggleCloseTabOnCtrlW())
# Create menu
menuBar = self.menuBar()
fileMenu = menuBar.addMenu("File")
fileMenu.addAction(openAction)
commandMenu = menuBar.addMenu("Command")
commandMenu.addAction(windowsRAction)
commandMenu.addAction(windowsLAction)
commandMenu.addAction(windowsEAction)
commandMenu.addAction(typeTextAction)
optionsMenu = menuBar.addMenu("Options")
optionsMenu.addAction(focusTabAction)
optionsMenu.addAction(closeTabOnCtrlW)
# If there's only one replay, play it immediately.
if len(filesToRead) == 1:
self.tabManager.setCurrentWidget(self.replayWindow)
self.replayWindow.openFile(filesToRead[0], autoplay=True)
else:
for f in filesToRead:
self.replayWindow.openFile(f)
def onOpenFile(self):
fileNames, _ = QFileDialog.getOpenFileNames(self, "Open File(s)")
if fileNames:
self.tabManager.setCurrentWidget(self.replayWindow)
for fileName in fileNames:
self.replayWindow.openFile(fileName)
def sendKeySequence(self, keys: [Qt.Key]):
if self.tabManager.currentWidget() is self.liveWindow:
self.liveWindow.sendKeySequence(keys)
def sendText(self):
if self.tabManager.currentWidget() is not self.liveWindow:
return
text, success = QInputDialog.getMultiLineText(self, "Type text...",
"Text to type:")
if not success:
return
self.liveWindow.sendText(text)
def toggleFocusNewTab(self):
self.options["focusNewTab"] = not self.options.get("focusNewTab")
def toggleCloseTabOnCtrlW(self):
self.options["closeTabOnCtrlW"] = not self.options.get(
"closeTabOnCtrlW")
def updateTabConnectionCount(self):
"""
Update the first tab (Live connections) with the current number of tabs
"""
self.tabManager.setTabText(
0, "Live connections (%d)" % self.liveWindow.count())