main.go

package main

import (
	"context"
	"log"
	"os"
	"strings"

	"github.com/sierrasoftworks/vault-azfn/agent"
	"go.opentelemetry.io/otel"
)

var version = "0.0.0-dev"

func main() {
	if len(os.Args) < 2 {
		log.Fatal("Usage: vault-launcher <app> [args...]")
	}

	ctx := context.Background()
	tp := agent.NewTraceProvider(ctx, version)
	defer func() { _ = tp.Shutdown(ctx) }()

	otel.SetTracerProvider(tp)

	for i, arg := range os.Args[1:] {
		if strings.HasSuffix(arg, ".tpl") {
			f, err := os.Stat(arg)
			if err != nil {
				log.Println("Failed to render template file ", arg, ": ", err)
				continue
			}

			if f.IsDir() {
				log.Println("Failed to render template directory ", arg)
			}

			agent.ApplyTemplate(arg, arg[:len(arg)-len(".tpl")])
			os.Args[i+1] = arg[:len(arg)-len(".tpl")]
		}
	}

	if os.Getenv("VAULT_AGENT_SET_EXECUTABLE_PATTERN") != "" {
		agent.SetExecutablePattern(ctx, os.Getenv("VAULT_AGENT_SET_EXECUTABLE_PATTERN"))
	}

	binary := os.Args[1]
	args := os.Args[2:]

	err := agent.RunApp(ctx, binary, args)
	if err != nil {
		log.Fatal(err)
	}
}