mirror of
https://github.com/wailsapp/wails.git
synced 2025-05-03 06:51:26 +08:00
8a7c098041
* [316-multi-bridge-conn] feat: use callback func for bridge response * [316-multi-bridge-conn] feat: implement multiple session support * split client handling portion into 'session' * keep track of sessions by remote address (ip & port) * notify each of the sessions anytime an event comes across the bus * [316-multi-bridge-conn] chore: move bridge files to package * [316-multi-bridge-conn] chore: remove deprecated Callback function The Callback function is no longer needed for the operation of the frontend callback since the ipc.Dispatch function now requires a callback function to be provided as an argument. This function can be a private function since it is passed by reference. * [316-multi-bridge-conn] chore: make webview.Callback private * [316-multi-bridge-conn] chore: remove unused injectCSS function I believe a slightly better method of doing this might need to be devised if it is needed in the future. I presume it should collect the values into a cache and then inject it into each sesssion as it appears. * [316-multi-bridge-conn] ensure wails:ready event is emitted Event is only emitted for the first session created from the Bridge. * [316-multi-bridge-conn] emit events for session lifecycle Emit an event for each session started and ended. * [316-multi-bridge-conn] fix: session handling fixes Co-authored-by: Lea Anthony <lea.anthony@gmail.com>
91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
package renderer
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/leaanthony/mewn"
|
|
"github.com/wailsapp/wails/lib/interfaces"
|
|
"github.com/wailsapp/wails/lib/logger"
|
|
)
|
|
|
|
// TODO Move this back into bridge.go
|
|
|
|
// session represents a single websocket session
|
|
type session struct {
|
|
bindingCache []string
|
|
conn *websocket.Conn
|
|
eventManager interfaces.EventManager
|
|
log *logger.CustomLogger
|
|
ipc interfaces.IPCManager
|
|
|
|
// Mutex for writing to the socket
|
|
lock sync.Mutex
|
|
}
|
|
|
|
// Identifier returns a string identifier for the remote connection.
|
|
// Taking the form of the client's <ip address>:<port>.
|
|
func (s *session) Identifier() string {
|
|
if s.conn != nil {
|
|
return s.conn.RemoteAddr().String()
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (s *session) sendMessage(msg string) error {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
|
|
if err := s.conn.WriteMessage(websocket.TextMessage, []byte(msg)); err != nil {
|
|
s.log.Debug(err.Error())
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *session) start(firstSession bool) {
|
|
s.log.Infof("Connected to frontend.")
|
|
|
|
wailsRuntime := mewn.String("../../runtime/assets/wails.js")
|
|
s.evalJS(wailsRuntime, wailsRuntimeMessage)
|
|
|
|
// Inject bindings
|
|
for _, binding := range s.bindingCache {
|
|
s.evalJS(binding, bindingMessage)
|
|
}
|
|
s.eventManager.Emit("wails:bridge:session:started", s.Identifier())
|
|
|
|
// Emit that everything is loaded and ready
|
|
if firstSession {
|
|
s.eventManager.Emit("wails:ready")
|
|
}
|
|
|
|
for {
|
|
messageType, buffer, err := s.conn.ReadMessage()
|
|
if messageType == -1 {
|
|
return
|
|
}
|
|
if err != nil {
|
|
s.log.Errorf("Error reading message: %v", err)
|
|
continue
|
|
}
|
|
|
|
s.log.Debugf("Got message: %#v\n", string(buffer))
|
|
|
|
s.ipc.Dispatch(string(buffer), s.Callback)
|
|
}
|
|
}
|
|
|
|
// Callback sends a callback to the frontend
|
|
func (s *session) Callback(data string) error {
|
|
return s.evalJS(data, callbackMessage)
|
|
}
|
|
|
|
func (s *session) evalJS(js string, mtype messageType) error {
|
|
|
|
// Prepend message type to message
|
|
return s.sendMessage(mtype.toString() + js)
|
|
|
|
}
|