ms/packages/sponge/src/channel.ts

39 lines
1.5 KiB
TypeScript
Raw Normal View History

import { channel, plugin } from '@ms/api'
import { inject, injectable } from '@ms/container'
const Sponge = org.spongepowered.api.Sponge
const RawDataListener = Java.type("org.spongepowered.api.network.RawDataListener")
const ChannelRegistrar = Sponge.getChannelRegistrar()
const Consumer = Java.type("java.util.function.Consumer");
@injectable()
export class SpongeChannel extends channel.Channel {
@inject(plugin.PluginInstance)
private pluginInstance: any;
private channelMap = new Map<string, any>();
send(player: any, channel: string, data: any) {
if (!this.channelMap.has(channel)) { return }
this.channelMap.get(channel).sendTo(player, new Consumer({
accept: (channelBuf: any) => channelBuf.writeBytes(data)
}))
}
register(channel: string, listener: channel.ChannelListener) {
if (!this.channelMap.has(channel)) {
this.channelMap.set(channel, ChannelRegistrar.getOrCreateRaw(this.pluginInstance, channel))
}
let innerListener = new RawDataListener({
handlePayload: (/* ChannelBuf */ data: any, /**RemoteConnection */ connection: any, /**Platform.Type */ side: any) => {
listener(data.readBytes(data.available()))
}
})
this.channelMap.get(channel).addListener(innerListener);
return innerListener;
}
unregister(channel: string, listener: any) {
if (!this.channelMap.has(channel)) { return }
this.channelMap.get(channel).removeListener(listener);
}
}