jx/internal/source/container.go

// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.

package source

import (
	"context"
_	"encoding/json"
	"fmt"
_	"gopkg.in/yaml.v3"
	"net/url"
_	"path/filepath"
	"decl/internal/resource"
_	"os"
_	"io"
	"github.com/docker/docker/api/types/container"
	"github.com/docker/docker/client"
	"log/slog"
)

type Container struct {
	apiClient resource.ContainerClient
}

func NewContainer(containerClientApi resource.ContainerClient) *Container {
	var apiClient resource.ContainerClient = containerClientApi
	if apiClient == nil {
		var err error
		apiClient, err = client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
		if err != nil {
			panic(err)
		}
	}
	return &Container{
		apiClient: apiClient,
	}
}

func init() {
	SourceTypes.Register([]string{"container"}, func(u *url.URL) DocSource {
		c := NewContainer(nil)
		return c
	})

}

func (c *Container) Type() string { return "container" }

func (c *Container) ExtractResources(filter ResourceSelector) ([]*resource.Document, error) {
	var extractErr error
	ctx := context.Background()
	slog.Info("container source ExtractResources()", "container", c) 
	containers, err := c.apiClient.ContainerList(ctx, container.ListOptions{All: true})
	if err != nil {
		return nil, err
	}

	document := resource.NewDocument()
	for _, container := range containers {
		runningContainer := resource.NewContainer(nil)
		if inspectErr := runningContainer.Inspect(ctx, container.ID); inspectErr != nil {
			extractErr = fmt.Errorf("%w: %w", extractErr, inspectErr)
		}
		document.AddResourceDeclaration("container", runningContainer)
	}

	return []*resource.Document{document}, extractErr
}