kitchen: CURRENT_TIMESTAMP support

client/cmd/kitchen/recipe.go

 // Copyright 2016 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package main
 
 import (
         "fmt"
         "io/ioutil"
         "os"
@@ skipping 23 matching lines @@
 type recipeRun struct {
         repositoryPath string
 
         // The following are command line recipes.py command line arguments.
 
         recipe               string
         propertiesJSON       string
         propertiesFile       string
         outputResultJSONFile string
         workDir              string // Where to run the recipe.
+        timestamps           bool   // Whether to print CURRENT_TIMESTAMP annotations.
 }
 
 // Command creates a exec.Cmd for running a recipe.
 func (r *recipeRun) Command() (*exec.Cmd, error) {
         cfgPath := filepath.Join(r.repositoryPath, "infra/config/recipes.cfg")
         cfg, err := readConfig(cfgPath)
         if err != nil {
                 return nil, err
         }
 
         if cfg.RecipesPath == nil || *cfg.RecipesPath == "" {
                 return nil, fmt.Errorf("recipe_path is unspecified in %q", cfgPath)
         }
 
         recipesPy := path.Join(r.repositoryPath, *cfg.RecipesPath, "recipes.py")
         if _, err := os.Stat(recipesPy); os.IsNotExist(err) {
                 return nil, fmt.Errorf("%q does not exist", recipesPy)
         }
 
         cmd := exec.Command(
                 "python", recipesPy,
                 "run",
                 "--properties", r.propertiesJSON,
                 "--properties-file", r.propertiesFile,
                 "--workdir", r.workDir,
                 "--output-result-json", r.outputResultJSONFile,
-                r.recipe,
         )
+        if r.timestamps {
+                cmd.Args = append(cmd.Args, "--timestamps")
+        }
+        cmd.Args = append(cmd.Args, r.recipe)
+
         cmd.Stdout = os.Stdout
         cmd.Stderr = os.Stderr
         return cmd, nil
 }