etri-smartfarm-poc-controller-serial/cli.go

package manager

import (
	"context"
	"errors"
	"log"

	"git.godopu.net/lab/etri-smartfarm-poc-controller-serial/puserial"
	"github.com/rjeczalik/notify"
)

var ctx context.Context
var cancel context.CancelFunc
var ch_discover chan notify.EventInfo
var _managerObj *_manager
var registerHandleFunc func(uuid string)
var removeHandleFunc func(uuid string)

func init() {
	ctx, cancel = context.WithCancel(context.Background())
	ch_discover = make(chan notify.EventInfo)
	devWithUUID := map[string]*_device{}
	devWithIface := map[interface{}]*_device{}
	chanForSync := map[string]chan map[string]interface{}{}

	_managerObj = &_manager{
		devicesWithUUID:  devWithUUID,
		devicesWithIface: devWithIface,
		chanForSync:      chanForSync,
		SyncListener:     &SyncHandler{devices: devWithUUID, chanForSync: chanForSync},
		RecvListener:     &RecvHandler{devices: devWithIface, chanForSync: chanForSync},
	}

	registerHandleFunc = nil
	removeHandleFunc = nil
}

func AddRegisterHandleFunc(h func(uuid string)) {
	registerHandleFunc = h
}

func AddRemoveHandleFunc(h func(uuid string)) {
	removeHandleFunc = h
}

func Close() {
	cancel()
}

func Sync(key string, param map[string]interface{}) {
	_managerObj.onSync(key, param)
}

func AddRecvListener(h EventHandler) {
	_managerObj.addRecvListener(h)
}

func SetDevicePropsToSync(uuid string, propsToSync []string) error {
	device, ok := _managerObj.devicesWithUUID[uuid]
	if !ok {
		return errors.New("device not found")
	}
	device.propsToSync = propsToSync
	return nil
}

func Run() error {
	iface, err := puserial.InitDevice()
	if err != nil {
		if err.Error() != "USB Not found" {
			return err
		}
	} else {
		// _managerObj.onAdd(iface)
		_managerObj.onAdded(iface)
	}

	go puserial.WatchNewDevice(ctx, ch_discover)

	for {
		e, ok := <-ch_discover
		if !ok {
			log.Println("manager exit")
			return nil
		}
		switch e.Event() {
		case notify.Create:
			_managerObj.onAdded(e.Path())
			// case notify.Remove:
			// 	log.Println("USB Disconnected!!")
			// 	_managerObj.onAdded(e.Path())
		}

	}
}