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

build/delete.go

@@ -0,0 +1,28 @@
+package build
+
+import (
+	"fmt"
+	"log"
+
+	"git.secnex.io/secnex/pgson/schema"
+	"git.secnex.io/secnex/pgson/sql"
+)
+
+func Delete(s *schema.Table, where map[string]any) (*string, error) {
+	if err := sql.ValidateIdent(s.Name); err != nil {
+		return nil, err
+	}
+
+	if len(where) == 0 {
+		return nil, fmt.Errorf("WHERE clause is required for DELETE to prevent accidental deletion of all rows")
+	}
+
+	whereClause, err := sql.BuildWhereClause(where)
+	if err != nil {
+		return nil, err
+	}
+
+	ddl := fmt.Sprintf(sql.DML_DELETE, sql.QuoteIdent(s.Name), whereClause)
+	log.Printf("DDL: %s", ddl)
+	return &ddl, nil
+}

build/insert.go

@@ -2,27 +2,32 @@ package build
 
 import (
 	"fmt"
+	"log"
+	"sort"
 	"strings"
 
 	"git.secnex.io/secnex/pgson/schema"
 	"git.secnex.io/secnex/pgson/sql"
 )
 
-func Insert(s *schema.Table) (*string, error) {
-	if err := sql.ValidateIdent(s.Name); err != nil {
-		return nil, err
+func Insert(s *schema.Table, data map[string]any) (*string, error) {
+	cols := []string{}
+	values := []string{}
+	// Keys of the data map
+	keys := make([]string, 0, len(data))
+	for k := range data {
+		keys = append(keys, k)
 	}
-
-	var cols, values []string
-
-	for _, f := range s.Schema {
-		if err := sql.ValidateIdent(f.Name); err != nil {
+	sort.Strings(keys)
+	for _, k := range keys {
+		if err := sql.ValidateIdent(k); err != nil {
+			log.Printf("Invalid column identifier: %s", k)
 			return nil, err
 		}
-		cols = append(cols, sql.QuoteIdent(f.Name))
-		values = append(values, "?")
+		cols = append(cols, sql.QuoteIdent(k))
+		values = append(values, sql.QuoteValue(data[k]))
 	}
-
-	dmlParts := fmt.Sprintf(sql.DML_INSERT_INTO, sql.QuoteIdent(s.Name), strings.Join(cols, ", "), strings.Join(values, ", "))
-	return &dmlParts, nil
+	ddl := fmt.Sprintf(sql.DML_INSERT_INTO, sql.QuoteIdent(s.Name), strings.Join(cols, ", "), strings.Join(values, ", "))
+	log.Printf("DDL: %s", ddl)
+	return &ddl, nil
 }

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
+}