Added update to pull the latest BHCE images

Author: chrismaddalena

CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+### [0.1.4] - 2025-1-31
+
+### Added
+
+* Added an `update` command to pull the latest BloodHound images
+
 ## [0.1.3] - 2025-1-31
 
 ### Added

cmd/internal/docker.go

@@ -232,6 +232,16 @@ func RunDockerComposeDown(yaml string, volumes bool) {
 	}
 }
 
+// RunDockerComposePull executes the "docker compose" commands to pull the latest container images for
+// the specified YAML file ("yaml" parameter).
+func RunDockerComposePull(yaml string) {
+	fmt.Printf("[+] Running `%s` to pull container imahes with %s...\n", dockerCmd, yaml)
+	startErr := RunCmd(dockerCmd, []string{"-f", yaml, "pull"})
+	if startErr != nil {
+		log.Fatalf("Error trying to pull the container images with %s: %v\n", yaml, startErr)
+	}
+}
+
 // FetchLogs fetches logs from the container with the specified "name" label ("containerName" parameter).
 func FetchLogs(containerName string, lines string) []string {
 	var logs []string

cmd/update.go

@@ -0,0 +1,28 @@
+package cmd
+
+import (
+	"fmt"
+	docker "github.com/SpecterOps/BloodHound_CLI/cmd/internal"
+	"github.com/spf13/cobra"
+)
+
+// updateCmd represents the update command
+var updateCmd = &cobra.Command{
+	Use:   "update",
+	Short: "Update the BloodHound container images if an update is available",
+	Long:  `Updates the BloodHound container images if an update is available.`,
+	Run:   updateBloodHound,
+}
+
+func init() {
+	rootCmd.AddCommand(updateCmd)
+}
+
+func updateBloodHound(cmd *cobra.Command, args []string) {
+	err := docker.EvaluateDockerComposeStatus()
+	if err != nil {
+		return
+	}
+	fmt.Println("[+] Checking for BloodHound image updates...")
+	docker.RunDockerComposePull("docker-compose.yml")
+}