test version early test event

internal/cli/event.go

@@ -0,0 +1,13 @@
+package cli
+
+import (
+	"github.com/rarimo/geo-points-svc/internal/config"
+	"github.com/rarimo/geo-points-svc/internal/service/event"
+)
+
+func eventStart(cfg config.Config, data int) {
+	err := event.Run(cfg, data)
+	if err != nil {
+		cfg.Log().Error("Error starting event: %v", err)
+	}
+}

internal/cli/main.go

@@ -29,6 +29,10 @@ func Run(args []string) bool {
 		migrateCmd     = app.Command("migrate", "migrate command")
 		migrateUpCmd   = migrateCmd.Command("up", "migrate db up")
 		migrateDownCmd = migrateCmd.Command("down", "migrate db down")
+
+		event    = app.Command("event", "claim event command")
+		eventCmd = event.Command("create-early-test", "claim event command")
+		data     = eventCmd.Arg("-before", "data after ...").Required().Int()
 	)
 
 	cmd, err := app.Parse(args[1:])
@@ -48,6 +52,8 @@ func Run(args []string) bool {
 		err = MigrateUp(cfg)
 	case migrateDownCmd.FullCommand():
 		err = MigrateDown(cfg)
+	case eventCmd.FullCommand():
+		eventStart(cfg, *data)
 	default:
 		log.Errorf("unknown command %s", cmd)
 		return false

internal/data/balances.go

@@ -54,6 +54,8 @@ type BalancesQ interface {
 	FilterByInternalAID(aid string) BalancesQ
 	FilterByExternalAID(aid string) BalancesQ
 	FilterBySharedHash(hash string) BalancesQ
+	FilterByCreatedBefore(date int) BalancesQ
+	FilterVerified() BalancesQ
 }
 
 type WithoutPassportEventBalance struct {

internal/data/events.go

@@ -65,4 +65,5 @@ type EventsQ interface {
 	FilterByUpdatedAtBefore(int64) EventsQ
 	FilterByExternalID(string) EventsQ
 	FilterInactiveNotClaimed(types ...string) EventsQ
+	FilterByNullifiers(nullifiers []string) EventsQ
 }

internal/data/evtypes/models/extra.go

@@ -21,6 +21,7 @@ const (
 	TypeInternalPassportScan = "internal_passport_scan"
 	TypeExternalPassportScan = "external_passport_scan"
 	TypePollParticipation    = "poll_participation"
+	TypeEarlyTest            = "early_test"
 )
 
 const (

internal/data/pg/balances.go

@@ -222,10 +222,18 @@ func (q *balances) FilterBySharedHash(hash string) data.BalancesQ {
 	return q.applyCondition(squirrel.Eq{"shared_hash": hash})
 }
 
+func (q *balances) FilterByCreatedBefore(data int) data.BalancesQ {
+	return q.applyCondition(squirrel.Lt{"created_at": data})
+}
+
 func (q *balances) applyCondition(cond squirrel.Sqlizer) data.BalancesQ {
 	q.selector = q.selector.Where(cond)
 	q.updater = q.updater.Where(cond)
 	q.rank = q.rank.Where(cond)
 	q.counter = q.counter.Where(cond)
 	return q
 }
+
+func (q *balances) FilterVerified() data.BalancesQ {
+	return q.applyCondition(squirrel.Gt{"is_verified": true})
+}

internal/data/pg/events.go

@@ -197,6 +197,13 @@ func (q *events) FilterByNullifier(nullifier string) data.EventsQ {
 	return q.applyCondition(squirrel.Eq{"nullifier": nullifier})
 }
 
+func (q *events) FilterByNullifiers(nullifiers []string) data.EventsQ {
+	if len(nullifiers) == 0 {
+		return q
+	}
+	return q.applyCondition(squirrel.Eq{"nullifier": nullifiers})
+}
+
 func (q *events) FilterByStatus(statuses ...data.EventStatus) data.EventsQ {
 	if len(statuses) == 0 {
 		return q

internal/service/event/main.go

@@ -0,0 +1,61 @@
+package event
+
+import (
+	"fmt"
+
+	"github.com/rarimo/geo-points-svc/internal/config"
+	"github.com/rarimo/geo-points-svc/internal/data"
+	"github.com/rarimo/geo-points-svc/internal/data/evtypes/models"
+	"github.com/rarimo/geo-points-svc/internal/data/pg"
+)
+
+func Run(cfg config.Config, date int) error {
+	log := cfg.Log()
+	db := cfg.DB()
+
+	balances := pg.NewBalances(db)
+	events := pg.NewEvents(db)
+
+	users, err := balances.FilterByCreatedBefore(date).FilterVerified().Select()
+
+	if err != nil {
+		log.WithError(err).Error("failed to filter by updated before")
+		return err
+	}
+
+	if users == nil {
+		log.Infof("no users found")
+		return nil
+	}
+
+	var userNullifiers []string
+	for _, user := range users {
+		userNullifiers = append(userNullifiers, user.Nullifier)
+	}
+
+	existingEvents, err := events.FilterByType(models.TypeEarlyTest).FilterByNullifiers(userNullifiers).Select()
+	if err != nil {
+		log.WithError(err).Error("failed to filter events")
+		return err
+	}
+
+	existingEventsMap := make(map[string]data.Event)
+	for _, event := range existingEvents {
+		existingEventsMap[event.Nullifier] = event
+	}
+
+	for _, user := range users {
+		if _, exists := existingEventsMap[user.Nullifier]; !exists {
+			err = events.Insert(data.Event{
+				Nullifier: user.Nullifier,
+				Type:      models.TypeEarlyTest,
+				Status:    data.EventFulfilled,
+			})
+			if err != nil {
+				return fmt.Errorf("failed to insert `early_test` event: %w", err)
+			}
+		}
+	}
+
+	return nil
+}

internal/service/workers/leveljustice/main.go

@@ -37,5 +37,4 @@ func Run(cfg config.Config, sig chan struct{}) {
 	}
 
 	sig <- struct{}{}
-
 }