34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package build
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.secnex.io/secnex/pgson/schema"
|
|
"git.secnex.io/secnex/pgson/utils"
|
|
)
|
|
|
|
func UpdateSQL(s *schema.Table, data map[string]any, where string) (string, error) {
|
|
// Create a map for quick field lookup
|
|
fieldMap := make(map[string]*schema.Field)
|
|
for i := range s.Schema {
|
|
fieldMap[s.Schema[i].Name] = &s.Schema[i]
|
|
}
|
|
|
|
setClause := make([]string, 0, len(data))
|
|
for field, value := range data {
|
|
// Check if field is of type "hash" and needs hashing
|
|
if schemaField, exists := fieldMap[field]; exists && schemaField.Type == "hash" && schemaField.Algorithm != nil {
|
|
valueStr := fmt.Sprintf("%v", value)
|
|
hashed, err := utils.Hash(valueStr, *schemaField.Algorithm)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
value = hashed
|
|
}
|
|
setClause = append(setClause, fmt.Sprintf("%s = %s", utils.SQLQuoteIdent(field), utils.SQLQuoteValue(value)))
|
|
}
|
|
setClause = append(setClause, "updated_at = CURRENT_TIMESTAMP")
|
|
return fmt.Sprintf("UPDATE %s SET %s WHERE %s = %s", utils.SQLQuoteIdent(s.Name), strings.Join(setClause, ", "), utils.SQLQuoteIdent(s.PrimaryKey), utils.SQLQuoteValue(where)), nil
|
|
}
|