CML sugar 🍬 (#339)

* CML sugar

* temp remove rerun

* nil

* temp remove rerun

* temp remove env

* Constrain permissions for tpi-task-shutdown

* cosmetic

* add TPI_TASK

* nitpicks

* Restyled by gofmt

* notpicks

* no os

Co-authored-by: Helio Machado <0x2b3bfa0+git@googlemail.com>
Co-authored-by: Restyled.io <commits@restyled.io>

Author: 3 people

iterative/resource_task.go

@@ -236,6 +236,15 @@ func resourceTaskBuild(ctx context.Context, d *schema.ResourceData, m interface{
 		}
 	}
 
+	val := "true"
+	v["TPI_TASK"] = &val
+	v["CI"] = nil
+	v["CI_*"] = nil
+	v["GITHUB_*"] = nil
+	v["BITBUCKET_*"] = nil
+	v["CML_*"] = nil
+	v["REPO_TOKEN"] = nil
+
 	c := common.Cloud{
 		Provider: common.Provider(d.Get("cloud").(string)),
 		Region:   common.Region(d.Get("region").(string)),

task/common/machine/script.go

@@ -32,6 +32,15 @@ base64 --decode << END | sudo tee /usr/bin/tpi-task > /dev/null
 END
 chmod u=rwx,g=rx,a=rx /usr/bin/tpi-task
 
+sudo tee /usr/bin/tpi-task-shutdown << 'END'
+#!/bin/bash
+if ! test -z "$CI"; then
+  cml rerun-workflow
+fi
+(systemctl is-system-running | grep stopping) || tpi --stop;
+END
+chmod u=rwx,g=rx,o=rx /usr/bin/tpi-task-shutdown
+
 base64 --decode << END | sudo tee /tmp/tpi-environment > /dev/null
 %s
 END
@@ -47,7 +56,7 @@ sudo tee /etc/systemd/system/tpi-task.service > /dev/null <<END
 [Service]
   Type=simple
   ExecStart=/usr/bin/tpi-task
-  ExecStopPost=/bin/bash -c 'systemctl is-system-running | grep stopping || tpi --stop'
+  ExecStopPost=/usr/bin/tpi-task-shutdown
   Environment=HOME=/root
   EnvironmentFile=/tmp/tpi-environment
   WorkingDirectory=/tmp/tpi-task
@@ -61,6 +70,10 @@ sudo mv terraform-provider-iterative* /usr/bin/tpi
 sudo chmod u=rwx,g=rx,o=rx /usr/bin/tpi
 sudo chown root:root /usr/bin/tpi
 
+curl --location --remote-name https://github.com/iterative/cml/releases/latest/download/cml-linux
+chmod u=rwx,g=rx,o=rx cml-linux
+sudo mv cml-linux /usr/bin/cml
+
 extract_here(){
   if command -v unzip 2>&1 > /dev/null; then
     unzip "$1"