package controlplane

import (
	"context"

	controlplanev1 "proxy-pool/gen/controlplane/v1"
)

type GeneratedClient struct {
	client controlplanev1.CheckerControlPlaneClient
}

func NewGeneratedClient(client controlplanev1.CheckerControlPlaneClient) *GeneratedClient {
	if client == nil {
		return nil
	}
	return &GeneratedClient{client: client}
}

func (client *GeneratedClient) StreamCheckTasks(
	ctx context.Context,
	request *controlplanev1.StreamCheckTasksRequest,
) (TaskStream, error) {
	if client == nil || client.client == nil {
		return nil, ErrInvalidRunner
	}
	return client.client.StreamCheckTasks(ctx, request)
}

func (client *GeneratedClient) ReportObservations(
	ctx context.Context,
	batch *controlplanev1.ObservationBatch,
) (*controlplanev1.ReportObservationsResponse, error) {
	if client == nil || client.client == nil {
		return nil, ErrInvalidRunner
	}
	return client.client.ReportObservations(ctx, batch)
}

var _ Client = (*GeneratedClient)(nil)