dart_runner.go

package main

import (
	_ "embed"
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"

	"github.com/APTrust/dart-runner/constants"
	"github.com/APTrust/dart-runner/core"
	"github.com/APTrust/dart-runner/util"
)

//go:embed help.txt
var help string

// Version value is injected at build time.
var Version string

func main() {
	constants.AppVersion = Version
	exitCode := constants.ExitOK
	options := core.ParseOptions()
	if !options.AreValid() {
		ShowOptionsError()
		exitCode = constants.ExitUsageErr
	} else if options.ShowHelp {
		ShowHelp()
	} else if options.Version {
		ShowVersion()
	} else if len(options.StdinData) > 0 || core.StdinHasData() {
		exitCode = RunJob(options)
	} else {
		exitCode = RunWorkflow(options)
	}
	os.Exit(exitCode)
}

func RunJob(opts *core.Options) int {
	params, err := InitParams(opts)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error creating job: %s\n", err.Error())
		return constants.ExitRuntimeErr
	}
	return core.RunJob(params.ToJob(), opts.DeleteAfterUpload, true)
}

func RunWorkflow(opts *core.Options) int {
	runner, err := core.NewWorkflowRunner(
		opts.WorkflowFilePath,
		opts.BatchFilePath,
		opts.OutputDir,
		opts.DeleteAfterUpload,
		opts.Concurrency,
	)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Cannot start workflow: %s\n", err.Error())
		return constants.ExitRuntimeErr
	}
	return runner.Run()
}

func InitParams(opts *core.Options) (*core.JobParams, error) {
	if !util.FileExists(opts.OutputDir) {
		return nil, fmt.Errorf("Output directory '%s' does not exist. You must create it first.", opts.OutputDir)
	}
	workflow, err := core.WorkflowFromJson(opts.WorkflowFilePath)
	if err != nil {
		return nil, fmt.Errorf("Workflow JSON (%s): %s", opts.WorkflowFilePath, err.Error())
	}
	partialParams := &core.JobParams{}
	err = json.Unmarshal(opts.StdinData, partialParams)
	if err != nil {
		return nil, fmt.Errorf("JobParams JSON: %s", err.Error())
	}
	params := core.NewJobParams(
		workflow,
		partialParams.PackageName,
		filepath.Join(opts.OutputDir, partialParams.PackageName),
		partialParams.Files,
		partialParams.Tags)
	return params, nil
}

func ShowHelp() {
	fmt.Println(help)
}

func ShowVersion() {
	fmt.Println(Version)
}

func ShowOptionsError() {
	message := `
	Invalid arguments.

	Params --workflow and --output-dir are always required.
	For batch jobs, param --batch is also required.

	For more info: dart-runner --help
	`
	fmt.Fprintln(os.Stderr, message)
}