package repository

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

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

type FactoryRepo struct {
	db *sql.DB
}

func NewFactoryRepo(db *sql.DB) *FactoryRepo {
	return &FactoryRepo{db: db}
}

func (r *FactoryRepo) ListRelationships(ctx context.Context, companyID string) ([]model.CompanyRelationship, error) {
	query := `SELECT cr.id, cr.purchaser_company_id, cr.factory_company_id,
		cr.factory_type, cr.status, cr.created_at, cr.updated_at,
		COALESCE(c.name, '') as factory_name
		FROM ilm_company_relationship cr
		LEFT JOIN ilm_company c ON c.id = cr.factory_company_id
		WHERE cr.purchaser_company_id = ? AND cr.status = 'active'
		ORDER BY cr.created_at DESC`

	rows, err := r.db.QueryContext(ctx, query, companyID)
	if err != nil {
		return nil, fmt.Errorf("list relationships: %w", err)
	}
	defer rows.Close()

	var rels []model.CompanyRelationship
	for rows.Next() {
		var rel model.CompanyRelationship
		if err := rows.Scan(
			&rel.ID, &rel.PurchaserCompanyID, &rel.FactoryCompanyID,
			&rel.FactoryType, &rel.Status, &rel.CreatedAt, &rel.UpdatedAt,
			&rel.FactoryName,
		); err != nil {
			return nil, fmt.Errorf("scan relationship: %w", err)
		}
		rels = append(rels, rel)
	}
	return rels, nil
}