Add CIPD GAE SDK recipe module and packager.

scripts/slave/recipe_modules/gae_sdk/api.py

+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from recipe_engine import recipe_api
+
+
+class GaeSdkApi(recipe_api.RecipeApi):
+
+  PLAT_PYTHON = 'python'
+  PLAT_GO = 'go'
+
+  # Map of {Platform => {Arch => (base, dirname)}}. Platform names are
+  # "recipe_engine/platform" values.
+  _PKG_MAP = {
+      PLAT_PYTHON: {
+        'all': ('google_appengine_', 'google_appengine'),
+      },
+      PLAT_GO: {
+        'linux-amd64': ('go_appengine_sdk_linux_amd64-', 'go_appengine'),
+        'linux-386': ('go_appengine_sdk_linux_386-', 'go_appengine'),
+        'mac-amd64': ('go_appengine_sdk_darwin_amd64-', 'go_appengine'),
+      },
+  }
+
+  # Map of architecture bitness to CIPD bitness suffix.
+  _BITS_MAP = {
+      64: 'amd64',
+      32: '386',
+  }
+
+
+  class PackageNotFound(Exception):
+    def __init__(self, plat, arch):
+      super(GaeSdkApi.PackageNotFound, self).__init__(
+          'Package not found for %s on %s' % (plat, arch))
+      self.plat = plat
+      self.arch = arch
+
+  @property
+  def latest_ref(self):
+    return 'latest'
+
+  def package_spec(self, plat, arch):
+    plat_dict = self._PKG_MAP.get(plat, {})
+    for a in (arch, 'all'):
+      spec = plat_dict.get(a)
+      if spec:
+        download_base, dirname = spec
+        pkg_name = 'infra/gae_sdk/%s/%s' % (plat, a)
+        return pkg_name, download_base, dirname
+    raise self.PackageNotFound(plat, arch)
+
+  def package(self, plat, arch=None):
+    arch = arch or '%s-%s' % (
+        self.m.platform.name, self._BITS_MAP[self.m.platform.bits])
+    pkg_name, _, _ = self.package_spec(plat, arch)
+    return pkg_name
+
+  @property
+  def platforms(self):
+    return sorted(self._PKG_MAP.keys())
+
+  @property
+  def all_packages(self):
+    for plat, arch_dict in sorted(self._PKG_MAP.items()):
+      for arch in sorted(arch_dict.keys()):
+        yield plat, arch
+
+  def fetch(self, plat, dst):
+    """Fetch the AppEngine SDK for the specified platform.
+
+    Args:
+      plat (str): platform string, one of the PLAT_ local variables.
+      dst (path.Path): The destination directory to extract it.
+    """
+    pkg = self.package(plat)
+    self.m.cipd.ensure(dst, {pkg: self.latest_ref})