fix(proxmox): adding lxc support

Author: bl4ko

internal/constants/constants.go

@@ -127,6 +127,9 @@ const (
 
 	DeviceRoleServer      = "Server"
 	DeviceRoleServerColor = "00add8"
+
+	DeviceRoleContainer      = "Container"
+	DeviceRoleContainerColor = "0db7ed"
 )
 
 // Constants used for variables in our contexts.

internal/netbox/objects/virtualization.go

@@ -96,6 +96,8 @@ type VM struct {
 	Memory int `json:"memory,omitempty"`
 	// Disk is the amount of disk space allocated to the virtual machine in GB.
 	Disk int `json:"disk,omitempty"`
+	// Role of the virtual machine.
+	Role *DeviceRole `json:"role,omitempty"`
 
 	// Additional Comments
 	Comments string `json:"comments,omitempty"`

internal/source/proxmox/proxmox.go

@@ -24,6 +24,7 @@ type ProxmoxSource struct {
 	NodeNetworks map[string][]*proxmox.NodeNetwork       // NodeName -> NodeNetworks (interfaces)
 	Vms          map[string][]*proxmox.VirtualMachine    // NodeName -> VirtualMachines
 	VMNetworks   map[string][]*proxmox.AgentNetworkIface // VMName -> NetworkDevices
+	Containers   map[string][]*proxmox.Container         // NodeName -> Contatiners
 
 	// Netbox related data for easier access. Initialized in sync functions.
 	NetboxCluster *objects.Cluster
@@ -102,6 +103,7 @@ func (ps *ProxmoxSource) Sync(nbi *inventory.NetboxInventory) error {
 		ps.syncCluster,
 		ps.syncNodes,
 		ps.syncVMs,
+		ps.syncContainers,
 	}
 	for _, syncFunc := range syncFunctions {
 		startTime := time.Now()

internal/source/proxmox/proxmox_init.go

@@ -26,6 +26,7 @@ func (ps *ProxmoxSource) initNodes(ctx context.Context, c *proxmox.Client) error
 	ps.Nodes = make([]*proxmox.Node, 0, len(nodes))
 	ps.NodeNetworks = make(map[string][]*proxmox.NodeNetwork, len(nodes))
 	ps.Vms = make(map[string][]*proxmox.VirtualMachine, len(nodes))
+	ps.Containers = make(map[string][]*proxmox.Container, len(nodes))
 	for _, node := range nodes {
 		node, err := c.Node(ctx, node.Node)
 		if err != nil {
@@ -42,6 +43,11 @@ func (ps *ProxmoxSource) initNodes(ctx context.Context, c *proxmox.Client) error
 		if err != nil {
 			return fmt.Errorf("init nodeVMs: %s", err)
 		}
+
+		err = ps.initContainers(ctx, node)
+		if err != nil {
+			return fmt.Errorf("init node containers: %s", err)
+		}
 	}
 	return nil
 }
@@ -79,3 +85,20 @@ func (ps *ProxmoxSource) initNodeVMs(ctx context.Context, node *proxmox.Node) er
 	}
 	return nil
 }
+
+// Helper function for initNodes. It collects all containers for given node.
+func (ps *ProxmoxSource) initContainers(ctx context.Context, node *proxmox.Node) error {
+	containers, err := node.Containers(ctx)
+	if err != nil {
+		return err
+	}
+	ps.Containers[node.Name] = make([]*proxmox.Container, 0, len(containers))
+	// ps.VMNetworks = make(map[string][]*proxmox.AgentNetworkIface, len(containers))
+	for _, container := range containers {
+		ps.Containers[node.Name] = append(ps.Containers[node.Name], container)
+		// ifaces, _ := container.AgentGetNetworkIFaces(ctx)
+		// ps.VMNetworks[container.Name] = make([]*proxmox.AgentNetworkIface, 0, len(ifaces))
+		// ps.VMNetworks[container.Name] = append(ps.VMNetworks[container.Name], ifaces...)
+	}
+	return nil
+}

internal/source/proxmox/proxmox_sync.go

@@ -289,3 +289,62 @@ func (ps *ProxmoxSource) syncVMNetworks(nbi *inventory.NetboxInventory, nbVM *ob
 	}
 	return nil
 }
+
+// Function that synces proxmox containers to the netbox inventory.
+func (ps *ProxmoxSource) syncContainers(nbi *inventory.NetboxInventory) error {
+	if len(ps.Containers) > 0 {
+		// Create container role
+		containerRole, err := nbi.AddDeviceRole(ps.Ctx, &objects.DeviceRole{
+			Name:   constants.DeviceRoleContainer,
+			Slug:   utils.Slugify(constants.DeviceRoleContainer),
+			Color:  constants.DeviceRoleContainerColor,
+			VMRole: true,
+		})
+		if err != nil {
+			return fmt.Errorf("create container role: %s", err)
+		}
+		for nodeName, containers := range ps.Containers {
+			nbHost := ps.NetboxNodes[nodeName]
+			for _, container := range containers {
+				// Determine Container status
+				containerStatus := &objects.VMStatusActive
+				if container.Status == "stopped" {
+					containerStatus = &objects.VMStatusOffline
+				}
+				// Determine Container tenant
+				vmTenant, err := common.MatchVMToTenant(ps.Ctx, nbi, container.Name, ps.VMTenantRelations)
+				if err != nil {
+					return fmt.Errorf("match vm to tenant: %s", err)
+				}
+				_, err = nbi.AddVM(ps.Ctx, &objects.VM{
+					NetboxObject: objects.NetboxObject{
+						Tags: ps.SourceTags,
+						CustomFields: map[string]string{
+							constants.CustomFieldSourceName:   ps.SourceConfig.Name,
+							constants.CustomFieldSourceIDName: fmt.Sprintf("%d", container.VMID),
+						},
+					},
+					Host:    nbHost,
+					Role:    containerRole,
+					Cluster: ps.NetboxCluster, // Default single proxmox cluster
+					Tenant:  vmTenant,
+					VCPUs:   float32(container.CPUs),
+					Memory:  int(container.MaxMem / constants.MiB),  // Memory is in MB
+					Disk:    int(container.MaxDisk / constants.GiB), // Disk is in GB
+					Site:    nbHost.Site,
+					Name:    container.Name,
+					Status:  containerStatus,
+				})
+				if err != nil {
+					return fmt.Errorf("new vm: %s", err)
+				}
+
+				// err = ps.syncContainerNetworks(nbi, nbContainer)
+				// if err != nil {
+				// 	return fmt.Errorf("sync container networks: %s", err)
+				// }
+			}
+		}
+	}
+	return nil
+}