Function addAllowedToNetwork

pkg/cli/codemod_mcp_network.go:365–418

addAllowedToNetwork adds an allowed field to an existing network block

(lines []string, domains []string)

// addAllowedToNetwork adds an allowed field to an existing network block
func addAllowedToNetwork(lines []string, domains []string) []string {
	var result []string
	var inNetworkBlock bool
	var networkIndent string
	var insertIndex = -1

	for i, line := range lines {
		trimmedLine := strings.TrimSpace(line)

		if strings.HasPrefix(trimmedLine, "network:") {
			inNetworkBlock = true
			networkIndent = getIndentation(line)
		}

		if inNetworkBlock && len(trimmedLine) > 0 && !strings.HasPrefix(trimmedLine, "#") {
			if hasExitedBlock(line, networkIndent) {
				// Found the end of network block
				insertIndex = i
				break
			}
		}

		result = append(result, line)
	}

	if insertIndex > 0 {
		// Insert allowed before the next top-level block
		allowedLines := []string{
			networkIndent + "  allowed:",
		}
		for _, domain := range domains {
			allowedLines = append(allowedLines, fmt.Sprintf("%s    - %s", networkIndent, domain))
		}

		result = append(result, allowedLines...)
		result = append(result, lines[insertIndex:]...)
	} else {
		// Append at the end of network block
		networkIndentStr := ""
		for i := range slices.Backward(result) {
			trimmed := strings.TrimSpace(result[i])
			if strings.HasPrefix(trimmed, "network:") {
				networkIndentStr = getIndentation(result[i])
				break
			}
		}
		result = append(result, networkIndentStr+"  allowed:")
		for _, domain := range domains {
			result = append(result, fmt.Sprintf("%s    - %s", networkIndentStr, domain))
		}
	}

	return result
}