package repository

import (
	"context"
	"database/sql"
	"time"

	"github.com/huandu/go-sqlbuilder"
)

var (
	timeZero = time.Time{}
)

func deleteFromTable(ctx context.Context, conn *sql.DB, tableName string, id int64) (int64, error) {
	b := sqlbuilder.NewDeleteBuilder()
	b.DeleteFrom(tableName)
	b.Where(
		b.Equal("Id", id),
	)
	query, args := b.Build()

	_, err := conn.ExecContext(ctx, query, args...)
	if err != nil {
		return 0, err
	}

	return 1, nil
}

func restoreRow(ctx context.Context, conn *sql.DB, tableName string, id int64) (int64, error) {
	timeZero := time.Time{}
	b := sqlbuilder.NewUpdateBuilder()
	b.Update(tableName)
	b.Set(
		b.Assign("UpdatedAt", time.Now()),
		b.Assign("DeletedAt", timeZero),
	)
	b.Where(
		b.Equal("Id", id),
	)
	query, args := b.Build()

	_, err := conn.ExecContext(ctx, query, args...)
	if err != nil {
		return 0, err
	}

	return 1, nil
}

func softDeleteRow(ctx context.Context, conn *sql.DB, tableName string, id int64) (int64, error) {
	now := time.Now()
	b := sqlbuilder.NewUpdateBuilder()
	b.Update(tableName)
	b.Set(
		b.Assign("UpdatedAt", now),
		b.Assign("DeletedAt", now),
	)
	b.Where(
		b.Equal("Id", id),
	)
	query, args := b.Build()

	_, err := conn.ExecContext(ctx, query, args...)
	if err != nil {
		return 0, err
	}

	return 1, nil
}