router.go

// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
// or more contributor license agreements. Licensed under the Elastic License;
// you may not use this file except in compliance with the Elastic License.

package fleet

import (
	"net/http"

	"github.com/elastic/fleet-server/v7/internal/pkg/bulk"
	"github.com/elastic/fleet-server/v7/internal/pkg/logger"
	"github.com/elastic/fleet-server/v7/internal/pkg/policy"
	"github.com/julienschmidt/httprouter"
	"github.com/rs/zerolog/log"
)

const (
	ROUTE_STATUS    = "/api/status"
	ROUTE_ENROLL    = "/api/fleet/agents/:id"
	ROUTE_CHECKIN   = "/api/fleet/agents/:id/checkin"
	ROUTE_ACKS      = "/api/fleet/agents/:id/acks"
	ROUTE_ARTIFACTS = "/api/fleet/artifacts/:id/:sha2"
)

type Router struct {
	bulker bulk.Bulk
	ver    string
	ct     *CheckinT
	et     *EnrollerT
	at     *ArtifactT
	ack    *AckT
	sm     policy.SelfMonitor
}

func NewRouter(bulker bulk.Bulk, ct *CheckinT, et *EnrollerT, at *ArtifactT, ack *AckT, sm policy.SelfMonitor) *httprouter.Router {

	r := Router{
		bulker: bulker,
		ct:     ct,
		et:     et,
		sm:     sm,
		at:     at,
		ack:    ack,
	}

	routes := []struct {
		method  string
		path    string
		handler httprouter.Handle
	}{
		{
			http.MethodGet,
			ROUTE_STATUS,
			r.handleStatus,
		},
		{
			http.MethodPost,
			ROUTE_ENROLL,
			r.handleEnroll,
		},
		{
			http.MethodPost,
			ROUTE_CHECKIN,
			r.handleCheckin,
		},
		{
			http.MethodPost,
			ROUTE_ACKS,
			r.handleAcks,
		},
		{
			http.MethodGet,
			ROUTE_ARTIFACTS,
			r.handleArtifacts,
		},
	}

	router := httprouter.New()

	// Install routes
	for _, rte := range routes {
		log.Info().
			Str("method", rte.method).
			Str("path", rte.path).
			Msg("Server install route")

		router.Handle(
			rte.method,
			rte.path,
			logger.HttpHandler(rte.handler),
		)
	}

	return router
}