package client

import (
	"context"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/api/types/filters"
	"github.com/docker/docker/client"
)

func GetVolumes(cl *client.Client, ctx context.Context, server string, fs filters.Args) ([]*types.Volume, error) {
	volumeListOKBody, err := cl.VolumeList(ctx, fs)
	volumeList := volumeListOKBody.Volumes
	if err != nil {
		return volumeList, err
	}

	return volumeList, nil
}

func GetVolumeNames(volumes []*types.Volume) []string {
	var volumeNames []string

	for _, vol := range volumes {
		volumeNames = append(volumeNames, vol.Name)
	}

	return volumeNames
}

func RemoveVolumes(cl *client.Client, ctx context.Context, server string, volumeNames []string, force bool) error {
	for _, volName := range volumeNames {
		err := cl.VolumeRemove(ctx, volName, force)
		if err != nil {
			return err
		}
	}

	return nil
}