ooni-probe-cli/internal/cli/run/run.go

package run

import (
	"time"

	"github.com/alecthomas/kingpin"
	"github.com/apex/log"
	"github.com/openobservatory/gooni/internal/cli/root"
	"github.com/openobservatory/gooni/internal/database"
	"github.com/openobservatory/gooni/nettests"
	"github.com/openobservatory/gooni/nettests/groups"
)

func init() {
	cmd := root.Command("run", "Run a test group or OONI Run link")

	nettestGroup := cmd.Arg("name", "the nettest group to run").Required().String()

	cmd.Action(func(_ *kingpin.ParseContext) error {
		log.Infof("Starting %s", *nettestGroup)
		_, ctx, err := root.Init()
		if err != nil {
			log.Errorf("%s", err)
			return err
		}
		group := groups.NettestGroups[*nettestGroup]
		log.Debugf("Running test group %s", group.Label)

		result, err := database.CreateResult(ctx.DB, database.Result{
			Name:      *nettestGroup,
			StartTime: time.Now().UTC(), // XXX get this from MK
		})
		if err != nil {
			log.Errorf("%s", err)
			return err
		}

		for _, nt := range group.Nettests {
			ctl := nettests.NewController(ctx, result)
			nt.Run(ctl)
			// XXX
			// 1. Generate the summary
			// 2. Link the measurement to the Result (this should probably happen in
			// the nettest class)
			// 3. Update the summary of the result and the other metadata in the db
		}
		result.Update(ctx.DB)
		return nil
	})
}