mirror of
https://github.com/letic/terraform-provider-proxmox.git
synced 2024-10-05 02:01:04 +00:00
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package proxmox
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
pxapi "github.com/Telmate/proxmox-api-go/proxmox"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
)
|
|
|
|
func Provisioner() terraform.ResourceProvisioner {
|
|
return &schema.Provisioner{
|
|
Schema: map[string]*schema.Schema{
|
|
"action": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
},
|
|
"net1": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
},
|
|
},
|
|
|
|
ApplyFunc: applyFn,
|
|
}
|
|
}
|
|
|
|
var currentClient *pxapi.Client = nil
|
|
|
|
func applyFn(ctx context.Context) error {
|
|
data := ctx.Value(schema.ProvConfigDataKey).(*schema.ResourceData)
|
|
state := ctx.Value(schema.ProvRawStateKey).(*terraform.InstanceState)
|
|
|
|
connInfo := state.Ephemeral.ConnInfo
|
|
|
|
act := data.Get("action").(string)
|
|
targetNode, _, vmId, err := parseResourceId(state.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
vmr := pxapi.NewVmRef(vmId)
|
|
vmr.SetNode(targetNode)
|
|
client := currentClient
|
|
if client == nil {
|
|
client, err = getClient(connInfo["pm_api_url"], connInfo["pm_user"], connInfo["pm_password"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
currentClient = client
|
|
}
|
|
switch act {
|
|
case "sshbackward":
|
|
return pxapi.RemoveSshForwardUsernet(vmr, client)
|
|
default:
|
|
return fmt.Errorf("Unkown action: %s", act)
|
|
}
|
|
return nil
|
|
}
|