utils

func ContainsKeyword ¶

func ContainsKeyword(sql string, keyword string) bool

ContainsKeyword checks if SQL contains a specific keyword (case-insensitive). Both sql and keyword are normalized to uppercase before comparison.

Example:

ContainsKeyword("create table foo", "CREATE") // true
ContainsKeyword("SELECT * FROM bar", "insert") // false

func ExtractAllBalancedParentheses ¶

func ExtractAllBalancedParentheses(text string) []string

ExtractAllBalancedParentheses finds all balanced parentheses groups in the text. Returns a slice of strings, each containing a balanced parentheses group.

Example:

ExtractAllBalancedParentheses("foo(a) bar(b, c)") // ["(a)", "(b, c)"]

func ExtractBalancedParentheses ¶

func ExtractBalancedParentheses(text string) string

ExtractBalancedParentheses extracts text within balanced parentheses from the given string. Starts from the first '(' and returns everything up to and including the matching ')'.

This is commonly used for:

• Extracting RLS policy USING clauses
• Extracting RLS policy WITH CHECK clauses
• Extracting function parameters
• Extracting composite type definitions

Example:

ExtractBalancedParentheses("USING (id = current_user_id())") // "(id = current_user_id())"
ExtractBalancedParentheses("FUNCTION foo(a INT, b INT) RETURNS") // "(a INT, b INT)"

Returns:

• The balanced parentheses substring (including the parentheses)
• Empty string if no balanced parentheses found

func ExtractFunctionName ¶

func ExtractFunctionName(sql string) string

ExtractFunctionName extracts the function name from a CREATE FUNCTION statement. Handles both simple names and schema-qualified names.

Example:

ExtractFunctionName("CREATE FUNCTION auth.jwt() RETURNS ...") // "jwt"
ExtractFunctionName("CREATE OR REPLACE FUNCTION my_func() ...") // "my_func"

func ExtractIndexName ¶

func ExtractIndexName(sql string) string

ExtractIndexName extracts the index name from a CREATE INDEX statement.

Example:

ExtractIndexName("CREATE INDEX idx_users_email ON users(email)") // "idx_users_email"
ExtractIndexName("CREATE UNIQUE INDEX CONCURRENTLY idx_pk ON t(id)") // "idx_pk"

func ExtractPolicyTargetTable ¶

func ExtractPolicyTargetTable(sql string) string

ExtractPolicyTargetTable extracts the target table name from a CREATE POLICY statement. Pattern: CREATE POLICY ... ON [schema.]table

Example:

ExtractPolicyTargetTable("CREATE POLICY p1 ON public.users ...") // "public.users"
ExtractPolicyTargetTable("CREATE POLICY p1 ON users FOR SELECT ...") // "users"

func ExtractSchemaName ¶

func ExtractSchemaName(sql string) string

ExtractSchemaName extracts the schema name from a CREATE SCHEMA statement.

Example:

ExtractSchemaName("CREATE SCHEMA IF NOT EXISTS auth") // "auth"
ExtractSchemaName("CREATE SCHEMA public AUTHORIZATION ...") // "public"

func ExtractTableName ¶

func ExtractTableName(sql string) string

ExtractTableName extracts the table name from a CREATE TABLE or ALTER TABLE statement. Handles schema-qualified names and IF NOT EXISTS clauses.

Example:

ExtractTableName("CREATE TABLE public.users (...)") // "users"
ExtractTableName("CREATE TABLE IF NOT EXISTS accounts (...)") // "accounts"
ExtractTableName("ALTER TABLE users ADD COLUMN ...") // "users"

func HasBalancedParentheses ¶

func HasBalancedParentheses(text string) bool

HasBalancedParentheses checks if the text has properly balanced parentheses. Returns true if all parentheses are balanced, false otherwise.

Example:

HasBalancedParentheses("(a (b c))") // true
HasBalancedParentheses("(a (b c)") // false

func HasClause ¶

func HasClause(sql string, clause string) bool

HasClause checks if SQL contains a specific SQL clause (case-insensitive). Adds a space after the clause to avoid partial matches.

