Add information from active machine with info command

Author: qu35t-code

cmd/info.go

@@ -8,6 +8,7 @@ import (
 	"text/tabwriter"
 	"time"
 
+	"github.com/AlecAivazis/survey/v2"
 	"github.com/GoToolSharing/htb-cli/utils"
 	"github.com/kyokomi/emoji/v2"
 	"github.com/spf13/cobra"
@@ -16,6 +17,47 @@ import (
 var machineParam []string
 var challengeParam []string
 
+func checkActiveMachine() {
+	machine_id := utils.GetActiveMachineID(proxyParam)
+	status := "Not defined"
+	retired_status := "Not defined"
+	if machine_id != "" {
+		log.Println("Active machine found !")
+		log.Println("Machine ID:", machine_id)
+		w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', tabwriter.Debug)
+		fmt.Fprintln(w, "Name\tOS\tActive\tDifficulty\tStars\tIP\tStatus\tRelease")
+		url := "https://www.hackthebox.com/api/v4/machine/profile/" + machine_id
+		resp, err := utils.HtbRequest(http.MethodGet, url, proxyParam, nil)
+		if err != nil {
+			log.Fatal(err)
+		}
+		info := utils.ParseJsonMessage(resp, "info")
+		data := info.(map[string]interface{})
+		if data["authUserInUserOwns"] == nil && data["authUserInRootOwns"] == nil {
+			status = emoji.Sprint(":x:User - :x:Root")
+		} else if data["authUserInUserOwns"] == true && data["authUserInRootOwns"] == nil {
+			status = emoji.Sprint(":white_check_mark:User - :x:Root")
+		} else if data["authUserInUserOwns"] == nil && data["authUserInRootOwns"] == true {
+			status = emoji.Sprint(":x:User - :white_check_mark:Root")
+		} else if data["authUserInUserOwns"] == true && data["authUserInRootOwns"] == true {
+			status = emoji.Sprint(":white_check_mark:User - :white_check_mark:Root")
+		}
+		if data["retired"].(float64) == 0 {
+			retired_status = emoji.Sprint(":white_check_mark:")
+		} else {
+			retired_status = emoji.Sprint(":x:")
+		}
+		t, err := time.Parse(time.RFC3339Nano, data["release"].(string))
+		if err != nil {
+			fmt.Println("Erreur when date parsing :", err)
+			return
+		}
+		datetime := t.Format("2006-01-02")
+		fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\n", data["name"], data["os"], retired_status, data["difficultyText"], data["stars"], data["ip"], status, datetime)
+		w.Flush()
+	}
+}
+
 var infoCmd = &cobra.Command{
 	Use:   "info",
 	Short: "Showcase detailed machine information",
@@ -26,12 +68,24 @@ var infoCmd = &cobra.Command{
 			cmd.Help()
 			os.Exit(1)
 		}
+		var confirmation bool
+		confirmation_message := "Do you want to check for active machine ?"
+		prompt := &survey.Confirm{
+			Message: confirmation_message,
+		}
+		if err := survey.AskOne(prompt, &confirmation); err != nil {
+			log.Fatal(err)
+		}
+		if confirmation {
+			checkActiveMachine()
+		}
 
 		// Machines search
 		if len(machineParam) > 0 {
 			w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', tabwriter.Debug)
 			fmt.Fprintln(w, "Name\tOS\tActive\tDifficulty\tStars\tFirstUserBlood\tFirstRootBlood\tStatus\tRelease")
 			status := "Not defined"
+			retired_status := "Not defined"
 			log.Println(machineParam)
 			for index, _ := range machineParam {
 				machine_id := utils.SearchItemIDByName(machineParam[index], proxyParam, "Machine")
@@ -53,15 +107,20 @@ var infoCmd = &cobra.Command{
 				} else if data["authUserInUserOwns"] == true && data["authUserInRootOwns"] == true {
 					status = emoji.Sprint(":white_check_mark:User - :white_check_mark:Root")
 				}
+				if data["retired"].(float64) == 0 {
+					retired_status = emoji.Sprint(":white_check_mark:")
+				} else {
+					retired_status = emoji.Sprint(":x:")
+				}
 				t, err := time.Parse(time.RFC3339Nano, data["release"].(string))
 				if err != nil {
 					fmt.Println("Erreur when date parsing :", err)
 					return
 				}
 				datetime := t.Format("2006-01-02")
-				fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\n", data["name"], data["os"], data["active"], data["difficultyText"], data["stars"], data["firstUserBloodTime"], data["firstRootBloodTime"], status, datetime)
+				fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\n", data["name"], data["os"], retired_status, data["difficultyText"], data["stars"], data["firstUserBloodTime"], data["firstRootBloodTime"], status, datetime)
+				w.Flush()
 			}
-			w.Flush()
 		}
 
 		// Challenges search
@@ -98,8 +157,8 @@ var infoCmd = &cobra.Command{
 				}
 				datetime := t.Format("2006-01-02")
 				fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\n", data["name"], data["category_name"], retired_status, data["difficulty"], data["stars"], data["solves"], status, datetime)
+				w.Flush()
 			}
-			w.Flush()
 		}
 	},
 }

cmd/stop.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"os"
 
 	"github.com/GoToolSharing/htb-cli/utils"
 	"github.com/spf13/cobra"
@@ -14,6 +15,10 @@ var stopCmd = &cobra.Command{
 	Short: "Stop the current machine",
 	Run: func(cmd *cobra.Command, args []string) {
 		machine_id := utils.GetActiveMachineID(proxyParam)
+		if machine_id == "" {
+			fmt.Println("No machine is running")
+			os.Exit(0)
+		}
 		log.Println("Machine ID :", machine_id)
 		machine_type := utils.GetMachineType(machine_id, "")
 		log.Println("Machine Type :", machine_type)

utils/utils.go

@@ -207,8 +207,7 @@ func GetActiveMachineID(proxyURL string) string {
 	}
 	info := ParseJsonMessage(resp, "info")
 	if info == nil {
-		fmt.Println("No machine is running")
-		os.Exit(0)
+		return ""
 	}
 	return fmt.Sprintf("%.0f", info.(map[string]interface{})["id"].(float64))
 }