types

var CloudInitAttributes = map[string]schema.Attribute{
	"user": schema.SingleNestedAttribute{
		Optional:   true,
		Attributes: CloudInitUserAttributes,
	},
	"ip": schema.SetNestedAttribute{
		Optional:     true,
		CustomType:   NewCloudInitIpSetType(),
		NestedObject: CloudInitIpSchema,
		Validators: []validator.Set{
			setvalidator.SizeAtLeast(1),
		},
	},
	"dns": schema.SingleNestedAttribute{
		Optional:   true,
		Attributes: CloudInitDnsAttributes,
	},
}

var CloudInitDataSourceAttributes = map[string]dschema.Attribute{
	"user": dschema.SingleNestedAttribute{
		Computed:   true,
		Attributes: CloudInitUserDataSourceAttributes,
	},
	"ip": dschema.SetNestedAttribute{
		Computed:     true,
		CustomType:   NewCloudInitIpSetType(),
		NestedObject: CloudInitIpDataSourceSchema,
	},
	"dns": dschema.SingleNestedAttribute{
		Computed:   true,
		Attributes: CloudInitDnsDataSourceAttributes,
	},
}

var CloudInitDnsAttributes = map[string]schema.Attribute{
	"nameserver": schema.StringAttribute{
		Optional:    true,
		Description: "The nameserver to use for the machine.",
	},
	"domain": schema.StringAttribute{
		Optional:    true,
		Description: "The domain to use for the machine.",
	},
}

var CloudInitDnsDataSourceAttributes = map[string]dschema.Attribute{
	"nameserver": dschema.StringAttribute{
		Computed:    true,
		Description: "The nameserver to use for the machine.",
	},
	"domain": dschema.StringAttribute{
		Computed:    true,
		Description: "The domain to use for the machine.",
	},
}

var CloudInitIp = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"position": types.Int64Type,
		"v4":       CloudInitIpConfig,
		"v6":       CloudInitIpConfig,
	},
}

var CloudInitIpConfig = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"dhcp":    types.BoolType,
		"address": types.StringType,
		"netmask": types.StringType,
		"gateway": types.StringType,
	},
}

var CloudInitIpConfigDataSourceSchema = map[string]dschema.Attribute{
	"dhcp": dschema.BoolAttribute{
		Computed:    true,
		Description: "Whether to use DHCP to get the IP address.",
	},
	"address": dschema.StringAttribute{
		Computed:    true,
		Description: "The IP address to use for the machine.",
	},
	"netmask": dschema.StringAttribute{
		Computed:    true,
		Description: "The IP address netmask to use for the machine.",
	},
	"gateway": dschema.StringAttribute{
		Computed:    true,
		Description: "The gateway to use for the machine.",
	},
}

var CloudInitIpConfigSchema = map[string]schema.Attribute{
	"dhcp": schema.BoolAttribute{
		Optional:    true,
		Computed:    true,
		Description: "Whether to use DHCP to get the IP address.",
		PlanModifiers: []planmodifier.Bool{
			boolplanmodifier.UseStateForUnknown(),
		},
	},
	"address": schema.StringAttribute{
		Optional:    true,
		Description: "The IP address to use for the machine.",
	},
	"netmask": schema.StringAttribute{
		Optional:    true,
		Description: "The IP address netmask to use for the machine.",
	},
	"gateway": schema.StringAttribute{
		Optional:    true,
		Description: "The gateway to use for the machine.",
	},
}

var CloudInitIpConfigTypes = map[string]attr.Type{
	"dhcp":    types.BoolType,
	"address": types.StringType,
	"netmask": types.StringType,
	"gateway": types.StringType,
}

var CloudInitIpDataSourceSchema = dschema.NestedAttributeObject{
	Attributes: map[string]dschema.Attribute{
		"position": dschema.Int64Attribute{
			Computed:    true,
			Description: "The position of the network interface in the VM as an int. Used to determine the interface name (net0, net1, etc).",
		},
		"v4": dschema.SingleNestedAttribute{
			Computed:   true,
			Attributes: CloudInitIpConfigDataSourceSchema,
		},
		"v6": dschema.SingleNestedAttribute{
			Computed:   true,
			Attributes: CloudInitIpConfigDataSourceSchema,
		},
	},
}

var CloudInitIpSchema = schema.NestedAttributeObject{
	Attributes: map[string]schema.Attribute{
		"position": schema.Int64Attribute{
			Required:    true,
			Description: "The position of the network interface in the VM as an int. Used to determine the interface name (net0, net1, etc).",
		},
		"v4": schema.SingleNestedAttribute{
			Optional:   true,
			Attributes: CloudInitIpConfigSchema,
		},
		"v6": schema.SingleNestedAttribute{
			Optional:   true,
			Attributes: CloudInitIpConfigSchema,
		},
	},
}

