package model

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

	"github.com/zeromicro/go-zero/core/stores/builder"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
	"github.com/zeromicro/go-zero/core/stringx"
)

var (
	purPurchaseReceiptFieldNames          = builder.RawFieldNames(&PurPurchaseReceipt{})
	purPurchaseReceiptRows                = strings.Join(purPurchaseReceiptFieldNames, ",")
	purPurchaseReceiptRowsExpectAutoSet   = strings.Join(stringx.Remove(purPurchaseReceiptFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
	purPurchaseReceiptRowsWithPlaceHolder = strings.Join(stringx.Remove(purPurchaseReceiptFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
)

type (
	purPurchaseReceiptModel interface {
		Insert(ctx context.Context, data *PurPurchaseReceipt) (sql.Result, error)
		FindOne(ctx context.Context, id int64) (*PurPurchaseReceipt, error)
		FindOneByReceiptId(ctx context.Context, receiptId string) (*PurPurchaseReceipt, error)
		Update(ctx context.Context, data *PurPurchaseReceipt) error
		Delete(ctx context.Context, id int64) error
	}

	defaultPurPurchaseReceiptModel struct {
		conn  sqlx.SqlConn
		table string
	}

	PurPurchaseReceipt struct {
		Id          int64          `db:"id"`
		ReceiptId   string         `db:"receipt_id"`
		TenantId    string         `db:"tenant_id"`
		ReceiptNo   string         `db:"receipt_no"`
		OrderId     string         `db:"order_id"`
		ReceiptDate time.Time      `db:"receipt_date"`
		ReceivedBy  string         `db:"received_by"`
		Status      int64          `db:"status"`
		Remark      sql.NullString `db:"remark"`
		CreatedAt   time.Time      `db:"created_at"`
	}
)

func newPurPurchaseReceiptModel(conn sqlx.SqlConn) *defaultPurPurchaseReceiptModel {
	return &defaultPurPurchaseReceiptModel{
		conn:  conn,
		table: "`pur_purchase_receipt`",
	}
}

func (m *defaultPurPurchaseReceiptModel) Insert(ctx context.Context, data *PurPurchaseReceipt) (sql.Result, error) {
	query := fmt.Sprintf("INSERT INTO %s (%s) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", m.table, purPurchaseReceiptRowsExpectAutoSet)
	return m.conn.ExecCtx(ctx, query,
		data.ReceiptId, data.TenantId, data.ReceiptNo, data.OrderId, data.ReceiptDate, data.ReceivedBy, data.Status, data.Remark)
}

func (m *defaultPurPurchaseReceiptModel) FindOne(ctx context.Context, id int64) (*PurPurchaseReceipt, error) {
	query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", purPurchaseReceiptRows, m.table)
	var resp PurPurchaseReceipt
	if err := m.conn.QueryRowCtx(ctx, &resp, query, id); err != nil {
		return nil, err
	}
	return &resp, nil
}

func (m *defaultPurPurchaseReceiptModel) FindOneByReceiptId(ctx context.Context, receiptId string) (*PurPurchaseReceipt, error) {
	query := fmt.Sprintf("SELECT %s FROM %s WHERE `receipt_id` = ? LIMIT 1", purPurchaseReceiptRows, m.table)
	var resp PurPurchaseReceipt
	if err := m.conn.QueryRowCtx(ctx, &resp, query, receiptId); err != nil {
		return nil, err
	}
	return &resp, nil
}

func (m *defaultPurPurchaseReceiptModel) Update(ctx context.Context, data *PurPurchaseReceipt) error {
	query := fmt.Sprintf("UPDATE %s SET %s WHERE `id` = ?", m.table, purPurchaseReceiptRowsWithPlaceHolder)
	_, err := m.conn.ExecCtx(ctx, query,
		data.ReceiptId, data.TenantId, data.ReceiptNo, data.OrderId, data.ReceiptDate, data.ReceivedBy, data.Status, data.Remark, data.Id)
	return err
}

func (m *defaultPurPurchaseReceiptModel) Delete(ctx context.Context, id int64) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE `id` = ?", m.table)
	_, err := m.conn.ExecCtx(ctx, query, id)
	return err
}

func (m *defaultPurPurchaseReceiptModel) tableName() string {
	return m.table
}