package db

import (
	"context"
	"fmt"
	"strings"

	"github.com/uptrace/bun"
)

type Repository[T any] struct {
	db *bun.DB

	pkFieldList []string
}

func NewRepository[T any](db *bun.DB, pkFieldList []string) *Repository[T] {
	return &Repository[T]{db: db, pkFieldList: pkFieldList}
}

func (r *Repository[T]) CreateOrUpdateMany(ctx context.Context, list []T) error {
	if len(list) == 0 {
		return nil
	}
	_, err := r.db.NewInsert().Model(&list).
		On(fmt.Sprintf("CONFLICT (%s) DO UPDATE", strings.Join(r.pkFieldList, ","))).
		Exec(ctx)
	return err
}

func (r *Repository[T]) GetAll(ctx context.Context) ([]T, error) {
	var list []T
	err := r.db.NewSelect().Model(&list).Scan(ctx)
	return list, err
}