talos

1
// This Source Code Form is subject to the terms of the Mozilla Public
2
// License, v. 2.0. If a copy of the MPL was not distributed with this
3
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
4

5
package cluster
6

7
import (
8
	"net/netip"
9

10
	"github.com/siderolabs/talos/pkg/machinery/config/machine"
11
	"github.com/siderolabs/talos/pkg/provision"
12
)
13

14
// MapProvisionNodeInfosToClusterNodeInfos maps provision.NodeInfos to cluster.NodeInfos.
15
func MapProvisionNodeInfosToClusterNodeInfos(nodes []provision.NodeInfo) ([]NodeInfo, error) {
16
	result := make([]NodeInfo, len(nodes))
17

18
	for i, info := range nodes {
19
		clusterNodeInfo, err := toClusterNodeInfo(info)
20
		if err != nil {
21
			return nil, err
22
		}
23

24
		result[i] = *clusterNodeInfo
25
	}
26

27
	return result, nil
28
}
29

30
// MapProvisionNodeInfosToNodeInfosByType maps provision.NodeInfos
31
// to cluster.NodeInfos, grouping them by machine type.
32
func MapProvisionNodeInfosToNodeInfosByType(nodes []provision.NodeInfo) (map[machine.Type][]NodeInfo, error) {
33
	result := make(map[machine.Type][]NodeInfo)
34

35
	for _, info := range nodes {
36
		clusterNodeInfo, err := toClusterNodeInfo(info)
37
		if err != nil {
38
			return nil, err
39
		}
40

41
		result[info.Type] = append(result[info.Type], *clusterNodeInfo)
42
	}
43

44
	return result, nil
45
}
46

47
func toClusterNodeInfo(info provision.NodeInfo) (*NodeInfo, error) {
48
	ips := make([]netip.Addr, len(info.IPs))
49

50
	for i, ip := range info.IPs {
51
		parsed, err := netip.ParseAddr(ip.String())
52
		if err != nil {
53
			return nil, err
54
		}
55

56
		ips[i] = parsed
57
	}
58

59
	return &NodeInfo{
60
		InternalIP: ips[0],
61
		IPs:        ips,
62
	}, nil
63
}
64