From 56ae6d5055959eb291231f6d905b126a36231328 Mon Sep 17 00:00:00 2001
From: Rolf Neugebauer <rolf.neugebauer@docker.com>
Date: Wed, 16 Aug 2017 14:26:08 +0100
Subject: [PATCH] cmd: Allow re-using an existing packet.net machine

This add a -device option which enables the re-use of an existing
device. The device state is update to use a potentially new base-url
ro PXE boot from. This maybe faster than provisioning a new server.

Signed-off-by: Rolf Neugebauer <rolf.neugebauer@docker.com>
---
 src/cmd/linuxkit/run_packet.go | 68 ++++++++++++++++++++++++++--------
 1 file changed, 52 insertions(+), 16 deletions(-)

diff --git a/src/cmd/linuxkit/run_packet.go b/src/cmd/linuxkit/run_packet.go
index 6c97f65e2..3de5a89c3 100644
--- a/src/cmd/linuxkit/run_packet.go
+++ b/src/cmd/linuxkit/run_packet.go
@@ -60,6 +60,7 @@ func runPacket(args []string) {
 	machineFlag := flags.String("machine", packetDefaultMachine, "Packet Machine Type (or "+packetMachineVar+")")
 	apiKeyFlag := flags.String("api-key", "", "Packet API key (or "+packetAPIKeyVar+")")
 	projectFlag := flags.String("project-id", "", "Packet Project ID (or "+packetProjectIDVar+")")
+	deviceFlag := flags.String("device", "", "The ID of an existing device")
 	hostNameFlag := flags.String("hostname", packetDefaultHostname, "Hostname of new instance (or "+packetHostnameVar+")")
 	nameFlag := flags.String("img-name", "", "Overrides the prefix used to identify the files. Defaults to [name] (or "+packetNameVar+")")
 	alwaysPXE := flags.Bool("always-pxe", true, "Reboot from PXE every time.")
@@ -150,20 +151,52 @@ func runPacket(args []string) {
 
 	client := packngo.NewClient("", apiKey, nil)
 	tags := []string{}
-	req := packngo.DeviceCreateRequest{
-		HostName:     hostname,
-		Plan:         plan,
-		Facility:     facility,
-		OS:           osType,
-		BillingCycle: billing,
-		ProjectID:    projectID,
-		UserData:     userData,
-		Tags:         tags,
-		AlwaysPXE:    *alwaysPXE,
-	}
-	dev, _, err := client.Devices.Create(&req)
-	if err != nil {
-		log.Fatal(err)
+
+	var dev *packngo.Device
+	var err error
+	if *deviceFlag != "" {
+		dev, _, err = client.Devices.Get(*deviceFlag)
+		if err != nil {
+			log.Fatalf("Getting info for device %s failed: %v", *deviceFlag, err)
+		}
+		b, err := json.MarshalIndent(dev, "", "    ")
+		if err != nil {
+			log.Fatal(err)
+		}
+		log.Debugf("%s\n", string(b))
+
+		req := packngo.DeviceUpdateRequest{
+			HostName:     hostname,
+			BillingCycle: dev.BillingCycle,
+			UserData:     userData,
+			Locked:       dev.Locked,
+			Tags:         dev.Tags,
+			AlwaysPXE:    *alwaysPXE,
+		}
+		dev, _, err = client.Devices.Update(*deviceFlag, &req)
+		if err != nil {
+			log.Fatalf("Update device %s failed: %v", *deviceFlag, err)
+		}
+		if _, err := client.Devices.Reboot(*deviceFlag); err != nil {
+			log.Fatalf("Rebooting device %s failed: %v", *deviceFlag, err)
+		}
+	} else {
+		// Create a new device
+		req := packngo.DeviceCreateRequest{
+			HostName:     hostname,
+			Plan:         plan,
+			Facility:     facility,
+			OS:           osType,
+			BillingCycle: billing,
+			ProjectID:    projectID,
+			UserData:     userData,
+			Tags:         tags,
+			AlwaysPXE:    *alwaysPXE,
+		}
+		dev, _, err = client.Devices.Create(&req)
+		if err != nil {
+			log.Fatalf("Creating device failed: %v", err)
+		}
 	}
 	b, err := json.MarshalIndent(dev, "", "    ")
 	if err != nil {
@@ -200,12 +233,15 @@ func runPacket(args []string) {
 		httpServer.Shutdown(ctx)
 	}
 
-	if !*keepFlag {
+	if *keepFlag {
+		log.Printf("The machine is kept...")
+		log.Printf("Device ID: %s", dev.ID)
+		log.Printf("Serial:    ssh %s@%s", dev.ID, sshHost)
+	} else {
 		if _, err := client.Devices.Delete(dev.ID); err != nil {
 			log.Fatalf("Unable to delete device: %v", err)
 		}
 	}
-
 }
 
 // validateHTTPURL does a sanity check that a URL returns a 200 or 300 response