package scan

import (
	"context"
	"fmt"
	"time"

	"git.tcp.direct/perp/shogo/internal/utils"
)

// Submit an internet scan
func Internet(port int, protocol string) {
	// Get results
	results, err := utils.Client.ScanInternet(context.Background(), port, protocol)
	if err != nil {
		fmt.Printf("%s: %s\n", utils.Red("Error"), err.Error())
		return
	}

	// Print results
	utils.Title.Printf("Scanning internet for %d %s\n", port, protocol)

	fmt.Printf("%s: %s\n", utils.Green("ID"), utils.Blue(results))

	for {
		time.Sleep(time.Second * 3)

		// Get results
		results, err := utils.Client.GetScanStatus(context.Background(), results)
		if err != nil {
			fmt.Printf("%s: %s\n", utils.Red("Error"), err.Error())
			return
		}

		// Get status
		switch results.Status {
		case "DONE":
			fmt.Printf("%s: %s\n", utils.Blue("Status"), utils.Green(results.Status))
			return
		}
	}
}