UNPKG

madoi-client

Version:
1,209 lines (1,118 loc) 35.2 kB
import { TypedCustomEventListenerOrObject, TypedCustomEventTarget } from "tcet"; // ---- message definitions ---- export type CastType = "UNICAST" | "MULTICAST" | "BROADCAST" | "SELFCAST" | "OTHERCAST" | "PEERTOSERVER" | "SERVERTOPEER"; export interface Message{ type: string; sender: string; castType: CastType; recipients: string[] | undefined; [name: string]: any; } export interface RoomSpec{ maxLog: number; } export interface RoomInfo{ id: string; spec: RoomSpec; profile: {[key: string]: any}; } export interface PeerInfo{ id: string; order: number; profile: {[key: string]: any}; } // CastType別メッセージ用interface。 export interface ServerToPeerMessage extends Message{ sender: "__SERVER__"; castType: "SERVERTOPEER"; recipients: undefined; } export interface PeerToServerMessage extends Message{ castType: "PEERTOSERVER"; recipients: undefined; } const peerToServerMessageDefault = { sender: "__PEER__", castType: "PEERTOSERVER" as "PEERTOSERVER", recipients: undefined }; export interface PeerToPeerMessage extends Message{ castType: "UNICAST" | "MULTICAST" | "BROADCAST" | "SELFCAST" | "OTHERCAST"; } export interface BroadcastMessage extends PeerToPeerMessage{ castType: "BROADCAST"; recipients: undefined; } const broadcastMessageDefault = { sender: "__PEER__", castType: "BROADCAST" as "BROADCAST", recipients: undefined }; export interface BroadcastOrOthercastMessage extends PeerToPeerMessage{ castType: "BROADCAST" | "OTHERCAST"; recipients: undefined; } const broadcastOrOthercastMessageDefault = { sender: "__PEER__", recipients: undefined }; export interface Ping extends PeerToServerMessage{ type: "Ping"; body: object | undefined; } export function newPing(body = undefined): Ping{ return { type: "Ping", ...peerToServerMessageDefault, body: body }; } export interface Pong extends ServerToPeerMessage{ type: "Pong"; body: object | undefined; } export interface EnterRoomBody{ room?: { spec: RoomSpec; profile: {[key: string]: any}; }; selfPeer?: PeerInfo; } export interface EnterRoom extends PeerToServerMessage, EnterRoomBody{ type: "EnterRoom"; } export function newEnterRoom(body: EnterRoomBody): EnterRoom{ return { type: "EnterRoom", ...peerToServerMessageDefault, ...body }; } export interface EnterRoomAllowed extends ServerToPeerMessage{ type: "EnterRoomAllowed"; room: RoomInfo; selfPeer: PeerInfo; otherPeers: PeerInfo[]; histories: StoredMessageType[]; } export interface EnterRoomDenied extends ServerToPeerMessage{ type: "EnterRoomDenied"; message: string; } export interface LeaveRoomBody{ } export interface LeaveRoom extends PeerToServerMessage, LeaveRoomBody{ type: "LeaveRoom"; } export function newLeaveRoom(body: LeaveRoomBody): LeaveRoom{ return { type: "LeaveRoom", ...peerToServerMessageDefault, ...body }; } export interface LeaveRoomDone extends ServerToPeerMessage{ type: "LeaveRoomDone"; } export interface UpdateRoomProfileBody{ updates?: {[key: string]: any}; deletes?: string[]; } export interface UpdateRoomProfile extends BroadcastMessage, UpdateRoomProfileBody{ type: "UpdateRoomProfile" } export function newUpdateRoomProfile(body: UpdateRoomProfileBody): UpdateRoomProfile{ return { type: "UpdateRoomProfile", ...broadcastMessageDefault, ...body }; } export interface PeerEntered extends ServerToPeerMessage{ type: "PeerEntered"; peer: PeerInfo; } export interface PeerLeaved extends ServerToPeerMessage{ type: "PeerLeaved"; peerId: string; } export interface UpdatePeerProfileBody{ updates?: {[key: string]: any}; deletes?: string[]; } export interface UpdatePeerProfile extends BroadcastMessage, UpdatePeerProfileBody{ type: "UpdatePeerProfile" } export function newUpdatePeerProfile(body: UpdatePeerProfileBody): UpdatePeerProfile{ return { type: "UpdatePeerProfile", ...broadcastMessageDefault, ...body }; } export interface FunctionDefinition{ funcId: number; name: string; config: MethodConfig; } export interface DefineFunctionBody{ definition: FunctionDefinition; } export interface DefineFunction extends PeerToServerMessage, DefineFunctionBody{ type: "DefineFunction"; } export function newDefineFunction(body: DefineFunctionBody): DefineFunction{ return { type: "DefineFunction", ...peerToServerMessageDefault, ...body }; } export interface MethodDefinition{ methodId: number; name: string; config: MethodConfig; } export interface ObjectDefinition{ objId: number; className: string; methods: MethodDefinition[]; } export interface DefineObjectBody{ definition: ObjectDefinition; } export interface DefineObject extends PeerToServerMessage, DefineObjectBody{ type: "DefineObject"; } export function newDefineObject(body: DefineObjectBody): DefineObject{ return { type: "DefineObject", ...peerToServerMessageDefault, ...body } } export interface InvokeFunctionBody{ funcId: number; args: any[]; } export interface InvokeFunction extends BroadcastOrOthercastMessage, InvokeFunctionBody{ type: "InvokeFunction"; } export function newInvokeFunction(castType: "BROADCAST" | "OTHERCAST", body: InvokeFunctionBody): InvokeFunction{ return { type: "InvokeFunction", castType: castType, ...broadcastOrOthercastMessageDefault, ...body }; } export interface UpdateObjectStateBody{ objId: number; objRevision: number; state: string; } export interface UpdateObjectState extends PeerToServerMessage{ type: "UpdateObjectState"; } export function newUpdateObjectState(body: UpdateObjectStateBody): UpdateObjectState{ return { type: "UpdateObjectState", ...peerToServerMessageDefault, ...body }; } export interface InvokeMethodBody{ objId: number; objRevision: number; // メソッド実行前のクライアントのオブジェクトリビジョン methodId: number; args: any[]; serverObjRevision?: number; // サーバー側のオブジェクトリビジョン } export interface InvokeMethod extends BroadcastOrOthercastMessage, InvokeMethodBody{ type: "InvokeMethod"; } export function newInvokeMethod(castType: "BROADCAST" | "OTHERCAST", body: InvokeMethodBody): InvokeMethod{ return { type: "InvokeMethod", castType: castType, ...broadcastOrOthercastMessageDefault, ...body }; } export interface UserMessage extends Message{ content: any; } export type UpstreamMessageType = Ping | EnterRoom | LeaveRoom | UpdateRoomProfile | UpdatePeerProfile | DefineFunction | DefineObject | InvokeFunction | UpdateObjectState | InvokeMethod; export type DownStreamMessageType = Pong | EnterRoomAllowed | EnterRoomDenied | LeaveRoomDone | UpdateRoomProfile | PeerEntered | PeerLeaved | UpdatePeerProfile | InvokeFunction | UpdateObjectState | InvokeMethod | UserMessage; export type StoredMessageType = InvokeMethod | InvokeFunction | UpdateObjectState; //---- decorators ---- // Decorator export function ShareClass(config: {className?: string} = {}){ return (target: any) => { target.madoiClassConfig_ = config; }; } // Decorator export interface ShareConfig{ type?: "beforeExec" | "afterExec" maxLog?: number allowedTo?: string[] update?: {freq?: number, interpolateBy?: number, reckonUntil?: number} } export const shareConfigDefault: ShareConfig = { type: "beforeExec", maxLog: 0, allowedTo: ["USER"] }; export function Share(config: ShareConfig = shareConfigDefault) { const c = config; if(!c.type) c.type = "beforeExec"; if(!c.maxLog) c.maxLog = 0; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {share: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface NotifyConfig{ type?: "beforeExec" | "afterExec" } export const notifyConfigDefault: NotifyConfig = { type: "beforeExec" }; export function Notify(config: NotifyConfig = notifyConfigDefault) { const c = {...config}; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {notify: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface GetStateConfig{ maxInterval?: number maxUpdates?: number } export const getStateConfigDefault: GetStateConfig = { maxInterval: 5000 }; export function GetState(config: GetStateConfig = getStateConfigDefault){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {getState: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface SetStateConfig{ } export function SetState(config: SetStateConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {setState: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface HostOnlyConfig{ } export function HostOnly(config: HostOnlyConfig = {}){ return (target: any, name: string, _descriptor: PropertyDescriptor) => { const c = config; const mc: MethodConfig = {hostOnly: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface BeforeEnterRoomConfig{ } export function BeforeEnterRoom(config: BeforeEnterRoomConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {beforeEnterRoom: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface EnterRoomAllowedConfig{ } export function EnterRoomAllowed(config: EnterRoomAllowedConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {enterRoomAllowed: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface EnterRoomDeniedConfig{ } export function EnterRoomDenied(config: EnterRoomDeniedConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {enterRoomDenied: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface LeaveRoomDoneConfig{ } export function LeaveRoomDone(config: LeaveRoomDoneConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {leaveRoomDone: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface RoomProfileUpdatedConfig{ } export function RoomProfileUpdated(config: RoomProfileUpdatedConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {roomProfileUpdated: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface PeerEnteredConfig{ } export function PeerEntered(config: PeerEnteredConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {peerEntered: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface PeerLeavedConfig{ } export function PeerLeaved(config: PeerLeavedConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {peerLeaved: c}; target[name].madoiMethodConfig_ = mc; } } // Decorator export interface PeerProfileUpdatedConfig{ } export function PeerProfileUpdated(config: PeerProfileUpdatedConfig = {}){ const c = config; return (target: any, name: string, _descriptor: PropertyDescriptor) => { const mc: MethodConfig = {peerProfileUpdated: c}; target[name].madoiMethodConfig_ = mc; } } export type MethodConfig = {share: ShareConfig} | {notify: NotifyConfig} | {hostOnly: HostOnlyConfig} | {getState: GetStateConfig} | {setState: SetStateConfig} | {beforeEnterRoom: BeforeEnterRoomConfig} | {enterRoomAllowed: EnterRoomAllowedConfig} | {enterRoomDenied: EnterRoomDeniedConfig} | {leaveRoomDone: LeaveRoomDoneConfig} | {roomProfileUpdated: RoomProfileUpdatedConfig} | {peerEntered: PeerEnteredConfig} | {peerLeaved: PeerLeavedConfig} | {peerProfileUpdated: PeerProfileUpdatedConfig}; // ---- madoi ---- export type MethodAndConfigParam = {method: Function} & MethodConfig; interface FunctionEntry { promise?: Promise<any>; resolve?: Function; reject?: Function; original: Function; config: {share?: ShareConfig, notify?: NotifyConfig}; } type MadoiObject = {[key: string]: any, madoiClassConfig_: {className?: string}, madoiObjectId_: number}; interface ObjectEntry { instance: MadoiObject; modification: number; revision: number; } interface MethodEntry { promise?: Promise<any>; resolve?: Function; reject?: Function; original: Function; config: {share?: ShareConfig, notify?: NotifyConfig}; } //---- events ---- export interface EnterRoomAllowedDetail{ room: RoomInfo; selfPeer: PeerInfo; otherPeers: PeerInfo[]; } export interface EnterRoomDeniedDetail{ message: string; } export interface LeaveRoomDoneDetail{ } export interface RoomProfileUpdatedDetail{ updates?: {[key: string]: any}; deletes?: string[]; } export interface PeerEnteredDetail{ peer: PeerInfo; } export interface PeerLeavedDetail{ peerId: string; } export interface PeerProfileUpdatedDetail{ peerId: string; updates?: {[key: string]: any}; deletes?: string[]; } export interface UserMessageDetail<T>{ type: string; sender?: string; castType?: CastType; recipients?: string[]; content: T; } interface ErrorDetail{ error: any; } export type UserMessageListener<D> = TypedCustomEventListenerOrObject<Madoi, UserMessageDetail<D>> | null; export class Madoi extends TypedCustomEventTarget<Madoi, { enterRoomAllowed: EnterRoomAllowedDetail, enterRoomDenied: EnterRoomDeniedDetail, leaveRoomDone: LeaveRoomDoneDetail, roomProfileUpdated: RoomProfileUpdatedDetail, peerEntered: PeerEnteredDetail, peerProfileUpdated: PeerProfileUpdatedDetail, peerLeaved: PeerLeavedDetail, error: ErrorDetail }>{ private connecting: boolean = false; private interimQueue: Array<object>; private shareOrNotifyFunctions = new Map<string, FunctionEntry>(); private shareObjects = new Map<number, ObjectEntry>(); private shareOrNotifyMethods = new Map<string, MethodEntry>(); // annotated methods private getStateMethods = new Map<number, {method: (madoi: Madoi)=>any, config: GetStateConfig, lastGet: number}>(); private setStateMethods = new Map<number, (state: any, madoi: Madoi)=>void>(); // objectId -> @SetState method private beforeEnterRoomMethods = new Map<number, (selfProfile: {[key: string]: string}, madoi: Madoi)=>void>(); private enterRoomAllowedMethods = new Map<number, (detail: EnterRoomAllowedDetail, madoi: Madoi)=>void>(); private enterRoomDeniedMethods = new Map<number, (detail: EnterRoomDeniedDetail, madoi: Madoi)=>void>(); private leaveRoomDoneMethods = new Map<number, (madoi: Madoi)=>void>(); private roomProfileUpdatedMethods = new Map<number, (detail: RoomProfileUpdatedDetail, madoi: Madoi)=>void>(); private peerEnteredMethods = new Map<number, (detail: PeerEnteredDetail, madoi: Madoi)=>void>(); private peerLeavedMethods = new Map<number, (detail: PeerLeavedDetail, madoi: Madoi)=>void>(); private peerProfileUpdatedMethods = new Map<number, (detail: PeerProfileUpdatedDetail, madoi: Madoi)=>void>(); private url: string; private ws: WebSocket | null = null; private room: RoomInfo = {id: "", spec: {maxLog: 1000}, profile: {}}; private selfPeer: PeerInfo = {id: "", order: -1, profile: {}}; private peers = new Map<string, PeerInfo>(); private currentSender: string | null = null; constructor(roomIdOrUrl: string, authToken: string, selfPeer?: {id: string, profile: {[key: string]: any}}, room?: {spec: RoomSpec, profile: {[key: string]: any}}){ super(); if(room) this.room = {...this.room, ...room}; if(selfPeer) this.selfPeer = {...this.selfPeer, ...selfPeer, order: -1}; this.interimQueue = new Array(); const sep = roomIdOrUrl.indexOf("?") != -1 ? "&" : "?"; if(roomIdOrUrl.match(/^wss?:\/\//)){ this.url = `${roomIdOrUrl}${sep}authToken=${authToken}`; this.room.id = roomIdOrUrl.split("rooms/")[1].split("?")[0]; } else{ const p = (document.querySelector("script[src$='madoi.js']") as HTMLScriptElement).src.split("\/", 5); const contextUrl = (p[0] == "http:" ? "ws:" : "wss:") + "//" + p[2] + "/" + p[3]; this.url = `${contextUrl}/rooms/${roomIdOrUrl}${sep}authToken=${authToken}`; this.room.id = roomIdOrUrl; } this.ws = new WebSocket(this.url); this.ws.onopen = e => this.handleOnOpen(e); this.ws.onclose = e => this.handleOnClose(e); this.ws.onerror = e => this.handleOnError(e); this.ws.onmessage = e => this.handleOnMessage(e); setInterval(()=>{this.saveStates();}, 1000); setInterval(()=>{this.sendPing();}, 30000); } getRoomId(){ return this.room.id; } getRoomProfile(){ return this.room?.profile; } setRoomProfile(name: string, value: any){ const m: {[key: string]: any} = {}; m[name] = value; this.sendMessage(newUpdateRoomProfile( {updates: m} )); } removeRoomProfile(name: string){ this.sendMessage(newUpdateRoomProfile( {deletes: [name]} )); } getSelfPeerId(){ return this.selfPeer?.id; } getSelfPeerProfile(){ return this.selfPeer.profile; } updateSelfPeerProfile(name: string, value: any){ this.selfPeer.profile[name] = value; const m: {[key: string]: any} = {}; m[name] = value; this.sendMessage(newUpdatePeerProfile( {updates: m} )); const v: PeerProfileUpdatedDetail = {updates: m, peerId: this.selfPeer.id}; for(const [_, f] of this.peerProfileUpdatedMethods){ f(v, this); } this.dispatchCustomEvent("peerProfileUpdated", v); } removeSelfPeerProfile(name: string){ delete this.selfPeer.profile[name]; this.sendMessage(newUpdatePeerProfile( {deletes: [name]} )); const v: PeerProfileUpdatedDetail = {deletes: [name], peerId: this.selfPeer.id}; for(const [_, f] of this.peerProfileUpdatedMethods){ f(v, this); } this.dispatchCustomEvent("peerProfileUpdated", v); } getCurrentSender(){ if(!this.currentSender) return null; return this.peers.get(this.currentSender); } isCurrentSenderSelf(){ return this.currentSender === this.selfPeer.id; } close(){ this.ws?.close(); this.ws = null; } private sendPing(){ this.ws?.send(JSON.stringify(newPing())); } private handleOnOpen(_e: Event){ this.connecting = true; for(const [_, f] of this.beforeEnterRoomMethods){ f(this.selfPeer.profile, this); } this.doSendMessage(newEnterRoom({ room: this.room, selfPeer: this.selfPeer })); for(let m of this.interimQueue){ this.ws?.send(JSON.stringify(m)); } this.interimQueue = []; } private handleOnClose(e: CloseEvent){ console.debug(`websocket closed because: ${e.reason}.`); this.connecting = false; this.ws = null; } private handleOnError(_e: Event){ } private handleOnMessage(e: MessageEvent){ const msg = JSON.parse(e.data); this.currentSender = msg.sender; this.data(msg); } private data(msg: DownStreamMessageType){ if(msg.type == "Pong"){ } else if(msg.type === "EnterRoomAllowed"){ const m: EnterRoomAllowedDetail = msg as EnterRoomAllowed; for(const [_, f] of this.enterRoomAllowedMethods){ f(m, this); } this.room = msg.room; this.selfPeer.order = msg.selfPeer.order; this.peers.set(m.selfPeer.id, {...m.selfPeer, profile: this.selfPeer.profile}); for(const p of m.otherPeers){ this.peers.set(p.id, p); } this.dispatchCustomEvent("enterRoomAllowed", m); if(msg.histories) for(const h of msg.histories){ this.data(h); } } else if(msg.type === "EnterRoomDenied"){ const m = msg as EnterRoomDenied; const d: EnterRoomDeniedDetail = m; for(const [_, f] of this.enterRoomDeniedMethods){ f(d, this); } this.dispatchCustomEvent("enterRoomDenied", d); } else if(msg.type == "LeaveRoomDone"){ for(const [_, f] of this.leaveRoomDoneMethods){ f(this); } this.dispatchCustomEvent("leaveRoomDone"); } else if(msg.type === "UpdateRoomProfile"){ const m = msg as UpdateRoomProfile; if(msg.updates) for(const [key, value] of Object.entries(msg.updates)) { this.room.profile[key] = value; } if(msg.deletes) for(const key of msg.deletes){ delete this.room.profile[key]; } for(const [_, f] of this.roomProfileUpdatedMethods){ f(m, this); } this.dispatchCustomEvent("roomProfileUpdated", m); } else if(msg.type === "PeerEntered"){ const m: PeerEnteredDetail = msg as PeerEntered; this.peers.set(m.peer.id, m.peer); for(const [_, f] of this.peerEnteredMethods){ f(m, this); } this.dispatchCustomEvent("peerEntered", m); } else if(msg.type === "PeerLeaved"){ const m: PeerLeavedDetail = msg as PeerLeaved; this.peers.delete(msg.peerId); for(const [_, f] of this.peerLeavedMethods){ f(m, this); } this.dispatchCustomEvent("peerLeaved", m); } else if(msg.type === "UpdatePeerProfile"){ const p = this.peers.get(msg.sender!); if(msg.sender && p){ if(msg.updates) for(const [key, value] of Object.entries(msg.updates)) { p.profile[key] = value; } if(msg.deletes) for(const key of msg.deletes){ delete p.profile[key]; } const v: PeerProfileUpdatedDetail = {...msg, peerId: msg.sender}; for(const [_, f] of this.peerProfileUpdatedMethods){ f(v, this); } this.dispatchCustomEvent("peerProfileUpdated", v); } } else if(msg.type === "InvokeFunction"){ const id = `${msg.funcId}`; const f = this.shareOrNotifyFunctions.get(id); if(f === undefined){ console.warn("no suitable function for ", msg); return; } const ret = this.applyInvocation(f.original, msg.args); if(ret instanceof Promise){ ret.then(()=>{ f.resolve?.apply(null, arguments); }).catch(()=>{ f.reject?.apply(null, arguments); }); } } else if(msg.type === "UpdateObjectState"){ const f = this.setStateMethods.get(msg.objId); if(f) f(msg.state, msg.objRevision); const o = this.shareObjects.get(msg.objId); if(o) o.revision = msg.objRevision; } else if(msg.type === "InvokeMethod"){ const o = this.shareObjects.get(msg.objId); if(o === undefined){ console.error(`Object not found for id: ${msg.objId}.`, msg); return; } const id = `${msg.objId}:${msg.methodId}`; const m = this.shareOrNotifyMethods.get(id); if(m === undefined){ console.error(`Method not found for id: ${id}.`, msg); return; } if(m.config.share){ if(o.revision + 1 !== msg.serverObjRevision){ console.error(`Found inconsistency. serverObjRevision must be ${o.revision + 1} but ${msg.serverObjRevision}.`, msg); } o.revision = msg.serverObjRevision; } const ret = this.applyInvocation(m.original, msg.args); if(ret instanceof Promise){ ret.then(()=>{ m.resolve?.apply(null, arguments); }).catch(()=>{ m.reject?.apply(null, arguments); }); } } else if(msg.type){ this.dispatchEvent(new CustomEvent(msg.type, {detail: msg})); } else{ console.warn("Unknown message type.", msg); } } private systemMessageTypes = [ "Ping", "Pong", "EnterRoom", "EnterRoomAllowed", "EnterRoomDenied", "LeaveRoom", "LeaveRoomDone", "UpdateRoomProfile", "PeerArrived", "PeerLeaved", "UpdatePeerProfile", "DefineFunction", "DefineObject", "InvokeFunction", "UpdateObjectState", "InvokeMethod" ]; private isSystemMessageType(type: string){ return type in this.systemMessageTypes; } send(type: string, content: any, castType: "BROADCAST" | "SELFCAST" | "OTHERCAST" | "PEERTOSERVER" = "BROADCAST" ){ if(!this.ws) return; this.sendMessage({ type: type, sender: this.selfPeer.id, castType: castType, recipients: undefined, content: content, }); } unicast(type: string, content: any, recipient: string){ this.sendMessage({ type: type, sender: this.selfPeer.id, castType: "UNICAST", recipients: [recipient], content: content }); } multicast(type: string, content: any, recipients: string[]){ this.sendMessage({ type: type, sender: this.selfPeer.id, castType: "MULTICAST", recipients: recipients, content: content }); } broadcast(type: string, content: any){ this.sendMessage({ type: type, sender: this.selfPeer.id, castType: "BROADCAST", recipients: undefined, content: content }); } othercast(type: string, content: any){ this.sendMessage({ type: type, sender: this.selfPeer.id, castType: "OTHERCAST", recipients: undefined, content: content }); } sendMessage(msg: Message){ if(this.isSystemMessageType(msg.type)) throw new Error("システムメッセージは送信できません。"); this.doSendMessage(msg); } addReceiver<D>(type: string, listener: UserMessageListener<D>){ if(this.isSystemMessageType(type)) throw new Error("システムメッセージのレシーバは登録できません。"); this.addEventListener(type as any, listener as EventListener); } removeReceiver<D>(type: string, listener: UserMessageListener<D>){ this.removeEventListener(type as any, listener as EventListener); } private replacer(_: any, value: any) { if (value instanceof Map) { return Object.fromEntries(value); } else { return value; } } private doSendMessage(msg: Message){ if(this.connecting){ this.ws?.send(JSON.stringify(msg, this.replacer)); } else{ this.interimQueue.push(msg); } } registerFunction<T extends Function>(func: T, config: MethodConfig = {share: {}}): T{ if("hostOnly" in config){ return this.addHostOnlyFunction(func, config); } else if("notify" in config){ // デフォルト値チェック if(!config.notify.type) config.notify.type = notifyConfigDefault.type; const funcName = func.name; const funcId = this.shareOrNotifyFunctions.size; const f = this.createFunctionProxy(func, {notify: config.notify}, funcId); const ret = function(){ return f.apply(null, arguments); } as any; this.doSendMessage(newDefineFunction({ definition: { funcId: funcId, name: funcName, config: config } })); return ret; } else if("share" in config){ // デフォルト値チェック if(!config.share.type) config.share.type = shareConfigDefault.type; if(!config.share.maxLog) config.share.maxLog = shareConfigDefault.maxLog; const funcName = func.name; const funcId = this.shareOrNotifyFunctions.size; const f = this.createFunctionProxy(func, {share: config.share}, funcId); const ret = function(){ return f.apply(null, arguments); } as any; this.doSendMessage(newDefineFunction({ definition: { funcId: funcId, name: funcName, config: config } })); return ret; } return func; } register<T>(object: T, methodAndConfigs: MethodAndConfigParam[] = []): T{ if(!this.ws) return object; const obj = object as MadoiObject; if(obj.madoiObjectId_){ console.warn("Ignore object registration because it's already registered."); return object; } let className = obj.constructor.name; if(obj.__proto__.constructor.madoiClassConfig_){ className = obj.__proto__.constructor.madoiClassConfig_.className; } // 共有オブジェクトのidを確定 const objId = this.shareObjects.size; const objEntry = {instance: obj, revision: 0, modification: 0}; this.shareObjects.set(objId, objEntry); obj.madoiObjectId_ = objId; // コンフィグを集める const methods = new Array<Function>(); const methodDefinitions = new Array<MethodDefinition>(); const methodToIndex = new Map<string, number>(); // デコレータから Object.getOwnPropertyNames(Object.getPrototypeOf(obj)).forEach(methodName => { const f = obj[methodName]; if(typeof(f) != "function") return; if(!f.madoiMethodConfig_) return; const cfg: MethodConfig = f.madoiMethodConfig_; const mi = methods.length; methodToIndex.set(methodName, mi); methods.push(f); methodDefinitions.push({methodId: mi, name: methodName, config: cfg}); console.debug(`add config ${className}.${methodName}=${JSON.stringify(cfg)} from decorator`); }); // 引数から for(const mc of methodAndConfigs){ const f = mc.method; const c: MethodConfig = mc; const methodName = f.name; if("share" in c){ // デフォルト値チェック if(!c.share.type) c.share.type = shareConfigDefault.type; if(!c.share.maxLog) c.share.maxLog = shareConfigDefault.maxLog; } else if("notify" in c){ // デフォルト値チェック if(!c.notify.type) c.notify.type = notifyConfigDefault.type; } else if("hostOnly" in c){ } else if("getState" in c){ if(!c.getState.maxInterval) c.getState.maxInterval = getStateConfigDefault.maxInterval; } else if("setState" in c){ } else if("enterRoomAllowed" in c){ } else if("enterRoomDenied" in c){ } else if("leaveRoomDone" in c){ } else if("peerEntered" in c){ } else if("peerLeaved" in c){ } else{ continue; } const mi = methodToIndex.get(methodName); if(typeof mi === "undefined"){ // 追加 const mi = methods.length; methodToIndex.set(methodName, mi); methods.push(f); methodDefinitions.push({methodId: mi, name: mc.method.name, config: c}); console.debug(`add config ${className}.${methodName}=${JSON.stringify(mc)} from argument`); } else{ // 既にあれば設定を置き換え methodDefinitions[mi].config = mc; console.debug(`replace config ${className}.${methodName}=${JSON.stringify(mc)} from argument`); } } // 集めたコンフィグ内のメソッドに応じて登録や置き換え処理を行う。 for(let i = 0; i < methods.length; i++){ const f = methods[i]; const mc = methodDefinitions[i]; const c = mc.config; if("share" in c){ // @Shareの場合はメソッドを置き換え const newf = this.createMethodProxy( f.bind(obj), {share: c.share}, objId, mc.methodId); obj[mc.name] = function(){ objEntry.modification++; return newf.apply(null, arguments); }; } else if("notify" in c){ // @Notifyの場合はメソッドを置き換え const newf = this.createMethodProxy( f.bind(obj), {notify: c.notify}, objId, mc.methodId); obj[mc.name] = function(){ return newf.apply(null, arguments); }; } else if("hostOnly" in c){ // @HostOnlyの場合はメソッドを置き換え const newf = this.addHostOnlyFunction( f.bind(obj), c.hostOnly); obj[mc.name] = function(){ objEntry.modification++; return newf.apply(null, arguments); } } else if("getState" in c){ // @GetStateの場合はメソッドを登録 this.getStateMethods.set(objId, {method: f.bind(obj), config: c.getState, lastGet: 0}); } else if("setState" in c){ // @SetStateの場合はメソッドを登録 this.setStateMethods.set(objId, f.bind(obj)); } else if("beforeEnterRoom" in c){ // @BeforeEnterRoomの場合はメソッドを登録 this.beforeEnterRoomMethods.set(objId, f.bind(obj)); } else if("enterRoomAllowed" in c){ // @EnterRoomAllowedの場合はメソッドを登録 this.enterRoomAllowedMethods.set(objId, f.bind(obj)); } else if("enterRoomDenied" in c){ // @EnterRoomDeniedの場合はメソッドを登録 this.enterRoomDeniedMethods.set(objId, f.bind(obj)); } else if("leaveRoomDone" in c){ // @LeaveRoomDoneの場合はメソッドを登録 this.leaveRoomDoneMethods.set(objId, f.bind(obj)); } else if("peerEntered" in c){ // @PeerEnteredの場合はメソッドを登録 this.peerEnteredMethods.set(objId, f.bind(obj)); } else if("peerProfileUpdated" in c){ // @PeerProfileUpdatedの場合はメソッドを登録 this.peerProfileUpdatedMethods.set(objId, f.bind(obj)) } else if("peerLeaved" in c){ // @PeerLeavedの場合はメソッドを登録 this.peerLeavedMethods.set(objId, f.bind(obj)) } } const msg = newDefineObject({ definition: { objId: objId, className: className, methods: methodDefinitions } }); this.doSendMessage(msg); return object; } private createFunctionProxy(f: Function, config: {share?: ShareConfig, notify?: NotifyConfig}, funcId: number): Function{ const id = `${funcId}`; const fe: FunctionEntry = {original: f, config}; this.shareOrNotifyFunctions.set(id, fe); fe.promise = new Promise((resolve, reject)=>{ fe.resolve = resolve; fe.reject = reject; }); const self = this; return function(){ if(self.ws === null){ if(f) return f.apply(null, arguments); } else{ let ret = null; let castType: CastType = "BROADCAST"; if(config.share?.type === "afterExec" || config.notify?.type === "afterExec"){ ret = f.apply(null, arguments); castType = "OTHERCAST"; } self.sendMessage(newInvokeFunction( castType, { funcId: funcId, args: Array.from(arguments) } )); return (ret != null) ? ret : fe.promise; } }; } private createMethodProxy(f: Function, config: {share?: ShareConfig, notify?: NotifyConfig}, objId: number, methodId: number): Function{ const id = `${objId}:${methodId}`; const me: MethodEntry = {original: f, config} this.shareOrNotifyMethods.set(id, me); me.promise = new Promise((resolve, reject)=>{ me.resolve = resolve; me.reject = reject; }); const self = this; return function(){ if(self.ws === null){ if(f) return f.apply(null, [...arguments, self]); } else{ let ret = null; let castType: CastType = "BROADCAST"; if(config.share?.type === "afterExec" || config.notify?.type === "afterExec"){ ret = f.apply(null, [...arguments, self]); castType = "OTHERCAST"; } self.sendMessage(newInvokeMethod( castType, { objId: objId, objRevision: self.shareObjects.get(objId)!.revision, methodId: methodId, args: Array.from(arguments), } )); return (ret != null) ? ret : me.promise; } }; } private addHostOnlyFunction<T extends Function>(f: T, _config: HostOnlyConfig): T{ const self = this; return function(){ // orderが最も小さければ実行。そうでなければ無視 let minOrder = self.selfPeer.order; for(const p of self.peers.values()){ if(minOrder > p.order) minOrder = p.order; } if(self.selfPeer.order === minOrder){ f.apply(null, [...arguments, self]); } } as any; } public saveStates(){ if(!this.ws || !this.connecting) return; for(let [objId, oe] of this.shareObjects){ if(oe.modification == 0) continue; const info = this.getStateMethods.get(objId); if(!info) continue; const curTick = performance.now(); if(info.config.maxUpdates && info.config.maxUpdates <= oe.modification || info.config.maxInterval && info.config.maxInterval <= (curTick - info.lastGet)){ this.doSendMessage(newUpdateObjectState({ objId: objId, objRevision: oe.revision, state: info.method(this) })); info.lastGet = curTick; oe.modification = 0; console.debug(`state saved: ${objId}`) } } } private applyInvocation(method: Function, args: any[]){ return method.apply(null, args); } }