ms/packages/sponge/src/channel.ts
MiaoWoo 0584ec9d85 refactor: rename scope to @ccms
Signed-off-by: MiaoWoo <admin@yumc.pw>
2020-05-03 00:14:47 +08:00

39 lines
1.6 KiB
TypeScript

import { channel, plugin } from '@ccms/api'
import { inject, provideSingleton } from '@ccms/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");
@provideSingleton(channel.Channel)
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()), { data, connection, side })
}
})
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);
}
}