status-go-monitor/control.go

package main

import (
	"log"
)

type StateController struct {
	State  *AppState
	Client *StatusGoClient
}

// For fetching current state of peers from status-go
func (s *StateController) Fetch() {
	peers, err := s.Client.getPeers()
	if err != nil {
		log.Panicln(err)
	}
	ps := s.State.GetData()
	s.State.UpdatePeers(peers)
	if ps.Current == -1 {
		s.State.SetCurrentPeer(0)
	}
}

// For removing a selected peer from connected to status-go
func (s *StateController) TrustPeer(peer *Peer) error {
	success, err := s.Client.trustPeer(peer.Enode)
	if err != nil {
		log.Panicln(err)
	}
	if success != true {
		log.Panicln("failed to trust whisper peer")
	}
	return nil
}

// For removing a selected peer from connected to status-go
func (s *StateController) RemovePeer(peer *Peer) error {
	success, err := s.Client.removePeer(peer.Enode)
	if err != nil {
		log.Panicln(err)
	}
	if success != true {
		log.Panicln("failed to remove peer")
	}
	s.Fetch()
	return nil
}

func (s *StateController) GetInfo() error {
	info, err := s.Client.nodeInfo()
	if err != nil {
		return err
	}
	s.State.UpdateInfo(info)
	return nil
}