Feature/callback messages (#220)

Author: pchampio

embedder/embedder.go

@@ -3,13 +3,18 @@ package embedder
 // #include "embedder.h"
 // FlutterEngineResult runFlutter(void *user_data, FlutterEngine *engine, FlutterProjectArgs * Args,
 //						 const char *const * vmArgs, int nVmAgrs);
+// FlutterEngineResult
+// createMessageResponseHandle(FlutterEngine engine, void *user_data,
+//                             FlutterPlatformMessageResponseHandle **reply);
 // char** makeCharArray(int size);
 // void setArrayString(char **a, char *s, int n);
 // const int32_t kFlutterSemanticsNodeIdBatchEnd = -1;
 // const int32_t kFlutterSemanticsCustomActionIdBatchEnd = -1;
 import "C"
 import (
+	"errors"
 	"fmt"
+	"runtime"
 	"runtime/debug"
 	"sync"
 	"unsafe"
@@ -257,7 +262,6 @@ type PlatformMessage struct {
 	Message []byte
 
 	// ResponseHandle is only set when receiving a platform message.
-	// https://github.com/flutter/flutter/issues/18852
 	ResponseHandle PlatformMessageResponseHandle
 }
 
@@ -357,8 +361,40 @@ func (flu *FlutterEngine) MarkExternalTextureFrameAvailable(textureID int64) Res
 	return (Result)(res)
 }
 
-// FlutterEngineGetCurrentTime gets the current time in nanoseconds from the
-// clock used by the flutter engine.
+// DataCallback is a function called when a PlatformMessage response send back
+// to the embedder.
+type DataCallback func(binaryReply []byte)
+
+// CreatePlatformMessageResponseHandle creates a platform message response
+// handle that allows the embedder to set a native callback for a response to a
+// message.
+func (flu *FlutterEngine) CreatePlatformMessageResponseHandle(callback DataCallback) (PlatformMessageResponseHandle, error) {
+	var responseHandle *C.FlutterPlatformMessageResponseHandle
+
+	callbackPointer := uintptr(unsafe.Pointer(&callback))
+	defer func() {
+		runtime.KeepAlive(callbackPointer)
+	}()
+
+	res := C.createMessageResponseHandle(flu.Engine, unsafe.Pointer(&callbackPointer), &responseHandle)
+	if (Result)(res) != ResultSuccess {
+		return 0, errors.New("failed to create a response handle")
+	}
+	return PlatformMessageResponseHandle(unsafe.Pointer(responseHandle)), nil
+}
+
+// ReleasePlatformMessageResponseHandle collects a platform message response
+// handle.
+func (flu *FlutterEngine) ReleasePlatformMessageResponseHandle(responseHandle PlatformMessageResponseHandle) {
+	cResponseHandle := (*C.FlutterPlatformMessageResponseHandle)(unsafe.Pointer(responseHandle))
+	res := C.FlutterPlatformMessageReleaseResponseHandle(flu.Engine, cResponseHandle)
+	if (Result)(res) != ResultSuccess {
+		fmt.Printf("go-flutter: failed to collect platform response message handle")
+	}
+}
+
+// FlutterEngineGetCurrentTime gets the current time in nanoseconds from the clock used by the flutter
+// engine.
 func FlutterEngineGetCurrentTime() uint64 {
 	return uint64(C.FlutterEngineGetCurrentTime())
 }

embedder/embedder_helper.c

@@ -21,6 +21,9 @@ bool proxy_runs_task_on_current_thread_callback(void *user_data);
 void proxy_post_task_callback(FlutterTask task, uint64_t target_time_nanos,
                               void *user_data);
 
+void proxy_desktop_binary_reply(const uint8_t *data, size_t data_size,
+                                void *user_data);
+
 // C helper
 FlutterEngineResult runFlutter(void *user_data, FlutterEngine *engine,
                                FlutterProjectArgs *Args,
@@ -62,3 +65,11 @@ FlutterEngineResult runFlutter(void *user_data, FlutterEngine *engine,
 char **makeCharArray(int size) { return calloc(sizeof(char *), size); }
 
 void setArrayString(char **a, char *s, int n) { a[n] = s; }
+
+FlutterEngineResult
+createMessageResponseHandle(FlutterEngine engine, void *user_data,
+                            FlutterPlatformMessageResponseHandle **reply) {
+
+  return FlutterPlatformMessageCreateResponseHandle(
+      engine, proxy_desktop_binary_reply, user_data, reply);
+}

embedder/embedder_proxy.go

@@ -91,3 +91,10 @@ func proxy_post_task_callback(task C.FlutterTask, targetTimeNanos C.uint64_t, us
 	flutterEngine := (*FlutterEngine)(unsafe.Pointer(flutterEnginePointer))
 	flutterEngine.TaskRunnerPostTask(task, uint64(targetTimeNanos))
 }
+
+//export proxy_desktop_binary_reply
+func proxy_desktop_binary_reply(data *C.uint8_t, dataSize C.size_t, userData unsafe.Pointer) {
+	callbackPointer := *(*uintptr)(userData)
+	handler := *(*DataCallback)(unsafe.Pointer(callbackPointer))
+	handler(C.GoBytes(unsafe.Pointer(data), C.int(dataSize)))
+}

key-events.go

@@ -81,7 +81,7 @@ func (p *keyeventPlugin) sendKeyEvent(window *glfw.Window, key glfw.Key, scancod
 		ScanCode:  scancode,
 		Modifiers: int(mods),
 	}
-	_, err := p.keyEventChannel.Send(event)
+	err := p.keyEventChannel.Send(event)
 	if err != nil {
 		fmt.Printf("go-flutter: Failed to send raw_keyboard event %v: %v\n", event, err)
 	}

lifecycle.go

@@ -33,7 +33,7 @@ func (p *lifecyclePlugin) glfwIconifyCallback(w *glfw.Window, iconified bool) {
 	case false:
 		state = "AppLifecycleState.resumed"
 	}
-	_, err := p.channel.Send(state)
+	err := p.channel.Send(state)
 	if err != nil {
 		fmt.Printf("go-flutter: Failed to send lifecycle event %s: %v\n", state, err)
 	}

messenger.go

@@ -1,13 +1,13 @@
 package flutter
 
 import (
-	"errors"
 	"fmt"
 	"sync"
 
 	"github.com/go-flutter-desktop/go-flutter/embedder"
 	"github.com/go-flutter-desktop/go-flutter/internal/tasker"
 	"github.com/go-flutter-desktop/go-flutter/plugin"
+	"github.com/pkg/errors"
 )
 
 type messenger struct {
@@ -30,13 +30,25 @@ func newMessenger(engine *embedder.FlutterEngine) *messenger {
 	}
 }
 
-// Send pushes a binary message on a channel to the Flutter side. Replies are
-// not supported yet (https://github.com/flutter/flutter/issues/18852). This
-// means that currently, binaryReply will be nil on success.
+// Send pushes a binary message on a channel to the Flutter side and wait for a
+// reply.
+// NOTE: If no value are returned by the flutter handler, the function will
+// wait forever. In case you don't want to wait for reply, use SendNoReply.
 func (m *messenger) Send(channel string, binaryMessage []byte) (binaryReply []byte, err error) {
+	reply := make(chan []byte)
+	defer close(reply)
+	responseHandle, err := m.engine.CreatePlatformMessageResponseHandle(func(binaryMessage []byte) {
+		reply <- binaryMessage
+	})
+	if err != nil {
+		return nil, err
+	}
+	defer m.engine.ReleasePlatformMessageResponseHandle(responseHandle)
+
 	msg := &embedder.PlatformMessage{
-		Channel: channel,
-		Message: binaryMessage,
+		Channel:        channel,
+		Message:        binaryMessage,
+		ResponseHandle: responseHandle,
 	}
 	res := m.engine.SendPlatformMessage(msg)
 	if err != nil {
@@ -48,9 +60,28 @@ func (m *messenger) Send(channel string, binaryMessage []byte) (binaryReply []by
 		return nil, errors.New("failed to send message")
 	}
 
-	// NOTE: Response from engine is not yet supported by embedder.
-	// https://github.com/flutter/flutter/issues/18852
-	return nil, nil
+	// wait for a reply and return
+	return <-reply, nil
+}
+
+// SendNoReply pushes a binary message on a channel to the Flutter side without
+// expecting replies.
+func (m *messenger) SendNoReply(channel string, binaryMessage []byte) (err error) {
+	msg := &embedder.PlatformMessage{
+		Channel: channel,
+		Message: binaryMessage,
+	}
+	res := m.engine.SendPlatformMessage(msg)
+	if err != nil {
+		if ferr, ok := err.(*plugin.FlutterError); ok {
+			return ferr
+		}
+	}
+	if res != embedder.ResultSuccess {
+		return errors.New("failed to send message")
+	}
+
+	return nil
 }
 
 // SetChannelHandler satisfies plugin.BinaryMessenger

plugin/basic-message-channel.go

@@ -53,11 +53,28 @@ func NewBasicMessageChannel(messenger BinaryMessenger, channelName string, codec
 	return b
 }
 
-// Send encodes and sends the specified message to the Flutter application and
-// returns the reply, or an error. Results from the Flutter side are not yet
-// implemented in the embedder. Until then, InvokeMethod will always return nil
-// as reult. https://github.com/flutter/flutter/issues/18852
-func (b *BasicMessageChannel) Send(message interface{}) (reply interface{}, err error) {
+// Send encodes and sends the specified message to the Flutter application
+// without waiting for a reply.
+func (b *BasicMessageChannel) Send(message interface{}) error {
+	encodedMessage, err := b.codec.EncodeMessage(message)
+	if err != nil {
+		return errors.Wrap(err, "failed to encode outgoing message")
+	}
+	err = b.messenger.SendNoReply(b.channelName, encodedMessage)
+	if err != nil {
+		return errors.Wrap(err, "failed to send outgoing message")
+	}
+	return nil
+}
+
+// SendWithReply encodes and sends the specified message to the Flutter
+// application and returns the reply, or an error.
+//
+// NOTE: If no value are returned by the handler setted in the
+// setMessageHandler flutter method, the function will wait forever. In case
+// you don't want to wait for reply, use Send or launch the
+// function in a goroutine.
+func (b *BasicMessageChannel) SendWithReply(message interface{}) (reply interface{}, err error) {
 	encodedMessage, err := b.codec.EncodeMessage(message)
 	if err != nil {
 		return nil, errors.Wrap(err, "failed to encode outgoing message")

plugin/basic-message-channel_test.go

@@ -31,7 +31,7 @@ func TestBasicMethodChannelStringCodecSend(t *testing.T) {
 		return nil
 	})
 	channel := NewBasicMessageChannel(messenger, "ch", codec)
-	reply, err := channel.Send("hello")
+	reply, err := channel.SendWithReply("hello")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -100,7 +100,7 @@ func TestBasicMethodChannelBinaryCodecSend(t *testing.T) {
 		return nil
 	})
 	channel := NewBasicMessageChannel(messenger, "ch", codec)
-	reply, err := channel.Send([]byte{0x01})
+	reply, err := channel.SendWithReply([]byte{0x01})
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -160,7 +160,7 @@ func TestBasicMethodChannelNilMockHandler(t *testing.T) {
 	messenger := NewTestingBinaryMessenger()
 	messenger.MockSetChannelHandler("ch", nil)
 	channel := NewBasicMessageChannel(messenger, "ch", codec)
-	reply, err := channel.Send("hello")
+	reply, err := channel.SendWithReply("hello")
 	Nil(t, reply)
 	NotNil(t, err)
 	Equal(t, "failed to send outgoing message: no handler set", err.Error())
@@ -170,7 +170,7 @@ func TestBasicMethodChannelEncodeFail(t *testing.T) {
 	codec := StringCodec{}
 	messenger := NewTestingBinaryMessenger()
 	channel := NewBasicMessageChannel(messenger, "ch", codec)
-	reply, err := channel.Send(int(42)) // invalid value
+	reply, err := channel.SendWithReply(int(42)) // invalid value
 	Nil(t, reply)
 	NotNil(t, err)
 	Equal(t, "failed to encode outgoing message: invalid type provided to message codec: expected message to be of type string", err.Error())

plugin/binary-messenger.go

@@ -5,6 +5,10 @@ type BinaryMessenger interface {
 	// Send sends a binary message to the Flutter application.
 	Send(channel string, binaryMessage []byte) (binaryReply []byte, err error)
 
+	// SendNoReply sends a binary message to the Flutter application without
+	// expecting replies.
+	SendNoReply(channel string, binaryMessage []byte) (err error)
+
 	// SetChannelHandler registers a handler to be invoked when the Flutter
 	// application sends a message to its host platform on given channel.
 	//

plugin/event-sink.go

@@ -36,7 +36,10 @@ func (es *EventSink) Success(event interface{}) {
 	if err != nil {
 		fmt.Printf("go-flutter: failed to encode success envelope for event channel '%s', error: %v\n", es.eventChannel.channelName, err)
 	}
-	es.eventChannel.messenger.Send(es.eventChannel.channelName, binaryMsg)
+	err = es.eventChannel.messenger.SendNoReply(es.eventChannel.channelName, binaryMsg)
+	if err != nil {
+		fmt.Printf("go-flutter: failed to send Success message on event channel '%s', error: %v\n", es.eventChannel.channelName, err)
+	}
 }
 
 // Error consumes an error event.
@@ -51,7 +54,10 @@ func (es *EventSink) Error(errorCode string, errorMessage string, errorDetails i
 	if err != nil {
 		fmt.Printf("go-flutter: failed to encode success envelope for event channel '%s', error: %v\n", es.eventChannel.channelName, err)
 	}
-	es.eventChannel.messenger.Send(es.eventChannel.channelName, binaryMsg)
+	err = es.eventChannel.messenger.SendNoReply(es.eventChannel.channelName, binaryMsg)
+	if err != nil {
+		fmt.Printf("go-flutter: failed to send Error message on event channel '%s', error: %v\n", es.eventChannel.channelName, err)
+	}
 }
 
 // EndOfStream consumes end of stream.
@@ -63,5 +69,8 @@ func (es *EventSink) EndOfStream() {
 	}
 	es.hasEnded = true
 
-	es.eventChannel.messenger.Send(es.eventChannel.channelName, nil)
+	err := es.eventChannel.messenger.SendNoReply(es.eventChannel.channelName, nil)
+	if err != nil {
+		fmt.Printf("go-flutter: failed to send EndOfStream message on event channel '%s', error: %v\n", es.eventChannel.channelName, err)
+	}
 }