Added mainloop

3 files changed, 30 insertions, 4 deletions

diff --git a/unicorn/FakeUnicorn.go b/unicorn/FakeUnicorn.go
index 85833c9..eaec4f2 100644
--- a/unicorn/FakeUnicorn.go
+++ b/unicorn/FakeUnicorn.go
@@ -96,3 +96,19 @@ func (f *FakeUnicorn) Show() {
 func (f *FakeUnicorn) Off() {
 	f.Close()
 }
+
+// MainLoop ...
+// Handle UI events so OS doesn't think we're frozen
+func (f *FakeUnicorn) MainLoop() {
+	running := true
+	for running {
+		for event := sdl.PollEvent(); event != nil; event = sdl.PollEvent() {
+			switch event.(type) {
+			case *sdl.QuitEvent:
+				println("Quit")
+				running = false
+				break
+			}
+		}
+	}
+}
diff --git a/unicorn/RealUnicorn.go b/unicorn/RealUnicorn.go
index 4e5c376..fb10a79 100644
--- a/unicorn/RealUnicorn.go
+++ b/unicorn/RealUnicorn.go
@@ -3,10 +3,9 @@
 package unicorn
 
 import (
-	//"golang.org/x/exp/io/spi"
-	"log"
-
 	"github.com/ecc1/spi"
+	"log"
+	"os"
 )
 
 type RealUnicorn struct {
@@ -50,7 +49,6 @@ func (u *RealUnicorn) Show() {
 		}
 	}
 	// Write to the device
-	//err := u.device.Tx(write, nil)
 	err := u.device.Transfer(write)
 	if err != nil {
 		log.Printf("Error writing to SPI device %v", err)
@@ -64,3 +62,12 @@ func (u *RealUnicorn) Off() {
 func (u *RealUnicorn) Close() error {
 	return u.device.Close()
 }
+
+// MainLoop ...
+// Do nothing until SIGTERM, then close the SPI library
+func MainLoop() {
+	c := make(chan os.Signal, 2)
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+	<- c
+	Close()
+}
diff --git a/unicorn/Unicorn.go b/unicorn/Unicorn.go
index 8a2b68d..07fa3c2 100644
--- a/unicorn/Unicorn.go
+++ b/unicorn/Unicorn.go
@@ -21,6 +21,9 @@ type Unicorn interface {
 
 	// Turns off the LEDs
 	Off()
+
+	// Unicorn needs to be in charge of the main thread
+	MainLoop()
 }
 
 // GetUnicorn ...