ble/ble_darwin.go

package ble

import "C"

import (
	"encoding/binary"
	"fmt"
	"runtime"
	"sync"
	"time"
	"unsafe"

	"git.wow.st/gmp/ble/gatt"
	"git.wow.st/gmp/ble/ns"
)

type BLE struct {
	state       ns.CBManagerState
	events      chan interface{}
	peripherals Peripherals
	cd          *ns.CBDelegate
	cm          *ns.CBCentralManager

	ready, wantScan bool
	sync.Mutex

	connections Connections
}

var cdLookup map[unsafe.Pointer]*BLE
var pdLookup map[unsafe.Pointer]*BLE
var pcache map[unsafe.Pointer]*Peripheral

func peripheralLookup(p Peripheral) *BLE {
	return pdLookup[p.p.Ptr()]
}

type State string

type Peripheral struct {
	Name       string
	RSSI       int
	Identifier string

	p *ns.CBPeripheral
}

type Service *ns.CBService
type Characteristic *ns.CBCharacteristic

//Init needs to be called before the BLE library can be used. No setup needed
//on Darwin.
func Init() {

}

func (x *Peripheral) Retain() {
//NOTE: ns.ObjectAtIndex() calls SetFinalizer for us, which will be a problem
//later when we call GC(), so we always first clear the finalizer before
//setting a new one.
	runtime.SetFinalizer(x.p, nil)
	x.p.Retain()
	x.p.GC()
}

func peripheralName(p *ns.CBPeripheral) string {
	var ret string
	nsname := p.Name()
	if nsname.Ptr() != nil {
		ret = nsname.String()
	}
	return ret
}

func newPeripheral(x *ns.CBPeripheral) Peripheral {
	if pcache == nil {
		pcache = make(map[unsafe.Pointer]*Peripheral)
	} else {
		if ret,ok := pcache[x.Ptr()]; ok {
			return *ret
		}
	}
	ret := Peripheral{
		Name:       peripheralName(x),
		Identifier: x.Identifier().UUIDString().String(),
		p:          x,
	}
	pcache[x.Ptr()] = &ret
	return ret
}

func (b *BLE) stringState() string {
	x := b.state
	switch x {
	case (ns.CBManagerState)(ns.CBManagerStateResetting):
		return "resetting"
	case (ns.CBManagerState)(ns.CBManagerStateUnsupported):
		return "unsupported"
	case (ns.CBManagerState)(ns.CBManagerStateUnauthorized):
		return "unauthorized"
	case (ns.CBManagerState)(ns.CBManagerStatePoweredOff):
		return "powered off"
	case (ns.CBManagerState)(ns.CBManagerStatePoweredOn):
		return "powered on"
	case (ns.CBManagerState)(ns.CBManagerStateUnknown):
		return "unknown"
	default:
		return "no state"
	}
}

func (b *BLE) readyToScan() bool {
	return b.state == (ns.CBManagerState)(ns.CBManagerStatePoweredOn)
}

func (b *BLE) scan() {
	b.cm.ScanForPeripheralsWithServices(nil, nil)
}

func (b *BLE) stopScan() {
	b.cm.StopScan()
}

func (b *BLE) setState(x ns.CBManagerState) {
	b.Lock()
	defer b.Unlock()
	b.state = x
	if b.ready && b.wantScan {
		fmt.Printf("Go: Scanning\n")
		b.scan()
		b.wantScan = false
	}
}

func (b *BLE) connectPeripheral(x Peripheral) {
	fmt.Printf("BLE.Connect(): calling cm.ConnectPeripheral(%p)\n", x.p.Ptr())
	b.cm.ConnectPeripheral(x.p, nil)
}

func CancelConnection(p Peripheral) {
	b := peripheralLookup(p)
	if b == nil {
		return
	}
	b.connections.Lock()
	var ch chan string;
	for _, item := range b.connections.items {
		if item.p.Identifier == p.Identifier {
			ch = item.state
			break
		}
	}
	b.connections.Unlock()
	if ch != nil {
		ch <- "cancel"
	}
}

func (b *BLE) cancelConnection(p Peripheral) {
	b.cm.CancelPeripheralConnection(p.p)
}

func (b *BLE) knownPeripheral(p Peripheral) (Peripheral, bool) {
	fmt.Printf("RetrievePeripheralsWithIdentifiers\n")
	ps := b.cm.RetrievePeripheralsWithIdentifiers(ns.NSArrayWithObjects(ns.NSUUIDAlloc().InitWithUUIDString(ns.NSStringWithGoString(p.Identifier))))
	if x := (int)(ps.Count()); x > 0 {
		fmt.Printf("--found %d\n", x)
		cbp := ps.ObjectAtIndex(0).CBPeripheral()
		return newPeripheral(cbp), true
	} else {
		fmt.Printf("--none found\n")
		return Peripheral{}, false
	}
}

func didUpdateState(c *ns.CBCentralManager) {
	b := cdLookup[c.Ptr()]
	st := c.CBManager.State()
	if st == (ns.CBManagerState)(ns.CBManagerStatePoweredOn) {
		b.ready = true
		b.connections.close = make(chan struct{})
	} else {
		if b.ready {
			close(b.connections.close)
			b.connections.Lock()
			for _, item := range b.connections.items {
				fmt.Printf("Closing connection to %s\n", item.p.Name)
				b.cm.CancelPeripheralConnection(item.p.p)
			}
			b.connections.items = b.connections.items[:0]
			b.connections.Unlock()
			b.ready = false
		}
	}
	b.setState(st)
	b.events <- UpdateStateEvent{State: b.stringState()}
}

