Added sshproxy

master
Brainwrecked Tech 4 years ago
parent c80dca759c
commit 6c0c68dd2e

@ -19,6 +19,9 @@ mkminchroot
ssh-multirun ssh-multirun
: Runs the same command on multiple machines : Runs the same command on multiple machines
ssh-proxy
: Systray applet to enabled/disable an SSH proxy tunnel
ssh-reboot ssh-reboot
: Reboots a machine via `ssh` : Reboots a machine via `ssh`

@ -0,0 +1,146 @@
#!/usr/bin/env python
# ©2020 Shzublox, BrainreckedTech
# Licensed under the GPL v3
import subprocess, wx, wx.adv
PROGRAM_NAME = 'SSH Proxy'
STOPPED_ICON = "/usr/share/pixmaps/sshproxy-unused.png"
STARTED_ICON = "/usr/share/pixmaps/sshproxy-active.png"
DEFAULT_PORT = "8080"
DEFAULT_ADDR = "proxy"
def create_menu_item(menu, label, func):
item = wx.MenuItem(menu, -1, label)
menu.Bind(wx.EVT_MENU, func, id=item.GetId())
menu.Append(item)
return item
class MyTaskBarIcon(wx.adv.TaskBarIcon):
def __init__(self, frame):
wx.adv.TaskBarIcon.__init__(self)
self.frame = frame
self.SetIcon(self.frame.stopped_icon, PROGRAM_NAME)
self.Bind(wx.adv.EVT_TASKBAR_LEFT_DOWN, self.leftclicked)
def CreatePopupMenu(self):
menu = wx.Menu()
create_menu_item(menu, 'Configure', self.configure)
create_menu_item(menu, 'Exit', self.frame.close)
return menu
def leftclicked(self, event):
if self.frame.ssh:
self.frame.on_stop(None)
else:
self.frame.on_start(None)
def configure(self, event):
self.frame.Show()
self.frame.Restore()
class MyFrame(wx.Frame):
def __init__(self, parent, stopped_icon, started_icon):
wx.Frame.__init__(self, parent, title=PROGRAM_NAME,
style = wx.DEFAULT_FRAME_STYLE &
~(wx.RESIZE_BORDER | wx.MINIMIZE_BOX
| wx.MAXIMIZE_BOX | wx.CLOSE_BOX))
self.ssh = None
self.stopped_icon = stopped_icon
self.started_icon = started_icon
self.SetIcon(stopped_icon)
self.tray = MyTaskBarIcon(self)
self.Bind(wx.EVT_ICONIZE, self.iconize)
self.Bind(wx.EVT_CLOSE, self.close)
panel_sizer = wx.FlexGridSizer(cols=1)
text_sizer = wx.FlexGridSizer(cols=2)
button_sizer = wx.BoxSizer()
panel = wx.Panel(self)
portlabel = wx.StaticText(panel, label="Port: ")
self.portbox = wx.TextCtrl(panel, value=DEFAULT_PORT, size=(300, -1))
remotelabel = wx.StaticText(panel, label="Remote: ")
self.remotebox = wx.TextCtrl(panel, value=DEFAULT_ADDR, size=(300, -1))
self.startbutton = wx.Button(panel, label="Start")
self.stopbutton = wx.Button(panel, label="Stop")
self.stopbutton.Disable()
text_sizer.Add(portlabel, 1, wx.EXPAND)
text_sizer.Add(self.portbox, 1, wx.EXPAND)
text_sizer.Add(remotelabel, 1, wx.EXPAND)
text_sizer.Add(self.remotebox, 1, wx.EXPAND)
text_sizer.Fit(self)
button_sizer.Add(self.startbutton, 1, wx.EXPAND)
button_sizer.Add(self.stopbutton, 1, wx.EXPAND)
button_sizer.Fit(self)
panel_sizer.Add(text_sizer, 1, wx.EXPAND)
panel_sizer.Add(button_sizer, 1, wx.EXPAND)
panel_sizer.Fit(self)
self.startbutton.Bind(wx.EVT_BUTTON, self.on_start)
self.stopbutton.Bind(wx.EVT_BUTTON, self.on_stop)
self.SetSizer(panel_sizer)
def close(self, event):
self.tray.RemoveIcon()
self.tray.Destroy()
self.Destroy()
def iconize(self, event):
if self.IsIconized():
self.Show(False)
self.Restore()
def on_start(self, event):
if self.portbox.GetValue() == "" or self.remotebox.GetValue() == "":
return
self.startbutton.Disable()
self.stopbutton.Enable()
self.tray.SetIcon(self.started_icon)
self.ssh = subprocess.Popen(["ssh", "-ND",
"%s" % self.portbox.GetValue(),
"%s" % self.remotebox.GetValue()],
stdin = subprocess.PIPE,
stdout = subprocess.PIPE,
stderr = subprocess.PIPE,
universal_newlines = True)
def on_stop(self, event):
self.startbutton.Enable()
self.stopbutton.Disable()
self.tray.SetIcon(self.stopped_icon)
if self.ssh:
self.ssh.terminate()
self.ssh = None
def main():
app = wx.App(False)
image = wx.Image(STOPPED_ICON, wx.BITMAP_TYPE_ANY)
bmp = wx.Bitmap(image)
stopped_icon = wx.Icon()
stopped_icon.CopyFromBitmap(bmp)
image = wx.Image(STARTED_ICON, wx.BITMAP_TYPE_ANY)
bmp = wx.Bitmap(image)
started_icon = wx.Icon()
started_icon.CopyFromBitmap(bmp)
frame = MyFrame(None, stopped_icon, started_icon)
app.MainLoop()
if __name__ == "__main__":
main()

Binary file not shown.

After

Width:  |  Height:  |  Size: 4.6 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 4.8 KiB

Loading…
Cancel
Save