onyx_online / VBAN SendText

#!/usr/bin/env python3 # MIT License # # Copyright (c) 2025 Onyx and Iris # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in all # copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE # SOFTWARE. """ sendtext.py This script provides functionality to send text messages over a network using the VBAN protocol. It includes classes and functions to construct and send VBAN packets with text data, read configuration from a TOML file, and handle command-line arguments for customization. Classes: RTHeader: Represents the VBAN header for a text stream. RequestPacket: Encapsulates a VBAN request packet with text data. VbanSendText: Manages the sending of text messages over VBAN with rate limiting. Functions: ratelimit: Decorator to enforce a rate limit on a function. conn_from_toml: Reads a TOML configuration file and returns its contents as a dictionary. parse_args: Parses command-line arguments. main: Main function to send text using VbanSendText. Usage: Run the script with appropriate command-line arguments to send text messages. Example: python sendtext.py --config /path/to/config.toml --log-level DEBUG "strip[0].mute=0 strip[1].mute=0" """ import argparse import functools import logging import socket import sys import time from dataclasses import dataclass, field from pathlib import Path from typing import Callable import tomllib logger = logging.getLogger(__name__) @dataclass class RTHeader: name: str bps: int channel: int VBAN_PROTOCOL_TXT = 0x40 framecounter: bytes = (0).to_bytes(4, "little") # fmt: off BPS_OPTS: list[int] = field(default_factory=lambda: [ 0, 110, 150, 300, 600, 1200, 2400, 4800, 9600, 14400, 19200, 31250, 38400, 57600, 115200, 128000, 230400, 250000, 256000, 460800, 921600, 1000000, 1500000, 2000000, 3000000 ]) # fmt: on def __post_init__(self): if len(self.name) > 16: raise ValueError( f"Stream name got: '{self.name}', want: must be 16 characters or fewer" ) try: self.bps_index = self.BPS_OPTS.index(self.bps) except ValueError as e: ERR_MSG = f"Invalid bps: {self.bps}, must be one of {self.BPS_OPTS}" e.add_note(ERR_MSG) raise def __sr(self) -> bytes: return (RTHeader.VBAN_PROTOCOL_TXT + self.bps_index).to_bytes(1, "little") def __nbc(self) -> bytes: return (self.channel).to_bytes(1, "little") def build(self) -> bytes: header = "VBAN".encode("utf-8") header += self.__sr() header += (0).to_bytes(1, "little") header += self.__nbc() header += (0x10).to_bytes(1, "little") header += self.name.encode() + bytes(16 - len(self.name)) header += RTHeader.framecounter return header class RequestPacket: def __init__(self, header: RTHeader): self.header = header def encode(self, text: str) -> bytes: return self.header.build() + text.encode("utf-8") def bump_framecounter(self) -> None: self.header.framecounter = ( int.from_bytes(self.header.framecounter, "little") + 1 ).to_bytes(4, "little") logger.debug( f"framecounter: {int.from_bytes(self.header.framecounter, 'little')}" ) def ratelimit(func: Callable) -> Callable: """ Decorator to enforce a rate limit on a function. This decorator ensures that the decorated function is not called more frequently than the specified delay. If the function is called before the delay has passed since the last call, it will wait for the remaining time before executing. Args: func (callable): The function to be decorated. Returns: callable: The wrapped function with rate limiting applied. Example: @ratelimit def send_message(self, message): # Function implementation pass """ @functools.wraps(func) def wrapper(self, *args, **kwargs): now = time.time() if now - self.lastsent < self.delay: time.sleep(self.delay - (now - self.lastsent)) self.lastsent = time.time() return func(self, *args, **kwargs) return wrapper class VbanSendText: def __init__(self, **kwargs): defaultkwargs = { "host": "localhost", "port": 6980, "streamname": "Command1", "bps": 256000, "channel": 0, "delay": 0.02, } defaultkwargs.update(kwargs) self.__dict__.update(defaultkwargs) self._request = RequestPacket(RTHeader(self.streamname, self.bps, self.channel)) self.lastsent = 0 def __enter__(self): self._sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) return self def __exit__(self, exc_type, exc_value, traceback): if self._sock: self._sock.close() self._sock = None @ratelimit def sendtext(self, text: str): """ Sends a text message to the specified host and port. Args: text (str): The text message to be sent. """ self._sock.sendto(self._request.encode(text), (self.host, self.port)) self._request.bump_framecounter() def conn_from_toml(filepath: str = "config.toml") -> dict: """ Reads a TOML configuration file and returns its contents as a dictionary. Args: filepath (str): The path to the TOML file. Defaults to "config.toml". Returns: dict: The contents of the TOML file as a dictionary. Example: # config.toml host = "localhost" port = 6980 streamname = "Command1" """ pn = Path(filepath) if not pn.exists(): logger.info( f"no {pn} found, using defaults: localhost:6980 streamname: Command1" ) return {} try: with open(pn, "rb") as f: return tomllib.load(f) except tomllib.TOMLDecodeError as e: raise ValueError(f"Error decoding TOML file: {e}") from e def parse_args() -> argparse.Namespace: """ Parse command-line arguments. Returns: argparse.Namespace: Parsed command-line arguments. Command-line arguments: --log-level (str): Set the logging level. Choices are "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL". Default is "INFO". --config (str): Path to config file. Default is "config.toml". -i, --input-file (argparse.FileType): Input file to read from. Default is sys.stdin. text (str, optional): Text to send. """ parser = argparse.ArgumentParser(description="Voicemeeter VBAN Send Text CLI") parser.add_argument( "--log-level", type=str, choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], default="INFO", help="Set the logging level", ) parser.add_argument( "--config", type=str, default="config.toml", help="Path to config file" ) parser.add_argument( "-i", "--input-file", type=argparse.FileType("r"), default=sys.stdin, ) parser.add_argument("text", nargs="?", type=str, help="Text to send") return parser.parse_args() def main(config: dict): """ Main function to send text using VbanSendText. Args: config (dict): Configuration dictionary for VbanSendText. Behavior: - If 'args.text' is provided, sends the text and returns. - Otherwise, reads lines from 'args.input_file', strips whitespace, and sends each line. - Stops reading and sending if a line equals "Q". - Logs each line being sent at the debug level. """ with VbanSendText(**config) as vban: if args.text: vban.sendtext(args.text) return for line in args.input_file: line = line.strip() if line.upper() == "Q": break logger.debug(f"Sending {line}") vban.sendtext(line) if __name__ == "__main__": args = parse_args() logging.basicConfig(level=args.log_level) main(conn_from_toml(args.config))