init

iac/cluster/utils/ansible-configuration.go

@@ -0,0 +1,60 @@
+package utils
+
+import (
+	"fmt"
+	"regexp"
+	"strings"
+
+	"github.com/pulumi/pulumi-command/sdk/go/command/remote"
+	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
+)
+
+type SwarmJoinTokens struct {
+	ManagerToken string
+	WorkerToken  string
+}
+
+func InstallAnsibleDependencies(ctx *pulumi.Context, connArgs remote.ConnectionArgs, uniqueness string) error {
+	_, err := remote.NewCommand(ctx, strings.Join([]string{uniqueness, "Install Ansible Dependencies"}, ": "),
+		&remote.CommandArgs{
+			Connection: connArgs,
+			Create:     pulumi.String("apt-get update && apt-get install -y python3-pip python3-jsondiff"),
+		})
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func InitDockerSwarm(ctx *pulumi.Context, connArgs remote.ConnectionArgs, advertiseAddr pulumi.StringOutput) (pulumi.StringOutput, error) {
+	var tokens SwarmJoinTokens
+
+	fullCommand := advertiseAddr.ApplyT(func(addr string) *string {
+		initCommand := fmt.Sprintf("docker swarm init --advertise-addr %s", addr)
+		fullCommand := strings.Join([]string{initCommand, "echo \"Manager Token: $(docker swarm join-token -q manager)\"", "echo \"Worker Token: $(docker swarm join-token -q worker)\""}, " && ")
+		return &fullCommand
+	}).(pulumi.StringPtrOutput)
+
+	out, err := remote.NewCommand(ctx, "Init docker swarm",
+		&remote.CommandArgs{
+			Connection: connArgs,
+			Create:     fullCommand,
+		})
+	if err != nil {
+		return pulumi.StringOutput{}, err
+	}
+
+	return out.Stdout.ApplyT(func(output string) string {
+		searchWorker := "Worker Token: "
+		pattern := regexp.MustCompile(searchWorker + `(\S+)`)
+
+		matches := pattern.FindStringSubmatch(output)
+		if len(matches) > 1 {
+			extracted := matches[1]
+			tokens.WorkerToken = extracted
+			return extracted
+		}
+		fmt.Println(tokens.WorkerToken)
+		return ""
+	}).(pulumi.StringOutput), nil
+}

iac/cluster/utils/resource-creation.go

@@ -0,0 +1,92 @@
+package utils
+
+import (
+	"fmt"
+	"gc-infra/config"
+	"net"
+
+	"github.com/pulumi/pulumi-hcloud/sdk/go/hcloud"
+	"github.com/pulumi/pulumi-tls/sdk/v5/go/tls"
+	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
+)
+
+func CreateSshKey(ctx *pulumi.Context) (*tls.PrivateKey, error) {
+	return tls.NewPrivateKey(ctx, "sshKey", &tls.PrivateKeyArgs{
+		Algorithm: pulumi.String("ED25519"),
+	}, pulumi.AdditionalSecretOutputs([]string{"privKey"}))
+}
+
+func CreatePlacementGroup(ctx *pulumi.Context, name string) (*hcloud.PlacementGroup, error) {
+	pg, err := hcloud.NewPlacementGroup(ctx, name, &hcloud.PlacementGroupArgs{
+		Name: pulumi.String(name),
+		Type: pulumi.String("spread"),
+	})
+	return pg, err
+}
+
+func CreateClusterNet(ctx *pulumi.Context, cfg config.InfraConfig) (*pulumi.IDOutput, error) {
+	var id pulumi.IDOutput
+	network, err := hcloud.NewNetwork(ctx, cfg.SwarmNetworkName, &hcloud.NetworkArgs{
+		Name:    pulumi.String(cfg.SwarmNetworkName),
+		IpRange: pulumi.String(cfg.SwarmIpRange),
+	})
+	if err != nil {
+		return nil, err
+	}
+	id = network.ID()
+	_, err = hcloud.NewNetworkSubnet(ctx, "network-subnet", &hcloud.NetworkSubnetArgs{
+		Type:        pulumi.String("cloud"),
+		NetworkId:   IDtoIntOutput(id),
+		NetworkZone: pulumi.String("eu-central"),
+		IpRange:     pulumi.String(cfg.SwarmSubnetIpRange),
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &id, nil
+}
+
+type CreateServerArgs struct {
+	NetworkId        pulumi.IDInput
+	PlacementGroupId pulumi.IDInput
+	// The first assignable IP of the network
+	NetworkFirstIP string
+	Basename       string
+	Count          int
+	SshKey         *hcloud.SshKey
+}
+
+func CreateServer(ctx *pulumi.Context, cfg CreateServerArgs) ([]*hcloud.Server, error) {
+	var nodes []*hcloud.Server
+	nextIp := cfg.NetworkFirstIP
+	for i := range cfg.Count {
+		sn := fmt.Sprintf("%s-%d", cfg.Basename, i+1)
+		s, err := hcloud.NewServer(ctx, sn, &hcloud.ServerArgs{
+			Name:       pulumi.String(sn),
+			Image:      pulumi.String("docker-ce"),
+			ServerType: pulumi.String("cpx21"),
+			Location:   pulumi.StringPtr("fsn1"),
+			// Datacenter: pulumi.StringPtr("fsn1"),
+			Networks: hcloud.ServerNetworkTypeArray{
+				&hcloud.ServerNetworkTypeArgs{
+					NetworkId: IDtoIntOutput(cfg.NetworkId),
+					Ip:        pulumi.String(nextIp),
+				},
+			},
+			PlacementGroupId: IDtoIntPtrOutput(cfg.PlacementGroupId),
+			PublicNets: hcloud.ServerPublicNetArray{
+				&hcloud.ServerPublicNetArgs{
+					// Ipv4Enabled: pulumi.Bool(true),
+					Ipv6Enabled: pulumi.Bool(true),
+				},
+			},
+			SshKeys: pulumi.StringArray{cfg.SshKey.ID()},
+		})
+		if err != nil {
+			return nodes, err
+		}
+		nodes = append(nodes, s)
+		nextIp = IncrementIP(net.ParseIP(nextIp)).String()
+	}
+	return nodes, nil
+}

iac/cluster/utils/utils.go

@@ -0,0 +1,41 @@
+package utils
+
+import (
+	"net"
+	"strconv"
+
+	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
+)
+
+func IncrementIP(ip net.IP) net.IP {
+	if ip == nil {
+		return ip
+	}
+	ip = ip.To4()
+	ip[3]++
+	return ip
+}
+
+func IDtoInt(val string) (int, error) {
+	i, err := strconv.Atoi(val)
+	return i, err
+}
+
+func IDtoIntPtr(val string) (*int, error) {
+	i, err := strconv.Atoi(val)
+	return &i, err
+}
+
+// Convert pulumi.IDOutput to pulumi.InOutput
+//
+// Some hcloud ID's are int based, but pulumi.IDOutput doesn't have a ToIntOutput method
+func IDtoIntOutput(val pulumi.IDInput) pulumi.IntOutput {
+	return val.ToIDOutput().ToStringOutput().ApplyT(IDtoInt).(pulumi.IntOutput)
+}
+
+// Convert pulumi.IDOutput to pulumi.InOutput
+//
+// Some hcloud ID's are int based, but pulumi.IDOutput doesn't have a ToIntOutput method
+func IDtoIntPtrOutput(val pulumi.IDInput) pulumi.IntPtrOutput {
+	return val.ToIDOutput().ToStringOutput().ApplyT(IDtoIntPtr).(pulumi.IntPtrOutput)
+}