package knowledgeqa

import (
	"context"

	"github.com/tidbcloud/AgentTeam/internal/store/organization"
)

type organizationSummaryResolver interface {
	GetWorkspaceOrganizationForUser(
		context.Context,
		string,
		string,
	) (organization.OrganizationSummary, error)
}

type organizationResolver struct {
	next organizationSummaryResolver
}

func NewOrganizationResolver(next organizationSummaryResolver) OrganizationResolver {
	return organizationResolver{next: next}
}

func (resolver organizationResolver) GetWorkspaceOrganizationForUser(
	ctx context.Context,
	userID string,
	workspaceID string,
) (Organization, error) {
	summary, err := resolver.next.GetWorkspaceOrganizationForUser(ctx, userID, workspaceID)
	return Organization{ID: summary.ID}, err
}