feat: added /jailscount command

pkg/database/database.go

@@ -470,3 +470,44 @@ func (d *Database) FindLastEventsByValidator(
 
 	return events, nil
 }
+
+func (d *Database) FindAllJailsCount(chain string) ([]types.ValidatorWithJailsCount, error) {
+	d.MaybeMutexLock()
+	defer d.MaybeMutexUnlock()
+
+	jailsCount := []types.ValidatorWithJailsCount{}
+
+	rows, err := d.client.Query(
+		"SELECT validator, count(*) from events where chain = $1 AND event = $2 GROUP BY chain, validator ORDER BY count DESC",
+		chain,
+		constants.EventValidatorJailed,
+	)
+	if err != nil {
+		d.logger.Error().Err(err).Msg("Error getting events by operator address and type")
+		return jailsCount, err
+	}
+	defer func() {
+		_ = rows.Close()
+		_ = rows.Err()
+	}()
+
+	for rows.Next() {
+		var (
+			validator string
+			count     int
+		)
+
+		err = rows.Scan(&validator, &count)
+		if err != nil {
+			d.logger.Error().Err(err).Msg("Error fetching jails count")
+			return jailsCount, err
+		}
+
+		jailsCount = append(jailsCount, types.ValidatorWithJailsCount{
+			Validator:  validator,
+			JailsCount: count,
+		})
+	}
+
+	return jailsCount, nil
+}

pkg/reporters/telegram/jails_count.go

@@ -0,0 +1,46 @@
+package telegram
+
+import (
+	"main/pkg/constants"
+
+	tele "gopkg.in/telebot.v3"
+)
+
+func (reporter *Reporter) HandleJailsCount(c tele.Context) error {
+	reporter.Logger.Info().
+		Str("sender", c.Sender().Username).
+		Str("text", c.Text()).
+		Msg("Got jails count query")
+
+	reporter.MetricsManager.LogReporterQuery(reporter.Config.Name, constants.TelegramReporterName, "jailscount")
+
+	snapshot, found := reporter.SnapshotManager.GetNewerSnapshot()
+	if !found {
+		reporter.Logger.Info().
+			Str("sender", c.Sender().Username).
+			Str("text", c.Text()).
+			Msg("No older snapshot on telegram events query!")
+		return reporter.BotReply(c, "Error getting validator events!")
+	}
+
+	jailsCount, err := reporter.Manager.FindAllJailsCount()
+	if err != nil {
+		return reporter.BotReply(c, "Error searching for jails count!")
+	}
+
+	jailsCountRendered := make([]renderedJailsCount, len(jailsCount))
+
+	for index, validatorJailsCount := range jailsCount {
+		validatorEntries := snapshot.Entries.ByValidatorAddresses([]string{validatorJailsCount.Validator})
+		if len(validatorEntries) == 0 {
+			return reporter.BotReply(c, "Validator is not found!")
+		}
+
+		jailsCountRendered[index] = renderedJailsCount{
+			ValidatorLink: reporter.Config.ExplorerConfig.GetValidatorLink(validatorEntries[0].Validator),
+			JailsCount:    validatorJailsCount.JailsCount,
+		}
+	}
+
+	return reporter.ReplyRender(c, "JailsCount", jailsCountRendered)
+}

pkg/reporters/telegram/telegram.go

@@ -111,6 +111,7 @@ func (reporter *Reporter) Init() {
 	bot.Handle("/config", reporter.HandleParams)
 	bot.Handle("/jails", reporter.HandleJailsList)
 	bot.Handle("/events", reporter.HandleValidatorEventsList)
+	bot.Handle("/jailscount", reporter.HandleJailsCount)
 
 	reporter.TelegramBot = bot
 }

pkg/reporters/telegram/types.go

@@ -107,3 +107,8 @@ type eventsRender struct {
 	ValidatorLink types.Link
 	Events        []renderedHistoricalEvent
 }
+
+type renderedJailsCount struct {
+	ValidatorLink types.Link
+	JailsCount    int
+}

pkg/state/manager.go

@@ -227,6 +227,10 @@ func (m *Manager) FindLastEventsByValidator(operatorAddress string) ([]types.His
 	)
 }
 
+func (m *Manager) FindAllJailsCount() ([]types.ValidatorWithJailsCount, error) {
+	return m.database.FindAllJailsCount(m.config.Name)
+}
+
 func (m *Manager) GetValidators() types.ValidatorsMap {
 	return m.state.GetValidators()
 }

pkg/types/types.go

@@ -34,3 +34,8 @@ func (validatorsMap ValidatorsMap) ToSlice() Validators {
 
 	return validators
 }
+
+type ValidatorWithJailsCount struct {
+	Validator  string
+	JailsCount int
+}

templates/telegram/JailsCount.html

@@ -0,0 +1,8 @@
+{{- if . }}
+<strong>Validators jails count since the app was started:</strong>
+{{- range . }}
+{{ SerializeLink .ValidatorLink }}: {{ .JailsCount }}
+{{- end }}
+{{- else }}
+Nobody has been jailed since the app launch.
+{{- end }}