main.go

package main

import (
	"fmt"
	"log"
	"math"
	"net/http"
	"strings"
	"time"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"

	"github.com/Chainflow/solana-mission-control/alerter"
	"github.com/Chainflow/solana-mission-control/config"
	"github.com/Chainflow/solana-mission-control/exporter"
	"github.com/Chainflow/solana-mission-control/monitor"
	"github.com/Chainflow/solana-mission-control/utils"
)

func main() {
	cfg, err := config.ReadFromFile() // Read config file
	if err != nil {
		log.Fatal(err)
	}

	collector := exporter.NewSolanaCollector(cfg)

	go collector.WatchSlots(cfg)

	// Calling command based alerting
	go func() {
		for {
			monitor.TelegramAlerting(cfg)
			time.Sleep(2 * time.Second)
		}
	}()

	go func() {
		for {
			monitor.SkipRateAlerts(cfg)
			time.Sleep(60 * time.Second)
		}
	}()

	if strings.EqualFold(cfg.AlerterPreferences.StartupAlerts, "yes") {
		currEpoch := monitor.GetEpochDetails(cfg)

		activatedStake := float64(-1)
		voteAccs, err := monitor.GetVoteAccounts(cfg, utils.Network)
		if err != nil {
			log.Printf("Error while getting vote accounts: %v", err)
		} else {
			for _, vote := range voteAccs.Result.Current {
				if vote.NodePubkey == cfg.ValDetails.PubKey {
					activatedStake = float64(vote.ActivatedStake) / math.Pow(10, 9)
					break
				}
			}
		}

		// send alert
		msg := fmt.Sprintf("Solana Mission Control started up. Current Epoch Info:\n%s\nActivated Stake: %.4f", currEpoch, activatedStake)
		err = alerter.SendTelegramAlert(msg, cfg)
		if err != nil {
			log.Printf("Error while sending startup alert to telegram: %v", err)
		}
		// send email alert
		err = alerter.SendEmailAlert(msg, cfg)
		if err != nil {
			log.Printf("Error while sending startup alert to email: %v", err)
		}
		// send slack alert
		err = alerter.SendSlackAlert(msg, cfg)
		if err != nil {
			log.Printf("Error while sending startup alert to slack: %v", err)
		}
	}

	prometheus.MustRegister(collector)
	http.Handle("/metrics", promhttp.Handler()) // exported metrics can be seen in /metrics
	err = http.ListenAndServe(fmt.Sprintf("%s", cfg.Prometheus.ListenAddress), nil)
	if err != nil {
		log.Printf("Error while listening on server : %v", err)
	}
}