mirror of
https://github.com/empayre/fleet.git
synced 2024-11-06 17:05:18 +00:00
fa10dbe0a8
This change allows the images in Fleet emails to load properly from any device with connectivity to github.com. Previously, emails might try to load resources from a Kolide server not accessible from the email client. The asset URL will be based on the most recent git tag to accomodate backwards-compatibility if the assets in the repo change. Closes #1471
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
// Package service holds the implementation of the kolide service interface and the HTTP endpoints
|
|
// for the API
|
|
package service
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/WatchBeam/clock"
|
|
kitlog "github.com/go-kit/kit/log"
|
|
"github.com/kolide/fleet/server/config"
|
|
"github.com/kolide/fleet/server/kolide"
|
|
"github.com/kolide/fleet/server/logging"
|
|
"github.com/kolide/fleet/server/sso"
|
|
"github.com/kolide/kit/version"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// NewService creates a new service from the config struct
|
|
func NewService(ds kolide.Datastore, resultStore kolide.QueryResultStore,
|
|
logger kitlog.Logger, config config.KolideConfig, mailService kolide.MailService,
|
|
c clock.Clock, sso sso.SessionStore) (kolide.Service, error) {
|
|
var svc kolide.Service
|
|
|
|
osqueryLogger, err := logging.New(config, logger)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "initializing osquery logging")
|
|
}
|
|
|
|
svc = service{
|
|
ds: ds,
|
|
resultStore: resultStore,
|
|
logger: logger,
|
|
config: config,
|
|
clock: c,
|
|
osqueryLogWriter: osqueryLogger,
|
|
mailService: mailService,
|
|
ssoSessionStore: sso,
|
|
metaDataClient: &http.Client{
|
|
Timeout: 5 * time.Second,
|
|
},
|
|
}
|
|
svc = validationMiddleware{svc, ds, sso}
|
|
return svc, nil
|
|
}
|
|
|
|
type service struct {
|
|
ds kolide.Datastore
|
|
resultStore kolide.QueryResultStore
|
|
logger kitlog.Logger
|
|
config config.KolideConfig
|
|
clock clock.Clock
|
|
|
|
osqueryLogWriter *logging.OsqueryLogger
|
|
|
|
mailService kolide.MailService
|
|
ssoSessionStore sso.SessionStore
|
|
metaDataClient *http.Client
|
|
}
|
|
|
|
func (s service) SendEmail(mail kolide.Email) error {
|
|
return s.mailService.SendEmail(mail)
|
|
}
|
|
|
|
func (s service) Clock() clock.Clock {
|
|
return s.clock
|
|
}
|
|
|
|
type validationMiddleware struct {
|
|
kolide.Service
|
|
ds kolide.Datastore
|
|
ssoSessionStore sso.SessionStore
|
|
}
|
|
|
|
// getAssetURL gets the URL prefix used for retrieving assets from Github. This
|
|
// function will determine the appropriate version to use, and create a URL
|
|
// prefix for retrieving assets from that tag.
|
|
func getAssetURL() template.URL {
|
|
v := version.Version().Version
|
|
tag := strings.Split(v, "-")[0]
|
|
if tag == "unknown" {
|
|
tag = "master"
|
|
}
|
|
|
|
return template.URL("https://github.com/kolide/fleet/blob/" + tag)
|
|
}
|