68 lines
1.6 KiB
Go
68 lines
1.6 KiB
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
|
|
}
|