74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package proxmox
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
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"], connInfo["pm_tls_insecure"] == "true")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
currentClient = client
|
|
}
|
|
switch act {
|
|
case "sshbackward":
|
|
return pxapi.RemoveSshForwardUsernet(vmr, client)
|
|
|
|
case "reconnect":
|
|
err = pxapi.RemoveSshForwardUsernet(vmr, client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
time.Sleep(10 * time.Second)
|
|
vmParams := map[string]string{
|
|
"net1": data.Get("net1").(string),
|
|
}
|
|
_, err = client.SetVmConfig(vmr, vmParams)
|
|
time.Sleep(10 * time.Second)
|
|
return err
|
|
default:
|
|
return fmt.Errorf("Unkown action: %s", act)
|
|
}
|
|
return nil
|
|
}
|