func didDiscoverPeripheral(c *ns.CBCentralManager, p *ns.CBPeripheral, d *ns.NSDictionary, rssi *ns.NSNumber) {
	b := cdLookup[c.Ptr()]
	peripheral := newPeripheral(p)
	peripheral.RSSI = (int)(rssi.IntValue())
	_didDiscoverPeripheral(b, peripheral)
}

func _didDiscoverPeripheral(b *BLE, peripheral Peripheral) {
	if peripheral.Name == "" {
		return
	}
	if ok := b.peripherals.Add(peripheral); ok {
		pdLookup[peripheral.p.Ptr()] = b
		b.events <- DiscoverPeripheralEvent{Peripheral: peripheral}
	}
}

func didConnectPeripheral(c *ns.CBCentralManager, p *ns.CBPeripheral) {
	fmt.Printf("Did connect peripheral\n")
	b := cdLookup[c.Ptr()]

	// set ourselves up as a peripheral delegate
	p.SetDelegate(b.cd)

	uuidstring := p.Identifier().UUIDString().String()
	peripheral := newPeripheral(p)
	found := false

	b.peripherals.Lock()
	for _, item := range b.peripherals.items {
		if item.p.Identifier == uuidstring {
			peripheral = item.p
			found = true
			break
		}
	}
	b.peripherals.Unlock()

	if !found {
		peripheral.Name = peripheralName(p)
		if ok := b.peripherals.Add(peripheral); ok {
			pdLookup[p.Ptr()] = b
		}
	}

	b.connections.UpdateState(peripheral, "connected")
	b.events <- ConnectEvent{peripheral}
	fmt.Printf("Go: didConnectPeripheral returning\n")
}

func (x Peripheral) DiscoverServices() {
	fmt.Printf("Discovering services on %s\n", x.Name)
	p := x.p

	// discover all services on this device

	p.DiscoverServices(nil)
}

func didDiscoverServices(p *ns.CBPeripheral, e *ns.NSError) {
	b := pdLookup[p.Ptr()]
	fmt.Printf("Did discover services\n")
	p.Services().ObjectEnumerator().ForIn(func(o *ns.Id) bool {
		serv := o.CBService()
		uuid := serv.UUID().UUIDString().String()
		b.events <-DiscoverServiceEvent{
			Peripheral: newPeripheral(p),
			Gatt: gatt.Service{uuid},
			Service: serv,
		}
		return true
	})
	fmt.Printf("Go: didDiscoverServices returning\n")
}

func hr(d *ns.NSData) int {
	if l := int(d.Length()); l < 4 {
		return 0
	}
	x := C.GoBytes(d.Bytes(), 4)
	flags := x[0]
	if flags&0x80 != 0 { // uint16 format
		return int(binary.BigEndian.Uint16(x[1:2]))
	} else {
		return int(x[1])
	}
}

func (p Peripheral) DiscoverCharacteristics(serv *ns.CBService) {
	p.p.DiscoverCharacteristics(nil, serv)
}

func didDiscoverCharacteristics(p *ns.CBPeripheral, s *ns.CBService, e *ns.NSError) {
	b := pdLookup[p.Ptr()]
	fmt.Printf("Did discover characteristics\n")
	s.Characteristics().ObjectEnumerator().ForIn(func(o *ns.Id) bool {
		chr := o.CBCharacteristic()
		chuuid := chr.UUID()
		fmt.Printf("------%s\n", chuuid.UUIDString())
		b.events <-DiscoverCharacteristicEvent{
			Peripheral: newPeripheral(p),
			Service: s,
			Characteristic: chr,
			Gatt: gatt.Characteristic{chuuid.UUIDString().String()},
		}
		return true
	})
	fmt.Printf("Go: didDiscoverCharacteristics returning\n")
}

func (p Peripheral) SetNotifyValue(c *ns.CBCharacteristic) {
	p.p.SetNotifyValue(1, c)
}

func didUpdateValue(p *ns.CBPeripheral, chr *ns.CBCharacteristic, e *ns.NSError) {
	b := pdLookup[p.Ptr()]
	v := chr.Value()
	b.events <-UpdateValueEvent{
		Peripheral: newPeripheral(p),
		Characteristic: chr,
		Data: C.GoBytes(v.Bytes(), (C.int)(v.Length())),
	}
}

func NewBLE() *BLE {
	ps := Peripherals{items: make([]PeripheralListItem, 0)}
	ble := &BLE{events: make(chan interface{}), peripherals: ps}

	queue := ns.DispatchQueueCreate(ns.CharWithGoString("go_hrm_queue"), nil)

	cd := ns.CBDelegateAlloc()

	cd.CentralManagerDidUpdateStateCallback(didUpdateState)
	cd.CentralManagerDidDiscoverPeripheralCallback(didDiscoverPeripheral)
	cd.CentralManagerDidConnectPeripheralCallback(didConnectPeripheral)
	cd.PeripheralDidDiscoverServicesCallback(didDiscoverServices)
	cd.PeripheralDidDiscoverCharacteristicsForServiceCallback(didDiscoverCharacteristics)
	cd.PeripheralDidUpdateValueForCharacteristicCallback(didUpdateValue)


	ble.cd = cd
	if cdLookup == nil {
		cdLookup = make(map[unsafe.Pointer]*BLE, 0)
	}
	if pdLookup == nil {
		pdLookup = make(map[unsafe.Pointer]*BLE, 0)
	}

	// We defined our own queue because this won't work on the main queue.
	ble.cm = ns.CBCentralManagerAlloc().InitWithDelegateQueue(cd, queue)
	cdLookup[ble.cm.Ptr()] = ble

	// For debugging purposes, run GC every second to make sure things are
	// not over-released.
	go func() {
		for {
			runtime.GC()
			time.Sleep(time.Second * 30)
		}
	}()
	return ble
}