Adds mongo ssh debug methods

nucleuscloud · Jun 20, 2024 · 20eb452 · 20eb452
1 parent cfa7596
commit 20eb452
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 7 deletions.
diff --git a/backend/pkg/mongoconnect/connector.go b/backend/pkg/mongoconnect/connector.go
@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"log/slog"
+	"net/url"
 	"sync"
 
 	mgmtv1alpha1 "github.com/nucleuscloud/neosync/backend/gen/go/protos/mgmt/v1alpha1"
@@ -58,12 +59,14 @@ func (w *WrappedMongoClient) Open(ctx context.Context) (*mongo.Client, error) {
 			return nil, err
 		}
 		<-ready
+		w.logger.Info("tunnel is now ready", "isopen", w.details.Tunnel.IsOpen())
 	}
 	serverAPI := options.ServerAPI(options.ServerAPIVersion1)
+	w.logger.Info("connecting to mongo instance", "url", w.details.String())
 	opts := options.Client().ApplyURI(w.details.String()).SetServerAPIOptions(serverAPI)
 	client, err := mongo.Connect(ctx, opts)
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to connect to mongo instance: %w", err)
 	}
 	w.client = client
 	return client, nil
@@ -79,6 +82,7 @@ func (w *WrappedMongoClient) Close(ctx context.Context) error {
 	w.client = nil
 	err := client.Disconnect(ctx)
 	if w.details.Tunnel != nil && w.details.Tunnel.IsOpen() {
+		w.logger.Debug("closing tunnel...")
 		w.details.Tunnel.Close()
 	}
 	return err
@@ -119,9 +123,13 @@ func (c *ConnectionDetails) GetTunnel() *sshtunnel.Sshtunnel {
 func (c *ConnectionDetails) String() string {
 	if c.Tunnel != nil && c.Tunnel.IsOpen() {
 		localhost, port := c.Tunnel.GetLocalHostPort()
-		newDetails := *c.Details
-		newDetails.Hosts = []string{fmt.Sprintf("%s:%d", localhost, port)}
-		return newDetails.String()
+		parseUrl, err := url.Parse(c.Details.String())
+		if err != nil {
+			return "" // todo
+		}
+		parseUrl.Host = fmt.Sprintf("%s:%d", localhost, port)
+		parseUrl.Scheme = "mongodb"
+		return parseUrl.String()
 	}
 	return c.Details.String()
 }
@@ -165,7 +173,7 @@ func GetConnectionDetails(
 		return nil, err
 	}
 	var publickey ssh.PublicKey
-	if tunnelCfg.GetKnownHostPublicKey() == "" {
+	if tunnelCfg.GetKnownHostPublicKey() != "" {
 		publickey, err = sshtunnel.ParseSshKey(tunnelCfg.GetKnownHostPublicKey())
 		if err != nil {
 			return nil, err

diff --git a/backend/pkg/sshtunnel/tunnel.go b/backend/pkg/sshtunnel/tunnel.go
@@ -112,6 +112,7 @@ func (t *Sshtunnel) serve(listener net.Listener, ready chan<- any, logger *slog.
 			t.isOpen = false
 			go func() {
 				t.shutdowns.Range(func(key, value any) bool {
+					logger.Debug("shutting down tunnel session", "key", key)
 					sd, ok := value.(chan any)
 					if ok {
 						sd <- struct{}{}
@@ -212,7 +213,7 @@ func (s *Sshtunnel) getSshClient(
 	if err != nil {
 		return nil, err
 	}
-	logger.Debug(fmt.Sprintf("conntected to %s", addr))
+	logger.Debug(fmt.Sprintf("[ssh-client] conntected to %s", addr))
 	s.sshclient = client
 	return client, nil
 }

diff --git a/backend/pkg/sshtunnel/utils.go b/backend/pkg/sshtunnel/utils.go
@@ -32,7 +32,7 @@ func getPlaintextPrivateKeyAuthMethod(keyBytes []byte) (ssh.AuthMethod, error) {
 
 func ParseSshKey(keyString string) (ssh.PublicKey, error) {
 	// Parse the key
-	publicKey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(keyString)) //nolint
+	publicKey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(keyString)) //nolint:dogsled
 	if err != nil {
 		return nil, fmt.Errorf("failed to parse public key: %v", err)
 	}