import {bool, int, str, True, uint32} from "../../cactus/types/_python_types_";
import {TRPCAgent} from "../../../rpc";
import {Plot} from "../../cactus/harvester/harvester";
import {GetMessageType, ResType} from "../../types";
import {TDaemon} from "../../../daemon/index";

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

export const get_plots_command = "get_plots";
export type get_plots_command = typeof get_plots_command;
export type TGetPlotsRequest = {
};
export type TGetPlotsResponse = {
  plots: Plot[];
  failed_to_open_filenames: str[];
  not_found_filenames: str[];
};
export type WsGetPlotsMessage = GetMessageType<cactus_harvester_service, get_plots_command, TGetPlotsResponse>;
export async function get_plots<T extends TRPCAgent | TDaemon>(agent: T) {
  type R = ResType<T, TGetPlotsResponse, WsGetPlotsMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, get_plots_command);
}




export const refresh_plots_command = "refresh_plots";
export type refresh_plots_command = typeof refresh_plots_command;
export type TRefreshPlotsRequest = {
};
export type TRefreshPlotsResponse = {
};
export type WsRefreshPlotsMessage = GetMessageType<cactus_harvester_service, refresh_plots_command, TRefreshPlotsResponse>;
export async function refresh_plots<T extends TRPCAgent | TDaemon>(agent: T) {
  type R = ResType<T, TRefreshPlotsResponse, WsRefreshPlotsMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, refresh_plots_command);
}




export const delete_plot_command = "delete_plot";
export type delete_plot_command = typeof delete_plot_command;
export type TDeletePlotRequest = {
  filename: str;
};
export type TDeletePlotResponse = {
};
export type WsDeletePlotMessage = GetMessageType<cactus_harvester_service, delete_plot_command, TDeletePlotResponse>;
export async function delete_plot<T extends TRPCAgent | TDaemon>(agent: T, data: TDeletePlotRequest) {
  type R = ResType<T, TDeletePlotResponse, WsDeletePlotMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, delete_plot_command, data);
}




export const add_plot_directory_command = "add_plot_directory";
export type add_plot_directory_command = typeof add_plot_directory_command;
export type TAddPlotDirectoryRequest = {
  dirname: str;
};
export type TAddPlotDirectoryResponse = {
};
export type WsAddPlotDirectoryMessage = GetMessageType<cactus_harvester_service, add_plot_directory_command, TAddPlotDirectoryResponse>;
export async function add_plot_directory<T extends TRPCAgent | TDaemon>(agent: T, data: TAddPlotDirectoryRequest) {
  type R = ResType<T, TAddPlotDirectoryResponse, WsAddPlotDirectoryMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, add_plot_directory_command, data);
}




export const get_plot_directories_command = "get_plot_directories";
export type get_plot_directories_command = typeof get_plot_directories_command;
export type TGetPlotDirectoriesRequest = {
};
export type TGetPlotDirectoriesResponse = {
  directories: str[];
};
export type WsGetPlotDirectoriesMessage = GetMessageType<cactus_harvester_service, get_plot_directories_command, TGetPlotDirectoriesResponse>;
export async function get_plot_directories<T extends TRPCAgent | TDaemon>(agent: T) {
  type R = ResType<T, TGetPlotDirectoriesResponse, WsGetPlotDirectoriesMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, get_plot_directories_command);
}




export const remove_plot_directory_command = "remove_plot_directory";
export type remove_plot_directory_command = typeof remove_plot_directory_command;
export type TRemovePlotDirectoryRequest = {
  dirname: str;
};
export type TRemovePlotDirectoryResponse = {
};
export type WsRemovePlotDirectoryMessage = GetMessageType<cactus_harvester_service, remove_plot_directory_command, TRemovePlotDirectoryResponse>;
export async function remove_plot_directory<T extends TRPCAgent | TDaemon>(agent: T, data: TRemovePlotDirectoryRequest) {
  type R = ResType<T, TRemovePlotDirectoryResponse, WsRemovePlotDirectoryMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, remove_plot_directory_command, data);
}


export const get_harvester_config_command = "get_harvester_config";
export type get_harvester_config_command = typeof get_harvester_config_command;
export type TGetHarvesterConfigResponse = {
  success: True;
  use_gpu_harvesting: bool;
  gpu_index: int;
  enforce_gpu_index: bool;
  disable_cpu_affinity: bool;
  parallel_decompressor_count: int;
  decompressor_thread_count: int;
  recursive_plot_scan: bool;
  refresh_parameter_interval_seconds: int;
};
export type WsGetHarvesterConfigMessage = GetMessageType<cactus_harvester_service, get_harvester_config_command, TGetHarvesterConfigResponse>;
export async function get_harvester_config<T extends TRPCAgent | TDaemon>(agent: T) {
  type R = ResType<T, TGetHarvesterConfigResponse, WsGetHarvesterConfigMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, get_harvester_config_command);
}


export const update_harvester_config_command = "update_harvester_config";
export type update_harvester_config_command = typeof update_harvester_config_command;
export type TUpdateHarvesterConfigRequest = {
  use_gpu_harvesting?: bool;
  gpu_index?: int;
  enforce_gpu_index?: bool;
  disable_cpu_affinity?: bool;
  parallel_decompressor_count?: int;
  decompressor_thread_count?: int;
  recursive_plot_scan?: bool;
  refresh_parameter_interval_seconds?: uint32;
};
export type TUpdateHarvesterConfigResponse = {
};
export type WsUpdateHarvesterConfigMessage = GetMessageType<cactus_harvester_service, update_harvester_config_command, TUpdateHarvesterConfigResponse>;
export async function update_harvester_config<T extends TRPCAgent | TDaemon>(agent: T, data: TUpdateHarvesterConfigRequest) {
  type R = ResType<T, TUpdateHarvesterConfigResponse, WsUpdateHarvesterConfigMessage>;
  return agent.sendMessage<R>(cactus_harvester_service, update_harvester_config_command, data);
}

export type RpcHarvesterMessage =
  TAddPlotDirectoryResponse
  | TDeletePlotResponse
  | TGetPlotDirectoriesResponse
  | TGetPlotsResponse
  | TRefreshPlotsResponse
  | TRemovePlotDirectoryResponse
  | TGetHarvesterConfigResponse
  | TUpdateHarvesterConfigResponse
;

export type RpcHarvesterMessageOnWs =
  WsAddPlotDirectoryMessage
  | WsDeletePlotMessage
  | WsGetPlotDirectoriesMessage
  | WsGetPlotsMessage
  | WsRefreshPlotsMessage
  | WsRemovePlotDirectoryMessage
  | WsGetHarvesterConfigMessage
  | WsUpdateHarvesterConfigMessage
;