hybridgroup.gobot/platforms/mavlink/mavlink_driver.go

package mavlink

import (
	"time"

	"gobot.io/x/gobot/v2"
	common "gobot.io/x/gobot/v2/platforms/mavlink/common"
)

const (
	// PacketEvent event
	PacketEvent = "packet"
	// MessageEvent event
	MessageEvent = "message"
	// ErrorIOEEvent event
	ErrorIOEvent = "errorIO"
	// ErrorMAVLinkEvent event
	ErrorMAVLinkEvent = "errorMAVLink"
)

type Driver struct {
	name       string
	connection gobot.Connection
	interval   time.Duration
	gobot.Eventer
}

type MavlinkInterface interface{}

// NewDriver creates a new mavlink driver.
//
// It add the following events:
//
//	"packet" - triggered when a new packet is read
//	"message" - triggered when a new valid message is processed
func NewDriver(a BaseAdaptor, v ...time.Duration) *Driver {
	m := &Driver{
		name:       "Mavlink",
		connection: a,
		Eventer:    gobot.NewEventer(),
		interval:   10 * time.Millisecond,
	}

	if len(v) > 0 {
		m.interval = v[0]
	}

	m.AddEvent(PacketEvent)
	m.AddEvent(MessageEvent)
	m.AddEvent(ErrorIOEvent)
	m.AddEvent(ErrorMAVLinkEvent)

	return m
}

func (m *Driver) Connection() gobot.Connection { return m.connection }
func (m *Driver) Name() string                 { return m.name }
func (m *Driver) SetName(n string)             { m.name = n }

// adaptor returns driver associated adaptor
func (m *Driver) adaptor() BaseAdaptor {
	//nolint:forcetypeassert // ok here
	return m.Connection().(BaseAdaptor)
}

// Start begins process to read mavlink packets every m.Interval
// and process them
func (m *Driver) Start() error {
	go func() {
		for {
			packet, err := m.adaptor().ReadMAVLinkPacket()
			if err != nil {
				m.Publish(ErrorIOEvent, err)
				continue
			}
			m.Publish(PacketEvent, packet)
			message, err := packet.MAVLinkMessage()
			if err != nil {
				m.Publish(ErrorMAVLinkEvent, err)
				continue
			}
			m.Publish(MessageEvent, message)
			time.Sleep(m.interval)
		}
	}()
	return nil
}

// Halt returns true if device is halted successfully
func (m *Driver) Halt() error { return nil }

// SendPacket sends a packet to mavlink device
func (m *Driver) SendPacket(packet *common.MAVLinkPacket) error {
	_, err := m.adaptor().Write(packet.Pack())
	return err
}