Example:

HasClause("CREATE TABLE users (...)", "CREATE TABLE") // true
HasClause("ALTER TABLE users ADD", "ALTER") // true
HasClause("SELECT * FROM altered", "ALTER") // false (no space after)

func HasVolatilityMarker ¶

func HasVolatilityMarker(sql string) bool

HasVolatilityMarker checks if SQL contains a PostgreSQL function volatility marker. Returns true if IMMUTABLE, STABLE, or VOLATILE is found in the string.

Used by plugin transformation systems to avoid adding duplicate volatility markers.

func IsDDLOperation ¶

func IsDDLOperation(op types.Operation) bool

IsDDLOperation checks if an operation is a DDL operation (CREATE, ALTER, DROP)

func IsDDLStatement ¶

func IsDDLStatement(sql string) bool

IsDDLStatement checks if SQL is a Data Definition Language statement. Returns true for CREATE, ALTER, DROP, TRUNCATE.

func IsDMLOperation ¶

func IsDMLOperation(op types.Operation) bool

IsDMLOperation checks if an operation is a DML operation (INSERT, UPDATE, DELETE)

func IsDMLStatement ¶

func IsDMLStatement(sql string) bool

IsDMLStatement checks if SQL is a Data Manipulation Language statement. Returns true for INSERT, UPDATE, DELETE, SELECT.

func IsSecurityOperation ¶

func IsSecurityOperation(op types.Operation) bool

IsSecurityOperation checks if an operation is a security operation (GRANT, REVOKE)

func IsValidCategory ¶

func IsValidCategory(c types.Category) bool

IsValidCategory checks if a Category is valid

func IsValidObjectType ¶

func IsValidObjectType(t types.ObjectType) bool

IsValidObjectType checks if an ObjectType is valid

func IsValidOperation ¶

func IsValidOperation(op types.Operation) bool

IsValidOperation checks if an Operation is valid

func NormalizeObjectName ¶

func NormalizeObjectName(name string) string

NormalizeObjectName normalizes database object names (tables, columns, functions) by converting to lowercase and trimming whitespace. This provides consistent object name comparison across the codebase.

func NormalizeSQLWhitespace ¶

func NormalizeSQLWhitespace(sql string) string

NormalizeSQLWhitespace collapses multiple spaces into single spaces and removes leading/trailing whitespace. Using precompiled pattern for performance.

func SetDefaultLogger ¶

func SetDefaultLogger(logger *Logger)

SetDefaultLogger sets the global default logger

func StripOutermostParentheses ¶

func StripOutermostParentheses(text string) string

StripOutermostParentheses removes the outermost pair of parentheses if present. If the text doesn't start and end with matching parentheses, returns the original text.

Example:

StripOutermostParentheses("(id = 1)") // "id = 1"
StripOutermostParentheses("((id = 1))") // "(id = 1)"
StripOutermostParentheses("id = 1") // "id = 1"

func TrimSQLComments ¶

func TrimSQLComments(sql string) string

TrimSQLComments removes single-line SQL comments (--) from the beginning of a string. Multi-line comments (/* */) are not handled.

Example:

TrimSQLComments("-- Comment\nCREATE TABLE") // "CREATE TABLE"
TrimSQLComments("CREATE TABLE") // "CREATE TABLE"

func ValidCategories ¶

func ValidCategories() []types.Category

ValidCategories returns all valid Category values

func ValidObjectTypes ¶

func ValidObjectTypes() []types.ObjectType

ValidObjectTypes returns all valid ObjectType values

func ValidOperations ¶

func ValidOperations() []types.Operation

ValidOperations returns all valid Operation values

func ValidateCategory ¶

func ValidateCategory(c types.Category) error

ValidateCategory returns an error if the Category is invalid

func ValidateObjectType ¶

func ValidateObjectType(t types.ObjectType) error

ValidateObjectType returns an error if the ObjectType is invalid

func ValidateOperation ¶

func ValidateOperation(op types.Operation) error

ValidateOperation returns an error if the Operation is invalid