helper.go

package pgxadapter

import (
	"context"
	"fmt"

	"github.com/jackc/pgx/v5"
)

func (a *Adapter) genRow(ptype string, rule []string) []interface{} {
	row := make([]interface{}, 7)
	row[0] = ptype
	for i := 0; i < len(rule) && i < 6; i++ {
		row[i+1] = rule[i]
	}
	// Verify defaults for remaining columns (already nil/interface{} nil which maps to NULL)
	// If we want empty strings instead of NULLs:
	for i := len(rule) + 1; i < 7; i++ {
		row[i] = ""
	}
	return row
}

func (a *Adapter) removePolicy(ctx context.Context, tx pgx.Tx, ptype string, rule []string) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE ptype = $1", a.tableName)
	args := []interface{}{ptype}

	for i, v := range rule {
		query += fmt.Sprintf(" AND v%d = $%d", i, i+2)
		args = append(args, v)
	}

	_, err := tx.Exec(ctx, query, args...)
	return err
}

func (a *Adapter) updatePolicy(ctx context.Context, tx pgx.Tx, ptype string, oldRule, newRule []string) error {
	query := fmt.Sprintf("UPDATE %s SET ptype = $1", a.tableName)
	args := []interface{}{ptype}
	argIndex := 2

	// Construct UPDATE clause
	// We need to update v0..v5 based on newRule
	// and WHERE based on oldRule

	// Set clause
	for i := 0; i < 6; i++ {
		val := ""
		if i < len(newRule) {
			val = newRule[i]
		}
		query += fmt.Sprintf(", v%d = $%d", i, argIndex)
		args = append(args, val)
		argIndex++
	}

	// Where clause
	query += fmt.Sprintf(" WHERE ptype = $%d", argIndex)
	args = append(args, ptype)
	argIndex++

	for i, v := range oldRule {
		query += fmt.Sprintf(" AND v%d = $%d", i, argIndex)
		args = append(args, v)
		argIndex++
	}

	_, err := tx.Exec(ctx, query, args...)
	return err
}