1 files changed, 14 insertions, 15 deletions
diff --git a/uitask_darwin.go b/uitask_darwin.go
index 754418c..695b5c5 100644
--- a/uitask_darwin.go
+++ b/uitask_darwin.go
@@ -16,32 +16,31 @@ import "C"
 
 var uitask chan func()
 
-func ui(main func()) error {
-	runtime.LockOSThread()
-
-	uitask = make(chan func())
-
+func uiinit() error {
 	err := initCocoa()
 	if err != nil {
 		return err
 	}
 
-	// Cocoa must run on the first thread created by the program, so we run our dispatcher on another thread instead
-	go func() {
-		for f := range uitask {
-			C.douitask(appDelegate, unsafe.Pointer(&f))
-		}
-	}()
+	// do this at the end in case something goes wrong
+	uitask = make(chan func())
+	return nil
+}
 
+func ui() {
+	// Cocoa must run on the first thread created by the program, so we run our dispatcher on another thread instead
 	go func() {
-		main()
-		uitask <- func() {
-			C.breakMainLoop()
+		for {
+			select {
+			case f := range uitask:
+				C.douitask(appDelegate, unsafe.Pointer(&f))
+			case <-Stop:
+				C.breakMainLoop()
+			}
 		}
 	}()
 
 	C.cocoaMainLoop()
-	return nil
 }
 
 func initCocoa() (err error) {