基本逻辑完成

internal/middleware/bgm.go

@@ -0,0 +1,65 @@
+package middleware
+
+import (
+	"bytes"
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"game-driver/pkg/audio"
+	"io"
+	"log"
+	"net/http"
+	"net/url"
+	"os"
+	"strings"
+)
+
+func open(u string) io.ReadCloser {
+	p, _ := strings.CutPrefix(u, "file://")
+	f, e := os.Open(p)
+	if e != nil {
+		log.Printf("BGM 文件 [%v] 打开错误: %v\n", u, e)
+		return nil
+	}
+	return f
+}
+
+func get(u string) io.ReadCloser {
+	resp, e := http.Get(u)
+	if e != nil {
+		log.Printf("BGM 文件 [%v] 下载失败: %v\n", u, e)
+		return nil
+	}
+	return resp.Body
+}
+
+func toBuffer(b io.ReadCloser) io.ReadCloser {
+	data := &bytes.Buffer{}
+	_, _ = data.ReadFrom(b)
+	defer b.Close()
+	return io.NopCloser(data)
+}
+
+func PlayBgm() leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
+		u, err := url.Parse(pm.BGM)
+		if err != nil {
+			log.Println("BGM URL 解析错误: ", err)
+		} else {
+			var bgm io.ReadCloser
+			if u.Scheme == "file" {
+				bgm = open(u.String())
+			} else if u.Scheme == "http" || u.Scheme == "https" {
+				bgm = get(u.String())
+			} else {
+				log.Printf("不支持的BGM文件协议: %v\n", u.String())
+			}
+			// 获取到了资源进行播放
+			if bgm != nil {
+				go audio.PlayMP3(c, toBuffer(bgm))
+			}
+		}
+
+		c.Next()
+	}
+}

internal/middleware/device.go

@@ -0,0 +1,16 @@
+package middleware
+
+import (
+	"game-driver/internal/common"
+	"game-driver/leaf"
+)
+
+// DeviceLock 设备锁中间件
+func DeviceLock(d *common.Device) leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		d.Lock()
+		defer d.Unlock()
+
+		c.Next()
+	}
+}

internal/middleware/json.go

@@ -0,0 +1,25 @@
+package middleware
+
+import (
+	"encoding/json"
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"log"
+)
+
+type JSONKey string
+
+const PayloadJSONKey JSONKey = "payload_json"
+
+// PayloadJSON 解析报文
+func PayloadJSON() leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := &schema.PlayModal{}
+		err := json.Unmarshal(c.Payload, pm)
+		if err != nil {
+			log.Panicf("报文解析错误: %v\n", err)
+		}
+		leaf.WithValue[*schema.PlayModal](c, PayloadJSONKey, pm)
+		c.Next()
+	}
+}

internal/middleware/relay.go

@@ -0,0 +1,19 @@
+package middleware
+
+import (
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"game-driver/pkg/relay"
+)
+
+// RelayMaster 继电器中间件
+func RelayMaster(r *relay.Device) leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
+		if r != nil && pm.Power {
+			r.On(1)
+			defer r.Off(1)
+		}
+		c.Next()
+	}
+}

internal/middleware/sound_start.go

@@ -0,0 +1,26 @@
+package middleware
+
+import (
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"game-driver/pkg/tts"
+)
+
+// SoundStart 开始词播报
+func SoundStart(t *tts.AliTTS) leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
+		t.Sound(pm.TTS.Start)
+
+		defer func() {
+			switch leaf.Value[leaf.EndType](c, leaf.EndKey) {
+			case leaf.EndTimer:
+				t.Sound(pm.TTS.End)
+			case leaf.EndStop:
+				t.Sound(pm.TTS.Stop)
+			}
+		}()
+
+		c.Next()
+	}
+}

internal/middleware/stop.go

@@ -0,0 +1,32 @@
+package middleware
+
+import (
+	"game-driver/internal/common"
+	"game-driver/leaf"
+)
+
+// EmergencyStop 紧急停止中间件
+func EmergencyStop() leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		cancel := leaf.WithCancel(c)
+		defer common.GlobalStopper.Reset()
+
+		// 结束信号通道
+		a := make(chan struct{})
+		// 发送结束信号
+		defer close(a)
+
+		go func() {
+			select {
+			case <-a:
+			case <-common.GlobalStopper.Done():
+				{
+					cancel()
+					leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndStop)
+				}
+			}
+		}()
+
+		c.Next()
+	}
+}

internal/middleware/timer.go

@@ -0,0 +1,80 @@
+package middleware
+
+import (
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"game-driver/pkg/tts"
+	"sync"
+	"time"
+)
+
+// TimerAction 定时器中间件，用于定时触发屏幕打印和语音播报。 t 是语音播报实例
+func TimerAction(t *tts.AliTTS, maxTimeout int) leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
+
+		// 构建打印和语音播报的时间映射
+		printMap := make(map[int]schema.PrintModal, len(pm.Print))
+		for _, p := range pm.Print {
+			printMap[p.Time] = p
+		}
+		ttsMap := make(map[int]schema.TTSTimer, len(pm.TTS.Timer))
+		for _, t := range pm.TTS.Timer {
+			ttsMap[t.Time] = t
+		}
+
+		// 定时器
+		var timer *time.Timer
+		if pm.Timeout != 0 {
+			timer = time.NewTimer(time.Second * time.Duration(pm.Timeout))
+		} else {
+			timer = time.NewTimer(time.Second * time.Duration(maxTimeout))
+		}
+		defer timer.Stop()
+
+		// 等待组
+		var wait sync.WaitGroup
+		defer wait.Wait()
+
+		// 结束信号通道
+		a := make(chan struct{})
+		// 发送结束信号
+		defer close(a)
+
+		cancel := leaf.WithCancel(c)
+		go func() {
+			start := time.Now()
+			// 等待结束
+			wait.Add(1)
+			defer wait.Done()
+			// 定时器
+			ticker := time.NewTicker(time.Second)
+			defer ticker.Stop()
+			// 结束标志
+			for over := false; !over; {
+				select {
+				case <-a:
+					over = true
+				case <-timer.C: // 定时器结束
+					{
+						cancel()
+						over = true
+						leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndTimer)
+					}
+				case m := <-ticker.C:
+					{
+						s := int(m.Sub(start).Seconds())
+						if _, ok := printMap[s]; ok {
+							//TODO: 屏幕打印
+						}
+						if to, ok := ttsMap[s]; ok {
+							t.Sound(to.Value)
+						}
+					}
+				}
+			}
+		}()
+
+		c.Next()
+	}
+}