166 lines
4.4 KiB
Python
166 lines
4.4 KiB
Python
import time
|
|
from typing import Any
|
|
import socketio
|
|
import socketio.exceptions
|
|
from . import storage
|
|
|
|
class DockgeConnection:
|
|
"""
|
|
Provider class for Dockge
|
|
Provides all the functionality for connecting, logging in and executing commands
|
|
"""
|
|
class LoginException(BaseException):
|
|
"""
|
|
Special exception when login fails too often
|
|
"""
|
|
|
|
_sio: socketio.Client
|
|
_host: str
|
|
_logged_in: bool
|
|
|
|
_stacklist: Any
|
|
|
|
def __init__(self):
|
|
self._logged_in = False
|
|
self._host = storage.get("host")
|
|
if self._host is None:
|
|
raise ValueError("Host for Dockge is not defined!")
|
|
self._sio = socketio.Client(logger=False, engineio_logger=False)
|
|
self._stacklist = None
|
|
self._init_events()
|
|
|
|
def _init_events(self):
|
|
@self._sio.event
|
|
def connect():
|
|
self.connect()
|
|
print("Connected!")
|
|
|
|
@self._sio.event
|
|
def disconnect():
|
|
self._logged_in = False
|
|
print("Disconnected!")
|
|
|
|
@self._sio.event
|
|
def connect_error(data):
|
|
print("The connection failed!")
|
|
print(data)
|
|
|
|
@self._sio.on('info')
|
|
def info(data):
|
|
if all(k in data for k in ("version", "latestVersion", "isContainer", "primaryHostname")):
|
|
print(f"Dockge Version: {data['version']}")
|
|
|
|
@self._sio.on('agent')
|
|
def agent(*args):
|
|
if args[0] == "stackList":
|
|
self._stacklist = args[1]
|
|
|
|
# Callbacks
|
|
def _login_process(self, data):
|
|
success = False
|
|
if "ok" in data and "token" in data:
|
|
print(f"Login was {'successful' if data['ok'] else 'unsuccessful'}!")
|
|
success = True
|
|
else:
|
|
print("Issue with login procedure")
|
|
return success
|
|
|
|
# Functions
|
|
def connect_and_login(self):
|
|
"""
|
|
Connect to the websocket
|
|
"""
|
|
self._sio.connect(f"https://{self._host}/socket.io/", transports=['websocket'])
|
|
self.connect()
|
|
|
|
def connect(self):
|
|
"""
|
|
Log into dockge using basicauth
|
|
Retries 5 times when timeouts occur
|
|
"""
|
|
if self._logged_in:
|
|
return
|
|
|
|
data = None
|
|
retry, count = True, 0
|
|
|
|
while retry and count < 5:
|
|
try:
|
|
data = self._sio.call(
|
|
"login",
|
|
{
|
|
"username": storage.get("username", encoded=True),
|
|
"password": storage.get("password", encoded=True),
|
|
"token":""
|
|
},
|
|
timeout=5
|
|
)
|
|
retry = False
|
|
except socketio.exceptions.TimeoutError:
|
|
retry = True
|
|
count += 1
|
|
|
|
if retry or not self._login_process(data):
|
|
raise self.LoginException
|
|
self._logged_in = True
|
|
|
|
def list_stacks(self):
|
|
"""
|
|
Requests stack list from dockge, returns list when event was sent
|
|
"""
|
|
self._sio.emit("agent", ("", "requestStackList"))
|
|
while self._stacklist is None:
|
|
time.sleep(0.5)
|
|
retval = self._stacklist
|
|
self._stacklist = None
|
|
return retval
|
|
|
|
def list_stack(self, name: str):
|
|
"""
|
|
Lists status for a stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "serviceStatusList", name), timeout=5)
|
|
return ret
|
|
|
|
def restart(self, name):
|
|
"""
|
|
Restarts a given stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "restartStack", name), timeout=10)
|
|
return ret
|
|
|
|
def update(self, name):
|
|
"""
|
|
Updates a given stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "updateStack", name), timeout=10)
|
|
return ret
|
|
|
|
def stop(self, name):
|
|
"""
|
|
Stops a given stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "stopStack", name), timeout=10)
|
|
return ret
|
|
|
|
def start(self, name):
|
|
"""
|
|
Starts a given stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "startStack", name), timeout=10)
|
|
return ret
|
|
|
|
def down(self, name):
|
|
"""
|
|
Stops and downs a given stack
|
|
"""
|
|
ret = self._sio.call("agent", ("", "downStack", name), timeout=10)
|
|
return ret
|
|
|
|
def disconnect(self):
|
|
"""
|
|
Logs out of dockge
|
|
"""
|
|
self._sio.emit("logout")
|
|
self._sio.disconnect()
|