package device

import (
	"context"
	"net/http"

	"github.com/go-kit/kit/endpoint"
	"github.com/micromdm/micromdm/pkg/httputil"
)

func (svc *DeviceService) RemoveDevices(ctx context.Context, opt RemoveDevicesOptions) error {
	for _, udid := range opt.UDIDs {
		err := svc.store.DeleteByUDID(ctx, udid)
		if err != nil {
			return err
		}
	}

	for _, serial := range opt.Serials {
		err := svc.store.DeleteBySerial(ctx, serial)
		if err != nil {
			return err
		}
	}

	return nil
}

type removeDevicesRequest struct{ Opts RemoveDevicesOptions }

type removeDevicesResponse struct {
	Err error `json:"err,omitempty"`
}

func (r removeDevicesResponse) Failed() error { return r.Err }

func decodeRemoveDevicesRequest(ctx context.Context, r *http.Request) (interface{}, error) {
	var req removeDevicesRequest
	err := httputil.DecodeJSONRequest(r, &req)
	return req, err
}

func decodeRemoveDevicesResponse(_ context.Context, r *http.Response) (interface{}, error) {
	var resp removeDevicesResponse
	err := httputil.DecodeJSONResponse(r, &resp)
	return resp, err
}

func MakeRemoveDevicesEndpoint(svc Service) endpoint.Endpoint {
	return func(ctx context.Context, request interface{}) (response interface{}, err error) {
		req := request.(removeDevicesRequest)
		err = svc.RemoveDevices(ctx, req.Opts)
		return removeDevicesResponse{
			Err: err,
		}, nil
	}
}

func (e Endpoints) RemoveDevices(ctx context.Context, Opts RemoveDevicesOptions) error {
	request := removeDevicesRequest{Opts: Opts}
	resp, err := e.RemoveDevicesEndpoint(ctx, request)
	if err != nil {
		return err
	}
	return resp.(removeDevicesResponse).Err
}