updated to use the new HealthCheck engine API endpoint

Author: caffix

cmd/ae_isready/main.go

@@ -5,14 +5,12 @@
 package main
 
 import (
-	"context"
 	"flag"
 	"fmt"
 	"os"
 	"path"
-	"time"
 
-	amasshttp "github.com/owasp-amass/amass/v5/internal/net/http"
+	"github.com/owasp-amass/amass/v5/engine/api/client"
 )
 
 func main() {
@@ -27,24 +25,18 @@ func main() {
 	flag.BoolVar(&help1, "h", false, "Show the program usage message")
 	flag.BoolVar(&help2, "help", false, "Show the program usage message")
 	flag.StringVar(&hostname, "host", "", "Hostname or IP address of the Amass Engine")
-	//flag.BoolVar(&version, "version", false, "Print the version number of this Amass binary")
 	flag.Parse()
 
 	if (help1 || help2) || hostname == "" {
 		fmt.Fprintf(os.Stderr, "Usage: %s %s\n\n", path.Base(os.Args[0]), "--host HOSTNAME")
 		flag.PrintDefaults()
 		return
 	}
-	/*if version {
-		fmt.Fprintf(color.Error, "%s\n", format.Version)
-		return
-	}*/
 
-	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
-	defer cancel()
+	c := client.NewClient("http://" + hostname + ":4000")
+	defer c.Close()
 
-	u := "http://" + hostname + ":4000"
-	if _, err := amasshttp.RequestWebPage(ctx, &amasshttp.Request{URL: u}); err != nil {
+	if !c.HealthCheck() {
 		// a failure to respond indicates that the server is not yet available
 		os.Exit(1)
 	}