#! /usr/bin/env python3 # -*- coding: utf-8 -*- """ Copyright 2020, Nils Hilbricht, Germany ( https://www.hilbricht.net ) The Non-Session-Manager by Jonathan Moore Liles : http://non.tuxfamily.org/nsm/ New Session Manager, by LinuxAudio.org: https://github.com/linuxaudio/new-session-manager With help from code fragments from https://github.com/attwad/python-osc ( DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE v2 ) This file is part of the Laborejo Software Suite ( https://www.laborejo.org ). This application is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . """ import logging; logger = logging.getLogger(__name__); logger.info("import") #Standard Library import struct import socket from os import getenv #to get NSM env var from shutil import rmtree as shutilrmtree from shutil import copytree as shutilcopytree from urllib.parse import urlparse #to convert NSM env var import subprocess import atexit import pathlib import json from uuid import uuid4 from datetime import datetime from sys import exit as sysexit class _IncomingMessage(object): """Representation of a parsed datagram representing an OSC message. An OSC message consists of an OSC Address Pattern followed by an OSC Type Tag String followed by zero or more OSC Arguments. """ def __init__(self, dgram): #NSM Broadcasts are bundles, but very simple ones. We only need to care about the single message it contains. #Therefore we can strip the bundle prefix and handle it as normal message. if b"#bundle" in dgram: bundlePrefix, singleMessage = dgram.split(b"/", maxsplit=1) dgram = b"/" + singleMessage # / eaten by split self.LENGTH = 4 #32 bit self._dgram = dgram self._parameters = [] self.parse_datagram() def get_int(self, dgram, start_index): """Get a 32-bit big-endian two's complement integer from the datagram. Args: dgram: A datagram packet. start_index: An index where the integer starts in the datagram. Returns: A tuple containing the integer and the new end index. Raises: ValueError if the datagram could not be parsed. """ try: if len(dgram[start_index:]) < self.LENGTH: raise ValueError('Datagram is too short') return ( struct.unpack('>i', dgram[start_index:start_index + self.LENGTH])[0], start_index + self.LENGTH) except (struct.error, TypeError) as e: raise ValueError('Could not parse datagram %s' % e) def get_string(self, dgram, start_index): """Get a python string from the datagram, starting at pos start_index. We receive always the full string, but handle only the part from the start_index internally. In the end return the offset so it can be added to the index for the next parameter. Each subsequent call handles less of the same string, starting further to the right. According to the specifications, a string is: "A sequence of non-null ASCII characters followed by a null, followed by 0-3 additional null characters to make the total number of bits a multiple of 32". Args: dgram: A datagram packet. start_index: An index where the string starts in the datagram. Returns: A tuple containing the string and the new end index. Raises: ValueError if the datagram could not be parsed. """ #First test for empty string, which is nothing, followed by a terminating \x00 padded by three additional \x00. if dgram[start_index:].startswith(b"\x00\x00\x00\x00"): return "", start_index + 4 #Otherwise we have a non-empty string that must follow the rules of the docstring. offset = 0 try: while dgram[start_index + offset] != 0: offset += 1 if offset == 0: raise ValueError('OSC string cannot begin with a null byte: %s' % dgram[start_index:]) # Align to a byte word. if (offset) % self.LENGTH == 0: offset += self.LENGTH else: offset += (-offset % self.LENGTH) # Python slices do not raise an IndexError past the last index, # do it ourselves. if offset > len(dgram[start_index:]): raise ValueError('Datagram is too short') data_str = dgram[start_index:start_index + offset] return data_str.replace(b'\x00', b'').decode('utf-8'), start_index + offset except IndexError as ie: raise ValueError('Could not parse datagram %s' % ie) except TypeError as te: raise ValueError('Could not parse datagram %s' % te) def get_float(self, dgram, start_index): """Get a 32-bit big-endian IEEE 754 floating point number from the datagram. Args: dgram: A datagram packet. start_index: An index where the float starts in the datagram. Returns: A tuple containing the float and the new end index. Raises: ValueError if the datagram could not be parsed. """ try: return (struct.unpack('>f', dgram[start_index:start_index + self.LENGTH])[0], start_index + self.LENGTH) except (struct.error, TypeError) as e: raise ValueError('Could not parse datagram %s' % e) def parse_datagram(self): try: self._address_regexp, index = self.get_string(self._dgram, 0) if not self._dgram[index:]: # No params is legit, just return now. return # Get the parameters types. type_tag, index = self.get_string(self._dgram, index) if type_tag.startswith(','): type_tag = type_tag[1:] # Parse each parameter given its type. for param in type_tag: if param == "i": # Integer. val, index = self.get_int(self._dgram, index) elif param == "f": # Float. val, index = self.get_float(self._dgram, index) elif param == "s": # String. val, index = self.get_string(self._dgram, index) else: logger.warning("Unhandled parameter type: {0}".format(param)) continue self._parameters.append(val) except ValueError as pe: #raise ValueError('Found incorrect datagram, ignoring it', pe) # Raising an error is not ignoring it! logger.warning("Found incorrect datagram, ignoring it. {}".format(pe)) @property def oscpath(self): """Returns the OSC address regular expression.""" return self._address_regexp @staticmethod def dgram_is_message(dgram): """Returns whether this datagram starts as an OSC message.""" return dgram.startswith(b'/') @property def size(self): """Returns the length of the datagram for this message.""" return len(self._dgram) @property def dgram(self): """Returns the datagram from which this message was built.""" return self._dgram @property def params(self): """Convenience method for list(self) to get the list of parameters.""" return list(self) def __iter__(self): """Returns an iterator over the parameters of this message.""" return iter(self._parameters) class _OutgoingMessage(object): def __init__(self, oscpath): self.LENGTH = 4 #32 bit self.oscpath = oscpath self._args = [] def write_string(self, val): dgram = val.encode('utf-8') diff = self.LENGTH - (len(dgram) % self.LENGTH) dgram += (b'\x00' * diff) return dgram def write_int(self, val): return struct.pack('>i', val) def write_float(self, val): return struct.pack('>f', val) def add_arg(self, argument): t = {str:"s", int:"i", float:"f"}[type(argument)] self._args.append((t, argument)) def build(self): dgram = b'' #OSC Path dgram += self.write_string(self.oscpath) if not self._args: dgram += self.write_string(',') return dgram # Write the parameters. arg_types = "".join([arg[0] for arg in self._args]) dgram += self.write_string(',' + arg_types) for arg_type, value in self._args: f = {"s":self.write_string, "i":self.write_int, "f":self.write_float}[arg_type] dgram += f(value) return dgram class NsmServerControl(object): """ The ServerControl can be started in three modes, regarding nsmd. We expect that starting our own nsmd will be the majority of cases. SessionRoot parameter is only honored if we start nsmd ourselves. Ascending lookup priority: 1) Default is to start our own nsmd. A single-instance watcher will prevent multiple programs on the same system. 2) When $NSM_URL is found as environment we will connect to that nsmd. 3) When hostname and portnumber are given explicitely as instance variables we will first test if a server is running at that URL, if not we will start our own with these parameters. This is not only a pure implemenation of the protocol. It is extended by us reacting to and storing incoming data. This data can be interpreted and enhanced by looking at the session dir ourselves. However, we don't do anything that is not possible by the original nsmd + human interaction. 100% Compatibility is the highest priority. The big problems are the async nature of communication, message come out of order or interleaved, and nsm is not consistent in its usage of osc-paths. For example it starts listing sessions with /nsm/gui/server/message, but sends the content with /reply [/nsm/server/list, nsmSessionName] and then ends it with /nsm/server/list [0, Done] (no reply!). So three message types, three callbacks for one logically connected process. To update our internal session information we therefore need to split the functionality into severall seemingly unconnected callbacks and you need to know how the protocol works to actually know the order of operations. Switch logging to info to learn more. We have a mix between NSM callbacks and our own functions. Most important there is a watchdog that looks at the session directory and creates its own callbacks if something changes. A typical operation, say sessionDelete or sessionCopy looks like this: * Ask (blocking) nsmd for a current list of sessions, update our internal state * Perform a file operation, like copy or delete or lift a lock * Let our watchdog discover the changes in the file system and trigger another (non-blocking) request for a list of sessions to adjust our internal state to reality. Granted, we could just call our blocking query again at the end, but we would still need to let the watchdog run for operations that the user does with a filemanager, which would end up in redundant calls. Bottom line: _updateSessionListBlocking is called at the beginning of a function, but not at the end. Docs: http://non.tuxfamily.org/nsm/ http://non.tuxfamily.org/wiki/Non%20Session%20Manager http://non.tuxfamily.org/wiki/ApplicationsSupportingNsm http://non.tuxfamily.org/nsm/API.html """ def __init__(self, sessionOpenReadyHook, sessionOpenLoadingHook, sessionClosedHook, clientStatusHook, singleInstanceActivateWindowHook, dataClientNamesHook, dataClientDescriptionHook, parameterNsmOSCUrl=None, sessionRoot=None, useCallbacks=True): """If useCallbacks is False you will see every message in the log. This is just a development mode to see all messages, unfiltered. Normally we have special hook functions that save and interpret data, so they don't show in the logs""" self._queue = list() #Incoming OSC messages are buffered here. #Status variables that are set by our callbacks self.internalState = { "sessions" : set(), #nsmSessionNames:str . We use set for unqiue, just in case. But we also clear that on /nsm/gui/server/message ['Listing sessions'] to handle deleted sessions "currentSession" : None, "port" : None, #Our GUI port "serverPort" : None, #nsmd port "nsmUrl" : None, #the environment variable "clients" : {}, #clientId:dict see self._initializeEmptyClient . Gets replaced with a new dict instance on session changes. "broadcasts" : [], #in the order they appeared "datastorage" : None, #URL, if present in the session } self.dataStorage = None #Becomes DataStorage() every time a datastorage client does a broadcast announce. self._addToNextSession = [] #A list of executables in PATH. Filled by new, waits for reply that session is created and then will send clientNew and clear the list. #Hooks for api callbacks self.sessionOpenReadyHook = sessionOpenReadyHook #nsmSessionName as parameter self.sessionOpenLoadingHook = sessionOpenLoadingHook #nsmSessionName as parameter self.sessionClosedHook = sessionClosedHook #no parameter. This is also "choose a session" mode self.clientStatusHook = clientStatusHook #all client status is done via this single hook. GUIs need to check if they already know the client or not. self.dataClientNamesHook = dataClientNamesHook self.dataClientDescriptionHook = dataClientDescriptionHook self.singleInstanceActivateWindowHook = singleInstanceActivateWindowHook #added to self.processSingleInstance() to listen for a message from another wannabe-instance if useCallbacks: self.callbacks = { "/nsm/gui/session/name" : self._reactCallback_activeSessionChanged, #"/nsm/gui/session/root" #Session root is an active blocking call in init "/nsm/gui/client/label" : self._reactCallback_ClientLabelChanged, "/nsm/gui/client/new" : self._reactCallback_ClientNew, "/nsm/gui/session/session" : self._reactCallback_SessionSession, "/nsm/gui/client/status" : self._reactCallback_statusChanged, #handles multiple status keywords "/reply" : self._reactCallback_reply, #handles multiple replies "/error" : self._reactCallback_error, "/nsm/gui/client/has_optional_gui" : self._reactCallback_clientHasOptionalGui, "/nsm/gui/client/gui_visible" : self._reactCallback_clientGuiVisible, "/nsm/gui/client/pid" : self._reactCallback_clientPid, "/nsm/gui/client/dirty" : self._reactCallback_clientDirty, "/nsm/gui/server/message" : self._reactCallback_serverMessage, "/nsm/gui/gui_announce" : self._reactCallback_guiAnnounce, "/nsm/server/list" : self._reactCallback_serverList, "/nsm/server/broadcast" : self._reactCallback_broadcast, } else: #This is just a development mode to see all messages, unfiltered self.callbacks = set() #empty set is easiest to check #Networking and Init for our control part, not for the server self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #internet, udp self.sock.bind(('', 0)) #pick a free port on localhost. self.sock.setblocking(False) self.internalState["port"] = self.sock.getsockname()[1] #only happens once, ports don't change during runtime #self.sock.close() Do not close, this runs until the end of the program ###Testing of existing servers, starting and connecting #First handle the NSM URL, or generate on. #self.nsmOSCUrl must be a tuple compatible to the result of urlparse. (hostname, port) self.singleInstanceSocket = None if parameterNsmOSCUrl: o = urlparse(parameterNsmOSCUrl) self.nsmOSCUrl = (o.hostname, o.port) else: envResult = self._getNsmOSCUrlFromEnvironment() if envResult: self.nsmOSCUrl = envResult else: #This is the default case. User just starts the GUI. The other modes are concious decisions to either start with URL as parameter or in an NSM environment. #But now we need to test if the user accidentaly opened a second GUI, which would start a second server. self._setupAndTestForSingleInstance() #This might quit the whole program and we will never see the next line. self.nsmOSCUrl = self._generateFreeNsmOSCUrl() assert self.nsmOSCUrl self.internalState["serverPort"] = self.nsmOSCUrl[1] #only happens once, ports don't change during runtime self.internalState["nsmUrl"] = f"osc.udp://{self.nsmOSCUrl[0]}:{self.nsmOSCUrl[1]}/" #only happens once, ports don't change during runtime #Then check if a server is running there. If not start one. self.ourOwnServer = None #Might become a subprocess handle if self._isNsmdRunning(self.nsmOSCUrl): #serverport = self.nsmOSCUrl[1] #No further action required. GUI announce below this testing. pass else: self._startNsmdOurselves(sessionRoot) #Session root can be a commandline parameter we forward to the server if we start it ourselves. assert type(self.ourOwnServer) is subprocess.Popen, (self.ourOwnServer, type(self.ourOwnServer)) #Wait for the server, or test if it is reacting. self._waitForPingResponseBlocking() logger.info("nsmd is ready @ {}".format(self.nsmOSCUrl)) #Tell nsmd that we are a GUI and want to receive general messages async, not only after we request something self.sessionRoot = self._initial_announce() #Triggers "hi" and session root self.internalState["sessionRoot"] = self.sessionRoot atexit.register(self.quit) #mostly does stuff when we started nsmd ourself self._receiverActive = True #Now an external event loop can add self.process #Internal Methods def _setupAndTestForSingleInstance(self): """on program startup trigger this if there is already another instance of us running. This socket is only """ self.singleInstanceSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) logger.info("Testing if another non-specific Argodejo is running.") try: ## Create an abstract socket, by prefixing it with null. # this relies on a feature only in linux, when current process quits, the # socket will be deleted. self.singleInstanceSocket.bind('\0' + "argodejo") self.singleInstanceSocket.listen(1) self.singleInstanceSocket.setblocking(False) logger.info("No other non-specific Argodejo found. Starting GUI") #Continue in self.processSingleInstance() return True except socket.error: logger.error("GUI for this nsmd server already running. Informing the existing application to show itself.") self.singleInstanceSocket.connect('\0' + "argodejo") self.singleInstanceSocket.send("argodejoactivate".encode()); self.singleInstanceSocket.close() sysexit(1) #triggers atexit #print ("not executed") return False def processSingleInstance(self): """Tests our unix socket for an incoming signal. if received forward to the engine->gui Can be added to a slower event loop, so it is not in self.process""" if self.singleInstanceSocket: try: connection, client_address = self.singleInstanceSocket.accept() #This blocks and waits for a message incoming = connection.recv(1024) if incoming and incoming == b"argodejoactivate": self.singleInstanceActivateWindowHook() except BlockingIOError: #happens while no data is received. Has nothing to do with blocking or not. In fact: this happens when in non-blocking mode. pass except socket.timeout: pass def _setPause(self, state:bool): """Set both the socket and the thread into waiting mode or not. With this we can wait for answers until we resume async operation""" if state: self.sock.setblocking(True) #explicitly wait. self.sock.settimeout(0.5) self._receiverActive = False logger.info("Suspending receiving async mode.") else: self.sock.setblocking(False) self._receiverActive = True logger.info("Resuming receiving async mode.") def process(self): """Use this in an external event loop""" if self._receiverActive: while True: try: data, addr = self.sock.recvfrom(1024) msg = _IncomingMessage(data) if msg: self._queue.append(msg) except BlockingIOError: #happens while no data is received. Has nothing to do with blocking or not. break except socket.timeout: break for msg in self._queue: if msg.oscpath in self.callbacks: self.callbacks[msg.oscpath](msg.params) else: logger.warning(f"Unhandled message with path {msg.oscpath} and parameters {msg.params}") self._queue.clear() def _getNsmOSCUrlFromEnvironment(self): """Return the nsm osc url or None""" nsmOSCUrl = getenv("NSM_URL") if not nsmOSCUrl: return None else: #osc.udp://hostname:portnumber/ o = urlparse(nsmOSCUrl) return o.hostname, o.port def _generateFreeNsmOSCUrl(self): #Instead of reading out the NSM port we get a free port ourselves and set up nsmd with that tempServerSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #internet, udp tempServerSock.bind(('', 0)) #pick a free port on localhost. address, tempServerSockPort = tempServerSock.getsockname() tempServerSock.close() #We need to close it because nsmd will open it right away. nsmOSCUrl = ("0.0.0.0", tempServerSockPort) #compatible to result of urlparse logger.info("Generated our own free NSM_URL to start a server @ {}".format(nsmOSCUrl)) return nsmOSCUrl def _isNsmdRunning(self, nsmOSCUrl): """Test if the port is open or not""" logger.info(f"Testing if a server is running @ {nsmOSCUrl}") hostname, port = nsmOSCUrl tempServerSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #internet, udp try: tempServerSock.bind((hostname, port)) logger.info(f"No external nsmd found (we tested if port is closed) @ {nsmOSCUrl}") return False except: logger.info(f"External nsmd found (we tested if port is closed) @ {nsmOSCUrl}") return True finally: tempServerSock.close() def _startNsmdOurselves(self, sessionRoot:str): assert self.nsmOSCUrl hostname, port = self.nsmOSCUrl if sessionRoot: self.ourOwnServer = subprocess.Popen(["nsmd","--osc-port", str(port), "--session-root", sessionRoot]) else: self.ourOwnServer = subprocess.Popen(["nsmd","--osc-port", str(port)]) def _blockingRequest(self, path:str, arguments:list, answerPath:str, answerArguments:list, repeat=False)->list: """During start-up we need to wait for replies. Also some operations only make sense if we got data back. This is an abstraction that deals with messages that may come out-of-order and keeps them for later, but at least prevents our side from sending messages out-of-order itself. Default is: send once, wait for answer. repeat=True sends multiple times until an answer arrives. Returns list of arguments, can be empty. """ assert not self._queue, [(m.oscpath, m.params) for m in self._queue] logger.info(f"[wait for answer]: Sending {path}: {arguments}") self._setPause(True) out_msg = _OutgoingMessage(path) for arg in arguments: out_msg.add_arg(arg) if not repeat: self.sock.sendto(out_msg.build(), self.nsmOSCUrl) #Wait for answer ready = False while not ready: if repeat: #we need to send multiple times. self.sock.sendto(out_msg.build(), self.nsmOSCUrl) try: data, addr = self.sock.recvfrom(1024) msg = _IncomingMessage(data) if answerArguments and msg.oscpath == answerPath and msg.params == answerArguments: result = msg.params logger.info(f"[wait from {path}] Received {answerPath}: {result}") ready = True elif msg.oscpath == answerPath: result = msg.params logger.info(f"[wait from {path}] Received {answerPath}: {result}") ready = True else: logger.warning(f"Waiting for {answerPath} from nsmd, but got: {msg.oscpath} with {msg.params}. Adding to queue for later.") self._queue.append(msg) except BlockingIOError: #happens while no data is received. Has nothing to do with blocking or not. continue except socket.timeout: continue self._setPause(False) return result def _waitForPingResponseBlocking(self): self._blockingRequest(path="/osc/ping", arguments=[], answerPath="/reply", answerArguments=["/osc/ping",], repeat=True) def _initial_announce(self)->pathlib.Path: """nsm/gui/gui_announce triggers a multi-stage reply. First we get "hi", then we get the session root. We wait for session root and then clean 'hi' from the queue. Returns session root as pathlib-path.""" resultArguments = self._blockingRequest(path="/nsm/gui/gui_announce", arguments=[], answerPath="/nsm/gui/session/root", answerArguments=[]) if len(self._queue) == 1 and self._queue[0].oscpath == "/nsm/gui/gui_announce" and self._queue[0].params == ["hi"]: self._queue.clear() else: logging.error(f"For ValueError below: {[(m.oscpath, m.params) for m in self._queue]}") raise ValueError("We were expecting a clean _queue with only 'hi' as leftover, but instead there were unhandled messages. see print above. Better abort than a wrong program state") #all ok return pathlib.Path(resultArguments[0]) #General Commands def send(self, arg): """ Intended for a text input / command line interface. Sends anything to nsmd, separated by semicolon. First part is the message address, the rest are string-parameters.""" args = arg.split() msg = _OutgoingMessage(args[0]) for p in args[1:]: msg.add_arg(p) self.sock.sendto(msg.build(), self.nsmOSCUrl) def gui_announce(self): """This is just the announce without any answer. This is a last-resort method if another GUI "stole" our slot. For our own initial announce we use self._initial_announce()""" msg = _OutgoingMessage("/nsm/gui/gui_announce") self.sock.sendto(msg.build(), self.nsmOSCUrl) def ping(self): msg = _OutgoingMessage("/osc/ping") self.sock.sendto(msg.build(), self.nsmOSCUrl) def list(self): msg = _OutgoingMessage("/nsm/server/list") self.sock.sendto(msg.build(), self.nsmOSCUrl) def _updateSessionListBlocking(self): """To ensure correct data on session operations we manage ourselves, like copy, rename and delete. Ask nsmd for projects in session root and update our internal state. This will wait for an answer and block all other operations. First is /nsm/gui/server/message ['Listing sessions'] Then session names come one reply at a time such as /reply ['/nsm/server/list', 'test3'] Finally /nsm/server/list [0, 'Done.'] , not a reply """ logger.info("Requesting project list from session server in blocking mode") self._setPause(True) msg = _OutgoingMessage("/nsm/server/list") self.sock.sendto(msg.build(), self.nsmOSCUrl) #Wait for /reply ready = False while True: try: data, addr = self.sock.recvfrom(1024) except socket.timeout: continue msg = _IncomingMessage(data) if not ready and msg.oscpath == "/nsm/gui/server/message" and msg.params == ["Listing sessions"]: self.internalState["sessions"].clear() # new clients are added as /reply /nsm/server/list callbacks ready = True else: if len(msg.params) != 2: logger.warning(f"Expected project but got path {msg.oscpath} with {msg.params}. Adding to queue for later.") self._queue.append(msg) continue #This is what we want: elif msg.oscpath == "/reply" and msg.params[0] == "/nsm/server/list": #/reply ['/nsm/server/list', 'test3'] self.internalState["sessions"].add(msg.params[1]) logger.info(f"Received session name: {msg.params[1]}") elif msg.params[0] == 0 and msg.params[1] == "Done.": break else: logger.warning(f"Expected project but got path {msg.oscpath} with {msg.params}. Adding to queue for later.") self._queue.append(msg) continue self._setPause(False) def quit(self): """Called through atexit. Thanks to start.py sys exception hook this will also trigger on PyQt crash""" if self.ourOwnServer: msg = _OutgoingMessage("/nsm/server/quit") self.sock.sendto(msg.build(), self.nsmOSCUrl) returncode = self.ourOwnServer.wait() logger.info("Stopped our own server with return code {}".format(returncode)) def broadcast(self, path:str, arguments:list): """/nsm/server/broadcast s:path [arguments...] http://non.tuxfamily.org/nsm/API.html 1.2.7.1 /nsm/server/broadcast s:path [arguments...] /nsm/server/broadcast /tempomap/update "0,120,4/4:12351234,240,4/4" All clients except the sender recive: /tempomap/update "0,120,4/4:12351234,240,4/4" """ logger.info(f"Sending broadcast with path {path} and args {arguments}") message = _OutgoingMessage("/nsm/server/broadcast") message.add_arg(path) for arg in arguments: message.add_arg(arg) #type autodetect self.sock.sendto(message.build(), self.nsmOSCUrl) #Primarily Without Session def open(self, nsmSessionName:str): if nsmSessionName in self.internalState["sessions"]: msg = _OutgoingMessage("/nsm/server/open") msg.add_arg(nsmSessionName) #s:project_name self.sock.sendto(msg.build(), self.nsmOSCUrl) else: logger.warning(f"Session {nsmSessionName} not found. Not forwarding to nsmd.") def new(self, newName:str, startClients:list=[])->str: """Saves the current session and creates a new session. Only works if dir does not exist yet. """ basePath = pathlib.Path(self.sessionRoot, newName) if basePath.exists(): return None self._addToNextSession = startClients msg = _OutgoingMessage("/nsm/server/new") msg.add_arg(newName) #s:project_name self.sock.sendto(msg.build(), self.nsmOSCUrl) #Only with Current Session def save(self): msg = _OutgoingMessage("/nsm/server/save") self.sock.sendto(msg.build(), self.nsmOSCUrl) def close(self, blocking=False): if not blocking: msg = _OutgoingMessage("/nsm/server/close") self.sock.sendto(msg.build(), self.nsmOSCUrl) else: msg = _OutgoingMessage("/nsm/server/close") self.sock.sendto(msg.build(), self.nsmOSCUrl) #Drive the process loop ourselves. This will still trigger updates but the mainloop will wait. while self.internalState["currentSession"]: self.process() def abort(self, blocking=False): if not blocking: msg = _OutgoingMessage("/nsm/server/abort") self.sock.sendto(msg.build(), self.nsmOSCUrl) else: msg = _OutgoingMessage("/nsm/server/abort") self.sock.sendto(msg.build(), self.nsmOSCUrl) #Drive the process loop ourselves. This will still trigger updates but the mainloop will wait. while self.internalState["currentSession"]: self.process() def duplicate(self, newName:str)->str: """Saves the current session and creates a new session. Requires an open session and uses nsmd to do the work. If you want to do copy of any session use our owns self.sessionCopy""" msg = _OutgoingMessage("/nsm/server/duplicate") msg.add_arg(newName) #s:project_name self.sock.sendto(msg.build(), self.nsmOSCUrl) #Client Commands for Loaded Session def clientAdd(self, executableName:str): """Adds a client to the current session. executable must be in $PATH. We do not trust NSM to perform the right checks. It will add an empty path or wrong path. """ if not pathlib.Path(executableName).name == executableName: logger.warning(f"{executableName} must be just an executable file in your $PATH. We expected: {pathlib.Path(executableName).name} . We will not ask nsmd to add it as client") return False allPaths = getenv("PATH") assert allPaths, allPaths binaryPaths = allPaths.split(":") #TODO: There is a corner case that NSMD runs in a different $PATH environment. executableInPath = any(pathlib.Path(bp, executableName).is_file() for bp in binaryPaths) if executableInPath: msg = _OutgoingMessage("/nsm/server/add") msg.add_arg(executableName) #s:executable_name self.sock.sendto(msg.build(), self.nsmOSCUrl) return True else: logger.warning("Executable {} not found. We will not ask nsmd to add it as client".format(executableName)) return False def clientStop(self, clientId:str): msg = _OutgoingMessage("/nsm/gui/client/stop") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) def clientResume(self, clientId:str): """Opposite of clientStop""" msg = _OutgoingMessage("/nsm/gui/client/resume") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) def clientRemove(self, clientId:str): """Client needs to be stopped already. We will do that and wait for an answer. Remove from the session. Will not delete the save-files, but make them inaccesible""" #We have a blocking operation in here so we need to be extra cautios that the client exists. if not clientId in self.internalState["clients"]: return False self.clientStop(clientId) #We need to wait for an answer. #Drive the process loop ourselves. This will still trigger updates but the mainloop will wait. logger.info(f"Waiting for {clientId} to be status 'stopped'") while not self.internalState["clients"][clientId]["lastStatus"] == "stopped": self.process() msg = _OutgoingMessage("/nsm/gui/client/remove") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) #Flood lazy lagging nsmd until it removed the client. #We will receive a few -10 "No such client." errors but that is ok. while True: if not clientId in self.internalState["clients"]: break if self.internalState["clients"][clientId]["lastStatus"] == "removed": break self.sock.sendto(msg.build(), self.nsmOSCUrl) self.process() def clientSave(self, clientId:str): """Saves only the given client""" msg = _OutgoingMessage("/nsm/gui/client/save") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) def clientHide(self, clientId:str): """Hides the client. Works only if client announced itself with this feature""" msg = _OutgoingMessage("/nsm/gui/client/hide_optional_gui") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) def clientShow(self, clientId:str): """Hides the client. Works only if client announced itself with this feature""" msg = _OutgoingMessage("/nsm/gui/client/show_optional_gui") msg.add_arg(clientId) #s:clientId self.sock.sendto(msg.build(), self.nsmOSCUrl) #Callbacks def _reactCallback_guiAnnounce(self, parameters:list): """Acknowledge""" assert parameters == ["hi"], parameters logger.info("We got acknowledged as current nsmd GUI.") def _reactCallback_error(self, parameters:list): logger.error(parameters) def _reactCallback_reply(self, parameters:list): """This is a difficult function because replies arrive for many unrelated things, like status. We do our best to send all replies on the right way""" success = False l = len(parameters) if l == 2: originalMessage, data = parameters logger.info(f"Got reply for {originalMessage} with {data}") reply = { "/nsm/server/list" : self._reactReply_nsmServerList, "/nsm/server/new" : self._reactReply_nsmServerNew, "/nsm/server/close" : self._reactReply_nsmServerClose, "/nsm/server/open" : self._reactReply_nsmServerOpen, "/nsm/server/save" : self._reactReply_nsmServerSave, "/nsm/server/abort" : self._reactReply_nsmServerAbort, "/nsm/server/duplicate" : self._reactReply_nsmServerDuplicate, } if originalMessage in reply: reply[originalMessage](data) success = True elif l == 3: originalMessage, errorCode, answer = parameters logger.info(f"Got reply for {originalMessage} with code {errorCode} saying {answer}") if originalMessage == "/nsm/server/add": assert errorCode == 0, parameters self._reactReply_nsmServerAdd(answer) success = True elif l == 1: singleMessage = parameters[0] """For unknown reasons these replies do not repeat the originalMessage""" if singleMessage == "/osc/ping": logger.info(singleMessage) success = True elif singleMessage == "Client removed.": self._reactReply_nsmServerRemoved() success = True elif singleMessage == "Client stopped.": self._reactReply_nsmServerStopped() success = True #After all these reactions and checks the function will eventually return here. if not success: raise NotImplementedError(parameters) def _reactCallback_serverMessage(self, parameters:list): """Messages are normally harmless and uninteresting. Howerver, we need to use some of them for actual tasks. In opposite to reply and status this all go in our function for now, until refactoring""" if parameters == ["Listing session"]: #this feels bad! A simple message is not a reliable state token and could change in the future. #we cannot put that into our own /list outgoing message because other actions like "new" also trigger this callback self.internalState["sessions"].clear() # new clients are added as /reply /nsm/server/list callbacks if parameters[0].startswith("Opening session"): #This gets send only when an existing session starts loading. It will not trigger on new sessions, be it really new or duplicate. #e.g. /nsm/gui/server/message ["Opening session FOO"] nsmSessionName = parameters[0].replace("Opening session ", "") logger.info(f"Starting to load clients of session: {nsmSessionName}") self.sessionOpenLoadingHook(self.sessionAsDict(nsmSessionName)) #notify the api->UI else: logger.info("/nsm/gui/server/message " + repr(parameters)) def _reactCallback_broadcast(self, parameters:list): """We have nothing to do with broadcast. But we save them, so they can be shown on request parameters[0] is an osc path:str without naming constraints the rest is a list of arguments. """ logger.info(f"Received broadcast. Saving in internal state: {parameters}") self.internalState["broadcasts"].append(parameters) #Our little trick. We know and like some clients better than others. #If we detect our own data-storage we remember our friends. #It is possible that another datastorage broadcasts, then we overwrite the URL. if parameters and parameters[0] == "/argodejo/datastorage/announce": path, clientId, messageSizeLimit, url = parameters assert "osc.udp" in url logger.info(f"Got announce from argodejo datastorage clientId {clientId} @ {url}") o = urlparse(url) self.dataStorage = DataStorage(self, clientId, messageSizeLimit, (o.hostname, o.port), self.sock) def _reactCallback_serverList(self, parameters:list): """This finalizes a new session list. Here we send new data to the GUI etc.""" l = len(parameters) if l == 2: errorCode, message = parameters assert errorCode == 0, errorCode assert message == "Done.", message #don't miss the dot after Done logger.info("/nsm/server/list is done and has transmitted all available sessions to us") else: raise NotImplementedError(parameters) def _reactCallback_activeSessionChanged(self, parameters:list): """This is called when the session has already changed. Shortly before we receive /nsm/gui/session/session which indicates the attempt to create, I guess! :) If you want to react to the attempt to open a session you need to use /nsm/gui/server/message ["Opening session FOO"] OR creating a new session, after which nsmd will open that session without a message. Empty string is "No session" or "Choose A Session" mode. """ l = len(parameters) if l == 2: shortName, nsmSessionName = parameters if not shortName and not nsmSessionName: #No session loaded. We are in session-choosing mode. self.internalState["currentSession"] = None self.sessionClosedHook() else: nsmSessionName = nsmSessionName.lstrip("/") logger.info(f"Current Session changed. We are now {shortName} under {nsmSessionName}") self.internalState["currentSession"] = nsmSessionName #This is after the session, received after all programs have loaded. #We have a counterpart as message reaction that signals the attempt to load. self.sessionOpenReadyHook(self.sessionAsDict(nsmSessionName)) #notify the api->UI for autoClientExecutableInPath in self._addToNextSession: self.clientAdd(autoClientExecutableInPath) self._addToNextSession = [] #reset elif l == 0: #Another way of "no session". self.internalState["currentSession"] = None self.sessionClosedHook() else: raise NotImplementedError(parameters) def _initializeEmptyClient(self, clientId:str): """NSM reuses signals. It is quite possible that this will be called multiple times, e.g. after opening a session""" if clientId in self.internalState["clients"]: return logger.info(f"Creating new internal entry for client {clientId}") client = { "clientId":clientId, #for convenience, included internally as well "executable":None, #For dumb clients this is the same as reportedName. "dumbClient":True, #Bool. Real nsm or just any old program? status "Ready" switches this. "reportedName":None, #str . The reported name is first the executable name, for status started. But for NSM clients it gets replaced with a reported name. "label":None, #str "lastStatus":None, #str "statusHistory":[], #list "hasOptionalGUI": False, #bool "visible": None, # bool "dirty": None, # bool } self.internalState["clients"][clientId] = client def _setClientData(self, clientId:str, parameter:str, value): if clientId in self.internalState["clients"]: self.internalState["clients"][clientId][parameter] = value return True else: logger.warning(f"Client {clientId} not found in internal status storage. If the session was just closed this is most likely a known race condition. Everything is fine in this case.") return False def _reactCallback_ClientLabelChanged(self, parameters:list): """osc->add_method( "/nsm/gui/client/label", "ss", osc_handler, osc, "path,display_name" ); """ l = len(parameters) if l == 2: clientId, label = parameters logger.info(f"Label for client {clientId} changed to {label}") self._setClientData(clientId, "label", label) self.clientStatusHook(self.internalState["clients"][clientId]) else: raise NotImplementedError(parameters) def _reactCallback_clientPid(self, parameters:list): clientId, pid = parameters self._setClientData(clientId, "pid", pid) def _reactCallback_SessionSession(self, parameters:list): """This is received only when a new session gets created and followed by /nsm/gui/client/new and then a reply for /reply /nsm/server/new Session created""" #This is the counterpart to Message "Opening Session", but for really new or freshly duplicated session. logger.info(f"Attempt to create session: {parameters}") self.sessionOpenLoadingHook(self.sessionAsDict(parameters[0])) #notify the api->UI def _reactCallback_ClientNew(self, parameters:list): """/nsm/gui/client/new ['nBAVO', 'jackpatch'] This is both add client or open. It appears in the session. And the message actually comes twice. Once when you add a client, then parameters will contain the executable name. If the client reports itself as NSM compatible through announce we will also get the Open message through this function. Then the name changes from executableName to a reportedName, which will remain for the rest of the session. A loaded session will directly start with the reported/announced name. """ l = len(parameters) if l == 2: clientId, executableName = parameters if not clientId in self.internalState["clients"]: self._initializeEmptyClient(clientId) self._setClientData(clientId, "executable", executableName) logger.info(f"Client started {executableName}:{clientId}") else: self._setClientData(clientId, "reportedName", executableName) logger.info(f"Client upgraded to NSM-compatible: {executableName}:{clientId}") self.clientStatusHook(self.internalState["clients"][clientId]) else: raise NotImplementedError(parameters) def _reactCallback_clientDirty(self, parameters:list): """/nsm/gui/client/dirty ['nMAJH', 1] """ l = len(parameters) if l == 2: clientId, dirty = parameters dirty = bool(dirty) self._setClientData(clientId, "dirty", dirty) logger.info(f"Client {clientId} save status dirty is now: {dirty}") self.clientStatusHook(self.internalState["clients"][clientId]) else: raise NotImplementedError(parameters) def _reactCallback_clientGuiVisible(self, parameters:list): """/nsm/gui/client/gui_visible ['nMAJH', 0] """ l = len(parameters) if l == 2: clientId, visible = parameters visible = bool(visible) self._setClientData(clientId, "visible", visible) logger.info(f"Client {clientId} visibility is now: {visible}") self.clientStatusHook(self.internalState["clients"][clientId]) else: raise NotImplementedError(parameters) def _reactCallback_clientHasOptionalGui(self, parameters:list): """/nsm/gui/client/has_optional_gui ['nFDBK'] nsmd sends us this as reaction to a clients announce capabilities list """ l = len(parameters) if l == 1: clientId = parameters[0] self._setClientData(clientId, "hasOptionalGUI", True) logger.info(f"Client {clientId} supports optional GUI") else: raise NotImplementedError(parameters) def _reactCallback_statusChanged(self, parameters:list): """ Handles all status messages. Some changes, like removed and quit, are only available as status. This means that status removed is the opposite of /nsm/gui/client/new, even if it doesn't read like it. /nsm/gui/client/status ['nFDBK', 'open'] /nsm/gui/client/status ['nMAJH', 'launch'] /nsm/gui/client/status ['nLUPX', 'ready'] /nsm/gui/client/status ['nLUPX', 'save'] /nsm/gui/client/status ['nFHLB', 'quit'] /nsm/gui/client/status ['nLUPX', 'removed'] /nsm/gui/client/status ['nLUPX', 'stopped'] /nsm/gui/client/status ['nLUPX', 'noop'] #For dumb clients! no nsm support! /nsm/gui/client/status ['nLUPX', 'switch'] /nsm/gui/client/status ['nLUPX', 'error'] """ l = len(parameters) if l == 2: clientId, status = parameters logger.info(f"Client status {clientId} now {status}") r = self._setClientData(clientId, "lastStatus", status) if r: #a known race condition at quit may delete this in between calls self.internalState["clients"][clientId]["statusHistory"].append(status) if status == "ready": #we need to check for this now. Below in actions is after the statusHook and too late. self._setClientData(clientId, "dumbClient", False) self.clientStatusHook(self.internalState["clients"][clientId]) else: raise NotImplementedError(parameters) #Now handle our actions. For better readability in separate functions. actions = { "open": self._reactStatus_open, "launch": self._reactStatus_launch, "ready": self._reactStatus_ready, "save": self._reactStatus_save, "quit": self._reactStatus_quit, "removed": self._reactStatus_removed, "stopped": self._reactStatus_stopped, "noop": self._reactStatus_noop, "switch": self._reactStatus_switch, "error": self._reactStatus_error, }[status](clientId) actions #pylint does not like temporary dicts for case-switch def _reactStatus_removed(self, clientId:str): """Remove the client entry from our internal state. This also covers crashes.""" if clientId in self.internalState["clients"]: #race condition at quit del self.internalState["clients"][clientId] if self.dataStorage and clientId == self.dataStorage.ourClientId: #We only care about the current data-storage, not another instance that was started before it. self.dataClientNamesHook(None) self.dataClientDescriptionHook(None) self.dataStorage = None def _reactStatus_stopped(self, clientId:str): """The client has stopped and can be restarted. The status is not saved. NSM will try to open all clients on session open and end in "ready" """ if self.dataStorage and clientId == self.dataStorage.ourClientId: #We only care about the current data-storage, not another instance that was started before it. self.dataClientNamesHook(None) self.dataClientDescriptionHook(None) self.dataStorage = None def _reactStatus_launch(self, clientId:str): """ Launch is a transitional status for NSM clients but the terminal status for dumb clients """ pass def _reactStatus_open(self, clientId:str): """ """ pass def _reactStatus_ready(self, clientId:str): """ This is sent after startup but also after every save. It signals that the client can react to nsm signals, not that it is ready for something else. Note that this is *After* the clientStatusHook, so any data changed here is not submitted to the api/GUI yet. E.g. you can't change dumbClient to True here if that is needed directly after start by the GUI. """ pass def _reactStatus_save(self, clientId:str): """ """ pass def _reactStatus_quit(self, clientId:str): """ """ pass def _reactStatus_noop(self, clientId:str): """ Dumb clients, or rather nsmd, react with noop on signals they cannot understand, like saving. """ pass def _reactStatus_switch(self, clientId:str): """ """ pass def _reactStatus_error(self, clientId:str): """ """ logger.error(f"{clientId} has error status!") def _reactReply_nsmServerOpen(self, answer:str): assert answer == "Loaded.", answer def _reactReply_nsmServerSave(self, answer:str): assert answer == "Saved.", answer def _reactReply_nsmServerClose(self, answer:str): assert answer == "Closed.", answer def _reactReply_nsmServerAbort(self, answer:str): assert answer == "Aborted.", answer def _reactReply_nsmServerAdd(self, answer:str): """Reaction to add client""" assert answer == "Launched.", answer def _reactReply_nsmServerRemoved(self): pass def _reactReply_nsmServerStopped(self): pass def _reactReply_nsmServerDuplicate(self, answer:str): """There are a lot of errors possible here, reported through nsmd /error, because we are dealing with the file system. Our own GUI and other safeguards should protect us from most though Positive answers are 'Duplicated.' when nsmd finished copying and 'Loaded.' when the new session is loaded. Or so one would think... the messages arrive the other way around. Anyway, both are needed to signify a succesful duplication. """ assert answer == "Loaded." or answer == "Duplicated.", answer #We don't need any callbacks here, nsmd sends a session change on top of the duplicate replies. def _reactReply_nsmServerNew(self, answer:str): """Created. arrives when a new session is created for the first time and directory is mkdir Session created arrives when a session was opened and nsm created its internal "session". We do not need to react to the new signal because we watch the dir for new sessions ourselves and the currently active session is send through "/nsm/gui/session/name" : self._reactCallback_activeSessionChanged, """ assert answer == 'Created.' or answer == "Session created", answer def _reactReply_nsmServerList(self, nsmSessionName:str): """Session names come one reply at a time. We reacted to the message /nsm/gui/server/message ['Listing sessions'] by clearing our internal session status and will save the new ones here /reply ['/nsm/server/list', 'test3'] Do not confuse reply server list with the message /nsm/server/list [0, 'Done.'] The latter is a top level message :( """ self.internalState["sessions"].add(nsmSessionName) #Our own functions def allClientsHide(self): for clientId, clientDict in self.internalState["clients"].items(): if clientDict["hasOptionalGUI"]: self.clientHide(clientId) def allClientsShow(self): for clientId, clientDict in self.internalState["clients"].items(): if clientDict["hasOptionalGUI"]: self.clientShow(clientId) def clientToggleVisible(self, clientId:str): if self.internalState["clients"][clientId]["hasOptionalGUI"]: if self.internalState["clients"][clientId]["visible"]: self.clientHide(clientId) else: self.clientShow(clientId) #data-storage / nsm-data def clientNameOverride(self, clientId:str, name:str): """An argodejo-specific function that requires the client nsm-data in the session. If nsm-data is not present this function will write nothing, not touch any data. It will still send a callback to revert any GUI changes back to the original name. We accept empty string as a name to remove the name override """ if self.dataStorage: assert clientId in self.internalState["clients"], self.internalState["clients"] self.dataStorage.setClientOverrideName(clientId, name) #triggers callback #data-storage / nsm-data def setDescription(self, text:str): if self.dataStorage: self.dataStorage.setDescription(text) def _checkDirectoryForSymlinks(self, path)->bool: for p in path.rglob("*"): if p.is_symlink(): return True return False def _checkIfLocked(self, nsmSessionName:str)->bool: basePath = pathlib.Path(self.sessionRoot, nsmSessionName) assert basePath.exists() lockFile = pathlib.Path(basePath, ".lock") return lockFile.exists() def forceLiftLock(self, nsmSessionName:str): """Removes lockfile, no matter if session is actually open or just a remainder from a crash. If no lock exist it does nothing.""" self._updateSessionListBlocking() if self._checkIfLocked(nsmSessionName): basePath = pathlib.Path(self.sessionRoot, nsmSessionName) assert basePath.exists() #implied by _checkIfLocked lockFile = pathlib.Path(basePath, ".lock") lockFile.unlink(missing_ok=True) logger.info(f"{nsmSessionName} was forced to unlock by us.") else: logger.info(f"Tried to unlock, but {nsmSessionName} is not locked") def getSessionFiles(self, nsmSessionName:str)->list: """Return all session files, useful to present to the user, e.g. as warning before deletion""" self._updateSessionListBlocking() basePath = pathlib.Path(self.sessionRoot, nsmSessionName) assert basePath.exists() return [f.as_posix() for f in basePath.rglob("*")] #Includes directories themselves #Only files, no directories themselves. #result = [] #for path, dirs, files in walk(basePath): # for file in files: # result.append(pathlib.Path(path, file).as_posix()) #return result def deleteSession(self, nsmSessionName:str): """Delete project directory with all data. No undo. Only if session is not locked""" self._updateSessionListBlocking() if not nsmSessionName in self.internalState["sessions"]: logger.warning(f"{nsmSessionName} is not a session") return False basePath = pathlib.Path(self.sessionRoot, nsmSessionName) assert basePath.exists() if not self._checkIfLocked(nsmSessionName): logger.info(f"Deleting session {nsmSessionName}: {self.getSessionFiles(nsmSessionName)}") shutilrmtree(basePath) else: logger.warning(f"Tried to delete {basePath} but it is locked") def renameSession(self, nsmSessionName:str, newName:str): """Only works if session is not locked and dir does not exist yet""" self._updateSessionListBlocking() newPath = pathlib.Path(self.sessionRoot, newName) oldPath = pathlib.Path(self.sessionRoot, nsmSessionName) assert oldPath.exists() if self._checkIfLocked(nsmSessionName): logger.warning(f"Can't rename {nsmSessionName} to {newName}. {nsmSessionName} is locked.") return False elif newPath.exists(): logger.warning(f"Can't rename {nsmSessionName} to {newName}. {newName} already exists.") return False else: logger.info(f"Renaming {nsmSessionName} to {newName}.") tmp = pathlib.Path(oldPath.name+str(uuid4())) #Can't move itself into a subdir in itself. move to temp first. We don't use tempdir because that could be on another partition. we already know we can write here. oldPath.rename(tmp) pathlib.Path(newPath).mkdir(parents=True, exist_ok=True) tmp.rename(newPath) assert newPath.exists() def copySession(self, nsmSessionName:str, newName:str): """Copy a whole tree. Keep symlinks as symlinks. Lift lock""" self._updateSessionListBlocking() source = pathlib.Path(self.sessionRoot, nsmSessionName) destination = pathlib.Path(self.sessionRoot, newName) if destination.exists(): logger.warning(f"Can't copy {nsmSessionName} to {newName}. {newName} already exists.") return False elif not nsmSessionName in self.internalState["sessions"]: logger.warning(f"{nsmSessionName} is not a session") return elif not source.exists(): logger.warning(f"Can't copy {nsmSessionName} because it does not exist.") return False #All is well. try: shutilcopytree(source, destination, symlinks=True, dirs_exist_ok=False) #raises an error if dir already exists. But we already test above. self.forceLiftLock(newName) except Exception as e: #we don't want to crash if user tries to copy to /root or so. logger.error(e) return False #Export to the User Interface def sessionAsDict(self, nsmSessionName:str)->dict: assert self.sessionRoot entry = {} entry["nsmSessionName"] = nsmSessionName entry["name"] = pathlib.Path(nsmSessionName).name basePath = pathlib.Path(self.sessionRoot, nsmSessionName) sessionFile = pathlib.Path(basePath, "session.nsm") if not sessionFile.exists(): logger.info("Got wrong session directory from nsmd. Race condition after delete? Project: " + repr(sessionFile)) return None timestamp = datetime.fromtimestamp(sessionFile.stat().st_mtime).isoformat(sep=" ", timespec='minutes') entry["lastSavedDate"] = timestamp entry["sessionFile"] = sessionFile entry["lockFile"] = pathlib.Path(basePath, ".lock") entry["fullPath"] = str(basePath) entry["sizeInBytes"] = sum(f.stat().st_size for f in basePath.glob('**/*') if f.is_file() ) entry["numberOfClients"] = len(open(sessionFile).readlines()) entry["hasSymlinks"] = self._checkDirectoryForSymlinks(basePath) entry["parents"] = basePath.relative_to(self.sessionRoot).parts[:-1] #tuple of each dir between NSM root and nsmSessionName/session.nsm, exluding the actual project name. This is the tree entry["locked"] = self._checkIfLocked(nsmSessionName) #not for direct display return entry def exportSessionsAsDicts(self)->list: """Return a list of dicts of projects with additional information: """ results = [] self._updateSessionListBlocking() for nsmSessionName in self.internalState["sessions"]: result = self.sessionAsDict(nsmSessionName) results.append(result) return results class DataStorage(object): """Interface to handle the external datastorage client url is pre-processed (host, port) Our init is the same as announcing the nsm-data client in the session. That means everytime nsm-data sends a new/open reply we get created. Thus we will send all our data to parent and subsequently to GUI-callbacks in init. Keys are strings, While nsmd OSC support int, str and float we use json exclusively. We send json string and parse the received data. Try to use only ints, floats, strin gs, lists and dicts. Client pretty names are limited to 512 chars, depending on our OSC message size. nsm-data will just cut to 512 chars. So a GUI should better protect that limit. """ def __init__(self, parent, ourClientId, messageSizeLimit:int, url:tuple, sock): logger.info("Create new DataStorage instance") self.parent = parent self.messageSizeLimit = messageSizeLimit # e.g. 512 self.ourClientId = ourClientId self.clients = parent.internalState["clients"] #shortcut. Mutable, persistent dict, until instance gets deleted. self.url = url self.sock = sock self.ip, self.port = self.sock.getsockname() self.data = self.getAll() #blocks. our local copy. = {"clientOverrideNames":{clientId:nameOverride}, "description":"str"} self.namesToParentAndCallbacks() self.descriptionToParentAndCallbacks() def namesToParentAndCallbacks(self): self.parent.dataClientNamesHook(self.data["clientOverrideNames"]) def descriptionToParentAndCallbacks(self): """Every char!!!""" self.parent.dataClientDescriptionHook(self.data["description"]) def _waitForMultipartMessage(self, pOscpath:str)->str: """Returns a json string, as if the message was sent as a single one. Can consist of only one part as well.""" logger.info(f"Waiting for multi message {pOscpath} in blocking mode") self.parent._setPause(True) jsonString = "" chunkNumberOfParts = float("+inf") #zero based currentPartNumber = float("-inf") #zero based while True: if currentPartNumber >= chunkNumberOfParts: break try: data, addr = self.sock.recvfrom(1024) except socket.timeout: break msg = _IncomingMessage(data) if msg.oscpath == pOscpath: currentPartNumber, l, jsonChunk = msg.params jsonString += jsonChunk chunkNumberOfParts = l #overwrite infinity the first time and redundant afterwards. else: self.parent._queue.append(msg) self.parent._setPause(False) logger.info(f"Message complete with {chunkNumberOfParts} chunks.") return jsonString def getAll(self): """Mirror everything from nsm-data""" msg = _OutgoingMessage("/argodejo/datastorage/getall") msg.add_arg(self.ip) msg.add_arg(self.port) self.sock.sendto(msg.build(), self.url) jsonString = self._waitForMultipartMessage("/argodejo/datastorage/reply/getall") return json.loads(jsonString) def setClientOverrideName(self, clientId:str, value): """We accept empty string as a name to remove the name override""" assert clientId in self.clients, self.clients msg = _OutgoingMessage("/argodejo/datastorage/setclientoverridename") msg.add_arg(clientId) msg.add_arg(json.dumps(value)) self.sock.sendto(msg.build(), self.url) self.getClientOverrideName(clientId) #verifies data and triggers callback def getClientOverrideName(self, clientId:str): msg = _OutgoingMessage("/argodejo/datastorage/getclientoverridename") msg.add_arg(clientId) msg.add_arg(self.ip) msg.add_arg(self.port) self.sock.sendto(msg.build(), self.url) #Wait in blocking mode self.parent._setPause(True) while True: try: data, addr = self.sock.recvfrom(1024) except socket.timeout: break msg = _IncomingMessage(data) if msg.oscpath == "/argodejo/datastorage/reply/getclient": replyClientId, jsonName = msg.params assert replyClientId == clientId, (replyClientId, clientId) break else: self.parent._queue.append(msg) self.parent._setPause(False) #Got answer answer = json.loads(jsonName) if answer: self.data["clientOverrideNames"][clientId] = answer else: #It is possible that a client not present in our storage will send an empty string. Protect. if clientId in self.data["clientOverrideNames"]: del self.data["clientOverrideNames"][clientId] self.namesToParentAndCallbacks() def _chunkstring(self, string): return [string[0+i:self.messageSizeLimit+i] for i in range(0, len(string), self.messageSizeLimit)] def setDescription(self, text:str): """This most likely arrives one char at time with the complete text""" chunks = self._chunkstring(text) descriptionId = str(id(text))[:8] for index, chunk in enumerate(chunks): msg = _OutgoingMessage("/argodejo/datastorage/setdescription") msg.add_arg(descriptionId) msg.add_arg(index) msg.add_arg(chunk) msg.add_arg(self.ip) msg.add_arg(self.port) self.sock.sendto(msg.build(), self.url) #No echo answer. #We cheat a bit and inform parents with the new text directly. self.data["description"] = text self.descriptionToParentAndCallbacks() #and back #Generic Functions. Not in use and not ready. def _test(self): self.readAll() self.setDescription("Ein Jäger aus Kurpfalz,\nDer reitet durch den grünen Wald,\nEr schießt das Wild daher,\nGleich wie es ihm gefällt.") self.read("welt") self.create("welt", "world") self.read("welt") self.create("str", "bar") self.create("int", 1) self.create("list", [1, 2, 3]) self.create("tuple", (1, 2, 3)) #no tuples, everything will be a list. self.create("dict", {1:2, 3:4, 5:6}) self.update("str", "rolf") self.delete("str") def read(self, key:str): """Request one value""" msg = _OutgoingMessage("/argodejo/datastorage/read") msg.add_arg(key) msg.add_arg(self.ip) msg.add_arg(self.port) self.sock.sendto(msg.build(), self.url) def readAll(self): """Request all data""" msg = _OutgoingMessage("/argodejo/datastorage/readall") msg.add_arg(self.ip) msg.add_arg(self.port) self.sock.sendto(msg.build(), self.url) def create(self, key:str, value): """Write/Create one value.""" msg = _OutgoingMessage("/argodejo/datastorage/create") msg.add_arg(key) msg.add_arg(json.dumps(value)) self.sock.sendto(msg.build(), self.url) def update(self, key:str, value): """Update a value, but only if it exists""" msg = _OutgoingMessage("/argodejo/datastorage/update") msg.add_arg(key) msg.add_arg(json.dumps(value)) self.sock.sendto(msg.build(), self.url) def delete(self, key:str): """Delete a key/value completely""" msg = _OutgoingMessage("/argodejo/datastorage/delete") msg.add_arg(key) self.sock.sendto(msg.build(), self.url)