main.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"os"

	"github.com/aws/aws-lambda-go/lambda"
)

// handler is the AWS Lambda entry point (invoked on a schedule by EventBridge).
func handler(ctx context.Context) (map[string]any, error) {
	cfg, err := loadConfig(ctx)
	if err != nil {
		return nil, err
	}
	results, runErr := runAll(ctx, cfg)
	return map[string]any{"results": results}, runErr
}

func main() {
	// On Lambda this env var is always set: run the runtime loop.
	if os.Getenv("AWS_LAMBDA_RUNTIME_API") != "" {
		lambda.Start(handler)
		return
	}

	// Local / CLI mode: run once, print JSON, exit non-zero on error.
	ctx := context.Background()
	cfg, err := loadConfig(ctx)
	if err != nil {
		log.Fatal(err)
	}
	results, runErr := runAll(ctx, cfg)
	out, _ := json.MarshalIndent(map[string]any{"results": results}, "", "  ")
	fmt.Println(string(out))
	if runErr != nil {
		os.Exit(1)
	}
}