feat(mssql): add WithScripts and GetSQLCmdPath functions

Author: MattiasMTS

modules/mssql/mssql.go

@@ -3,6 +3,8 @@ package mssql
 import (
 	"context"
 	"fmt"
+	"path/filepath"
+	"strconv"
 	"strings"
 
 	"github.com/testcontainers/testcontainers-go"
@@ -41,6 +43,47 @@ func WithPassword(password string) testcontainers.CustomizeRequestOption {
 	}
 }
 
+// WithScripts adds SQL scripts to be executed after the container is ready.
+// The scripts are executed in the order they are provided using sqlcmd tool.
+func WithScripts(scripts ...string) testcontainers.CustomizeRequestOption {
+	return func(req *testcontainers.GenericContainerRequest) error {
+		hooks := make([]testcontainers.ContainerHook, 0, len(scripts))
+		for _, script := range scripts {
+			hook := func(ctx context.Context, c testcontainers.Container) error {
+				password := defaultPassword
+				if req.Env["MSSQL_SA_PASSWORD"] != "" {
+					password = req.Env["MSSQL_SA_PASSWORD"]
+				}
+
+				targetPath := "/tmp/" + filepath.Base(script)
+				if err := c.CopyFileToContainer(ctx, script, targetPath, 0o644); err != nil {
+					return err
+				}
+
+				cmd := testcontainers.NewRawCommand([]string{
+					GetSQLCmdPath(req.Image),
+					"-S", "localhost",
+					"-U", defaultUsername,
+					"-P", password,
+					"-No",
+					"-i", targetPath,
+				})
+				if _, _, err := c.Exec(ctx, cmd.AsCommand(), cmd.Options()...); err != nil {
+					return fmt.Errorf("script %q: %w", script, err)
+				}
+				return nil
+			}
+			hooks = append(hooks, hook)
+		}
+
+		req.LifecycleHooks = append(req.LifecycleHooks, testcontainers.ContainerLifecycleHooks{
+			PostReadies: hooks,
+		})
+
+		return nil
+	}
+}
+
 // Deprecated: use Run instead
 // RunContainer creates an instance of the MSSQLServer container type
 func RunContainer(ctx context.Context, opts ...testcontainers.ContainerCustomizer) (*MSSQLServerContainer, error) {
@@ -99,3 +142,25 @@ func (c *MSSQLServerContainer) ConnectionString(ctx context.Context, args ...str
 
 	return connStr, nil
 }
+
+// GetSQLCmdPath helper function to return the
+// sqlcmd path based on the image version
+func GetSQLCmdPath(image string) string {
+	const sqlCmd, sqlCmdOld = "/opt/mssql-tools18/bin/sqlcmd", "/opt/mssql-tools/bin/sqlcmd"
+
+	if strings.Contains(image, "2019-") || strings.Contains(image, "2017-") {
+		return sqlCmdOld
+	}
+
+	parts := strings.Split(strings.ToLower(image), "-")
+	for _, part := range parts {
+		if strings.HasPrefix(part, "cu") {
+			cuNumber := strings.TrimPrefix(part, "cu")
+			if num, err := strconv.Atoi(cuNumber); err == nil && num >= 14 {
+				return sqlCmd
+			}
+		}
+	}
+
+	return sqlCmd
+}