Apply gerrit ref if it is a Gerrit patch

infra/bots/recipe_modules/vars/api.py

Index: infra/bots/recipe_modules/vars/api.py
diff --git a/infra/bots/recipe_modules/vars/api.py b/infra/bots/recipe_modules/vars/api.py
index e2d0ed99c02534a4130e807a3e5186acaef25b84..91d344656759fdb538b1348479de94fb448127db 100644
--- a/infra/bots/recipe_modules/vars/api.py
+++ b/infra/bots/recipe_modules/vars/api.py
@@ -110,13 +110,24 @@ class SkiaVarsApi(recipe_api.RecipeApi):
     self.default_env.update({'SKIA_OUT': self.skia_out,
                              'BUILDTYPE': self.configuration})
     self.is_trybot = self.builder_cfg['is_trybot']
+    self.patch_storage = self.m.properties.get('patch_storage')
+
+    self.gerrit_repository = None
+    self.gerrit_ref = None
+    self.gerrit_issue = None
+
     self.issue = None
     self.patchset = None
     self.rietveld = None
     if self.is_trybot:
-      self.issue = self.m.properties['issue']
-      self.patchset = self.m.properties['patchset']
-      self.rietveld = self.m.properties['rietveld']
+      if self.patch_storage == 'gerrit':
+        self.gerrit_repository = self.m.properties['repository']
+        self.gerrit_ref = self.m.properties['event.patchSet.ref']
+        self.gerrit_issue = self.m.properties['event.change.number']
+      else:
+        self.issue = self.m.properties['issue']
+        self.patchset = self.m.properties['patchset']
+        self.rietveld = self.m.properties['rietveld']
     self.dm_dir = self.m.path.join(
         self.swarming_out_dir, 'dm')
     self.perf_data_dir = self.m.path.join(self.swarming_out_dir,