Rewrote generators to use plugin system with default plugins

internal/client.go

@@ -6,49 +6,110 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+
+	"github.com/OpenCHAMI/configurator/internal/util"
 )
 
 type SmdClient struct {
 	http.Client
-	Host        string
-	Port        int
-	AccessToken string
+	Host        string `yaml:"host"`
+	Port        int    `yaml:"port"`
+	AccessToken string `yaml:"access-token"`
 }
 
-func (client *SmdClient) FetchDNS(config *Config) error {
-	// fetch DNS related information from SMD's endpoint:
-	return nil
+type Params = map[string]any
+type Option func(Params)
+
+func WithVerbose() Option {
+	return func(p util.Params) {
+		p["verbose"] = true
+	}
 }
 
-func (client *SmdClient) FetchEthernetInterfaces() ([]EthernetInterface, error) {
+func NewParams() Params {
+	return Params{
+		"verbose": false,
+	}
+}
+
+// Fetch the ethernet interfaces from SMD service using its API. An access token may be required if the SMD
+// service SMD_JWKS_URL envirnoment variable is set.
+func (client *SmdClient) FetchEthernetInterfaces(opts ...util.Option) ([]EthernetInterface, error) {
+	// make request to SMD endpoint
+	b, err := client.makeRequest("/Inventory/EthernetInterfaces")
+	if err != nil {
+		return nil, fmt.Errorf("failed to read HTTP response: %v", err)
+	}
+
+	// unmarshal response body JSON and extract in object
+	eths := []EthernetInterface{} // []map[string]any{}
+	err = json.Unmarshal(b, &eths)
+	if err != nil {
+		return nil, fmt.Errorf("failed to unmarshal response: %v", err)
+	}
+
+	// print what we got if verbose is set
+	params := util.GetParams(opts...)
+	if verbose, ok := params["verbose"].(bool); ok {
+		if verbose {
+			fmt.Printf("Ethernet Interfaces: %v\n", string(b))
+		}
+	}
+
+	return eths, nil
+}
+
+// Fetch the components from SMD using its API. An access token may be required if the SMD
+// service SMD_JWKS_URL envirnoment variable is set.
+func (client *SmdClient) FetchComponents(opts ...util.Option) ([]Component, error) {
+	// make request to SMD endpoint
+	b, err := client.makeRequest("/State/Components")
+	if err != nil {
+		return nil, fmt.Errorf("failed to read HTTP response: %v", err)
+	}
+
+	// unmarshal response body JSON and extract in object
+	comps := []Component{}
+	err = json.Unmarshal(b, &comps)
+	if err != nil {
+		return nil, fmt.Errorf("failed to unmarshal response: %v", err)
+	}
+
+	// print what we got if verbose is set
+	params := util.GetParams(opts...)
+	if verbose, ok := params["verbose"].(bool); ok {
+		if verbose {
+			fmt.Printf("Components: %v\n", string(b))
+		}
+	}
+
+	return comps, nil
+}
+
+func (client *SmdClient) makeRequest(endpoint string) ([]byte, error) {
 	if client == nil {
 		return nil, fmt.Errorf("client is nil")
 	}
-	// fetch DHCP related information from SMD's endpoint:
-	url := fmt.Sprintf("%s:%d/hsm/v2/Inventory/EthernetInterfaces", client.Host, client.Port)
-	req, err := http.NewRequest(http.MethodGet, url, bytes.NewBuffer([]byte{}))
 
-	// include access token in authorzation header if found
-	if client.AccessToken != "" {
-		req.Header.Add("Authorization", "Bearer "+client.AccessToken)
-	}
+	// fetch DHCP related information from SMD's endpoint:
+	url := fmt.Sprintf("%s:%d/hsm/v2%s", client.Host, client.Port, endpoint)
+	req, err := http.NewRequest(http.MethodGet, url, bytes.NewBuffer([]byte{}))
 	if err != nil {
 		return nil, fmt.Errorf("failed to create new HTTP request: %v", err)
 	}
+
+	// include access token in authorzation header if found
+	// NOTE: This shouldn't be needed for this endpoint since it's public
+	if client.AccessToken != "" {
+		req.Header.Add("Authorization", "Bearer "+client.AccessToken)
+	}
+
+	// make the request to SMD
 	res, err := client.Do(req)
 	if err != nil {
 		return nil, fmt.Errorf("failed to make request: %v", err)
 	}
 
-	b, err := io.ReadAll(res.Body)
-	if err != nil {
-		return nil, fmt.Errorf("failed to read HTTP response: %v", err)
-	}
-
-	// unmarshal JSON and extract
-	eths := []EthernetInterface{} // []map[string]any{}
-	json.Unmarshal(b, &eths)
-	fmt.Printf("ethernet interfaces: %v\n", string(b))
-
-	return eths, nil
+	// read the contents of the response body
+	return io.ReadAll(res.Body)
 }