main.go

// Copyright Josh Komoroske. All rights reserved.
// Use of this source code is governed by the MIT license,
// a copy of which can be found in the LICENSE.txt file.
// SPDX-License-Identifier: MIT

// Package main defines the entrypoint for the actions-docker-shim tool.
package main

import (
	"fmt"
	"os"
	"strings"

	"github.com/joshdk/actions-docker-shim/docker"
	flag "github.com/spf13/pflag"
)

func main() {
	if err := mainCmd(); err != nil {
		fmt.Fprintln(os.Stderr, "actions-docker-shim:", err)
		os.Exit(1)
	}
}

//nolint:forbidigo,wsl
func mainCmd() error {
	var imageRepo string
	flag.StringVar(&imageRepo, "shim-image", "", "ghcr.io image to run")
	var imageTag string
	flag.StringVar(&imageTag, "shim-image-tag", "", "ghcr.io image tag to run")
	var tokenEnv string
	flag.StringVar(&tokenEnv, "shim-token-env", "", "env var to use for github token")
	flag.Parse()

	if imageRepo == "" {
		imageRepo = os.Getenv("GITHUB_ACTION_REPOSITORY")
	}
	if imageTag == "" {
		imageTag = os.Getenv("GITHUB_ACTION_REF")
	}
	image := fmt.Sprintf("ghcr.io/%s:%s", strings.ToLower(imageRepo), imageTag)

	var token string
	if tokenEnv != "" {
		// Environment variable named by the token-env flag.
		token = os.Getenv(tokenEnv)
	} else if value := os.Getenv("GITHUB_TOKEN"); value != "" {
		// Environment variable named "GITHUB_TOKEN".
		token = value
	} else if value := os.Getenv("INPUT_GITHUB-TOKEN"); value != "" {
		// Input named "github-token".
		token = value
	} else if value := os.Getenv("INPUT_TOKEN"); value != "" {
		// Input named "token".
		token = value
	}

	fmt.Printf("::group::%s\n", "Docker login")
	err := docker.Login(token)
	fmt.Println("::endgroup::")
	if err != nil {
		return err
	}

	fmt.Printf("::group::%s\n", "Docker pull")
	err = docker.Pull(image)
	fmt.Println("::endgroup::")
	if err != nil {
		return err
	}

	return docker.Run(image, flag.Args())
}