var CloudInitIpType = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"position": types.Int64Type,
		"v4": types.ObjectType{
			AttrTypes: CloudInitIpConfigTypes,
		},
		"v6": types.ObjectType{
			AttrTypes: CloudInitIpConfigTypes,
		},
	},
}

var CloudInitUserAttributes = map[string]schema.Attribute{
	"name": schema.StringAttribute{
		Required:    true,
		Description: "The name of the user.",
	},
	"password": schema.StringAttribute{
		Optional:    true,
		Sensitive:   true,
		Description: "The password of the user.",
	},
	"public_keys": schema.SetAttribute{
		Optional:    true,
		Description: "The public ssh keys of the user.",
		ElementType: types.StringType,
		Validators: []validator.Set{
			setvalidator.SizeAtLeast(1),
		},
	},
}

var CloudInitUserDataSourceAttributes = map[string]dschema.Attribute{
	"name": dschema.StringAttribute{
		Computed:    true,
		Description: "The name of the user.",
	},
	"password": dschema.StringAttribute{
		Computed:    true,
		Sensitive:   true,
		Description: "The password of the user.",
	},
	"public_keys": dschema.SetAttribute{
		Computed:    true,
		Description: "The public ssh keys of the user.",
		ElementType: types.StringType,
	},
}

var VirtualMachineDataSourceType = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"id":          types.Int64Type,
		"node":        types.StringType,
		"name":        types.StringType,
		"description": types.StringType,
		"tags": types.SetType{
			ElemType: types.StringType,
		},
		"agent": types.ObjectType{
			AttrTypes: map[string]attr.Type{
				"enabled":    types.BoolType,
				"use_fstrim": types.BoolType,
				"type":       types.StringType,
			},
		},
		"bios": types.StringType,
		"cpu": types.ObjectType{
			AttrTypes: map[string]attr.Type{
				"architecture":  types.StringType,
				"cores":         types.Int64Type,
				"sockets":       types.Int64Type,
				"emulated_type": types.StringType,
				"cpu_units":     types.Int64Type,
			},
		},
		"disks":              NewVirtualMachineDiskSetType(),
		"network_interfaces": NewVirtualMachineNetworkInterfaceSetType(),
		"pci_devices":        NewVirtualMachinePCIDeviceSetType(),
		"memory": types.ObjectType{
			AttrTypes: map[string]attr.Type{
				"dedicated": types.Int64Type,
				"floating":  types.Int64Type,
				"shared":    types.Int64Type,
			},
		},
		"cloud_init": types.ObjectType{
			AttrTypes: map[string]attr.Type{
				"user": types.ObjectType{
					AttrTypes: map[string]attr.Type{
						"password": types.StringType,
						"public_keys": types.SetType{
							ElemType: types.StringType,
						},
						"name": types.StringType,
					},
				},
				"ip": NewCloudInitIpSetType(),
				"dns": types.ObjectType{
					AttrTypes: map[string]attr.Type{
						"nameserver": types.StringType,
						"domain":     types.StringType,
					},
				},
			},
		},
		"machine_type":       types.StringType,
		"kvm_arguments":      types.StringType,
		"keyboard_layout":    types.StringType,
		"type":               types.StringType,
		"resource_pool":      types.StringType,
		"start_on_node_boot": types.BoolType,
	},
}

var VirtualMachineDisk = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"storage":      types.StringType,
		"file_format":  types.StringType,
		"size":         types.Int64Type,
		"name":         types.StringType,
		"use_iothread": types.BoolType,
		"speed_limits": types.ObjectType{
			AttrTypes: map[string]attr.Type{
				"read":            types.Int64Type,
				"write":           types.Int64Type,
				"read_burstable":  types.Int64Type,
				"write_burstable": types.Int64Type,
			},
		},
		"interface_type": types.StringType,
		"position":       types.Int64Type,
		"ssd_emulation":  types.BoolType,
		"discard":        types.BoolType,
	},
}

var VirtualMachineNetworkInterface = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"bridge":       types.StringType,
		"enabled":      types.BoolType,
		"use_firewall": types.BoolType,
		"mac_address":  types.StringType,
		"model":        types.StringType,
		"rate_limit":   types.Int64Type,
		"position":     types.Int64Type,
		"vlan":         types.Int64Type,
		"mtu":          types.Int64Type,
	},
}

var VirtualMachinePCIDevice = types.ObjectType{
	AttrTypes: map[string]attr.Type{
		"name":        types.StringType,
		"id":          types.StringType,
		"pcie":        types.BoolType,
		"mdev":        types.StringType,
		"rombar":      types.BoolType,
		"rom_file":    types.StringType,
		"primary_gpu": types.BoolType,
	},
}