lxd-migrate: Support using the local server

Signed-off-by: Stéphane Graber <stgraber@stgraber.org>
(cherry picked from commit 09e4119da1ca24d2c08a0bd9597a7e6a16803a74)
Signed-off-by: Din Music <din.music@canonical.com>
License: Apache-2.0

lxd-migrate/main_migrate.go

@@ -111,6 +111,19 @@ func (c *cmdMigrateData) render() string {
 }
 
 func (c *cmdMigrate) askServer() (lxd.InstanceServer, string, error) {
+	// Detect local server.
+	local, err := c.connectLocal()
+	if err == nil {
+		useLocal, err := c.global.asker.AskBool("The local LXD server is the target [default=yes]: ", "yes")
+		if err != nil {
+			return nil, "", err
+		}
+
+		if useLocal {
+			return local, "", nil
+		}
+	}
+
 	// Server address
 	serverURL, err := c.global.asker.AskString("Please provide LXD server URL: ", "", nil)
 	if err != nil {

lxd-migrate/utils.go

@@ -142,6 +142,13 @@ func transferRootfs(ctx context.Context, dst lxd.InstanceServer, op lxd.Operatio
 	return nil
 }
 
+func (c *cmdMigrate) connectLocal() (lxd.InstanceServer, error) {
+	args := lxd.ConnectionArgs{}
+	args.UserAgent = fmt.Sprintf("LXD-MIGRATE %s", version.Version)
+
+	return lxd.ConnectLXDUnix("", &args)
+}
+
 func (c *cmdMigrate) connectTarget(url string, certPath string, keyPath string, authType string, token string) (lxd.InstanceServer, string, error) {
 	args := lxd.ConnectionArgs{
 		AuthType: authType,