- Create helpers.go

- Update provider.go
- Update resouce_node.go

Author: jfreed-dev

provider/helpers.go

@@ -0,0 +1,67 @@
+package provider
+
+import (
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"strings"
+	"time"
+)
+
+func checkPowerStatus(node int) string {
+	// Simulate checking power status
+	fmt.Printf("Checking power status for node %d\n", node)
+	// Replace this with an actual API call
+	return "off"
+}
+
+func turnOffNode(node int) {
+	fmt.Printf("Turning off node %d\n", node)
+	// Replace this with an API call to turn off the node
+}
+
+func turnOnNode(node int) {
+	fmt.Printf("Turning on node %d\n", node)
+	// Replace this with an API call to turn on the node
+}
+
+func flashNode(node int, firmware string) {
+	fmt.Printf("Flashing node %d with firmware %s\n", node, firmware)
+	// Replace this with an API call to flash the firmware
+}
+
+func checkBootStatus(node int, timeout int, token string) (bool, error) {
+	url := fmt.Sprintf("https://turingpi.local/api/bmc?opt=get&type=uart&node=%d", node)
+	client := &http.Client{}
+
+	deadline := time.Now().Add(time.Duration(timeout) * time.Second)
+
+	for time.Now().Before(deadline) {
+		req, err := http.NewRequest("GET", url, nil)
+		if err != nil {
+			return false, fmt.Errorf("failed to create UART request: %v", err)
+		}
+
+		req.Header.Set("Authorization", "Bearer "+token)
+		resp, err := client.Do(req)
+		if err != nil {
+			return false, fmt.Errorf("UART request failed: %v", err)
+		}
+
+		defer resp.Body.Close()
+		body, err := ioutil.ReadAll(resp.Body)
+		if err != nil {
+			return false, fmt.Errorf("failed to read UART response: %v", err)
+		}
+
+		// Simulate login prompt detection
+		if strings.Contains(string(body), "login:") {
+			fmt.Printf("Node %d booted successfully: login prompt detected.\n", node)
+			return true, nil
+		}
+
+		time.Sleep(5 * time.Second)
+	}
+
+	return false, fmt.Errorf("timeout reached: node %d did not boot successfully", node)
+}

provider/provider.go

@@ -21,7 +21,7 @@ func Provider() *schema.Provider {
 		ResourcesMap: map[string]*schema.Resource{
 			"turingpi_power": resourcePower(),
 			"turingpi_flash": resourceFlash(),
-			"turingpi_node":  resourceNode(),
+			"turingpi_node":  resourceNode(), // Add this resource
 		},
 	}
 }

provider/resource_node.go

@@ -2,7 +2,6 @@ package provider
 
 import (
 	"fmt"
-	"time"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 )
@@ -52,35 +51,32 @@ func resourceNodeProvision(d *schema.ResourceData, meta interface{}) error {
 	powerState := d.Get("power_state").(string)
 	bootCheck := d.Get("boot_check").(bool)
 	timeout := d.Get("login_prompt_timeout").(int)
+	token := meta.(string) // Assuming token is passed in metadata
 
-	// Step 1: Check and adjust power state
-	currentPower := checkPowerStatus(node)
-	if currentPower == "on" && powerState == "off" {
-		turnOffNode(node)
-	} else if currentPower == "off" && powerState == "on" {
+	// Step 1: Turn on the node
+	if powerState == "on" {
 		turnOnNode(node)
+	} else {
+		turnOffNode(node)
 	}
 
-	// Step 2: Flash firmware if needed
+	// Step 2: Flash firmware if provided
 	if firmware != "" {
-		fmt.Printf("Flashing node %d with firmware %s\n", node, firmware)
 		flashNode(node, firmware)
 	}
 
-	// Step 3: Boot the node
-	if powerState == "on" {
-		turnOnNode(node)
-	}
-
-	// Step 4: Optional boot check via UART
+	// Step 3: Boot check
 	if bootCheck {
-		fmt.Printf("Checking boot status for node %d\n", node)
-		if !waitForLoginPrompt(node, timeout) {
-			return fmt.Errorf("node %d failed to reach login prompt within %d seconds", node, timeout)
+		fmt.Printf("Checking boot status for node %d...\n", node)
+		success, err := checkBootStatus(node, timeout, token)
+		if err != nil {
+			return fmt.Errorf("boot status check failed for node %d: %v", node, err)
+		}
+		if !success {
+			return fmt.Errorf("node %d did not boot successfully", node)
 		}
 	}
 
-	// Set resource ID to uniquely identify the node
 	d.SetId(fmt.Sprintf("node-%d", node))
 	return nil
 }
@@ -95,39 +91,6 @@ func resourceNodeStatus(d *schema.ResourceData, meta interface{}) error {
 
 func resourceNodeDelete(d *schema.ResourceData, meta interface{}) error {
 	node := d.Get("node").(int)
-
-	// Turn off the node during resource deletion
 	turnOffNode(node)
 	return nil
 }
-
-// Helper functions
-func checkPowerStatus(node int) string {
-	// Simulate checking power status
-	fmt.Printf("Checking power status for node %d\n", node)
-	// Replace this with API call
-	return "off"
-}
-
-func turnOffNode(node int) {
-	fmt.Printf("Turning off node %d\n", node)
-	// Replace this with API call to turn off the node
-}
-
-func turnOnNode(node int) {
-	fmt.Printf("Turning on node %d\n", node)
-	// Replace this with API call to turn on the node
-}
-
-func flashNode(node int, firmware string) {
-	fmt.Printf("Flashing node %d with firmware %s\n", node, firmware)
-	// Replace this with API call to flash the firmware
-}
-
-func waitForLoginPrompt(node int, timeout int) bool {
-	fmt.Printf("Waiting for login prompt on node %d for up to %d seconds\n", node, timeout)
-	// Simulate waiting with a sleep (replace with UART API call)
-	time.Sleep(5 * time.Second)
-	// Simulate successful boot
-	return true
-}