0
0
mirror of https://github.com/bpg/terraform-provider-proxmox.git synced 2025-07-01 02:52:58 +00:00
terraform-provider-proxmox/proxmoxtf/datasource/vm.go
Pavel Boldyrev 1f006aa82b
feat: API client cleanup and refactoring (#323)
* cleanup 1

* continue refactoring

* more refactoring

* move VMs under nodes

* move container and other apis under nodes

* cleanups

* enabled revive.exported linter & add comments to exported stuff

* enable godot linter

* enable wsl linter

* enable thelper linter

* enable govet linter

* cleanup after rebase

* cleanup after rebase

* extract SSH ops into a separate interface

* fix linter error

* move ssh code to its own package

* cleaning up VirtualEnvironmentClient receivers

* on the finish line

* not sure what else I forgot... 🤔

* fix ssh connection and upload

* renaming client interfaces

* final cleanups
2023-05-26 01:32:51 +00:00

112 lines
2.7 KiB
Go

/*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
*/
package datasource
import (
"context"
"sort"
"strconv"
"strings"
"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/bpg/terraform-provider-proxmox/proxmoxtf"
)
const (
mkDataSourceVirtualEnvironmentVMName = "name"
mkDataSourceVirtualEnvironmentVMNodeName = "node_name"
mkDataSourceVirtualEnvironmentVMTags = "tags"
mkDataSourceVirtualEnvironmentVMVMID = "vm_id"
)
// VM returns a resource for a single Proxmox VM.
func VM() *schema.Resource {
return &schema.Resource{
Schema: map[string]*schema.Schema{
mkDataSourceVirtualEnvironmentVMName: {
Type: schema.TypeString,
Description: "The VM name",
Computed: true,
},
mkDataSourceVirtualEnvironmentVMNodeName: {
Type: schema.TypeString,
Description: "The node name",
Required: true,
},
mkDataSourceVirtualEnvironmentVMTags: {
Type: schema.TypeList,
Description: "Tags of the VM",
Computed: true,
Elem: &schema.Schema{Type: schema.TypeString},
},
mkDataSourceVirtualEnvironmentVMVMID: {
Type: schema.TypeInt,
Description: "The VM identifier",
Required: true,
},
},
ReadContext: vmRead,
}
}
// vmRead reads the data of a VM by ID.
func vmRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
var diags diag.Diagnostics
config := m.(proxmoxtf.ProviderConfiguration)
api, err := config.GetClient()
if err != nil {
return diag.FromErr(err)
}
nodeName := d.Get(mkDataSourceVirtualEnvironmentVMNodeName).(string)
vmID := d.Get(mkDataSourceVirtualEnvironmentVMVMID).(int)
vmStatus, err := api.Node(nodeName).VM(vmID).GetVMStatus(ctx)
if err != nil {
if strings.Contains(err.Error(), "HTTP 404") ||
(strings.Contains(err.Error(), "HTTP 500") && strings.Contains(err.Error(), "does not exist")) {
d.SetId("")
return nil
}
return diag.FromErr(err)
}
if vmStatus.Name != nil {
err = d.Set(mkDataSourceVirtualEnvironmentVMName, *vmStatus.Name)
} else {
err = d.Set(mkDataSourceVirtualEnvironmentVMName, "")
}
diags = append(diags, diag.FromErr(err)...)
var tags []string
if vmStatus.Tags != nil {
for _, tag := range strings.Split(*vmStatus.Tags, ";") {
t := strings.TrimSpace(tag)
if len(t) > 0 {
tags = append(tags, t)
}
}
sort.Strings(tags)
}
err = d.Set(mkDataSourceVirtualEnvironmentVMTags, tags)
diags = append(diags, diag.FromErr(err)...)
d.SetId(strconv.Itoa(vmID))
return diags
}