status-go/cmd/status-cli/simulate.go

package main

import (
	"context"
	"fmt"
	"os"
	"os/signal"
	"sync"
	"syscall"
	"time"

	"github.com/urfave/cli/v2"
	"go.uber.org/zap"
)

func simulate(cCtx *cli.Context) error {
	ctx, cancel := context.WithCancel(cCtx.Context)

	go func() {
		sig := make(chan os.Signal, 1)
		signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
		<-sig
		cancel()
		time.Sleep(1 * time.Second)
		os.Exit(1)
	}()

	logger, err := getSLogger(cCtx.Bool(DebugLevel))
	if err != nil {
		zap.S().Fatalf("Error initializing logger: %v", err)
	}

	logger.Infof("Running %v command, with:\n%v", cCtx.Command, flagsUsed(cCtx))

	// Start messengers
	apiModules := cCtx.String(APIModulesFlag)
	telemetryUrl := cCtx.String(TelemetryServerURLFlag)
	failMessages := cCtx.Bool(MessageFailureFlag)

	alice, err := start("Alice", 0, apiModules, telemetryUrl, "", logger.Named("alice"))
	if err != nil {
		return err
	}
	defer alice.stop()

	charlie, err := start("Charlie", 0, apiModules, telemetryUrl, "", logger.Named("charlie"))
	if err != nil {
		return err
	}
	defer charlie.stop()

	// Retrieve for messages
	msgCh := make(chan string)
	var wg sync.WaitGroup

	wg.Add(1)
	go alice.retrieveMessagesLoop(ctx, RetrieveInterval, nil, &wg)
	wg.Add(1)
	go charlie.retrieveMessagesLoop(ctx, RetrieveInterval, msgCh, &wg)

	// Send contact request from Alice to Charlie, charlie accept the request
	time.Sleep(WaitingInterval)
	destID := charlie.messenger.GetSelfContact().ID
	err = alice.sendContactRequest(ctx, destID)
	if err != nil {
		return err
	}

	msgID := <-msgCh
	err = charlie.sendContactRequestAcceptance(ctx, msgID)
	if err != nil {
		return err
	}
	time.Sleep(WaitingInterval)

	// Send DM between alice to charlie
	interactive := cCtx.Bool(InteractiveFlag)
	if interactive {
		interactiveSendMessageLoop(ctx, alice, charlie)
	} else {
		for i := 0; i < cCtx.Int(CountFlag); i++ {
			err = alice.sendDirectMessage(ctx, fmt.Sprintf("message from alice, number: %d", i+1), failMessages)
			if err != nil {
				return err
			}
			time.Sleep(WaitingInterval)

			err = charlie.sendDirectMessage(ctx, fmt.Sprintf("message from charlie, number: %d", i+1), failMessages)
			if err != nil {
				return err
			}
			time.Sleep(WaitingInterval)
		}
		cancel()
	}

	wg.Wait()
	logger.Info("Exiting")

	return nil
}