0
0
mirror of https://github.com/bpg/terraform-provider-proxmox.git synced 2025-06-30 18:42:58 +00:00
terraform-provider-proxmox/proxmoxtf/data_source_virtual_environment_roles.go

94 lines
2.3 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 proxmoxtf
import (
"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
)
const (
mkDataSourceVirtualEnvironmentRolesPrivileges = "privileges"
mkDataSourceVirtualEnvironmentRolesRoleIDs = "role_ids"
mkDataSourceVirtualEnvironmentRolesSpecial = "special"
)
func dataSourceVirtualEnvironmentRoles() *schema.Resource {
return &schema.Resource{
Schema: map[string]*schema.Schema{
mkDataSourceVirtualEnvironmentRolesPrivileges: {
Type: schema.TypeList,
Description: "The role privileges",
Computed: true,
Elem: &schema.Schema{
Type: schema.TypeSet,
Elem: &schema.Schema{Type: schema.TypeString},
},
},
mkDataSourceVirtualEnvironmentRolesRoleIDs: {
Type: schema.TypeList,
Description: "The role ids",
Computed: true,
Elem: &schema.Schema{Type: schema.TypeString},
},
mkDataSourceVirtualEnvironmentRolesSpecial: {
Type: schema.TypeList,
Description: "Whether the role is special (built-in)",
Computed: true,
Elem: &schema.Schema{Type: schema.TypeBool},
},
},
Read: dataSourceVirtualEnvironmentRolesRead,
}
}
func dataSourceVirtualEnvironmentRolesRead(d *schema.ResourceData, m interface{}) error {
config := m.(providerConfiguration)
veClient, err := config.GetVEClient()
if err != nil {
return err
}
list, err := veClient.ListRoles()
if err != nil {
return err
}
privileges := make([]interface{}, len(list))
roleIDs := make([]interface{}, len(list))
special := make([]interface{}, len(list))
for i, v := range list {
if v.Privileges != nil {
p := schema.NewSet(schema.HashString, []interface{}{})
for _, v := range *v.Privileges {
p.Add(v)
}
privileges[i] = p
} else {
privileges[i] = map[string]interface{}{}
}
roleIDs[i] = v.ID
if v.Special != nil {
special[i] = v.Special
} else {
special[i] = false
}
}
d.SetId("roles")
d.Set(mkDataSourceVirtualEnvironmentRolesPrivileges, privileges)
d.Set(mkDataSourceVirtualEnvironmentRolesRoleIDs, roleIDs)
d.Set(mkDataSourceVirtualEnvironmentRolesSpecial, special)
return nil
}