package resolver

// This file will be automatically regenerated based on the schema, any resolver implementations
// will be copied through when generating and any unknown code will be moved to the end.
// Code generated by github.com/99designs/gqlgen version v0.17.45

import (
	"context"

	"edge-infra.dev/pkg/edge/api/graph/model"
)

// CreateCapability is the resolver for the createCapability field.
func (r *mutationResolver) CreateCapability(ctx context.Context, capInput model.CapabilityInput) (*model.Capability, error) {
	return r.CapabilityService.CreateCapability(ctx, capInput)
}

// UpdateCapability is the resolver for the updateCapability field.
func (r *mutationResolver) UpdateCapability(ctx context.Context, capUpdateInput model.CapabilityUpdateInput) (bool, error) {
	if err := r.CapabilityService.UpdateCapability(ctx, &capUpdateInput); err != nil {
		return false, err
	}
	return true, nil
}

// AddCapability is the resolver for the addCapability field.
func (r *mutationResolver) AddCapability(ctx context.Context, capMapping model.CapabilityBannerMappingInput) (bool, error) {
	if err := r.CapabilityService.AddCapability(ctx, &capMapping); err != nil {
		return false, err
	}
	return true, nil
}

// RemoveCapability is the resolver for the removeCapability field.
func (r *mutationResolver) RemoveCapability(ctx context.Context, capMapping model.CapabilityBannerMappingInput) (bool, error) {
	if err := r.CapabilityService.RemoveCapability(ctx, &capMapping); err != nil {
		return false, err
	}
	return true, nil
}

// Capabilities is the resolver for the capabilities field.
func (r *queryResolver) Capabilities(ctx context.Context, bannerEdgeID *string) ([]*model.Capability, error) {
	if bannerEdgeID != nil {
		return r.CapabilityService.ListCapabilitiesByBanner(ctx, bannerEdgeID)
	}
	return r.CapabilityService.ListCapabilities(ctx)
}