99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"blegw/workers"
|
|
|
|
"github.com/currantlabs/gatt"
|
|
"github.com/currantlabs/gatt/examples/option"
|
|
)
|
|
|
|
var done = make(chan struct{})
|
|
|
|
type BleService struct {
|
|
workers []workers.Worker
|
|
}
|
|
|
|
// AddWorker - add new worker
|
|
func (s BleService) AddWorker(worker workers.Worker) {
|
|
if s.workers == nil {
|
|
s.workers = []workers.Worker{}
|
|
}
|
|
s.workers = append(s.workers, worker)
|
|
}
|
|
|
|
func (s BleService) Init() error {
|
|
d, err := gatt.NewDevice(option.DefaultClientOptions...)
|
|
if err != nil {
|
|
log.Fatalf("Failed to open device, err: %s\n", err)
|
|
return err
|
|
}
|
|
|
|
// Register handlers.
|
|
d.Handle(
|
|
gatt.PeripheralDiscovered(s.onPeriphDiscovered),
|
|
gatt.PeripheralConnected(s.onPeriphConnected),
|
|
gatt.PeripheralDisconnected(s.onPeriphDisconnected),
|
|
)
|
|
|
|
d.Init(s.onStateChanged)
|
|
<-done
|
|
fmt.Println("Done")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s BleService) onStateChanged(d gatt.Device, state gatt.State) {
|
|
fmt.Println("State:", state)
|
|
switch state {
|
|
case gatt.StatePoweredOn:
|
|
fmt.Println("scanning...")
|
|
d.Scan([]gatt.UUID{}, false)
|
|
return
|
|
default:
|
|
d.StopScanning()
|
|
}
|
|
}
|
|
|
|
func (s BleService) onPeriphDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) {
|
|
fmt.Printf("\nPeripheral ID:%s, NAME:(%s)\n", p.ID(), p.Name())
|
|
fmt.Println(" Local Name =", a.LocalName)
|
|
fmt.Println(" TX Power Level =", a.TxPowerLevel)
|
|
fmt.Println(" Manufacturer Data =", a.ManufacturerData)
|
|
fmt.Println(" Service Data =", a.ServiceData)
|
|
fmt.Println(" RSSI =", rssi)
|
|
|
|
isConnect := false
|
|
for _, worker := range s.workers {
|
|
ret := worker.OnDiscovery(p, a, rssi)
|
|
if ret && !isConnect {
|
|
isConnect = true
|
|
}
|
|
}
|
|
|
|
if isConnect {
|
|
p.Device().Connect(p)
|
|
}
|
|
}
|
|
|
|
func (s BleService) onPeriphConnected(p gatt.Peripheral, err error) {
|
|
fmt.Println("Connected")
|
|
defer p.Device().CancelConnection(p)
|
|
|
|
if err := p.SetMTU(500); err != nil {
|
|
fmt.Printf("Failed to set MTU, err: %s\n", err)
|
|
}
|
|
fmt.Printf("Set mtu done.\n")
|
|
|
|
for _, worker := range s.workers {
|
|
worker.OnConnect(p, err)
|
|
}
|
|
}
|
|
|
|
func (s BleService) onPeriphDisconnected(p gatt.Peripheral, err error) {
|
|
fmt.Println("Disconnected")
|
|
close(done)
|
|
}
|