From 63ea6a9d3a069b05abb86d20e7648c1780b0f676 Mon Sep 17 00:00:00 2001
From: Tony Garnock-Jones <tonygarnockjones@gmail.com>
Date: Wed, 20 Mar 2019 22:36:05 +0000
Subject: [PATCH] extend-ground-boot!

---
 syndicate/broker/main.rkt |  7 +------
 syndicate/ground.rkt      | 12 +++++++++++-
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/syndicate/broker/main.rkt b/syndicate/broker/main.rkt
index aea6f68..118fa91 100644
--- a/syndicate/broker/main.rkt
+++ b/syndicate/broker/main.rkt
@@ -24,9 +24,4 @@
            (server-facet/websocket id scope))))
 
 (module+ main
-  (let ((go (current-ground-dataspace)))
-    (current-ground-dataspace
-     (lambda (boot-proc)
-       (go (lambda ()
-             (boot-proc)
-             (main)))))))
+  (extend-ground-boot! main))
diff --git a/syndicate/ground.rkt b/syndicate/ground.rkt
index ea51cd5..6dde3f2 100644
--- a/syndicate/ground.rkt
+++ b/syndicate/ground.rkt
@@ -6,6 +6,7 @@
          ground-assert!
          ground-retract!
          signal-background-activity!
+         extend-ground-boot!
          run-ground)
 
 (define-logger syndicate/ground)
@@ -18,6 +19,7 @@
 (require "syntax.rkt")
 
 (define current-ground-event-async-channel (make-parameter #f))
+(define *ground-boot-extensions* '())
 
 (define (ground-enqueue! item)
   (async-channel-put (current-ground-event-async-channel) item))
@@ -34,6 +36,9 @@
 (define (signal-background-activity! delta)
   (ground-enqueue! delta))
 
+(define (extend-ground-boot! proc)
+  (set! *ground-boot-extensions* (cons proc *ground-boot-extensions*)))
+
 (define (run-ground* boot-proc)
   (define ch (make-async-channel))
   (parameterize ((current-ground-event-async-channel ch))
@@ -67,7 +72,12 @@
                                              (set! ground-event-relay-actor (current-actor))
                                              ;; v Adds a dummy endpoint to keep this actor alive
                                              (begin/dataflow (void)))))
-                  (schedule-script! (current-actor) boot-proc))))
+                  (schedule-script! (current-actor)
+                                    (lambda ()
+                                      (boot-proc)
+                                      (let ((extensions (reverse *ground-boot-extensions*)))
+                                        (set! *ground-boot-extensions* '())
+                                        (for [(p (in-list extensions))] (p))))))))
 
     (let loop ()
       (define work-remaining? (run-scripts! ds))