mirror of
https://github.com/hexedtech/codemp-sublime.git
synced 2024-12-23 05:04:52 +01:00
66 lines
No EOL
1.6 KiB
Python
66 lines
No EOL
1.6 KiB
Python
import logging
|
|
import codemp
|
|
|
|
from ..utils import some
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class SessionManager():
|
|
def __init__(self) -> None:
|
|
self._running = False
|
|
self._driver = None
|
|
self._client = None
|
|
|
|
def is_init(self):
|
|
return self._running and self._driver is not None
|
|
|
|
def is_active(self):
|
|
return self.is_init() and self._client is not None
|
|
|
|
@property
|
|
def client(self):
|
|
return some(self._client)
|
|
|
|
def get_or_init(self) -> codemp.Driver:
|
|
if self._driver:
|
|
return self._driver
|
|
|
|
self._driver = codemp.init()
|
|
logger.debug("registering logger callback...")
|
|
if not codemp.set_logger(lambda msg: logger.debug(msg), False):
|
|
logger.debug(
|
|
"could not register the logger... \
|
|
If reconnecting it's ok, \
|
|
the previous logger is still registered"
|
|
)
|
|
self._running = True
|
|
return self._driver
|
|
|
|
def stop(self):
|
|
if not self._driver:
|
|
return
|
|
|
|
self.drop_client()
|
|
self._driver.stop()
|
|
self._running = False
|
|
self._driver = None
|
|
|
|
def connect(self, config: codemp.Config) -> codemp.Client:
|
|
if not self.is_init():
|
|
self.get_or_init()
|
|
|
|
self._client = codemp.connect(config).wait()
|
|
self.config = config
|
|
logger.debug(f"Connected to '{self.config.host}' as user {self._client.current_user().name} (id: {self._client.current_user().id})")
|
|
return self._client
|
|
|
|
def get_workspaces(self, owned: bool = True, invited: bool = True):
|
|
owned_wss = self.client.fetch_owned_workspaces().wait() if owned else []
|
|
invited_wss = self.client.fetch_joined_workspaces().wait() if invited else []
|
|
|
|
return owned_wss + invited_wss
|
|
|
|
def drop_client(self):
|
|
self._client = None
|
|
|
|
session = SessionManager() |