niki/repository/mysql/kind_box_req/kind_box_req.go

package mysqlkindboxreq

import (
	"context"
	"database/sql"
	"errors"

	"git.gocasts.ir/ebhomengo/niki/entity"
	errmsg "git.gocasts.ir/ebhomengo/niki/pkg/err_msg"
	richerror "git.gocasts.ir/ebhomengo/niki/pkg/rich_error"
)

func (d DB) AddKindBoxReq(ctx context.Context, kindBoxReq entity.KindBoxReq) (entity.KindBoxReq, error) {
	const op = "mysqlkindboxreq.AddKindBoxReq"

	res, err := d.conn.Conn().ExecContext(ctx, `insert into kind_box_reqs(benefactor_id,kind_box_type,address_id,count_requested,refer_date,status) values (?,?,?,?,?,?)`,
		kindBoxReq.BenefactorID, kindBoxReq.KindBoxType.String(), kindBoxReq.AddressID, kindBoxReq.CountRequested, kindBoxReq.ReferDate, kindBoxReq.Status.String())
	if err != nil {
		return entity.KindBoxReq{}, richerror.New(op).WithErr(err).
			WithMessage(errmsg.ErrorMsgNotFound).WithKind(richerror.KindUnexpected)
	}

	//nolint
	// err is always nil
	id, _ := res.LastInsertId()
	kindBoxReq.ID = uint(id)

	return kindBoxReq, nil
}

func (d DB) AcceptKindBoxReq(ctx context.Context, kindBoxReqID, countAccepted uint) error {
	op := richerror.Op("mysqlkindboxreq.AcceptKindBoxReq")
	statement, err := d.conn.Conn().
		Prepare(`update kind_box_reqs set count_accepted = ? , status = ? where id = ?`)
	defer statement.Close()
	if err != nil {
		return richerror.New(op).WithErr(err).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}
	_, eErr := statement.ExecContext(ctx, countAccepted, entity.KindBoxReqAcceptedStatus.String(), kindBoxReqID)
	if eErr != nil {
		return richerror.New(op).WithErr(eErr).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}

	return nil
}

func (d DB) GetByID(ctx context.Context, id uint) (entity.KindBoxReq, error) {
	op := richerror.Op("mysqlkindboxreq.GetByID")
	row := d.conn.Conn().QueryRowContext(ctx, `select * from kind_box_reqs where id = ?`, id)
	k, err := scanKindBoxReq(row)
	if err != nil {
		return entity.KindBoxReq{}, richerror.New(op).WithErr(err).
			WithMessage(errmsg.ErrorMsgCantScanQueryResult).WithKind(richerror.KindUnexpected)
	}

	return k, nil
}

func (d DB) KindBoxRequestExist(id uint) (bool, error) {
	op := richerror.Op("mysqlkindboxreq.KindBoxRequestExist")
	row := d.conn.Conn().QueryRow(`select * from kind_box_reqs where id = ?`, id)
	_, sErr := scanKindBoxReq(row)
	if sErr != nil {
		if errors.Is(sErr, sql.ErrNoRows) {
			return false, nil
		}

		return false, richerror.New(op).WithErr(sErr).WithMessage(errmsg.ErrorMsgCantScanQueryResult).
			WithKind(richerror.KindUnexpected).WithMeta(map[string]any{"id": id})
	}

	return true, nil
}

func (d DB) CheckKindBoxReqStatusForAccepting(id uint) error {
	op := richerror.Op("mysqlkindboxreq.CheckKindBoxReqStatusForAccepting")

	k, err := d.GetByID(context.Background(), id)
	if err != nil {
		return richerror.New(op).WithErr(err)
	}
	if k.Status == entity.KindBoxReqAcceptedStatus {
		return richerror.New(op).WithMessage(errmsg.ErrorMsgReAcceptKindBoxReq).WithMeta(map[string]interface{}{
			"id": id,
		}).WithKind(richerror.KindInvalid)
	}

	return nil
}

func (d DB) CheckKindBoxReqStatusForRejecting(id uint) error {
	op := richerror.Op("mysqlkindboxreq.CheckKindBoxReqStatusForRejecting")
	k, err := d.GetByID(context.Background(), id)
	if err != nil {
		return richerror.New(op).WithErr(err)
	}
	if k.Status == entity.KindBoxReqRejectedStatus {
		return richerror.New(op).WithMessage(errmsg.ErrorMsgReRejectKindBoxReq).WithMeta(map[string]interface{}{
			"id": id,
		}).WithKind(richerror.KindInvalid)
	}

	return nil
}

func (d DB) RejectKindBoxReq(ctx context.Context, kindBoxReqID uint, description string) error {
	op := richerror.Op("mysqlkindboxreq.RejectKindBoxReq")
	statement, err := d.conn.Conn().
		Prepare(`update kind_box_reqs set description = ? , status = ? where id = ?`)
	defer statement.Close()
	if err != nil {
		return richerror.New(op).WithErr(err).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}
	_, eErr := statement.ExecContext(ctx, description, entity.KindBoxReqRejectedStatus.String(), kindBoxReqID)
	if eErr != nil {
		return richerror.New(op).WithErr(eErr).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}

	return nil
}

func (d DB) RollbackKindBoxRequestStatus(ctx context.Context, id uint) error {
	op := richerror.Op("mysqlkindboxreq.RollbackKindBoxRequestStatus")
	statement, err := d.conn.Conn().
		Prepare(`update kind_box_reqs set status = ? where id = ?`)
	defer statement.Close()
	if err != nil {
		return richerror.New(op).WithErr(err).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}
	_, eErr := statement.ExecContext(ctx, entity.KindBoxReqPendingStatus.String(), id)
	if eErr != nil {
		return richerror.New(op).WithErr(eErr).
			WithMessage(errmsg.ErrorMsgSomethingWentWrong).WithKind(richerror.KindUnexpected)
	}
	return nil
}