From 4da5cb38d396d76d8072815d150725f7c9a85078 Mon Sep 17 00:00:00 2001
From: Björn Linse <bjorn.linse@gmail.com>
Date: Thu, 20 Sep 2018 19:19:38 +0200
Subject: startup: always wait for UI with --embed, unless --headless also is
 supplied

---
 src/nvim/api/ui.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/nvim/api/ui.c')

diff --git a/src/nvim/api/ui.c b/src/nvim/api/ui.c
index 509032892b..4971753854 100644
--- a/src/nvim/api/ui.c
+++ b/src/nvim/api/ui.c
@@ -58,6 +58,21 @@ void remote_ui_disconnect(uint64_t channel_id)
   xfree(ui);
 }
 
+/// Wait until ui has connected on stdio channel.
+void remote_ui_wait_for_attach(void)
+  FUNC_API_NOEXPORT
+{
+  Channel *channel = find_channel(CHAN_STDIO);
+  if (!channel) {
+    // this function should only be called in --embed mode, stdio channel
+    // can be assumed.
+    abort();
+  }
+
+  LOOP_PROCESS_EVENTS_UNTIL(&main_loop, channel->events, -1,
+                            pmap_has(uint64_t)(connected_uis, CHAN_STDIO));
+}
+
 void nvim_ui_attach(uint64_t channel_id, Integer width, Integer height,
                     Dictionary options, Error *err)
   FUNC_API_SINCE(1) FUNC_API_REMOTE_ONLY
-- 
cgit