main.go

package main

import (
	"flag"
	"fmt"
	"os"
)

var (
	timeout  int
	services Services
	quiet    bool
	strict   bool
)

func init() {
	flag.IntVar(&timeout, "t", 15, "Timeout in seconds, zero for no timeout")
	flag.BoolVar(&quiet, "q", false, "Quiet, don't output any status messages")
	flag.BoolVar(&strict, "s", false, "Only execute subcommand if the test succeeds")
	flag.Var(&services, "w", "Services to be waiting for, in the form `host:port`")
}

// Log is used to log with prefix wait-for-it:
func Log(message string) {
	if quiet {
		return
	}

	fmt.Println("wait-for-it: " + message)
}

func main() {
	flag.Parse()
	args := os.Args

	if len(services) != 0 {
		Log(fmt.Sprintf("waiting %d seconds for %s", timeout, services.String()))
		ok := services.Wait(timeout)

		if !ok {
			Log(fmt.Sprintf("timeout occured after waiting for %d seconds", timeout))
			if strict {
				Log("strict mode, refusing to execute subprocess")
				os.Exit(1)
			}
		}
	}

	var command []string

	for i, arg := range args {
		if arg == "--" {
			if (i + 1) < len(args) {
				command = args[i+1:]
			}

			break
		}
	}

	if len(command) == 0 {
		return
	}

	os.Exit(Execute(command))
}