Add mssql server dag config command

Signed-off-by: Neaj Morshad <neaj@appscode.com>

Author: Neaj-Morshad-101

pkg/cmds/mssql.go

@@ -0,0 +1,152 @@
+/*
+Copyright AppsCode Inc. and Contributors
+
+Licensed under the AppsCode Community License 1.0.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    https://github.com/appscode/licenses/raw/1.0.0/AppsCode-Community-1.0.0.md
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmds
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	core "k8s.io/api/core/v1"
+	kerr "k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	cmdutil "k8s.io/kubectl/pkg/cmd/util"
+	"sigs.k8s.io/yaml"
+)
+
+// NewCmdMSSQL creates the parent `mssql` command
+func NewCmdMSSQL(f cmdutil.Factory) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "mssql",
+		Short: "MSSQLServer database commands",
+		Long:  "Commands for managing KubeDB MSSQLServer instances.",
+		Run: func(cmd *cobra.Command, args []string) {
+			_ = cmd.Help()
+		},
+	}
+	// Add your new command as a subcommand of mssql
+	cmd.AddCommand(NewCmdDAGConfig(f))
+
+	// If you create more mssql commands in the future, add them here.
+	// For example: cmd.AddCommand(mssql.NewCmdSomeOtherCommand(f))
+
+	return cmd
+}
+
+// NewCmdDAGConfig creates the `kubectl dba mssql dag-config` command.
+func NewCmdDAGConfig(f cmdutil.Factory) *cobra.Command {
+	var (
+		namespace  string
+		outputDir  string
+		desLong    = `Generates a YAML file with the necessary secrets for setting up a MSSQLServer Distributed Availability Group (DAG) remote replica.`
+		exampleStr = `  # Generate DAG configuration secrets from MSSQLServer 'ag1' in namespace 'demo' and save to the current directory
+  kubectl dba mssql dag-config ag1 -n demo
+
+  # Generate and save the configuration to a specific directory
+  kubectl dba mssql dag-config ag1 -n demo --output-dir=/tmp/configs`
+	)
+
+	cmd := &cobra.Command{
+		Use:     "dag-config [mssqlserver-name]",
+		Short:   "Generate Distributed Availability Group configuration from a source MSSQLServer",
+		Long:    desLong,
+		Example: exampleStr,
+		Args:    cobra.ExactArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			mssqlServerName := args[0]
+			cmdutil.CheckErr(runDAGConfig(f, namespace, outputDir, mssqlServerName))
+		},
+	}
+
+	cmd.Flags().StringVarP(&namespace, "namespace", "n", "default", "Namespace of the source MSSQLServer")
+	cmd.Flags().StringVar(&outputDir, "output-dir", ".", "Directory where the configuration YAML file will be saved")
+	return cmd
+}
+
+func runDAGConfig(f cmdutil.Factory, namespace, outputDir, mssqlServerName string) error {
+	client, err := f.KubernetesClientSet()
+	if err != nil {
+		return fmt.Errorf("failed to get kubernetes client: %w", err)
+	}
+
+	fmt.Printf("Generating DAG configuration for MSSQLServer '%s' in namespace '%s'...\n", mssqlServerName, namespace)
+
+	yamlBuffer, err := generateMSSQLDAGConfig(client, namespace, mssqlServerName)
+	if err != nil {
+		return err
+	}
+
+	if err := os.MkdirAll(outputDir, 0o755); err != nil {
+		return fmt.Errorf("failed to create output directory '%s': %w", outputDir, err)
+	}
+
+	outputFile := fmt.Sprintf("%s/%s-dag-config.yaml", outputDir, mssqlServerName)
+	if err := os.WriteFile(outputFile, yamlBuffer, 0o644); err != nil {
+		return fmt.Errorf("failed to write DAG config to file '%s': %w", outputFile, err)
+	}
+
+	fmt.Printf("✅ Successfully generated DAG configuration.\n")
+	fmt.Printf("   Apply this file in your remote cluster: kubectl apply -f %s\n", outputFile)
+
+	return nil
+}
+
+func generateMSSQLDAGConfig(client kubernetes.Interface, namespace, mssqlServerName string) ([]byte, error) {
+	var finalYAML []byte
+	secretNames := []string{
+		fmt.Sprintf("%s-dbm-login", mssqlServerName),
+		fmt.Sprintf("%s-master-key", mssqlServerName),
+		fmt.Sprintf("%s-endpoint-cert", mssqlServerName),
+	}
+
+	for _, secretName := range secretNames {
+		fmt.Printf("  - Fetching secret '%s'...\n", secretName)
+		secret, err := client.CoreV1().Secrets(namespace).Get(context.TODO(), secretName, metav1.GetOptions{})
+		if err != nil {
+			if kerr.IsNotFound(err) {
+				return nil, fmt.Errorf("required secret '%s' not found in namespace '%s'. Please ensure the source MSSQLServer is running correctly", secretName, namespace)
+			}
+			return nil, fmt.Errorf("failed to get secret '%s': %w", secretName, err)
+		}
+		cleanedSecret := cleanupSecretForExport(secret)
+		secretYAML, err := yaml.Marshal(cleanedSecret)
+		if err != nil {
+			return nil, fmt.Errorf("failed to marshal secret '%s' to YAML: %w", secretName, err)
+		}
+		finalYAML = append(finalYAML, secretYAML...)
+		finalYAML = append(finalYAML, []byte("---\n")...)
+	}
+	return finalYAML, nil
+}
+
+func cleanupSecretForExport(secret *core.Secret) *core.Secret {
+	return &core.Secret{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: core.SchemeGroupVersion.Version,
+			Kind:       "Secret",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      secret.Name,
+			Namespace: secret.Namespace,
+			Labels:    secret.Labels,
+		},
+		Data: secret.Data,
+		Type: secret.Type,
+	}
+}

pkg/cmds/root.go

@@ -108,6 +108,12 @@ func NewKubeDBCommand(in io.Reader, out, err io.Writer) *cobra.Command {
 				NewCmdGenApb(f),
 			},
 		},
+		{
+			Message: "MSSQLServer specific commands",
+			Commands: []*cobra.Command{
+				NewCmdMSSQL(f),
+			},
+		},
 		{
 			Message: "Metric related CMDs",
 			Commands: []*cobra.Command{