mirror of
				https://github.com/steveiliop56/tinyauth.git
				synced 2025-11-03 23:55:44 +00:00 
			
		
		
		
	* wip: add middlewares * refactor: use context fom middleware in handlers * refactor: use controller approach in handlers * refactor: move oauth providers into services (non-working) * feat: create oauth broker service * refactor: use a boostrap service to bootstrap the app * refactor: split utils into smaller files * refactor: use more clear name for frontend assets * feat: allow customizability of resources dir * fix: fix typo in ui middleware * fix: validate resource file paths in ui middleware * refactor: move resource handling to a controller * feat: add some logging * fix: configure middlewares before groups * fix: use correct api path in login mutation * fix: coderabbit suggestions * fix: further coderabbit suggestions
		
			
				
	
	
		
			101 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			101 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package service
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"strings"
 | 
						|
	"tinyauth/internal/config"
 | 
						|
	"tinyauth/internal/utils"
 | 
						|
 | 
						|
	"slices"
 | 
						|
 | 
						|
	container "github.com/docker/docker/api/types/container"
 | 
						|
	"github.com/docker/docker/client"
 | 
						|
	"github.com/rs/zerolog/log"
 | 
						|
)
 | 
						|
 | 
						|
type DockerService struct {
 | 
						|
	Client  *client.Client
 | 
						|
	Context context.Context
 | 
						|
}
 | 
						|
 | 
						|
func NewDockerService() *DockerService {
 | 
						|
	return &DockerService{}
 | 
						|
}
 | 
						|
 | 
						|
func (docker *DockerService) Init() error {
 | 
						|
	client, err := client.NewClientWithOpts(client.FromEnv)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	ctx := context.Background()
 | 
						|
	client.NegotiateAPIVersion(ctx)
 | 
						|
 | 
						|
	docker.Client = client
 | 
						|
	docker.Context = ctx
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func (docker *DockerService) GetContainers() ([]container.Summary, error) {
 | 
						|
	containers, err := docker.Client.ContainerList(docker.Context, container.ListOptions{})
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return containers, nil
 | 
						|
}
 | 
						|
 | 
						|
func (docker *DockerService) InspectContainer(containerId string) (container.InspectResponse, error) {
 | 
						|
	inspect, err := docker.Client.ContainerInspect(docker.Context, containerId)
 | 
						|
	if err != nil {
 | 
						|
		return container.InspectResponse{}, err
 | 
						|
	}
 | 
						|
	return inspect, nil
 | 
						|
}
 | 
						|
 | 
						|
func (docker *DockerService) DockerConnected() bool {
 | 
						|
	_, err := docker.Client.Ping(docker.Context)
 | 
						|
	return err == nil
 | 
						|
}
 | 
						|
 | 
						|
func (docker *DockerService) GetLabels(app string, domain string) (config.Labels, error) {
 | 
						|
	isConnected := docker.DockerConnected()
 | 
						|
 | 
						|
	if !isConnected {
 | 
						|
		log.Debug().Msg("Docker not connected, returning empty labels")
 | 
						|
		return config.Labels{}, nil
 | 
						|
	}
 | 
						|
 | 
						|
	containers, err := docker.GetContainers()
 | 
						|
	if err != nil {
 | 
						|
		return config.Labels{}, err
 | 
						|
	}
 | 
						|
 | 
						|
	for _, container := range containers {
 | 
						|
		inspect, err := docker.InspectContainer(container.ID)
 | 
						|
		if err != nil {
 | 
						|
			log.Warn().Str("id", container.ID).Err(err).Msg("Error inspecting container, skipping")
 | 
						|
			continue
 | 
						|
		}
 | 
						|
 | 
						|
		labels, err := utils.GetLabels(inspect.Config.Labels)
 | 
						|
		if err != nil {
 | 
						|
			log.Warn().Str("id", container.ID).Err(err).Msg("Error getting container labels, skipping")
 | 
						|
			continue
 | 
						|
		}
 | 
						|
 | 
						|
		// Check if the container matches the ID or domain
 | 
						|
		if slices.Contains(labels.Domain, domain) {
 | 
						|
			log.Debug().Str("id", inspect.ID).Msg("Found matching container by domain")
 | 
						|
			return labels, nil
 | 
						|
		}
 | 
						|
 | 
						|
		if strings.TrimPrefix(inspect.Name, "/") == app {
 | 
						|
			log.Debug().Str("id", inspect.ID).Msg("Found matching container by name")
 | 
						|
			return labels, nil
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	log.Debug().Msg("No matching container found, returning empty labels")
 | 
						|
	return config.Labels{}, nil
 | 
						|
}
 |