package azure

import (
	"context"

	"github.com/Azure/azure-sdk-for-go/services/containerinstance/mgmt/2018-10-01/containerinstance"
	"github.com/Azure/go-autorest/autorest/azure/auth"
	"github.com/pkg/errors"

	"github.com/docker/api/backend"
	"github.com/docker/api/containers"
	apicontext "github.com/docker/api/context"
	"github.com/docker/api/context/store"
)

type containerService struct {
	cgc containerinstance.ContainerGroupsClient
	ctx store.AciContext
}

func init() {
	backend.Register("aci", "aci", func(ctx context.Context) (interface{}, error) {
		return New(ctx)
	})
}

func getter() interface{} {
	return &store.AciContext{}
}

func New(ctx context.Context) (containers.ContainerService, error) {
	cc := apicontext.CurrentContext(ctx)
	s, err := store.New()
	if err != nil {
		return nil, err
	}
	m, err := s.Get(cc, getter)
	if err != nil {
		return nil, errors.Wrap(err, "wrong context type")
	}
	tc, _ := m.Metadata.Data.(store.AciContext)

	auth, _ := auth.NewAuthorizerFromCLI()
	containerGroupsClient := containerinstance.NewContainerGroupsClient(tc.SubscriptionID)
	containerGroupsClient.Authorizer = auth

	return &containerService{
		cgc: containerGroupsClient,
		ctx: tc,
	}, nil
}

func (cs *containerService) List(ctx context.Context) ([]containers.Container, error) {
	var cg []containerinstance.ContainerGroup
	result, err := cs.cgc.ListByResourceGroup(ctx, cs.ctx.ResourceGroup)

	if err != nil {
		return []containers.Container{}, err
	}

	for result.NotDone() {
		cg = append(cg, result.Values()...)
		if err := result.NextWithContext(ctx); err != nil {
			return []containers.Container{}, err
		}
	}

	res := []containers.Container{}
	for _, c := range cg {
		cc := *c.Containers
		for _, d := range cc {
			res = append(res, containers.Container{
				ID:    *c.Name,
				Image: *d.Image,
				// Command: strings.Join(*d.ContainerProperties.Command, " "), // TODO command can be null
			})
		}
	}

	return res, nil
}