change add-init-hook to add-startup-hook, new add-shutdown-hook word

[factor.git] / core / init / init.factor

diff --git a/core/init/init.factor b/core/init/init.factor
index 5d8e88b85f5b2ee4a78109e618f868d8773cf913..540768ee63ec4e8e94f7e91b464bd9547a466bb1 100644 (file)
--- a/core/init/init.factor
+++ b/core/init/init.factor
@@ -4,16 +4,26 @@ USING: continuations continuations.private kernel
 kernel.private sequences assocs namespaces namespaces.private ;
 IN: init
 
-SYMBOL: init-hooks
+SYMBOL: startup-hooks
+SYMBOL: shutdown-hooks
 
-init-hooks global [ drop V{ } clone ] cache drop
+startup-hooks global [ drop V{ } clone ] cache drop
+shutdown-hooks global [ drop V{ } clone ] cache drop
 
-: do-init-hooks ( -- )
-    init-hooks get [ nip call( -- ) ] assoc-each ;
+: do-hooks ( assoc -- )
+    [ nip call( -- ) ] assoc-each ;
 
-: add-init-hook ( quot name -- )
-    dup init-hooks get at [ over call( -- ) ] unless
-    init-hooks get set-at ;
+: do-startup-hooks ( -- ) startup-hooks get do-hooks ;
+
+: do-shutdown-hooks ( -- ) shutdown-hooks get do-hooks ;
+
+: add-startup-hook ( quot name -- )
+    startup-hooks get
+    [ at [ drop ] [ call( -- ) ] if ]
+    [ set-at ] 3bi ;
+
+: add-shutdown-hook ( quot name -- )
+    shutdown-hooks get set-at ;
 
 : boot ( -- ) init-namespaces init-catchstack init-error-handler ;