package service

import (
	"context"
	"fmt"

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

type PlanService struct {
	repo *repository.PlanRepo
}

func NewPlanService(repo *repository.PlanRepo) *PlanService {
	return &PlanService{repo: repo}
}

func (s *PlanService) List(ctx context.Context, companyID string) ([]model.WashingPlan, error) {
	return s.repo.List(ctx, companyID)
}

func (s *PlanService) GetByID(ctx context.Context, id, companyID string) (*model.WashingPlan, error) {
	plan, err := s.repo.GetByID(ctx, id)
	if err != nil {
		return nil, err
	}
	if plan == nil {
		return nil, fmt.Errorf("washing plan not found")
	}
	if plan.WashingFactoryID == nil || *plan.WashingFactoryID != companyID {
		return nil, fmt.Errorf("washing plan not found")
	}
	return plan, nil
}

func (s *PlanService) UpdateStatus(ctx context.Context, id, companyID, status string) error {
	plan, err := s.repo.GetByID(ctx, id)
	if err != nil {
		return err
	}
	if plan == nil || plan.WashingFactoryID == nil || *plan.WashingFactoryID != companyID {
		return fmt.Errorf("washing plan not found")
	}
	return s.repo.UpdateStatus(ctx, id, status)
}

func (s *PlanService) Pending(ctx context.Context, companyID string) ([]model.WashingPlan, error) {
	return s.repo.Pending(ctx, companyID)
}

func (s *PlanService) Completed(ctx context.Context, companyID string) ([]model.WashingPlan, error) {
	return s.repo.Completed(ctx, companyID)
}