package repository

import (
	"context"
	"database/sql"

	"github.com/muyuqingfeng/iloom/textile-service/internal/model"
)

type YarnRepo struct {
	db *sql.DB
}

func NewYarnRepo(db *sql.DB) *YarnRepo {
	return &YarnRepo{db: db}
}

func (r *YarnRepo) List(ctx context.Context, companyID string) ([]model.YarnStock, error) {
	query := `
		SELECT id, company_id, name, spec, quantity, unit, min_stock, warehouse_id, updated_at
		FROM ilm_yarn_stock
		WHERE company_id = ?
		ORDER BY updated_at DESC`

	rows, err := r.db.QueryContext(ctx, query, companyID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var stocks []model.YarnStock
	for rows.Next() {
		var s model.YarnStock
		if err := rows.Scan(
			&s.ID, &s.CompanyID, &s.Name, &s.Spec, &s.Quantity,
			&s.Unit, &s.MinStock, &s.WarehouseID, &s.UpdatedAt,
		); err != nil {
			return nil, err
		}
		stocks = append(stocks, s)
	}
	return stocks, rows.Err()
}

func (r *YarnRepo) Create(ctx context.Context, y *model.YarnStock) error {
	query := `
		INSERT INTO ilm_yarn_stock (id, company_id, name, spec, quantity, unit, min_stock, warehouse_id, updated_at)
		VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`

	_, err := r.db.ExecContext(ctx, query,
		y.ID, y.CompanyID, y.Name, y.Spec, y.Quantity,
		y.Unit, y.MinStock, y.WarehouseID, y.UpdatedAt,
	)
	return err
}

func (r *YarnRepo) Update(ctx context.Context, y *model.YarnStock) error {
	query := `
		UPDATE ilm_yarn_stock
		SET name = ?, spec = ?, unit = ?, min_stock = ?, warehouse_id = ?, updated_at = ?
		WHERE id = ? AND company_id = ?`

	_, err := r.db.ExecContext(ctx, query,
		y.Name, y.Spec, y.Unit, y.MinStock, y.WarehouseID, y.UpdatedAt,
		y.ID, y.CompanyID,
	)
	return err
}

func (r *YarnRepo) UpdateQuantity(ctx context.Context, tx *sql.Tx, id string, delta float64) error {
	query := `
		UPDATE ilm_yarn_stock
		SET quantity = quantity + ?, updated_at = NOW()
		WHERE id = ?`

	_, err := tx.ExecContext(ctx, query, delta, id)
	return err
}

func (r *YarnRepo) CreateRecord(ctx context.Context, tx *sql.Tx, rec *model.YarnStockRecord) error {
	query := `
		INSERT INTO ilm_yarn_stock_record
		(id, yarn_stock_id, company_id, record_type, quantity, unit, batch_no, plan_id, notes, operator_id, created_at)
		VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`

	_, err := tx.ExecContext(ctx, query,
		rec.ID, rec.YarnStockID, rec.CompanyID, rec.RecordType, rec.Quantity,
		rec.Unit, rec.BatchNo, rec.PlanID, rec.Notes, rec.OperatorID, rec.CreatedAt,
	)
	return err
}

func (r *YarnRepo) Count(ctx context.Context, companyID string) (int, error) {
	var count int
	err := r.db.QueryRowContext(ctx,
		`SELECT COUNT(*) FROM ilm_yarn_stock WHERE company_id = ?`, companyID,
	).Scan(&count)
	return count, err
}

func (r *YarnRepo) CountLowStock(ctx context.Context, companyID string) (int, error) {
	var count int
	err := r.db.QueryRowContext(ctx,
		`SELECT COUNT(*) FROM ilm_yarn_stock WHERE company_id = ? AND quantity <= min_stock`, companyID,
	).Scan(&count)
	return count, err
}