Pixel-Composer/scripts/node_websocket_sender/node_websocket_sender.gml
2023-06-05 18:27:53 +02:00

129 lines
3.8 KiB
Plaintext

function Node_Websocket_Sender(_x, _y, _group = noone) : Node(_x, _y, _group) constructor {
name = "Websocket Sender";
w = 128;
h = 128;
min_h = h;
inputs[| 0] = nodeValue("Port", self, JUNCTION_CONNECT.input, VALUE_TYPE.integer, 22800);
inputs[| 1] = nodeValue("Data type", self, JUNCTION_CONNECT.input, VALUE_TYPE.integer, 0)
.setDisplay(VALUE_DISPLAY.enum_scroll, [ "Struct", "Surface", "File" ]);
inputs[| 2] = nodeValue("Struct", self, JUNCTION_CONNECT.input, VALUE_TYPE.struct, {});
inputs[| 3] = nodeValue("Surface", self, JUNCTION_CONNECT.input, VALUE_TYPE.surface, noone);
inputs[| 4] = nodeValue("Path", self, JUNCTION_CONNECT.input, VALUE_TYPE.path, "")
.setDisplay(VALUE_DISPLAY.path_load)
.nonValidate();
inputs[| 5] = nodeValue("Target", self, JUNCTION_CONNECT.input, VALUE_TYPE.text, "127.0.0.1");
input_display_list = [ 5, 0, 1, 2, 3, 4 ];
port = 0;
url = "";
connected = false;
function connectTo(newPort, newUrl) {
if(ds_map_exists(PORT_MAP, port))
array_remove(PORT_MAP[? port], self);
port = newPort;
url = newUrl;
if(!ds_map_exists(PORT_MAP, port))
PORT_MAP[? port] = [];
array_push(PORT_MAP[? port], self);
if(ds_map_exists(NETWORK_CLIENTS, newPort))
return;
var s = network_create_socket(network_socket_ws);
network_connect_raw_async(s, newUrl, newPort);
connected = false;
if(s >= 0) NETWORK_CLIENTS[? newPort] = s;
}
insp1UpdateTooltip = __txt("Reconnect");
insp1UpdateIcon = [ THEME.refresh, 1, COLORS._main_value_positive ];
static onInspector1Update = function() {
var _port = inputs[| 0].getValue();
var _url = inputs[| 5].getValue();
connectTo(_port, _url);
}
function asyncPackets(_async_load) {
var type = async_load[? "type"];
switch(type) {
case network_type_non_blocking_connect :
noti_status($"Websocket client: Connected at port {port} on node {display_name}");
connected = true;
break;
}
}
static step = function() {
var _type = inputs[| 1].getValue();
inputs[| 2].setVisible(_type == 0, _type == 0);
inputs[| 3].setVisible(_type == 1, _type == 1);
inputs[| 4].setVisible(_type == 2, _type == 2);
}
static update = function(frame = ANIMATOR.current_frame) {
var _port = inputs[| 0].getValue();
var _target = inputs[| 5].getValue();
if(port != _port || url != _target)
connectTo(_port, _target);
var network = ds_map_try_get(NETWORK_CLIENTS, _port, noone);
if(network < 0) return;
var _type = inputs[| 1].getValue();
switch(_type) {
case 0 :
var _stru = inputs[| 2].getValue();
var _str = json_stringify(_stru);
var _buff = buffer_from_string(_str);
var res = network_send_raw(network, _buff, buffer_get_size(_buff), network_send_text);
break;
case 1 :
var _surf = inputs[| 3].getValue();
if(!is_surface(_surf)) return;
var _buff = buffer_from_surface(_surf);
var res = network_send_raw(network, _buff, buffer_get_size(_buff), network_send_binary);
break;
case 2 :
var _path = inputs[| 4].getValue();
if(!file_exists(_path)) return;
var _buff = buffer_from_file(_path);
var res = network_send_raw(network, _buff, buffer_get_size(_buff), network_send_binary);
break;
}
}
static onDrawNode = function(xx, yy, _mx, _my, _s, _hover, _focus) {
var bbox = drawGetBbox(xx, yy, _s);
var network = ds_map_try_get(NETWORK_CLIENTS, port, noone);
var cc = CDEF.red, aa = 1;
if(network >= 0) cc = CDEF.lime;
var _y0 = bbox.y0 + ui(16);
var _y1 = bbox.y1 - ui(16);
var _ts = _s * 0.75;
draw_set_text(f_code, fa_center, fa_top, COLORS._main_text);
draw_set_alpha(0.75);
draw_text_add(bbox.xc, bbox.y0, $"Port {port}", _ts);
draw_set_alpha(1);
draw_sprite_fit(THEME.node_websocket_send, 0, bbox.xc, (_y0 + _y1) / 2, bbox.w, _y1 - _y0, cc, aa);
}
}