adds new action in action map for creating a shipit release (mozilla-releng#75)

* adds new action in action map for creating a shipit release

* Fix tox by importing functions properly

* use shipitapi for interfacing with shipit

* fix create_new_release_action typo

* Add triggering phase logic

* fixes up tox

Author: Jordan Lund

shipitscript/shipitscript/script.py

@@ -12,6 +12,7 @@
     get_task_action,
     validate_task_schema,
 )
+from shipitscript.utils import get_buildnum_from_version
 
 log = logging.getLogger(__name__)
 
@@ -36,8 +37,47 @@ def mark_as_shipped_action(context):
     ship_actions.mark_as_shipped_v2(context.ship_it_instance_config, release_name)
 
 
+def create_new_release_action(context):
+    """Determine if there is a shippable release and create it if so in Shipit """
+    payload = context.task['payload']
+    shipit_config = context.ship_it_instance_config
+    # TODO actually include these in the payload from taskgraph
+    product = payload["product"]
+    channel = payload["channel"]
+    repo = payload["repo"]
+    phase = payload["phase"]  # release phase we want to trigger
+
+    log.info(
+        'Determining most recent shipped revision and next version / buildnum to release'
+    )
+    last_shipped_revision = ship_actions.get_most_recent_shipped_revision(
+        product, channel, shipit_config
+    )
+    next_version = ship_actions.get_next_release_version(
+        product, channel, shipit_config
+    )
+    log.info('Ensuring next version is a new version and not a buildnum increment')
+    if get_buildnum_from_version(next_version) != 1:
+        # TODO quit early, mark task as green though
+        pass
+    log.info('Determining most recent shippable revision')
+    shippable_revision = ship_actions.get_shippable_revision(
+        repo, last_shipped_revision
+    )
+    if not shippable_revision:
+        # TODO quit early, mark task as green though
+        pass
+    log.info('create a new release')
+    ship_actions.create_new_release(
+        product, repo, channel, next_version, shippable_revision, phase, shipit_config
+    )
+
+
 # ACTION_MAP {{{1
-ACTION_MAP = {'mark-as-shipped': mark_as_shipped_action}
+ACTION_MAP = {
+    'mark-as-shipped': mark_as_shipped_action,
+    'create-new-release': create_new_release_action,
+}
 
 
 def get_default_config():

shipitscript/shipitscript/ship_actions.py

@@ -10,24 +10,66 @@
 log = logging.getLogger(__name__)
 
 
-def mark_as_shipped_v2(ship_it_instance_config, release_name):
-    """Function to make a simple call to Ship-it API v2 to change a release
-    status to 'shipped'
-    """
+def get_shipit_api_instance(shipit_config):
     (
-        taskcluster_client_id,
-        taskcluster_access_token,
+        tc_client_id,
+        tc_access_token,
         api_root,
         timeout_in_seconds,
-    ) = get_auth_primitives_v2(ship_it_instance_config)
+    ) = get_auth_primitives_v2(shipit_config)
+
     release_api = Release_V2(
-        taskcluster_client_id=taskcluster_client_id,
-        taskcluster_access_token=taskcluster_access_token,
+        taskcluster_client_id=tc_client_id,
+        taskcluster_access_token=tc_access_token,
         api_root=api_root,
         timeout=timeout_in_seconds,
     )
+    headers = get_request_headers(api_root)
+
+    return release_api, headers
+
+
+def get_shippable_revision(repo, last_shipped_revision):
+    pass
+
+
+def get_most_recent_shipped_revision(product, channel, shipit_config):
+    release_api, headers = get_shipit_api_instance(shipit_config)
+
+
+def get_next_release_version(product, channel, shipit_config):
+    release_api, headers = get_shipit_api_instance(shipit_config)
+
+
+def start_new_release(
+    self,
+    product,
+    channel,
+    release_name,
+    version,
+    revision,
+    phase,
+    shipit_config,
+    headers={},
+):
+    release_api, headers = get_shipit_api_instance(shipit_config)
+    release_name = ""
+
+    log.info('creating a new release...')
+    release_api.create_new_release(
+        product, channel, release_name, version, revision, headers=headers
+    )
+    release_api.trigger_release_phase(
+        product, channel, release_name, phase, headers=headers
+    )
+
+
+def mark_as_shipped_v2(shipit_config, release_name):
+    """Function to make a simple call to Ship-it API v2 to change a release
+    status to 'shipped'
+    """
+    release_api, headers = get_shipit_api_instance(shipit_config)
 
     log.info('Marking the release as shipped...')
-    headers = get_request_headers(api_root)
     release_api.update_status(release_name, status='shipped', headers=headers)
     check_release_has_values_v2(release_api, release_name, headers, status='shipped')

shipitscript/shipitscript/shipitapi.py

@@ -124,3 +124,31 @@ def update_status(self, name, status, rebuild_product_details=True, headers={}):
                 headers=headers,
             )
         return res
+
+    def releases_are_disabled(self, product, channel, headers={}):
+        pass
+
+    def get_next_release_version(self, product, channel, headers={}):
+        pass
+
+    def get_most_recent_shipped_revision(self, product, channel, headers={}):
+        pass
+
+    def create_new_release(
+        self, product, repo, channel, release_name, version, revision, headers={}
+    ):
+        if self.releases_are_disabled(product, channel):
+            return
+        pass
+
+    def trigger_release_phase(self, product, channel, release_name, phase, headers={}):
+        """Trigger a push phase for a specific release"""
+        if self.releases_are_disabled(product, channel):
+            return
+
+        self._request(
+            api_endpoint='/releases/{}/{}'.format(release_name, phase),
+            method='PUT',
+            data=None,
+            headers=headers,
+        )

shipitscript/shipitscript/utils.py

@@ -66,3 +66,7 @@ def check_release_has_values_v2(release_api, release_name, headers, **kwargs):
             raise ScriptWorkerTaskException(err_msg)
 
     log.info("All release fields have been correctly updated in Ship-it!")
+
+
+def get_buildnum_from_version(next_version):
+    pass