package vfs

import (
	"context"
	"errors"

	"github.com/containerd/log"
	"github.com/docker/docker/quota"
)

type driverQuota struct {
	quotaCtl *quota.Control
	quotaOpt quota.Quota
}

func setupDriverQuota(driver *Driver) {
	if quotaCtl, err := quota.NewControl(driver.home); err == nil {
		driver.quotaCtl = quotaCtl
	} else if !errors.Is(err, quota.ErrQuotaNotSupported) {
		log.G(context.TODO()).Warnf("Unable to setup quota: %v\n", err)
	}
}

func (d *Driver) setQuotaOpt(size uint64) error {
	d.quotaOpt.Size = size
	return nil
}

func (d *Driver) getQuotaOpt() uint64 {
	return d.quotaOpt.Size
}

func (d *Driver) setupQuota(dir string, size uint64) error {
	return d.quotaCtl.SetQuota(dir, quota.Quota{Size: size})
}

func (d *Driver) quotaSupported() bool {
	return d.quotaCtl != nil
}
