Fixed import

Author: The-Jonsey

handlers/gif.go

@@ -1,9 +1,10 @@
 package handlers
 
 import (
+	"github.com/nfnt/resize"
 	"github.com/unix-streamdeck/api"
 	"github.com/unix-streamdeck/driver"
-	"github.com/nfnt/resize"
+	"image"
 	"image/gif"
 	"log"
 	"os"
@@ -18,24 +19,32 @@ func (s *GifIconHandler) Icon(page int, index int, key *api.Key, dev streamdeck.
 		return
 	}
 	gifs, err := gif.DecodeAll(f)
+	if err != nil {
+		log.Println(err)
+		return
+	}
 	timeDelay := gifs.Delay[0]
-	gifIndex := 0
-	go loop(gifs, gifIndex, timeDelay, page, index, dev, key, s)
+	frames := make([]image.Image, len(gifs.Image))
+	for i, frame := range gifs.Image {
+		frames[i] = resize.Resize(dev.Pixels, dev.Pixels, frame, resize.Lanczos3)
+	}
+	go loop(frames, timeDelay, page, index, dev, key, s)
 }
 
 func (s *GifIconHandler) Stop() {
 	s.Running = false
 }
 
-func loop(gifs *gif.GIF, gifIndex int, timeDelay int, page int, index int, dev streamdeck.Device, key *api.Key, s *GifIconHandler) {
+func loop(frames []image.Image, timeDelay int, page int, index int, dev streamdeck.Device, key *api.Key, s *GifIconHandler) {
+	gifIndex := 0
 	for s.Running {
-		img := resize.Resize(dev.Pixels, dev.Pixels, gifs.Image[gifIndex], resize.Lanczos3)
+		img := frames[gifIndex]
 		s.OnSetImage(img, index, page, dev)
 		key.Buff = img
 		gifIndex++
-		if gifIndex >= len(gifs.Image) {
+		if gifIndex >= len(frames) {
 			gifIndex = 0
 		}
 		time.Sleep(time.Duration(timeDelay * 10000000))
 	}
-}
+}

interface.go

@@ -1,12 +1,14 @@
 package main
 
 import (
+	"context"
 	"github.com/fogleman/gg"
 	"github.com/nfnt/resize"
 	"github.com/unix-streamdeck/api"
 	"github.com/unix-streamdeck/driver"
 	"github.com/unix-streamdeck/streamdeckd/handlers"
 	"golang.org/x/image/font/inconsolata"
+	"golang.org/x/sync/semaphore"
 	"image"
 	"image/color"
 	"image/draw"
@@ -16,6 +18,7 @@ import (
 )
 
 var p int
+var sem = semaphore.NewWeighted(int64(1))
 
 func LoadImage(path string) (image.Image, error) {
 	f, err := os.Open(path)
@@ -37,6 +40,13 @@ func ResizeImage(img image.Image) image.Image {
 }
 
 func SetImage(img image.Image, i int, page int, _ streamdeck.Device) {
+	ctx := context.Background()
+	err := sem.Acquire(ctx, 1)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer sem.Release(1)
 	if p == page && isOpen {
 		err := dev.SetImage(uint8(i), img)
 		if err != nil {
@@ -49,7 +59,7 @@ func SetImage(img image.Image, i int, page int, _ streamdeck.Device) {
 	}
 }
 
-func SetKey(currentKey *api.Key, i int) {
+func SetKeyImage(currentKey *api.Key, i int) {
 	if currentKey.Buff == nil {
 		if currentKey.Icon == "" {
 			img := image.NewRGBA(image.Rect(0, 0, int(dev.Pixels), int(dev.Pixels)))
@@ -82,30 +92,34 @@ func SetPage(config *api.Config, page int) {
 	currentPage := config.Pages[page]
 	for i := 0; i < len(currentPage); i++ {
 		currentKey := &currentPage[i]
-		if currentKey.Buff == nil {
-			if currentKey.IconHandler == "" {
-				SetKey(currentKey, i)
+		go SetKey(currentKey, i, page, dev)
+	}
+	EmitPage(p)
+}
 
-			} else if currentKey.IconHandlerStruct == nil {
-				var handler api.IconHandler
-				if currentKey.IconHandler == "Gif" {
-					handler = &handlers.GifIconHandler{Running:true, OnSetImage: SetImage}
-				} else if currentKey.IconHandler == "Counter" {
-					handler = &handlers.CounterIconHandler{Count:0, Running: true, OnSetImage: SetImage}
-				} else if currentKey.IconHandler == "Time" {
-					handler = &handlers.TimeIconHandler{Running:true, OnSetImage: SetImage}
-				}
-				if handler == nil {
-					continue
-				}
-				handler.Icon(page, i, currentKey, dev)
-				currentKey.IconHandlerStruct = handler
+func SetKey(currentKey *api.Key, i int, page int, dev streamdeck.Device) {
+	if currentKey.Buff == nil {
+		if currentKey.IconHandler == "" {
+			SetKeyImage(currentKey, i)
+
+		} else if currentKey.IconHandlerStruct == nil {
+			var handler api.IconHandler
+			if currentKey.IconHandler == "Gif" {
+				handler = &handlers.GifIconHandler{Running:true, OnSetImage: SetImage}
+			} else if currentKey.IconHandler == "Counter" {
+				handler = &handlers.CounterIconHandler{Count:0, Running: true, OnSetImage: SetImage}
+			} else if currentKey.IconHandler == "Time" {
+				handler = &handlers.TimeIconHandler{Running:true, OnSetImage: SetImage}
 			}
-		} else {
-			SetImage(currentKey.Buff, i, p, dev)
+			if handler == nil {
+				return
+			}
+			handler.Icon(page, i, currentKey, dev)
+			currentKey.IconHandlerStruct = handler
 		}
+	} else {
+		SetImage(currentKey.Buff, i, p, dev)
 	}
-	EmitPage(p)
 }
 
 func HandleInput(key *api.Key, page int, index int) {
@@ -120,7 +134,10 @@ func HandleInput(key *api.Key, page int, index int) {
 		SetPage(config, page)
 	}
 	if key.Brightness != 0 {
-		_ = dev.SetBrightness(uint8(key.Brightness))
+		err := dev.SetBrightness(uint8(key.Brightness))
+		if err != nil {
+			log.Println(err)
+		}
 	}
 	if key.Url != "" {
 		runCommand("xdg-open " + key.Url)