package main

import (
	"flag"
	"fmt"
	"log"
	"time"

	"github.com/bfirsh/funker-go"
	"github.com/docker/distribution/reference"
	"github.com/docker/docker/hack/integration-cli-on-swarm/agent/types"
)

func main() {
	if err := xmain(); err != nil {
		log.Fatalf("fatal error: %v", err)
	}
}

func validImageDigest(s string) bool {
	return reference.DigestRegexp.FindString(s) != ""
}

func xmain() error {
	workerImageDigest := flag.String("worker-image-digest", "", "Needs to be the digest of this worker image itself")
	dryRun := flag.Bool("dry-run", false, "Dry run")
	keepExecutor := flag.Bool("keep-executor", false, "Do not auto-remove executor containers, which is used for running privileged programs on Swarm")
	flag.Parse()
	if !validImageDigest(*workerImageDigest) {
		// Because of issue #29582.
		// `docker service create localregistry.example.com/blahblah:latest` pulls the image data to local, but not a tag.
		// So, `docker run localregistry.example.com/blahblah:latest` fails: `Unable to find image 'localregistry.example.com/blahblah:latest' locally`
		return fmt.Errorf("worker-image-digest must be a digest, got %q", *workerImageDigest)
	}
	executor := privilegedTestChunkExecutor(!*keepExecutor)
	if *dryRun {
		executor = dryTestChunkExecutor()
	}
	return handle(*workerImageDigest, executor)
}

func handle(workerImageDigest string, executor testChunkExecutor) error {
	log.Printf("Waiting for a funker request")
	return funker.Handle(func(args *types.Args) types.Result {
		log.Printf("Executing chunk %d, contains %d test filters",
			args.ChunkID, len(args.Tests))
		begin := time.Now()
		code, rawLog, err := executor(workerImageDigest, args.Tests)
		if err != nil {
			log.Printf("Error while executing chunk %d: %v", args.ChunkID, err)
			if code == 0 {
				// Make sure this is a failure
				code = 1
			}
			return types.Result{
				ChunkID: args.ChunkID,
				Code:    int(code),
				RawLog:  rawLog,
			}
		}
		elapsed := time.Since(begin)
		log.Printf("Finished chunk %d, code=%d, elapsed=%v", args.ChunkID, code, elapsed)
		return types.Result{
			ChunkID: args.ChunkID,
			Code:    int(code),
			RawLog:  rawLog,
		}
	})
}