cactus-agent/src/api/ws/harvester/index.ts

cactus rpc/websocket client library

import {Plot} from "../../cactus/harvester/harvester";
import {TDaemon} from "../../../daemon/index";
import {GetMessageType, TConnectionGeneral, wallet_ui_service, metrics_service} from "../../types";
import {float, int, None, str} from "../../cactus/types/_python_types_";

export const cactus_harvester_service = "cactus_harvester";
export type cactus_harvester_service = typeof cactus_harvester_service;

export const get_connections_command = "get_connections";
export type get_connections_command = typeof get_connections_command;
export type TGetConnectionsBroadCast = {
  connections: TConnectionGeneral[];
};
export type WsGetConnectionsHarvesterMessage = GetMessageType<cactus_harvester_service, get_connections_command, TGetConnectionsBroadCast>;
export async function on_get_connections(daemon: TDaemon, callback: (e: WsGetConnectionsHarvesterMessage) => unknown){
  await daemon.subscribe(wallet_ui_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if(e.origin === cactus_harvester_service && e.command === get_connections_command){
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}

export const get_plots_command = "get_plots";
export type get_plots_command = typeof get_plots_command;
export type TGetPlotsBroadCast = {
  plots: Plot[];
  failed_to_open_filenames: string[];
  not_found_filenames: string[];
};
export type WsGetPlotsMessage = GetMessageType<cactus_harvester_service, get_plots_command, TGetPlotsBroadCast>;
export async function on_get_plots(daemon: TDaemon, callback: (e: WsGetPlotsMessage) => unknown){
  await daemon.subscribe(wallet_ui_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if(e.origin === cactus_harvester_service && e.command === get_plots_command){
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}

export const farming_info_command = "farming_info";
export type farming_info_command = typeof farming_info_command;
export type TFarmingInfoBroadCast = {
  challenge_hash: str;
  total_plots: int;
  found_proofs: int;
  eligible_plots: int;
  time: float;
};
export type WsFarmingInfoMessage = GetMessageType<cactus_harvester_service, farming_info_command, TFarmingInfoBroadCast>;
export async function on_farming_info(daemon: TDaemon, callback: (e: GetMessageType<cactus_harvester_service, farming_info_command, TFarmingInfoBroadCast>) => unknown){
  await daemon.subscribe(metrics_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if(e.origin === cactus_harvester_service && e.command === farming_info_command){
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}

export const add_connection_command = "add_connection";
export type add_connection_command = typeof add_connection_command;
export type TAddConnectionBroadCast = None;
export type WsAddConnectionMessage = GetMessageType<cactus_harvester_service, add_connection_command, TAddConnectionBroadCast>;
export async function on_add_connection(daemon: TDaemon, callback: (e: GetMessageType<cactus_harvester_service, add_connection_command, TAddConnectionBroadCast>) => unknown) {
  await daemon.subscribe(metrics_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if (e.origin === cactus_harvester_service && e.command === add_connection_command) {
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}

export const close_connection_command = "close_connection";
export type close_connection_command = typeof close_connection_command;
export type TCloseConnectionBroadCast = None;
export type WsCloseConnectionMessage = GetMessageType<cactus_harvester_service, close_connection_command, TCloseConnectionBroadCast>;
export async function on_close_connection(daemon: TDaemon, callback: (e: GetMessageType<cactus_harvester_service, close_connection_command, TCloseConnectionBroadCast>) => unknown) {
  await daemon.subscribe(metrics_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if (e.origin === cactus_harvester_service && e.command === close_connection_command) {
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}

export type WsHarvesterMessage = WsGetConnectionsHarvesterMessage
  | WsGetPlotsMessage
  | WsFarmingInfoMessage
  | WsAddConnectionMessage
  | WsCloseConnectionMessage
;
// Whole commands for the service
export type cactus_harvester_commands = get_plots_command
  | farming_info_command
  | add_connection_command
  | close_connection_command
  | get_connections_command
  ;
export type TCactusHarvesterBroadcast = TGetPlotsBroadCast
  | TFarmingInfoBroadCast
  | TAddConnectionBroadCast
  | TCloseConnectionBroadCast
  | TGetConnectionsBroadCast
  ;
export async function on_message_from_harvester(daemon: TDaemon, callback: (e: WsHarvesterMessage) => unknown){
  await daemon.subscribe(wallet_ui_service);
  const messageListener = (e: WsHarvesterMessage) => {
    if(e.origin === cactus_harvester_service){
      callback(e);
    }
  };
  return daemon.addMessageListener(cactus_harvester_service, messageListener);
}