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)

	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)

	return &ddl, nil
}