package docker

import (
	"context"
	"strings"
	appTypes "tinyauth/internal/types"
	"tinyauth/internal/utils"

	apiTypes "github.com/docker/docker/api/types"
	containerTypes "github.com/docker/docker/api/types/container"
	"github.com/docker/docker/client"
	"github.com/rs/zerolog/log"
)

func NewDocker() *Docker {
	return &Docker{}
}

type Docker struct {
	Client  *client.Client
	Context context.Context
}

func (docker *Docker) Init() error {
	// Create a new docker client
	apiClient, err := client.NewClientWithOpts(client.FromEnv)

	// Check if there was an error
	if err != nil {
		return err
	}

	// Set the context and api client
	docker.Context = context.Background()
	docker.Client = apiClient

	// Done
	return nil
}

func (docker *Docker) GetContainers() ([]apiTypes.Container, error) {
	// Get the list of containers
	containers, err := docker.Client.ContainerList(docker.Context, containerTypes.ListOptions{})

	// Check if there was an error
	if err != nil {
		return nil, err
	}

	// Return the containers
	return containers, nil
}

func (docker *Docker) InspectContainer(containerId string) (apiTypes.ContainerJSON, error) {
	// Inspect the container
	inspect, err := docker.Client.ContainerInspect(docker.Context, containerId)

	// Check if there was an error
	if err != nil {
		return apiTypes.ContainerJSON{}, err
	}

	// Return the inspect
	return inspect, nil
}

func (docker *Docker) DockerConnected() bool {
	// Ping the docker client if there is an error it is not connected
	_, err := docker.Client.Ping(docker.Context)
	return err == nil
}

func (docker *Docker) ContainerAction(appId string, runCheck func(labels appTypes.TinyauthLabels) (bool, error)) (bool, error) {
	// Check if we have access to the Docker API
	isConnected := docker.DockerConnected()

	// If we don't have access, it is assumed that the check passed
	if !isConnected {
		log.Debug().Msg("Docker not connected, passing check")
		return true, nil
	}

	// Get the containers
	containers, containersErr := docker.GetContainers()

	// If there is an error, return false
	if containersErr != nil {
		return false, containersErr
	}

	log.Debug().Msg("Got containers")

	// Loop through the containers
	for _, container := range containers {
		// Inspect the container
		inspect, inspectErr := docker.InspectContainer(container.ID)

		// If there is an error, return false
		if inspectErr != nil {
			return false, inspectErr
		}

		// Get the container name (for some reason it is /name)
		containerName := strings.Split(inspect.Name, "/")[1]

		// There is a container with the same name as the app ID
		if containerName == appId {
			log.Debug().Str("container", containerName).Msg("Found container")

			// Get only the tinyauth labels in a struct
			labels := utils.GetTinyauthLabels(inspect.Config.Labels)

			log.Debug().Msg("Got labels")

			// Run the check
			return runCheck(labels)
		}

	}

	log.Debug().Msg("No matching container found, passing check")

	// If no matching container is found, pass check
	return true, nil
}