add Final Infrastructure Setup

scenario_generator/inventory.go

@@ -0,0 +1,83 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strings"
+)
+
+type HostConfig struct {
+	Name        string
+	IP          string
+	User        string
+	KeyPath     string
+	SSHArgs     string
+}
+
+type Inventory struct {
+	Router  HostConfig
+	Workers map[string]HostConfig
+}
+
+func ParseInventory(path string) (*Inventory, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, fmt.Errorf("could not open inventory: %w", err)
+	}
+	defer file.Close()
+
+	inv := &Inventory{
+		Workers: make(map[string]HostConfig),
+	}
+
+	var currentGroup string
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := strings.TrimSpace(scanner.Text())
+		if line == "" || strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") {
+			continue
+		}
+
+		if strings.HasPrefix(line, "[") && strings.HasSuffix(line, "]") {
+			currentGroup = strings.Trim(line, "[]")
+			continue
+		}
+
+		parts := strings.Fields(line)
+		if len(parts) < 2 {
+			continue
+		}
+
+		config := HostConfig{Name: parts[0]}
+		for _, part := range parts[1:] {
+			kv := strings.SplitN(part, "=", 2)
+			if len(kv) != 2 {
+				continue
+			}
+			val := strings.Trim(kv[1], "'\"")
+			switch kv[0] {
+			case "ansible_host":
+				config.IP = val
+			case "ansible_user":
+				config.User = val
+			case "ansible_ssh_private_key_file":
+				config.KeyPath = val
+			case "ansible_ssh_common_args":
+				config.SSHArgs = val
+			}
+		}
+
+		if currentGroup == "router" {
+			inv.Router = config
+		} else if currentGroup == "workers" {
+			inv.Workers[config.Name] = config
+		}
+	}
+
+	if err := scanner.Err(); err != nil {
+		return nil, err
+	}
+
+	return inv, nil
+}