How can I connect to a virtual machine using the Teleport API using a Go code?

[lucasfellipec]

Hey, I am have problems to connect a virtual machine in Teleport using a Go code. Has anyone ever done this?

My current code:

package main

import (
	"context"
	"fmt"
	"log"
	"os"
	"time"

	"github.com/gravitational/teleport/api/client"
	"golang.org/x/crypto/ssh"
)

func main() {
	ctx := context.Background()

	clt, err := client.New(ctx, client.Config{
		Addrs: []string{
			"teleport.test.com:443",
			"teleport.test.com:3022",
			"teleport.test.com:3025",
		},
		Credentials: []client.Credentials{
			client.LoadIdentityFile("/Users/lucasfellipe/.tsh/keys/teleport.test.com/lucas-fellipe-teleport"),
		},
	})

	if err != nil {
		log.Fatalf("failed to create client: %v", err)
	}

	defer clt.Close()

	resp, err := clt.Ping(ctx)
	if err != nil {
		log.Fatalf("failed to ping server: %v", err)
	}

	user, err := clt.GetCurrentUser(ctx)
	if err != nil {
		log.Fatalf("failed to get current user: %v", err)
	}

	fmt.Println(user)

	log.Printf("Example success!")
	log.Printf("Server version: %s", resp.ServerVersion)

	certPath := "/Users/lucasfellipe/.tsh/keys/teleport.test.com/lucas-fellipe-teleport"
	// certPath := "/Users/lucasfellipe/.ssh/id_rsa"
	certBytes, err := os.ReadFile(certPath)
	if err != nil {
		log.Fatalf("Failed to read SSH certificate: %v", err)
	}
	signer, err := ssh.ParsePrivateKey(certBytes)
	if err != nil {
		log.Fatalf("Failed to parse SSH certificate: %v", err)
	}

	fmt.Println(signer)

	config := &ssh.ClientConfig{
		User: "root",
		Auth: []ssh.AuthMethod{
			ssh.PublicKeys(signer),
		},
		HostKeyAlgorithms: []string{"[email protected]", "[email protected]", "[email protected]"},
		HostKeyCallback:   ssh.InsecureIgnoreHostKey(),
		Timeout:           10 * time.Second,
	}

	conn, err := ssh.Dial("tcp", "172.31.3.248:3022", config)
	if err != nil {
		log.Fatalf("Failed to dial: %s", err)
	}
	defer conn.Close()

	// Run a command on the virtual machine
	session, err := conn.NewSession()
	if err != nil {
		log.Fatalf("Failed to create session: %s", err)
	}
	defer session.Close()

	cmd := "ls"
	output, err := session.CombinedOutput(cmd)
	if err != nil {
		log.Fatalf("Failed to run command: %s", err)
	}

	fmt.Println(string(output))
}

Can someone help me?