Generate random name for HCloud server

stephank · Mar 7, 2021 · 2beeaf9 · 2beeaf9
1 parent bab5cbf
commit 2beeaf9
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/doc/providers/hcloud.md b/doc/providers/hcloud.md
@@ -10,6 +10,10 @@ target "<address>" "hcloud" {
   # The API token to use. (Required)
   token = "9vx8w..."
 
+  # The name of the server to launch. Will be
+  # suffixed by a random string. (Required)
+  name = "lazyssh-ubuntu"
+
   # The image to launch. (Required)
   image = "ubuntu-20.03"
 

diff --git a/providers/hcloud/impl.go b/providers/hcloud/impl.go
@@ -5,6 +5,7 @@ package hcloud
 import (
 	"fmt"
 	"log"
+	"math/rand"
 	"net"
 	"strings"
 	"time"
@@ -24,6 +25,7 @@ func init() {
 type Factory struct{}
 
 type Provider struct {
+	Name       string
 	Image      string
 	ServerType string
 	SSHKey     string
@@ -43,6 +45,7 @@ type state struct {
 
 type hclTarget struct {
 	Token      string            `hcl:"token,attr"`
+	Name       string            `hcl:"name,attr"`
 	Image      string            `hcl:"image,attr"`
 	ServerType string            `hcl:"server_type,attr"`
 	SSHKey     string            `hcl:"ssh_key,attr"`
@@ -70,6 +73,7 @@ func (factory *Factory) NewProvider(target string, hclBlock hcl.Body) (providers
 
 	prov := &Provider{
 		HCloud:     client,
+		Name:       parsed.Name,
 		Image:      parsed.Image,
 		ServerType: parsed.ServerType,
 		SSHKey:     parsed.SSHKey,
@@ -174,7 +178,7 @@ func (prov *Provider) start(mach *providers.Machine) bool {
 	}
 
 	opts := hcloud.ServerCreateOpts{
-		Name:             "lazyssh",
+		Name:             randomName(prov.Name),
 		ServerType:       serverType,
 		Image:            image,
 		SSHKeys:          []*hcloud.SSHKey{sshKey},
@@ -221,6 +225,17 @@ func (prov *Provider) start(mach *providers.Machine) bool {
 	return true
 }
 
+func randomName(p string) string {
+	var n = 5
+	var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
+
+	s := make([]rune, n)
+	for i := range s {
+		s[i] = letters[rand.Intn(len(letters))]
+	}
+	return fmt.Sprintf("%s-%s", p, string(s))
+}
+
 func serverIsStarting(server *hcloud.Server) bool {
 	return server.Status == hcloud.ServerStatusInitializing ||
 		server.Status == hcloud.ServerStatusStarting ||