res_levis_backend/internal/pkg/location/inference.go

package location

import (
	"context"
	"crypto/tls"
	"net/http"

	"github.com/AFASystems/presence/internal/pkg/apiclient"
	"github.com/AFASystems/presence/internal/pkg/config"
	"github.com/AFASystems/presence/internal/pkg/model"
)

// Inferencer returns inferred positions (e.g. from an AI/ML service).
type Inferencer interface {
	Infer(ctx context.Context, cfg *config.Config) (model.PositionResponse, error)
}

// DefaultInferencer uses apiclient to get token and call the inference API.
type DefaultInferencer struct {
	Client *http.Client
	Token  string
}

// NewDefaultInferencer creates an inferencer with optional TLS skip verify (e.g. from config.TLSInsecureSkipVerify).
func NewDefaultInferencer(skipTLSVerify bool) *DefaultInferencer {
	tr := &http.Transport{}
	tr.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
	return &DefaultInferencer{
		Client: &http.Client{Transport: tr},
		Token:  "",
	}
}

// Infer gets a token and calls the inference API.
func (d *DefaultInferencer) Infer(ctx context.Context, cfg *config.Config) (model.PositionResponse, error) {
	if d.Token == "" {
		token, err := apiclient.GetToken(ctx, cfg, d.Client)
		if err != nil {
			return model.PositionResponse{}, err
		}
		d.Token = token
	}

	response, err := apiclient.InferPosition(d.Token, d.Client, cfg)
	if err != nil {
		token, err := apiclient.GetToken(ctx, cfg, d.Client)
		if err != nil {
			return model.PositionResponse{}, err
		}
		d.Token = token
		response, err = apiclient.InferPosition(d.Token, d.Client, cfg)
		if err != nil {
			return model.PositionResponse{}, err
		}
	}

	return response, nil
}