122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"time"
|
|
|
|
"ble_simulator/internal/ble"
|
|
"ble_simulator/internal/device"
|
|
"ble_simulator/internal/tui"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"tinygo.org/x/bluetooth"
|
|
)
|
|
|
|
func main() {
|
|
// Create shared state and log buffer
|
|
state := device.NewDeviceState()
|
|
logBuffer := tui.NewLogBuffer(500)
|
|
|
|
logBuffer.Info("BLE Simulator v1.0")
|
|
|
|
adapter := bluetooth.DefaultAdapter
|
|
if err := adapter.Enable(); err != nil {
|
|
log.Fatalf("Failed to enable BLE: %v", err)
|
|
}
|
|
|
|
// Channel for alarm notifications from TUI
|
|
notifyCh := make(chan string, 10)
|
|
|
|
// Connection handler
|
|
adapter.SetConnectHandler(func(dev bluetooth.Device, connected bool) {
|
|
if connected {
|
|
state.IncrConnectedClients()
|
|
logBuffer.Conn("Client Connected: %s", dev.Address.String())
|
|
} else {
|
|
state.DecrConnectedClients()
|
|
logBuffer.Conn("Client Disconnected")
|
|
}
|
|
})
|
|
|
|
// Setup BLE service with logger
|
|
notifyChar, err := ble.SetupService(adapter, state, logBuffer)
|
|
if err != nil {
|
|
log.Fatalf("Failed to add service: %v", err)
|
|
}
|
|
|
|
// Configure advertising
|
|
adv := adapter.DefaultAdvertisement()
|
|
if err := adv.Configure(bluetooth.AdvertisementOptions{
|
|
LocalName: "GO_SIMULATOR",
|
|
ServiceUUIDs: []bluetooth.UUID{ble.ServiceUUID},
|
|
}); err != nil {
|
|
log.Fatalf("Failed to configure advertising: %v", err)
|
|
}
|
|
|
|
if err := adv.Start(); err != nil {
|
|
log.Fatalf("Failed to start advertising: %v", err)
|
|
}
|
|
|
|
logBuffer.Info("Advertising as GO_SIMULATOR...")
|
|
logBuffer.Info("Using Nordic UART Service (NUS) UUIDs")
|
|
logBuffer.Info("Service UUID: 6e400001-b5a3-f393-e0a9-e50e24dcca9e")
|
|
logBuffer.Info("Command Char: 6e400002-b5a3-f393-e0a9-e50e24dcca9e (Write/RX)")
|
|
logBuffer.Info("Notify Char: 6e400003-b5a3-f393-e0a9-e50e24dcca9e (Notify/TX)")
|
|
logBuffer.Info("Waiting for connections...")
|
|
|
|
// Start notification loop (500ms heartbeat with sensor data)
|
|
go notificationLoop(notifyChar, state, logBuffer, notifyCh)
|
|
|
|
// Create and run TUI
|
|
model := tui.NewModel(state, logBuffer, notifyCh)
|
|
p := tea.NewProgram(model, tea.WithAltScreen())
|
|
|
|
if _, err := p.Run(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error running TUI: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func notificationLoop(char *bluetooth.Characteristic, state *device.DeviceState, logger *tui.LogBuffer, notifyCh chan string) {
|
|
ticker := time.NewTicker(500 * time.Millisecond)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
// Check for alarm state and send alarm message if active
|
|
if alarmActive, alarmType := state.GetAlarmState(); alarmActive {
|
|
msg := fmt.Sprintf("ALARM: %s", alarmType)
|
|
_, err := char.Write([]byte(msg + "\n"))
|
|
if err == nil {
|
|
logger.TX("Sending: %s", msg)
|
|
}
|
|
}
|
|
|
|
// Send regular sensor data
|
|
value := state.GetSensorValue()
|
|
jitter := state.GetJitterRange()
|
|
jitterVal := 0
|
|
if jitter > 0 {
|
|
jitterVal = rand.Intn(jitter*2+1) - jitter
|
|
}
|
|
msg := fmt.Sprintf("SENSOR:%d", value+jitterVal)
|
|
|
|
_, err := char.Write([]byte(msg + "\n"))
|
|
if err != nil {
|
|
// Silently ignore write errors (no subscribers)
|
|
continue
|
|
}
|
|
logger.TX("Sending: %s", msg)
|
|
|
|
case alarmMsg := <-notifyCh:
|
|
// Handle alarm notifications from TUI
|
|
_, err := char.Write([]byte(alarmMsg + "\n"))
|
|
if err == nil {
|
|
logger.TX("Sending: %s", alarmMsg)
|
|
}
|
|
}
|
|
}
|
|
}
|