DMAC/server/pkg/system/reactormanager.go

package system

import (
	"FRMS/internal/pkg/manager"
	pb "dmac/pkg/grpc"
	"dmac/pkg/logging"
	"time"

	//"FRMS/internal/pkg/device"
	"context"
	"fmt"
	_ "log"

	"github.com/spf13/viper"
)

// manager stuff

type Manager interface {
	Start() error // status checks
	Exit() error
	Timeout() (time.Duration, error) // TO Generator
}

func NewManager(max int) Manager {
	// takes a heartbeat and max connection attempts
	return manager.New(max)
}

type ReactorManager struct {
	Manager // base manager interface
	// *ClientManager // client manager (OUTDATED)
	*Client // access to ID etc
	//	StatusMon *StatusMonitor putting on pause
	//	*ReactorDevices
	Config *viper.Viper // config to update
	Err    chan error
}

// type ReactorDevices struct {
// 	// device struct
// 	Devices map[int]DeviceManager
// 	sync.RWMutex
// }

func NewReactorManager(cl *Client, config *viper.Viper, errCh chan error) *ReactorManager {
	// making managers
	m := NewManager(6)
	r := &ReactorManager{
		Manager: m,
		Client:  cl,
		Config:  config,
		Err:     errCh,
	}
	return r
}

func (r *ReactorManager) Start() error {
	// allows for extra stuff
	logging.Debug(logging.DStart, "RMA %v starting", r.Id)
	return r.Manager.Start()
	//go r.StatusMon.Send(&DeviceInfo{Id: r.Id, Type: "Reactor", Status: "[green]ONLINE[white]"}, "Reactor")
}

func (r *ReactorManager) Exit() error {
	// allows for extra stuff
	logging.Debug(logging.DExit, "RMA %v exiting", r.Id)
	return r.Manager.Exit()
	//go r.StatusMon.Send(&DeviceInfo{Id: r.Id, Type: "Reactor", Status: "[red]OFFLINE[white]", Data: fmt.Sprintf("Last Seen %v", time.Now().Format("Mon at 03:04:05pm MST"))}, "Reactor")
}

func (r *ReactorManager) UpdateClient(cl *Client) error {
	// this is probably unnessecary
	fmt.Printf("Reactor Manager %d updating client!\n", r.Id)
	r.Client = cl
	return nil
}

func (r *ReactorManager) ReactorStatusHandler(ctx context.Context, req *pb.ReactorStatusPing) (*pb.ReactorStatusResponse, error) {
	// function client will call to update reactor information
	//go r.PingReset()
	fmt.Printf("Recieved ping from %d!\n", req.GetId())
	// update devices/sensors
	for _, dev := range req.GetDevices() {
		fmt.Printf("Device %d is %s ", dev.GetAddr(), dev.GetStatus().String())
	}
	fmt.Printf("\n")
	// go r.UpdateDevices(req.GetDevices())

	return &pb.ReactorStatusResponse{Id: int32(r.Id)}, nil
}

// // device stuff

// type DeviceManager interface {
// 	LoadConfig() error
// 	UpdateStatus(string) error
// 	String() string // printing
// }

// func NewDeviceManager(addr int, config *viper.Viper, prefix string) (DeviceManager, error) {
// 	// returns a manager struct
// 	return device.NewDeviceManager(addr, config, prefix)
// }

//func (r *ReactorManager) UpdateDevices(devs []*pb.Device) {
//	// pass updates to correct manager
//	r.ReactorDevices.RLock() // read lock only
//	defer r.ReactorDevices.RUnlock()

//	for _, dev := range devs {
//		// looping over devs
//		if dm, ok := r.ReactorDevices.Devices[int(dev.GetAddr())]; ok {
//			// device manager found
//			go dm.UpdateStatus(dev.GetStatus().String())
//			//fmt.Println(dm)
//		} else {
//			// not found
//			go r.AddDevice(dev, r.Id, r.Config, r.Err)
//		}
//	}
//}

// func (r *ReactorDevices) AddDevice(dev *pb.Device, id int, config *viper.Viper, errCh chan error) {

// 	// setting vars
// 	prefix := fmt.Sprintf("reactors.%d.", id)
// 	addr := int(dev.GetAddr())
// 	var dm DeviceManager
// 	var err error
// 	// write locking
// 	r.Lock()
// 	defer r.Unlock()

// 	if dm, err = NewDeviceManager(addr, config, prefix); err != nil {
// 		errCh <- err
// 	}

// 	// setting status
// 	if err = dm.UpdateStatus(dev.GetStatus().String()); err != nil {
// 		errCh <- err
// 	}

// 	// loading config
// 	if err = dm.LoadConfig(); err != nil {
// 		errCh <- err
// 	}
// 	r.Devices[int(addr)] = dm
// }