feat(sql): Add new DML (DELETE, UPDATE)

build/update.go

@@ -0,0 +1,64 @@
+package build
+
+import (
+	"fmt"
+	"log"
+	"sort"
+	"strings"
+
+	"git.secnex.io/secnex/pgson/schema"
+	"git.secnex.io/secnex/pgson/sql"
+)
+
+func Update(s *schema.Table, data map[string]any, where map[string]any) (*string, error) {
+	if err := sql.ValidateIdent(s.Name); err != nil {
+		return nil, err
+	}
+
+	if len(data) == 0 {
+		return nil, fmt.Errorf("no columns to update")
+	}
+
+	if len(where) == 0 {
+		return nil, fmt.Errorf("WHERE clause is required for UPDATE to prevent accidental updates")
+	}
+
+	var setParts []string
+	keys := make([]string, 0, len(data))
+	for k := range data {
+		keys = append(keys, k)
+	}
+	sort.Strings(keys)
+
+	for _, k := range keys {
+		if err := sql.ValidateIdent(k); err != nil {
+			log.Printf("Invalid column identifier: %s", k)
+			return nil, err
+		}
+		setParts = append(setParts, fmt.Sprintf("%s = %s", sql.QuoteIdent(k), sql.QuoteValue(data[k])))
+	}
+
+	whereClause, err := sql.BuildWhereClause(where)
+	if err != nil {
+		return nil, err
+	}
+
+	ddl := fmt.Sprintf(sql.DML_UPDATE, sql.QuoteIdent(s.Name), strings.Join(setParts, ", "), whereClause)
+	log.Printf("DDL: %s", ddl)
+	return &ddl, nil
+}
+
+func UpdateDeletedAt(s *schema.Table, where map[string]any) (*string, error) {
+	if err := sql.ValidateIdent(s.Name); err != nil {
+		return nil, err
+	}
+
+	whereClause, err := sql.BuildWhereClause(where)
+	if err != nil {
+		return nil, err
+	}
+
+	ddl := fmt.Sprintf(sql.DML_UPDATE, sql.QuoteIdent(s.Name), "deleted_at = CURRENT_TIMESTAMP", whereClause)
+	log.Printf("DDL: %s", ddl)
+	return &ddl, nil
+}