Files
terraform-provider-virtfusion/internal/provider/virtfusion_server_build_resource.go
2023-10-22 18:51:18 -04:00

286 lines
8.2 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package provider
import (
"bytes"
"context"
"encoding/json"
"fmt"
"github.com/hashicorp/terraform-plugin-framework/resource/schema/booldefault"
"github.com/hashicorp/terraform-plugin-framework/types"
"io"
"io/ioutil"
"net/http"
"github.com/hashicorp/terraform-plugin-framework/path"
"github.com/hashicorp/terraform-plugin-framework/resource"
"github.com/hashicorp/terraform-plugin-framework/resource/schema"
)
// Ensure provider defined types fully satisfy framework interfaces.
var _ resource.Resource = &VirtfusionServerBuildResource{}
var _ resource.ResourceWithImportState = &VirtfusionServerBuildResource{}
func NewVirtfusionServerBuildResource() resource.Resource {
return &VirtfusionServerBuildResource{}
}
// VirtfusionServerBuildResource defines the resource implementation.
type VirtfusionServerBuildResource struct {
client *http.Client
}
type VirtfusionServerBuildResourceModel struct {
ServerId int64 `tfsdk:"server_id"`
Name string `tfsdk:"name" json:"name"`
Hostname string `tfsdk:"hostname" json:"hostname"`
Osid int64 `tfsdk:"osid" json:"operatingSystemId"`
Vnc bool `tfsdk:"vnc" json:"vnc"`
Ipv6 bool `tfsdk:"ipv6" json:"ipv6"`
SshKeys []int64 `tfsdk:"ssh_keys" json:"sshKeys"`
Email bool `tfsdk:"email" json:"email"`
}
func (r *VirtfusionServerBuildResource) Metadata(ctx context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
resp.TypeName = req.ProviderTypeName + "_build"
}
func (r *VirtfusionServerBuildResource) Schema(ctx context.Context, req resource.SchemaRequest, resp *resource.SchemaResponse) {
resp.Schema = schema.Schema{
// This description is used by the documentation generator and the language server.
MarkdownDescription: "Virtfusion Server Build Resource",
Attributes: map[string]schema.Attribute{
"server_id": schema.Int64Attribute{
MarkdownDescription: "Server ID",
Required: true,
},
"name": schema.StringAttribute{
MarkdownDescription: "Server Name",
Required: true,
},
"hostname": schema.StringAttribute{
MarkdownDescription: "Server Hostname",
Optional: true,
},
"osid": schema.Int64Attribute{
MarkdownDescription: "Server Operating System ID",
Required: true,
},
"vnc": schema.BoolAttribute{
MarkdownDescription: "Server VNC",
Optional: true,
Computed: true,
Default: booldefault.StaticBool(false),
},
"ipv6": schema.BoolAttribute{
MarkdownDescription: "Server IPv6",
Optional: true,
Computed: true,
Default: booldefault.StaticBool(false),
},
"ssh_keys": schema.ListAttribute{
MarkdownDescription: "Server SSH Keys IDs",
ElementType: types.Int64Type,
Optional: true,
},
"email": schema.BoolAttribute{
MarkdownDescription: "Server Email",
Optional: true,
Computed: true,
Default: booldefault.StaticBool(false),
},
},
}
}
func (r *VirtfusionServerBuildResource) Configure(ctx context.Context, req resource.ConfigureRequest, resp *resource.ConfigureResponse) {
// Prevent panic if the provider has not been configured.
if req.ProviderData == nil {
return
}
client, ok := req.ProviderData.(*http.Client)
if !ok {
resp.Diagnostics.AddError(
"Unexpected Resource Configure Type",
fmt.Sprintf("Expected *http.Client, got: %T. Please report this issue to the provider developers.", req.ProviderData),
)
return
}
r.client = client
}
func (r *VirtfusionServerBuildResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
var data VirtfusionServerBuildResourceModel
// Read Terraform plan data into the model
resp.Diagnostics.Append(req.Plan.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
createReq := VirtfusionServerBuildResourceModel{
Name: data.Name,
Hostname: data.Hostname,
Osid: data.Osid,
Vnc: data.Vnc,
Ipv6: data.Ipv6,
SshKeys: data.SshKeys,
Email: data.Email,
}
httpReqBody, err := json.Marshal(createReq)
if err != nil {
resp.Diagnostics.AddError(
"Unable to Create Resource",
"An unexpected error occurred while creating the resource create request. "+
"Please report this issue to the provider developers.\n\n"+
"JSON Error: "+err.Error(),
)
return
}
httpReq, err := http.NewRequest("POST", fmt.Sprintf("/servers/%d/build", data.ServerId), bytes.NewBuffer(httpReqBody))
if err != nil {
resp.Diagnostics.AddError(
"Failed to Create Request",
fmt.Sprintf("Failed to create a new HTTP request: %s", err.Error()),
)
return
}
// Add any additional headers (Content-Type, etc.)
httpReq.Header.Set("Content-Type", "application/json")
httpResponse, err := r.client.Do(httpReq)
if err != nil {
resp.Diagnostics.AddError(
"Failed to Execute Request",
fmt.Sprintf("Failed to execute HTTP request: %s", err.Error()),
)
return
}
defer func(Body io.ReadCloser) {
err := Body.Close()
if err != nil {
resp.Diagnostics.AddError(
"Failed to Close Request",
fmt.Sprintf("Failed to close HTTP request: %s", err.Error()),
)
return
}
}(httpResponse.Body)
if httpResponse.StatusCode == 422 {
responseBody, err := ioutil.ReadAll(httpResponse.Body)
if err != nil {
resp.Diagnostics.AddError(
"Failed to Read Response",
fmt.Sprintf("Failed to read HTTP response body: %s", err.Error()),
)
return
}
var errorResponse map[string]interface{}
err = json.Unmarshal(responseBody, &errorResponse)
if err != nil {
resp.Diagnostics.AddError(
"Failed to Parse Error Response",
fmt.Sprintf("Failed to parse HTTP response body: %s", err.Error()),
)
return
}
if errors, exists := errorResponse["errors"]; exists {
resp.Diagnostics.AddError(
"Server Returned Errors",
fmt.Sprintf("Errors from server: %v", errors),
)
}
return
}
if httpResponse.StatusCode != 200 {
resp.Diagnostics.AddError(
"Failed to Create Resource",
fmt.Sprintf("Failed to create resource: %s", httpResponse.Status),
)
return
}
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}
func (r *VirtfusionServerBuildResource) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) {
var data VirtfusionServerBuildResourceModel
// Read Terraform prior state data into the model
resp.Diagnostics.Append(req.State.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
// If applicable, this is a great opportunity to initialize any necessary
// provider client data and make a call using it.
// httpResp, err := r.client.Do(httpReq)
// if err != nil {
// resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to read example, got error: %s", err))
// return
// }
// Save updated data into Terraform state
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}
func (r *VirtfusionServerBuildResource) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) {
var data VirtfusionServerBuildResourceModel
// Read Terraform plan data into the model
resp.Diagnostics.Append(req.Plan.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
// If applicable, this is a great opportunity to initialize any necessary
// provider client data and make a call using it.
// httpResp, err := r.client.Do(httpReq)
// if err != nil {
// resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to update example, got error: %s", err))
// return
// }
// Save updated data into Terraform state
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}
func (r *VirtfusionServerBuildResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
var data VirtfusionServerBuildResourceModel
// Read Terraform prior state data into the model
resp.Diagnostics.Append(req.State.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}
func (r *VirtfusionServerBuildResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
resource.ImportStatePassthroughID(ctx, path.Root("id"), req, resp)
}