package plugin

import (
	"context"
	"io"

	"github.com/moby/moby/api/types/plugin"
	"github.com/moby/moby/api/types/system"
	"github.com/moby/moby/client"
)

type fakeClient struct {
	client.Client
	pluginCreateFunc  func(createContext io.Reader, options client.PluginCreateOptions) error
	pluginDisableFunc func(name string, options client.PluginDisableOptions) error
	pluginEnableFunc  func(name string, options client.PluginEnableOptions) error
	pluginRemoveFunc  func(name string, options client.PluginRemoveOptions) error
	pluginInstallFunc func(name string, options client.PluginInstallOptions) (io.ReadCloser, error)
	pluginListFunc    func(options client.PluginListOptions) (plugin.ListResponse, error)
	pluginInspectFunc func(name string) (*plugin.Plugin, []byte, error)
	pluginUpgradeFunc func(name string, options client.PluginInstallOptions) (io.ReadCloser, error)
}

func (c *fakeClient) PluginCreate(_ context.Context, createContext io.Reader, options client.PluginCreateOptions) error {
	if c.pluginCreateFunc != nil {
		return c.pluginCreateFunc(createContext, options)
	}
	return nil
}

func (c *fakeClient) PluginEnable(_ context.Context, name string, options client.PluginEnableOptions) error {
	if c.pluginEnableFunc != nil {
		return c.pluginEnableFunc(name, options)
	}
	return nil
}

func (c *fakeClient) PluginDisable(_ context.Context, name string, options client.PluginDisableOptions) error {
	if c.pluginDisableFunc != nil {
		return c.pluginDisableFunc(name, options)
	}
	return nil
}

func (c *fakeClient) PluginRemove(_ context.Context, name string, options client.PluginRemoveOptions) error {
	if c.pluginRemoveFunc != nil {
		return c.pluginRemoveFunc(name, options)
	}
	return nil
}

func (c *fakeClient) PluginInstall(_ context.Context, name string, options client.PluginInstallOptions) (io.ReadCloser, error) {
	if c.pluginInstallFunc != nil {
		return c.pluginInstallFunc(name, options)
	}
	return nil, nil
}

func (c *fakeClient) PluginList(_ context.Context, options client.PluginListOptions) (plugin.ListResponse, error) {
	if c.pluginListFunc != nil {
		return c.pluginListFunc(options)
	}

	return plugin.ListResponse{}, nil
}

func (c *fakeClient) PluginInspectWithRaw(_ context.Context, name string) (*plugin.Plugin, []byte, error) {
	if c.pluginInspectFunc != nil {
		return c.pluginInspectFunc(name)
	}

	return nil, nil, nil
}

func (*fakeClient) Info(context.Context) (system.Info, error) {
	return system.Info{}, nil
}

func (c *fakeClient) PluginUpgrade(_ context.Context, name string, options client.PluginInstallOptions) (io.ReadCloser, error) {
	if c.pluginUpgradeFunc != nil {
		return c.pluginUpgradeFunc(name, options)
	}
	return nil, nil
}