fix: import path should be github.com/ooni/probe-cli/v3 (#200)

See https://github.com/ooni/probe/issues/1335#issuecomment-771499511
2021-02-02 10:32:46 +01:00
parent faa9308b1e
commit b1ce300c8d
68 changed files with 86 additions and 85 deletions
@@ -0,0 +1,128 @@
+package list
+
+import (
+	"github.com/alecthomas/kingpin"
+	"github.com/apex/log"
+	"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/cli/root"
+	"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/database"
+	"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/output"
+)
+
+func init() {
+	cmd := root.Command("list", "List results")
+	resultID := cmd.Arg("id", "the id of the result to list measurements for").Int64()
+	cmd.Action(func(_ *kingpin.ParseContext) error {
+		probeCLI, err := root.Init()
+		if err != nil {
+			log.WithError(err).Error("failed to initialize root context")
+			return err
+		}
+		if *resultID > 0 {
+			measurements, err := database.ListMeasurements(probeCLI.DB(), *resultID)
+			if err != nil {
+				log.WithError(err).Error("failed to list measurements")
+				return err
+			}
+			msmtSummary := output.MeasurementSummaryData{
+				TotalCount:         0,
+				AnomalyCount:       0,
+				DataUsageUp:        0.0,
+				DataUsageDown:      0.0,
+				TotalRuntime:       0,
+				ASN:                0,
+				NetworkName:        "",
+				NetworkCountryCode: "ZZ",
+			}
+			isFirst := true
+			isLast := false
+			for idx, msmt := range measurements {
+				if idx > 0 {
+					isFirst = false
+				}
+				if idx == len(measurements)-1 {
+					isLast = true
+				}
+				// We assume that since these are summary level information the first
+				// item will contain the information necessary.
+				if isFirst {
+					msmtSummary.TotalRuntime = msmt.Result.Runtime
+					msmtSummary.DataUsageUp = msmt.DataUsageUp
+					msmtSummary.DataUsageDown = msmt.DataUsageDown
+					msmtSummary.NetworkName = msmt.NetworkName
+					msmtSummary.NetworkCountryCode = msmt.Network.CountryCode
+					msmtSummary.ASN = msmt.ASN
+					msmtSummary.StartTime = msmt.Measurement.StartTime
+				}
+				if msmt.IsAnomaly.Bool == true {
+					msmtSummary.AnomalyCount++
+				}
+				msmtSummary.TotalCount++
+				output.MeasurementItem(msmt, isFirst, isLast)
+			}
+			output.MeasurementSummary(msmtSummary)
+		} else {
+			doneResults, incompleteResults, err := database.ListResults(probeCLI.DB())
+			if err != nil {
+				log.WithError(err).Error("failed to list results")
+				return err
+			}
+			if len(incompleteResults) > 0 {
+				output.SectionTitle("Incomplete results")
+			}
+			for idx, result := range incompleteResults {
+				output.ResultItem(output.ResultItemData{
+					ID:                      result.Result.ID,
+					Index:                   idx,
+					TotalCount:              len(incompleteResults),
+					Name:                    result.TestGroupName,
+					StartTime:               result.StartTime,
+					NetworkName:             result.Network.NetworkName,
+					Country:                 result.Network.CountryCode,
+					ASN:                     result.Network.ASN,
+					MeasurementCount:        0,
+					MeasurementAnomalyCount: 0,
+					TestKeys:                "{}", // FIXME this used to be Summary we probably need to use a list now
+					Done:                    result.IsDone,
+					DataUsageUp:             result.DataUsageUp,
+					DataUsageDown:           result.DataUsageDown,
+				})
+			}
+			resultSummary := output.ResultSummaryData{}
+			netCount := make(map[uint]int)
+			output.SectionTitle("Results")
+			for idx, result := range doneResults {
+				totalCount, anmlyCount, err := database.GetMeasurementCounts(probeCLI.DB(), result.Result.ID)
+				if err != nil {
+					log.WithError(err).Error("failed to list measurement counts")
+				}
+				testKeys, err := database.GetResultTestKeys(probeCLI.DB(), result.Result.ID)
+				if err != nil {
+					log.WithError(err).Error("failed to get testKeys")
+				}
+				output.ResultItem(output.ResultItemData{
+					ID:                      result.Result.ID,
+					Index:                   idx,
+					TotalCount:              len(doneResults),
+					Name:                    result.TestGroupName,
+					StartTime:               result.StartTime,
+					NetworkName:             result.Network.NetworkName,
+					Country:                 result.Network.CountryCode,
+					ASN:                     result.Network.ASN,
+					TestKeys:                testKeys,
+					MeasurementCount:        totalCount,
+					MeasurementAnomalyCount: anmlyCount,
+					Done:                    result.IsDone,
+					DataUsageUp:             result.DataUsageUp,
+					DataUsageDown:           result.DataUsageDown,
+				})
+				resultSummary.TotalTests++
+				netCount[result.Network.ASN]++
+				resultSummary.TotalDataUsageUp += result.DataUsageUp
+				resultSummary.TotalDataUsageDown += result.DataUsageDown
+			}
+			resultSummary.TotalNetworks = int64(len(netCount))
+			output.ResultSummary(resultSummary)
+		}
+		return nil
+	})
+}