package docker

import (
	"context"
	"sort"

	"github.com/cuigh/swirl/misc"
	"github.com/docker/docker/api/types"
	"github.com/docker/docker/api/types/filters"
	"github.com/docker/docker/api/types/volume"
	"github.com/docker/docker/client"
)

// VolumeList return volumes on the host.
func (d *Docker) VolumeList(ctx context.Context, name string, pageIndex, pageSize int) (volumes []*types.Volume, total int, err error) {
	var (
		c    *client.Client
		resp volume.VolumeListOKBody
	)

	c, err = d.client()
	if err != nil {
		return
	}

	f := filters.NewArgs()
	//f.Add("dangling", "true")
	//f.Add("driver", "xx")
	if name != "" {
		f.Add("name", name)
	}
	resp, err = c.VolumeList(ctx, f)
	if err != nil {
		return
	}
	sort.Slice(resp.Volumes, func(i, j int) bool {
		return resp.Volumes[i].Name < resp.Volumes[j].Name
	})

	total = len(resp.Volumes)
	start, end := misc.Page(total, pageIndex, pageSize)
	volumes = resp.Volumes[start:end]
	return
}

// VolumeCreate create a volume.
func (d *Docker) VolumeCreate(ctx context.Context, options *volume.VolumeCreateBody) error {
	return d.call(func(c *client.Client) (err error) {
		_, err = c.VolumeCreate(ctx, *options)
		return
	})
}

// VolumeRemove remove a volume.
func (d *Docker) VolumeRemove(ctx context.Context, name string) error {
	return d.call(func(c *client.Client) (err error) {
		return c.VolumeRemove(ctx, name, false)
	})
}

// VolumePrune remove all unused volumes.
func (d *Docker) VolumePrune(ctx context.Context, ) (report types.VolumesPruneReport, err error) {
	err = d.call(func(c *client.Client) (err error) {
		report, err = c.VolumesPrune(ctx, filters.NewArgs())
		return
	})
	return
}

// VolumeInspect return volume raw information.
func (d *Docker) VolumeInspect(ctx context.Context, name string) (vol types.Volume, raw []byte, err error) {
	err = d.call(func(c *client.Client) error {
		vol, raw, err = c.VolumeInspectWithRaw(ctx, name)
		return err
	})
	return
}