| package main |
| |
| import ( |
| "context" |
| |
| "github.com/spf13/cobra" |
| |
| "go.skia.org/infra/gold-client/go/goldclient" |
| "go.skia.org/infra/golden/go/types" |
| ) |
| |
| // diffEnv provides the environment for the diff command. |
| type diffEnv struct { |
| inputFile string |
| test string |
| corpus string |
| instanceID string |
| outDir string |
| workDir string |
| } |
| |
| // getDiffCmd returns the definition of the diff command. |
| func getDiffCmd() *cobra.Command { |
| env := &diffEnv{} |
| cmd := &cobra.Command{ |
| Use: "diff", |
| Short: "Compute the diff to the closest known image", |
| Long: ` |
| Downloads all images for a given test and compares them to the provided image. |
| |
| Outputs the closest of these images and the diff to the given folder. |
| `, |
| Run: env.runDiffCmd, |
| } |
| |
| cmd.Flags().StringVar(&env.inputFile, "input", "", "path to file to diff") |
| cmd.Flags().StringVar(&env.test, "test", "", "Test name") |
| cmd.Flags().StringVar(&env.corpus, "corpus", "", "Corpus name") |
| cmd.Flags().StringVar(&env.instanceID, "instance", "", "Instance (e.g. 'chrome', 'flutter')") |
| cmd.Flags().StringVar(&env.outDir, "out-dir", "", "Work directory that will contain the output") |
| cmd.Flags().StringVar(&env.workDir, fstrWorkDir, "", "Work directory for intermediate results") |
| // Everything is required for this command. |
| must(cmd.MarkFlagRequired(fstrWorkDir)) |
| must(cmd.MarkFlagRequired("input")) |
| must(cmd.MarkFlagRequired("test")) |
| must(cmd.MarkFlagRequired("corpus")) |
| must(cmd.MarkFlagRequired("out-dir")) |
| must(cmd.MarkFlagRequired("instance")) |
| |
| return cmd |
| } |
| |
| func (d *diffEnv) runDiffCmd(cmd *cobra.Command, _ []string) { |
| ctx := cmd.Context() |
| d.Diff(ctx) |
| } |
| |
| // Diff executes the diff logic for comparing a given image against all that Gold knows. |
| func (d *diffEnv) Diff(ctx context.Context) { |
| ctx = loadAuthenticatedClients(ctx, d.workDir) |
| |
| config := goldclient.GoldClientConfig{ |
| InstanceID: d.instanceID, |
| WorkDir: d.workDir, |
| } |
| |
| // overwrite any existing configs in this workdir. |
| goldClient, err := goldclient.NewCloudClient(config) |
| ifErrLogExit(ctx, err) |
| |
| err = goldClient.Diff(ctx, types.TestName(d.test), d.corpus, d.inputFile, d.outDir) |
| ifErrLogExit(ctx, err) |
| exitProcess(ctx, 0) |
| } |