mod.ts

// deno-lint-ignore-file no-explicit-any
import { TypedEmitter, EventEmitter, green } from "./deps.ts";

const env = Deno.env.toObject();
export const pluginName = env.VRCOSCM_PLUGIN;

type value = string | number | boolean;

interface ManagerEvent {
  method: string;
  plugin: string;
  options: Record<string, any>;
  parameterName: string;
  parameterValue: value[];
  avatar: string;
}

type Events = {
  parameters: (name: string, value: value[]) => void;
  options: (a: Record<string, any>) => void;
  avatar_change: (avatar: string) => void;
};

export class Manager extends (EventEmitter as new () => TypedEmitter<Events>) {
  public address: string;
  private ws: WebSocket | undefined;

  constructor(private addr: string = env.VRCOSCM_WS_ADDR ?? "localhost:8787") {
    super();
    const params = new URLSearchParams();
    params.append("plugin", pluginName!);
    this.address = `ws://${this.addr}/?${params.toString()}`;
  }

  public connect(): Promise<void> {
    return new Promise((resolve, reject) => {
      this.ws = new WebSocket(this.address);
      this.ws.onopen = () => {
        console.log(green("Manager Connected"));
        resolve();
      };
      //@ts-ignore ws
      this.ws.onmessage = (event: MessageEvent | WebSocket.MessageEvent) => {
        if (!event.data) return;
        const m = JSON.parse(event.data) as ManagerEvent;
        switch (m.method) {
          case "get_options":
            this.emit("options", m.options);
            break;
          case "avatar_change":
            this.emit("avatar_change", m.avatar);
            break;
          case "parameters":
            this.emit("parameters", m.parameterName, m.parameterValue);
            break;
          default:
            break;
        }
      };
      this.ws.onerror = (err) => reject(err);
      this.ws.onclose = () => setTimeout((_) => this.reconnect(), 1000);
    });
  }

  private reconnect() {
    const old = this.ws!;
    console.log(green("Manager Reconnect"));
    this.ws = new WebSocket(this.address);
    this.ws.onopen = old.onopen;
    this.ws.onmessage = old.onmessage;
    this.ws.onclose = old.onclose;
  }

  private wsSend(data: any) {
    this.ws?.readyState === WebSocket.OPEN &&
      this.ws.send(JSON.stringify(data));
  }

  getOptions(): Promise<Record<string, any>> {
    return new Promise((resolve) => {
      this.once("options", (options) => resolve(options));
      this.wsSend({ method: "get_options", plugin: pluginName });
    });
  }

  send(addr: string, value: string | number | boolean) {
    this.wsSend({ method: "send", plugin: pluginName, addr, value });
  }

  listenParameters(p: string[]) {
    this.wsSend({
      method: "listen_parameters",
      plugin: pluginName,
      parameters: p,
    });
  }

  listenAvatarChange() {
    this.wsSend({ method: "listen_avatar_change", plugin: pluginName });
  }
}