bitsealws

type BitSealWSConn struct {
	Conn    *websocket.Conn
	Session *rtc.Session

	// Extra 保存服务器握手响应中除 token/salt_s/ts/nonce 之外的所有字段，
	// 对应服务端 OnHandshakeResponse 注入的自定义数据。
	Extra map[string]any

	// OnMessage 若非 nil，则 Serve/ServeAsync 解包明文后调用；
	// 返回值非 nil ⇒ 自动 Encode + 发送；
	OnMessage func(sess *rtc.Session, plain []byte) ([]byte, error)
}

type Server struct {

	// OnMessage 为业务回调；若不为 nil，则在收到每条消息后调用以生成响应明文。
	// 回调返回的明文会再次加密后发回客户端；若返回 nil 则表示不需要回复。
	// 若 OnMessage 本身为 nil，则 Server 默认回显收到的明文（兼容旧逻辑）。
	OnMessage func(sess *rtc.Session, plain []byte) ([]byte, error)

	// OnSession 会在成功与客户端建立 BST2 会话后立即调用，
	// 以便业务层获取 Session（例如存入连接表、提取 peerPub 等）。
	// 如不需要可保持为 nil。
	OnSession func(sess *rtc.Session)

	// OnHandshakeResponse 允许业务层在握手阶段向返回给客户端的 JSON
	// 中添加额外的键值对。若回调返回的 map 不为 nil，则其中的所有键值对
	// 将被合并进默认的 respObj 中；若键已存在则以回调结果为准。
	// 典型用法是在创建 Server 后赋值，例如：
	//   srv.OnHandshakeResponse = func(r *http.Request, clientPub *ec.PublicKey, nonce string) map[string]any {
	//       return map[string]any{"welcome": "hello"}
	//   }
	OnHandshakeResponse func(r *http.Request, clientPub *ec.PublicKey, nonce string) map[string]any
	// contains filtered or unexported fields
}