Update token

Signed-off-by: Tamal Saha <tamal@appscode.com>

Author: tamalsaha

main.go

@@ -17,20 +17,21 @@ limitations under the License.
 package main
 
 import (
-	"os"
-
 	"github.com/appscodelabs/gh-ci-webhook/pkg/cmds"
 	_ "github.com/appscodelabs/gh-ci-webhook/pkg/providers/registry"
 
 	"gomodules.xyz/logs"
 )
 
 func main() {
+	if err := realMain(); err != nil {
+		panic(err)
+	}
+}
+
+func realMain() error {
 	logs.InitLogs()
 	defer logs.FlushLogs()
 
-	if err := cmds.NewRootCmd().Execute(); err != nil {
-		os.Exit(1)
-	}
-	os.Exit(0)
+	return cmds.NewRootCmd().Execute()
 }

pkg/backend/stream.go

@@ -121,7 +121,12 @@ func (mgr *Manager) onerun(subj string) error {
 	if err != nil {
 		return err
 	}
-	defer mgr.streamCompleted.DeleteConsumer(ctx, cons.CachedInfo().Name)
+	defer func() {
+		err = mgr.streamCompleted.DeleteConsumer(ctx, cons.CachedInfo().Name)
+		if err != nil {
+			klog.Errorln(err)
+		}
+	}()
 
 	// FetchNoWait will not wait for new messages if the whole batch is not available at the time of sending request.
 	msgs, err := cons.FetchNoWait(mgr.numWorkers)
@@ -134,7 +139,10 @@ func (mgr *Manager) onerun(subj string) error {
 			if err != nil {
 				klog.Errorln(err)
 			}
-			msg.DoubleAck(context.TODO())
+			err = msg.DoubleAck(context.TODO())
+			if err != nil {
+				klog.Errorln(err)
+			}
 
 			if slot, found := mgr.Provider.Next(); found {
 				err := mgr.Provider.StartRunner(slot) // Not 1-1 mapping for the VM shut down to restarted
@@ -188,7 +196,10 @@ func (mgr *Manager) ProcessCompletedJobs() error {
 				if err != nil {
 					klog.Errorln(err)
 				}
-				msg.DoubleAck(context.TODO())
+				err = msg.DoubleAck(context.TODO())
+				if err != nil {
+					klog.Errorln(err)
+				}
 
 				if slot, found := mgr.Provider.Next(); found {
 					err := mgr.Provider.StartRunner(slot) // Not 1-1 mapping for the VM shut down to restarted

pkg/cmds/wait_for_job.go

@@ -141,7 +141,12 @@ func consumeMsg(ctx context.Context, streamQueued jetstream.Stream, subj string)
 	if err != nil {
 		return nil, err
 	}
-	defer streamQueued.DeleteConsumer(ctx, cons.CachedInfo().Name)
+	defer func() {
+		err = streamQueued.DeleteConsumer(ctx, cons.CachedInfo().Name)
+		if err != nil {
+			klog.Errorln(err)
+		}
+	}()
 
 	/*
 			Double-acking is a mechanism used in JetStream to ensure exactly once semantics in message processing.

pkg/providers/firecracker/cloud_init.go

@@ -178,6 +178,12 @@ func BuildData(ghToken string, instanceID int, ghUsernames ...string) (*MMDSConf
 	//echo "Created by bash shell script" >> /test-userscript/userscript.txt
 	//`
 
+	encDockerHubToken := "ZGNrcl9wYXRfTGhOV3VDdTNYeml0cWd5bmxrSE9wM1VoQVRnCg=="
+	dockerHubToken, err := base64.StdEncoding.DecodeString(encDockerHubToken)
+	if err != nil {
+		return nil, err
+	}
+
 	script := fmt.Sprintf(`#! /bin/bash
 set -x
 
@@ -218,7 +224,7 @@ echo 127.0.1.1 $HOSTNAME.localdomain ${RUNNER_NAME} >> /etc/hosts
 chmod a+w /usr/local/bin
 
 # bypass docker hub rate limits
-docker login -u tigerworks -p dckr_pat_TQSHB3Z8CoNU8G4jtW7xXOxMefM
+docker login -u tigerworks -p %s
 
 # Prepare GitHun Runner user
 export USER=runner
@@ -246,7 +252,7 @@ export RUNNER_NAME=%s
 # ephemeral runner: https://docs.github.com/en/actions/hosting-your-own-runners/autoscaling-with-self-hosted-runners#using-ephemeral-runners-for-autoscaling
 # https://github.blog/changelog/2021-09-20-github-actions-ephemeral-self-hosted-runners-new-webhooks-for-auto-scaling/
 curl -fsSL https://gist.githubusercontent.com/tamalsaha/af2f99c80f84410253bd1e532bdfabc7/raw/a3fa2a82ae14d406fd934b90f2be7dc2945607a8/start-runner.sh | bash -s -- -n ${RUNNER_NAME} -f
-`, DefaultOptions.NatsURL, DefaultOptions.NatsUsername, DefaultOptions.NatsPassword, ghToken, runnerName)
+`, string(dockerHubToken), DefaultOptions.NatsURL, DefaultOptions.NatsUsername, DefaultOptions.NatsPassword, ghToken, runnerName)
 
 	udBytes, err := PrepareCloudInitUserData(userData, script)
 	if err != nil {

pkg/providers/linode/linode.go

@@ -165,7 +165,7 @@ func createInstance(c *linodego.Client, machineName string) (int, error) {
 func waitForStatus(c *linodego.Client, id int, status linodego.InstanceStatus) error {
 	attempt := 0
 	klog.Infoln("waiting for instance status", "status", status)
-	return wait.PollImmediate(RetryInterval, RetryTimeout, func() (bool, error) {
+	return wait.PollUntilContextTimeout(context.TODO(), RetryInterval, RetryTimeout, true, func(ctx context.Context) (bool, error) {
 		attempt++
 
 		instance, err := c.GetInstance(context.Background(), id)