mirror of
https://github.com/gravitl/netmaker.git
synced 2026-04-23 00:17:10 +08:00
292af315dd
* feat(go): add user schema; * feat(go): migrate to user schema; * feat(go): add audit fields; * feat(go): remove unused fields from the network model; * feat(go): add network schema; * feat(go): migrate to network schema; * refactor(go): add comment to clarify migration logic; * fix(go): test failures; * fix(go): test failures; * feat(go): change membership table to store memberships at all scopes; * feat(go): add schema for access grants; * feat(go): remove nameservers from new networks table; ensure db passed for schema functions; * feat(go): set max conns for sqlite to 1; * fix(go): issues updating user account status; * NM-236: streamline operations in HA mode * NM-236: only master pod should subscribe to updates from clients * refactor(go): remove converters and access grants; * refactor(go): add json tags in schema models; * refactor(go): rename file to migrate_v1_6_0.go; * refactor(go): add user groups and user roles tables; use schema tables; * refactor(go): inline get and list from schema package; * refactor(go): inline get network and list users from schema package; * fix(go): staticcheck issues; * fix(go): remove test not in use; fix test case; * fix(go): validate network; * fix(go): resolve static checks; * fix(go): new models errors; * fix(go): test errors; * fix(go): handle no records; * fix(go): add validations for user object; * fix(go): set correct extclient status; * fix(go): test error; * feat(go): make schema the base package; * feat(go): add host schema; * feat(go): use schema host everywhere; * feat(go): inline get host, list hosts and delete host; * feat(go): use non-ptr value; * feat(go): use save to upsert all fields; * feat(go): use save to upsert all fields; * feat(go): save turn endpoint as string; * feat(go): check for gorm error record not found; * fix(go): test failures; * fix(go): update all network fields; * fix(go): update all network fields; * feat(go): add paginated list networks api; * feat(go): add paginated list users api; * feat(go): add paginated list hosts api; * feat(go): add pagination to list groups api; * fix(go): comment; * fix(go): implement marshal and unmarshal text for custom types; * fix(go): implement marshal and unmarshal json for custom types; * fix(go): just use the old model for unmarshalling; * fix(go): implement marshal and unmarshal json for custom types; * NM-271:Import swap: compress/gzip replaced with github.com/klauspost/compress/gzip (2-4x faster, wire-compatible output). Added sync import. Two sync.Pool variables (gzipWriterPool, bufferPool): reuse gzip.Writer and bytes.Buffer across calls instead of allocating fresh ones per publish. compressPayload rewritten: pulls writer + buffer from pools, resets them, compresses at gzip.BestSpeed (level 1), copies the result out of the pooled buffer, and returns both objects to the pools. * feat(go): remove paginated list networks api; * feat(go): use custom paginated response object; * NM-271: Improve server scalability under high host count - Replace stdlib compress/gzip with klauspost/compress at BestSpeed and pool gzip writers and buffers via sync.Pool to eliminate compression as the dominant CPU hotspot. - Debounce peer update broadcasts with a 500ms resettable window capped at 3s max-wait, coalescing rapid-fire PublishPeerUpdate calls into a single broadcast cycle. - Cache HostPeerInfo (batch-refreshed by debounce worker) and HostPeerUpdate (stored as side-effect of each publish) so the pull API and peer_info API serve from pre-computed maps instead of triggering expensive per-host computations under thundering herd conditions. - Warm both caches synchronously at startup before the first publish cycle so early pull requests are served instantly. - Bound concurrent MQTT publishes to 5 via semaphore to prevent broker TCP buffer overflows that caused broken pipe disconnects. - Remove manual Disconnect+SetupMQTT from ConnectionLostHandler and rely on the paho client's built-in AutoReconnect; add a 5s retry wait in publish() to ride out brief reconnection windows. * NM-271: Reduce server CPU contention under high concurrent load - Cache ServerSettings with atomic.Value to eliminate repeated DB reads on every pull request (was 32+ goroutines blocked on read lock) - Batch UpdateNodeCheckin writes in memory, flush every 30s to reduce per-checkin write lock contention (was 88+ goroutines blocked) - Enable SQLite WAL mode + busy_timeout and remove global dbMutex; let SQLite handle concurrency natively (reads no longer block writes) - Move ResetFailedOverPeer/ResetAutoRelayedPeer to async in pull() handler since results don't affect the cached response - Skip no-op UpsertNode writes in failover/relay reset functions (early return when node has no failover/relay state) - Remove CheckHostPorts from hostUpdateFallback hot path - Switch to pure-Go SQLite driver (glebarez/sqlite), set CGO_ENABLED=0 * fix(go): ensure default values for page and per_page are used when not passed; * fix(go): rename v1.6.0 to v1.5.1; * fix(go): check for gorm.ErrRecordNotFound instead of database.IsEmptyRecord; * fix(go): use host id, not pending host id; * NM-271: Revert pure-Go SQLite and FIPS disable to verify impact Revert to CGO-based mattn/go-sqlite3 driver and re-enable FIPS to isolate whether these changes are still needed now that the global dbMutex has been removed and WAL mode is enabled. Keep WAL mode pragma with mattn-compatible DSN format. * feat(go): add filters to paginated apis; * feat(go): add filters to paginated apis; * feat(go): remove check for max username length; * feat(go): add filters to count as well; * feat(go): use library to check email address validity; * feat(go): ignore pagination if params not passed; * fix(go): pagination issues; * fix(go): check exists before using; * fix(go): remove debug log; * NM-271: rm debug logs * NM-271: check if caching is enabled * NM-271: add server sync mq topic for HA mode * NM-271: fix build * NM-271: push metrics in batch to exproter over api * NM-271: use basic auth for exporter metrics api * fix(go): use gorm err record not found; * NM-271: Add monitoring stack on demand * NM-271: -m arg for install script should only add monitoring stack * fix(go): use gorm err record not found; * NM-271: update docker compose file for prometheus * NM-271: update docker compose file for prometheus * fix(go): use user principal name when creating pending user; * fix(go): use schema package for consts; * NM-236: rm duplicate network hook * NM-271: add server topic to reset idp hooks on master node * fix(go): prevent disabling superadmin user; Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): swap is admin and is superadmin; Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): remove dead code block; https://github.com/gravitl/netmaker/pull/3910#discussion_r2928837937 * fix(go): incorrect message when trying to disable self; https://github.com/gravitl/netmaker/pull/3910#discussion_r2928837934 * NM-271: fix stale peers on reset_failovered pull and add HTTP timeout to metrics exporter Run the failover/relay reset synchronously in the pull handler so the response reflects post-reset topology instead of serving stale cached peers. Add a 30s timeout to the metrics exporter HTTP client to prevent PushAllMetricsToExporter from blocking the Keepalive loop. * NM-271: fix gzip pool corruption, MQTT topic mismatch, stale settings cache, and reduce redundant DB fetches - Only return gzip.Writer to pool after successful Close to prevent silently malformed MQTT payloads from a previously errored writer. - Fix serversync subscription to exact topic match since syncType is now in the message payload, not the topic path. - Prevent zero-value ServerSettings from being cached indefinitely when the DB record is missing or unmarshal fails on startup. - Return fetched hosts/nodes from RefreshHostPeerInfoCache so warmPeerCaches reuses them instead of querying the DB twice. - Compute fresh HostPeerUpdate on reset_failovered pull instead of serving stale cache, and store result back for subsequent requests. * NM-271: fix gzip writer pool leak, log checkin flush errors, and fix master pod ordinal parsing - Reset gzip.Writer to io.Discard before returning to pool so errored writers are never leaked or silently reused with corrupt state. - Track and log failed DB inserts in FlushNodeCheckins so operators have visibility when check-in timestamps are lost. - Parse StatefulSet pod ordinal as integer instead of using HasSuffix to prevent netmaker-10 from being misidentified as master pod. * NM-271: simplify masterpod logic * fix(go): use correct header; Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): return after error response; Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): use correct order of params; https://github.com/gravitl/netmaker/pull/3910#discussion_r2929593036 * fix(go): set default values for page and page size; use v2 instead of /list; * NM-271: use host name * Update mq/serversync.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * NM-271: fix duplicate serversynce case * NM-271: streamline gw updates * Update logic/auth.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * Update schema/user_roles.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): syntax error; * fix(go): set default values when page and per_page are not passed or 0; * fix(go): use uuid.parse instead of uuid.must parse; * fix(go): review errors; * fix(go): review errors; * Update controllers/user.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * Update controllers/user.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * NM-163: fix errors: * Update db/types/options.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * fix(go): persist return user in event; * Update db/types/options.go Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com> * NM-271: signal pull on ip changes * NM-163: duplicate lines of code * NM-163: fix(go): fix missing return and filter parsing in user controller - Add missing return after error response in updateUserAccountStatus to prevent double-response and spurious ext-client side-effects - Use switch statements in listUsers to skip unrecognized account_status and mfa_status filter values * NM-271: signal pull req on node ip change * fix(go): check for both min and max page size; * NM-271: refresh node object before update * fix(go): enclose transfer superadmin in transaction; * fix(go): review errors; * fix(go): remove free tier checks; * fix(go): review fixes; * NM-271: streamline ip pool ops * NM-271: fix tests, set max idle conns * NM-271: fix(go): fix data races in settings cache and peer update worker - Use pointer type in atomic.Value for serverSettingsCache to avoid replacing the variable non-atomically in InvalidateServerSettingsCache - Swap peerUpdateReplace flag before draining the channel to prevent a concurrent replacePeers=true from being consumed by the wrong cycle --------- Co-authored-by: VishalDalwadi <dalwadivishal26@gmail.com> Co-authored-by: Vishal Dalwadi <51291657+VishalDalwadi@users.noreply.github.com> Co-authored-by: tenki-reviewer[bot] <262613592+tenki-reviewer[bot]@users.noreply.github.com>
1590 lines
48 KiB
Go
1590 lines
48 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/gorilla/mux"
|
|
"github.com/gravitl/netmaker/database"
|
|
"github.com/gravitl/netmaker/db"
|
|
dbtypes "github.com/gravitl/netmaker/db/types"
|
|
"github.com/gravitl/netmaker/logger"
|
|
"github.com/gravitl/netmaker/logic"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/mq"
|
|
"github.com/gravitl/netmaker/schema"
|
|
"github.com/gravitl/netmaker/servercfg"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"golang.org/x/exp/slog"
|
|
)
|
|
|
|
func hostHandlers(r *mux.Router) {
|
|
r.HandleFunc("/api/hosts", logic.SecurityCheck(true, http.HandlerFunc(getHosts))).
|
|
Methods(http.MethodGet)
|
|
r.HandleFunc("/api/v1/hosts", logic.SecurityCheck(true, http.HandlerFunc(listHosts))).
|
|
Methods(http.MethodGet)
|
|
r.HandleFunc("/api/hosts/keys", logic.SecurityCheck(true, http.HandlerFunc(updateAllKeys))).
|
|
Methods(http.MethodPut)
|
|
r.HandleFunc("/api/hosts/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHosts))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/hosts/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHosts))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/hosts/{hostid}/keys", logic.SecurityCheck(true, http.HandlerFunc(updateKeys))).
|
|
Methods(http.MethodPut)
|
|
r.HandleFunc("/api/hosts/{hostid}/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHost))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/hosts/{hostid}", logic.SecurityCheck(true, http.HandlerFunc(updateHost))).
|
|
Methods(http.MethodPut)
|
|
// used by netclient
|
|
r.HandleFunc("/api/hosts/{hostid}", AuthorizeHost(http.HandlerFunc(deleteHost))).
|
|
Methods(http.MethodDelete)
|
|
// used by UI
|
|
r.HandleFunc("/api/v1/ui/hosts/{hostid}", logic.SecurityCheck(true, http.HandlerFunc(deleteHost))).
|
|
Methods(http.MethodDelete)
|
|
r.HandleFunc("/api/hosts/{hostid}/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHost))).
|
|
Methods(http.MethodPut)
|
|
r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(addHostToNetwork))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(deleteHostFromNetwork))).
|
|
Methods(http.MethodDelete)
|
|
r.HandleFunc("/api/hosts/adm/authenticate", authenticateHost).Methods(http.MethodPost)
|
|
r.HandleFunc("/api/v1/host", AuthorizeHost(http.HandlerFunc(pull))).
|
|
Methods(http.MethodGet)
|
|
r.HandleFunc("/api/v1/host/{hostid}/signalpeer", AuthorizeHost(http.HandlerFunc(signalPeer))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/v1/fallback/host/{hostid}", AuthorizeHost(http.HandlerFunc(hostUpdateFallback))).
|
|
Methods(http.MethodPut)
|
|
r.HandleFunc("/api/v1/host/{hostid}/peer_info", AuthorizeHost(http.HandlerFunc(getHostPeerInfo))).
|
|
Methods(http.MethodGet)
|
|
r.HandleFunc("/api/v1/pending_hosts", logic.SecurityCheck(true, http.HandlerFunc(getPendingHosts))).
|
|
Methods(http.MethodGet)
|
|
r.HandleFunc("/api/v1/pending_hosts/approve/{id}", logic.SecurityCheck(true, http.HandlerFunc(approvePendingHost))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/v1/pending_hosts/reject/{id}", logic.SecurityCheck(true, http.HandlerFunc(rejectPendingHost))).
|
|
Methods(http.MethodPost)
|
|
r.HandleFunc("/api/emqx/hosts", logic.SecurityCheck(true, http.HandlerFunc(delEmqxHosts))).
|
|
Methods(http.MethodDelete)
|
|
r.HandleFunc("/api/v1/auth-register/host", socketHandler)
|
|
}
|
|
|
|
// @Summary Requests all the hosts to upgrade their version
|
|
// @Router /api/hosts/upgrade [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param force query bool false "Force upgrade"
|
|
// @Success 200 {string} string "upgrade all hosts request received"
|
|
func upgradeHosts(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
action := models.Upgrade
|
|
|
|
if r.URL.Query().Get("force") == "true" {
|
|
action = models.ForceUpgrade
|
|
}
|
|
|
|
user := r.Header.Get("user")
|
|
|
|
go func() {
|
|
slog.Info("requesting all hosts to upgrade", "user", user)
|
|
|
|
hosts, err := (&schema.Host{}).ListAll(r.Context())
|
|
if err != nil {
|
|
slog.Error("failed to retrieve all hosts", "user", user, "error", err)
|
|
return
|
|
}
|
|
|
|
for _, host := range hosts {
|
|
go func(host schema.Host) {
|
|
hostUpdate := models.HostUpdate{
|
|
Action: action,
|
|
Host: host,
|
|
}
|
|
if err = mq.HostUpdate(&hostUpdate); err != nil {
|
|
slog.Error("failed to request host to upgrade", "user", user, "host", host.ID.String(), "error", err)
|
|
} else {
|
|
slog.Info("host upgrade requested", "user", user, "host", host.ID.String())
|
|
}
|
|
}(host)
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.UpgradeAll,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: "All Hosts",
|
|
Name: "All Hosts",
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
slog.Info("upgrade all hosts request received", "user", user)
|
|
logic.ReturnSuccessResponse(w, r, "upgrade all hosts request received")
|
|
}
|
|
|
|
// @Summary Upgrade a host
|
|
// @Router /api/hosts/{hostid}/upgrade [put]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param force query bool false "Force upgrade"
|
|
// @Success 200 {string} string "passed message to upgrade host"
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
// upgrade host is a handler to send upgrade message to a host
|
|
func upgradeHost(w http.ResponseWriter, r *http.Request) {
|
|
hostIDStr := mux.Vars(r)["hostid"]
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(r.Context())
|
|
if err != nil {
|
|
slog.Error("failed to find host", "error", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
|
|
return
|
|
}
|
|
|
|
action := models.Upgrade
|
|
|
|
if r.URL.Query().Get("force") == "true" {
|
|
action = models.ForceUpgrade
|
|
}
|
|
|
|
if err := mq.HostUpdate(&models.HostUpdate{Action: action, Host: *host}); err != nil {
|
|
slog.Error("failed to upgrade host", "error", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
logic.ReturnSuccessResponse(w, r, "passed message to upgrade host")
|
|
}
|
|
|
|
// @Summary List all hosts
|
|
// @Router /api/hosts [get]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Success 200 {array} models.ApiHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func getHosts(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
currentHosts, err := (&schema.Host{}).ListAll(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
apiHosts := logic.GetAllHostsAPI(currentHosts[:])
|
|
logger.Log(2, r.Header.Get("user"), "fetched all hosts")
|
|
logic.SortApiHosts(apiHosts[:])
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(apiHosts)
|
|
}
|
|
|
|
// @Summary List all hosts
|
|
// @Router /api/v1/hosts [get]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param os query []string false "Filter by OS" Enums(windows, linux, darwin)
|
|
// @Param page query int false "Page number"
|
|
// @Param per_page query int false "Items per page"
|
|
// @Success 200 {array} models.ApiHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func listHosts(w http.ResponseWriter, r *http.Request) {
|
|
var osFilters []interface{}
|
|
for _, filter := range r.URL.Query()["os"] {
|
|
osFilters = append(osFilters, filter)
|
|
}
|
|
|
|
var page, pageSize int
|
|
page, _ = strconv.Atoi(r.URL.Query().Get("page"))
|
|
if page == 0 {
|
|
page = 1
|
|
}
|
|
|
|
pageSize, _ = strconv.Atoi(r.URL.Query().Get("per_page"))
|
|
if pageSize < 1 || pageSize > 100 {
|
|
pageSize = 10
|
|
}
|
|
|
|
currentHosts, err := (&schema.Host{}).ListAll(
|
|
r.Context(),
|
|
dbtypes.WithFilter("os", osFilters...),
|
|
dbtypes.InAscOrder("name"),
|
|
dbtypes.WithPagination(page, pageSize),
|
|
)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
apiHosts := logic.GetAllHostsAPI(currentHosts[:])
|
|
logger.Log(2, r.Header.Get("user"), "fetched all hosts")
|
|
|
|
total, err := (&schema.Host{}).Count(
|
|
r.Context(),
|
|
dbtypes.WithFilter("os", osFilters...),
|
|
)
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.Internal))
|
|
return
|
|
}
|
|
|
|
totalPages := (total + pageSize - 1) / pageSize
|
|
if totalPages == 0 {
|
|
totalPages = 1
|
|
}
|
|
|
|
response := models.PaginatedResponse{
|
|
Data: apiHosts,
|
|
Page: page,
|
|
PerPage: pageSize,
|
|
Total: total,
|
|
TotalPages: totalPages,
|
|
}
|
|
|
|
logic.ReturnSuccessResponseWithJson(w, r, response, "fetched hosts")
|
|
}
|
|
|
|
// @Summary Used by clients for "pull" command
|
|
// @Router /api/v1/host [get]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Success 200 {object} models.HostPull
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func pull(w http.ResponseWriter, r *http.Request) {
|
|
hostIDStr := r.Header.Get(hostIDHeader) // return JSON/API formatted keys
|
|
if len(hostIDStr) == 0 {
|
|
logger.Log(0, "no host authorized to pull")
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(fmt.Errorf("no host authorized to pull"), "internal"),
|
|
)
|
|
return
|
|
}
|
|
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, "no host found during pull", hostIDStr)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
resetFailovered := r.URL.Query().Get("reset_failovered") == "true"
|
|
if resetFailovered {
|
|
for _, nodeID := range host.Nodes {
|
|
node, err := logic.GetNodeByID(nodeID)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
logic.ResetFailedOverPeer(&node)
|
|
logic.ResetAutoRelayedPeer(&node)
|
|
}
|
|
go mq.PublishPeerUpdate(false)
|
|
}
|
|
|
|
hPU, ok := logic.GetCachedHostPeerUpdate(hostID.String())
|
|
if !ok || resetFailovered {
|
|
allNodes, err := logic.GetAllNodes()
|
|
if err != nil {
|
|
logger.Log(0, "failed to get nodes: ", hostID.String())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
hPU, err = logic.GetPeerUpdateForHost("", host, allNodes, nil, nil)
|
|
if err != nil {
|
|
logger.Log(0, "could not pull peers for host", hostID.String(), err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
logic.StoreHostPeerUpdate(hostID.String(), hPU)
|
|
}
|
|
|
|
response := models.HostPull{
|
|
Host: *host,
|
|
Nodes: logic.GetHostNodes(host),
|
|
ServerConfig: hPU.ServerConfig,
|
|
Peers: hPU.Peers,
|
|
PeerIDs: hPU.PeerIDs,
|
|
HostNetworkInfo: hPU.HostNetworkInfo,
|
|
EgressRoutes: hPU.EgressRoutes,
|
|
FwUpdate: hPU.FwUpdate,
|
|
ChangeDefaultGw: hPU.ChangeDefaultGw,
|
|
DefaultGwIp: hPU.DefaultGwIp,
|
|
IsInternetGw: hPU.IsInternetGw,
|
|
NameServers: hPU.NameServers,
|
|
EgressWithDomains: hPU.EgressWithDomains,
|
|
EndpointDetection: logic.IsEndpointDetectionEnabled(),
|
|
DnsNameservers: hPU.DnsNameservers,
|
|
ReplacePeers: hPU.ReplacePeers,
|
|
AutoRelayNodes: hPU.AutoRelayNodes,
|
|
GwNodes: hPU.GwNodes,
|
|
AddressIdentityMap: hPU.AddressIdentityMap,
|
|
}
|
|
|
|
logger.Log(1, hostIDStr, host.Name, "completed a pull")
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(&response)
|
|
}
|
|
|
|
// @Summary Updates a Netclient host on Netmaker server
|
|
// @Router /api/hosts/{hostid} [put]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param body body models.ApiHost true "New host data"
|
|
// @Success 200 {object} models.ApiHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func updateHost(w http.ResponseWriter, r *http.Request) {
|
|
var newHostData models.ApiHost
|
|
err := json.NewDecoder(r.Body).Decode(&newHostData)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
hostID, err := uuid.Parse(newHostData.ID)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
|
|
// confirm host exists
|
|
currHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = currHost.Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
newHost := newHostData.ConvertAPIHostToNMHost(currHost)
|
|
|
|
logic.UpdateHost(newHost, currHost) // update the in memory struct values
|
|
if newHost.DNS != "yes" {
|
|
// check if any node is internet gw
|
|
for _, nodeID := range newHost.Nodes {
|
|
node, err := logic.GetNodeByID(nodeID)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
if node.IsInternetGateway {
|
|
newHost.DNS = "yes"
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if err = logic.UpsertHost(newHost); err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
// publish host update through MQ
|
|
if err := mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.UpdateHost,
|
|
Host: *newHost,
|
|
}); err != nil {
|
|
logger.Log(
|
|
0,
|
|
r.Header.Get("user"),
|
|
"failed to send host update: ",
|
|
currHost.ID.String(),
|
|
err.Error(),
|
|
)
|
|
}
|
|
go func() {
|
|
if newHost.IsDefault && !currHost.IsDefault {
|
|
addDefaultHostToNetworks(newHost)
|
|
}
|
|
if err := mq.PublishPeerUpdate(false); err != nil {
|
|
logger.Log(0, "fail to publish peer update: ", err.Error())
|
|
}
|
|
if newHost.Name != currHost.Name {
|
|
if servercfg.IsDNSMode() {
|
|
logic.SetDNS()
|
|
}
|
|
}
|
|
}()
|
|
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.Update,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: currHost.ID.String(),
|
|
Name: newHost.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Diff: models.Diff{
|
|
Old: currHost,
|
|
New: newHost,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
apiHostData := models.NewApiHostFromSchemaHost(newHost)
|
|
logger.Log(2, r.Header.Get("user"), "updated host", newHost.ID.String())
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(apiHostData)
|
|
}
|
|
|
|
// @Summary Updates a Netclient host on Netmaker server
|
|
// @Router /api/v1/fallback/host/{hostid} [put]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param body body models.HostUpdate true "Host update data"
|
|
// @Success 200 {string} string "updated host data"
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func hostUpdateFallback(w http.ResponseWriter, r *http.Request) {
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
currentHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = currentHost.Get(r.Context())
|
|
if err != nil {
|
|
slog.Error("error getting host", "id", hostIDStr, "error", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
var sendPeerUpdate, sendDeletedNodeUpdate, replacePeers bool
|
|
var hostUpdate models.HostUpdate
|
|
err = json.NewDecoder(r.Body).Decode(&hostUpdate)
|
|
if err != nil {
|
|
slog.Error("failed to update a host:", "user", r.Header.Get("user"), "error", err.Error(), "host", currentHost.Name)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
slog.Info("recieved host update", "name", hostUpdate.Host.Name, "id", hostUpdate.Host.ID, "action", hostUpdate.Action)
|
|
switch hostUpdate.Action {
|
|
case models.CheckIn:
|
|
sendPeerUpdate = mq.HandleHostCheckin(&hostUpdate.Host, currentHost)
|
|
case models.UpdateHost:
|
|
if hostUpdate.Host.PublicKey != currentHost.PublicKey {
|
|
//remove old peer entry
|
|
replacePeers = true
|
|
}
|
|
sendPeerUpdate = logic.UpdateHostFromClient(&hostUpdate.Host, currentHost)
|
|
err := logic.UpsertHost(currentHost)
|
|
if err != nil {
|
|
slog.Error("failed to update host", "id", currentHost.ID, "error", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.Internal))
|
|
return
|
|
}
|
|
case models.UpdateNode:
|
|
sendDeletedNodeUpdate, sendPeerUpdate = logic.UpdateHostNode(&hostUpdate.Host, &hostUpdate.Node)
|
|
case models.UpdateMetrics:
|
|
mq.UpdateMetricsFallBack(hostUpdate.Node.ID.String(), hostUpdate.NewMetrics)
|
|
case models.EgressUpdate:
|
|
e := schema.Egress{ID: hostUpdate.EgressDomain.ID}
|
|
err = e.Get(db.WithContext(r.Context()))
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
if len(hostUpdate.Node.EgressGatewayRanges) > 0 {
|
|
e.DomainAns = hostUpdate.Node.EgressGatewayRanges
|
|
e.Update(db.WithContext(r.Context()))
|
|
}
|
|
sendPeerUpdate = true
|
|
case models.SignalHost:
|
|
mq.SignalPeer(hostUpdate.Signal)
|
|
case models.DeleteHost:
|
|
go mq.DeleteAndCleanupHost(currentHost)
|
|
}
|
|
go func() {
|
|
if sendDeletedNodeUpdate {
|
|
mq.PublishDeletedNodePeerUpdate(&hostUpdate.Node)
|
|
}
|
|
if sendPeerUpdate {
|
|
err := mq.PublishPeerUpdate(replacePeers)
|
|
if err != nil {
|
|
slog.Error("failed to publish peer update", "error", err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
logic.ReturnSuccessResponse(w, r, "updated host data")
|
|
}
|
|
|
|
// @Summary Deletes a Netclient host from Netmaker server
|
|
// @Router /api/hosts/{hostid} [delete]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param force query bool false "Force delete"
|
|
// @Success 200 {object} models.ApiHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func deleteHost(w http.ResponseWriter, r *http.Request) {
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
forceDelete := r.URL.Query().Get("force") == "true"
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
// confirm host exists
|
|
currHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = currHost.Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
for _, nodeID := range currHost.Nodes {
|
|
node, err := logic.GetNodeByID(nodeID)
|
|
if err != nil {
|
|
slog.Error("failed to get node", "nodeid", nodeID, "error", err)
|
|
continue
|
|
}
|
|
go mq.PublishMqUpdatesForDeletedNode(node, false)
|
|
|
|
}
|
|
if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
|
|
// delete EMQX credentials for host
|
|
if err := mq.GetEmqxHandler().DeleteEmqxUser(currHost.ID.String()); err != nil {
|
|
slog.Error(
|
|
"failed to remove host credentials from EMQX",
|
|
"id",
|
|
currHost.ID,
|
|
"error",
|
|
err,
|
|
)
|
|
}
|
|
}
|
|
if err = mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.DeleteHost,
|
|
Host: *currHost,
|
|
}); err != nil {
|
|
logger.Log(
|
|
0,
|
|
r.Header.Get("user"),
|
|
"failed to send delete host update: ",
|
|
currHost.ID.String(),
|
|
err.Error(),
|
|
)
|
|
}
|
|
if err = logic.RemoveHost(currHost, forceDelete); err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
// delete if any pending reqs
|
|
(&schema.PendingHost{
|
|
HostID: currHost.ID.String(),
|
|
}).DeleteAllPendingHosts(db.WithContext(r.Context()))
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.Delete,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: currHost.ID.String(),
|
|
Name: currHost.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
Diff: models.Diff{
|
|
Old: currHost,
|
|
New: nil,
|
|
},
|
|
})
|
|
apiHostData := models.NewApiHostFromSchemaHost(currHost)
|
|
logger.Log(2, r.Header.Get("user"), "removed host", currHost.Name)
|
|
logic.ReturnSuccessResponseWithJson(w, r, apiHostData, "deleted host "+currHost.Name)
|
|
}
|
|
|
|
// @Summary To Add Host To Network
|
|
// @Router /api/hosts/{hostid}/networks/{network} [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param network path string true "Network name"
|
|
// @Success 200 {string} string "OK"
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func addHostToNetwork(w http.ResponseWriter, r *http.Request) {
|
|
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
network := params["network"]
|
|
if hostIDStr == "" || network == "" {
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(errors.New("hostid or network cannot be empty"), logic.BadReq),
|
|
)
|
|
return
|
|
}
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
|
|
// confirm host exists
|
|
currHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = currHost.Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to find host:", hostIDStr, err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.Internal))
|
|
return
|
|
}
|
|
|
|
violations, _ := logic.CheckPostureViolations(models.PostureCheckDeviceInfo{
|
|
ClientLocation: currHost.CountryCode,
|
|
ClientVersion: currHost.Version,
|
|
OS: currHost.OS,
|
|
OSFamily: currHost.OSFamily,
|
|
OSVersion: currHost.OSVersion,
|
|
KernelVersion: currHost.KernelVersion,
|
|
AutoUpdate: currHost.AutoUpdate,
|
|
}, schema.NetworkID(network))
|
|
if len(violations) > 0 {
|
|
logic.ReturnErrorResponseWithJson(w, r, violations, logic.FormatError(errors.New("posture check violations"), logic.BadReq))
|
|
return
|
|
}
|
|
newNode, err := logic.UpdateHostNetwork(currHost, network, true)
|
|
if err != nil {
|
|
logger.Log(
|
|
0,
|
|
r.Header.Get("user"),
|
|
"failed to add host to network:",
|
|
hostIDStr,
|
|
network,
|
|
err.Error(),
|
|
)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
|
|
if currHost.IsDefault {
|
|
// make host gateway
|
|
logic.CreateIngressGateway(network, newNode.ID.String(), models.IngressRequest{})
|
|
logic.CreateRelay(models.RelayRequest{
|
|
NodeID: newNode.ID.String(),
|
|
NetID: network,
|
|
})
|
|
}
|
|
go func() {
|
|
mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.JoinHostToNetwork,
|
|
Host: *currHost,
|
|
Node: *newNode,
|
|
})
|
|
mq.PublishPeerUpdate(false)
|
|
if servercfg.IsDNSMode() {
|
|
logic.SetDNS()
|
|
}
|
|
}()
|
|
logger.Log(
|
|
2,
|
|
r.Header.Get("user"),
|
|
fmt.Sprintf("added host %s to network %s", currHost.Name, network),
|
|
)
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.JoinHostToNet,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: currHost.ID.String(),
|
|
Name: currHost.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
NetworkID: schema.NetworkID(network),
|
|
Origin: schema.Dashboard,
|
|
})
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
// @Summary To Remove Host from Network
|
|
// @Router /api/hosts/{hostid}/networks/{network} [delete]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param network path string true "Network name"
|
|
// @Param force query bool false "Force delete"
|
|
// @Success 200 {string} string "OK"
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func deleteHostFromNetwork(w http.ResponseWriter, r *http.Request) {
|
|
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
network := params["network"]
|
|
forceDelete := r.URL.Query().Get("force") == "true"
|
|
if hostIDStr == "" || network == "" {
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
|
|
)
|
|
return
|
|
}
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
// confirm host exists
|
|
currHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = currHost.Get(r.Context())
|
|
if err != nil {
|
|
if database.IsEmptyRecord(err) {
|
|
// check if there is any daemon nodes that needs to be deleted
|
|
node, err := logic.GetNodeByHostRef(hostIDStr, network)
|
|
if err != nil {
|
|
slog.Error(
|
|
"couldn't get node for host",
|
|
"hostid",
|
|
hostIDStr,
|
|
"network",
|
|
network,
|
|
"error",
|
|
err,
|
|
)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
if err = logic.DeleteNodeByID(&node); err != nil {
|
|
slog.Error("failed to force delete daemon node",
|
|
"nodeid", node.ID.String(), "hostid", hostIDStr, "network", network, "error", err)
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(
|
|
fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
|
|
"internal",
|
|
),
|
|
)
|
|
return
|
|
}
|
|
logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
|
|
return
|
|
}
|
|
|
|
logger.Log(0, r.Header.Get("user"), "failed to find host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
node, err := logic.UpdateHostNetwork(currHost, network, false)
|
|
if err != nil {
|
|
if node == nil && forceDelete {
|
|
// force cleanup the node
|
|
node, err := logic.GetNodeByHostRef(hostIDStr, network)
|
|
if err != nil {
|
|
slog.Error(
|
|
"couldn't get node for host",
|
|
"hostid",
|
|
hostIDStr,
|
|
"network",
|
|
network,
|
|
"error",
|
|
err,
|
|
)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
if err = logic.DeleteNodeByID(&node); err != nil {
|
|
slog.Error("failed to force delete daemon node",
|
|
"nodeid", node.ID.String(), "hostid", hostIDStr, "network", network, "error", err)
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(
|
|
fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
|
|
"internal",
|
|
),
|
|
)
|
|
return
|
|
}
|
|
logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
|
|
return
|
|
}
|
|
logger.Log(
|
|
0,
|
|
r.Header.Get("user"),
|
|
"failed to remove host from network:",
|
|
hostIDStr,
|
|
network,
|
|
err.Error(),
|
|
)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
logger.Log(1, "deleting node", node.ID.String(), "from host", currHost.Name)
|
|
if err := logic.DeleteNode(node, forceDelete); err != nil {
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(fmt.Errorf("failed to delete node"), "internal"),
|
|
)
|
|
return
|
|
}
|
|
go func() {
|
|
mq.PublishMqUpdatesForDeletedNode(*node, true)
|
|
if servercfg.IsDNSMode() {
|
|
logic.SetDNS()
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.RemoveHostFromNet,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: currHost.ID.String(),
|
|
Name: currHost.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
NetworkID: schema.NetworkID(network),
|
|
Origin: schema.Dashboard,
|
|
})
|
|
logger.Log(
|
|
2,
|
|
r.Header.Get("user"),
|
|
fmt.Sprintf("removed host %s from network %s", currHost.Name, network),
|
|
)
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
// @Summary To Fetch Auth Token for a Host
|
|
// @Router /api/hosts/adm/authenticate [post]
|
|
// @Tags Auth
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body models.AuthParams true "Authentication parameters"
|
|
// @Success 200 {object} models.SuccessResponse
|
|
// @Failure 400 {object} models.ErrorResponse
|
|
// @Failure 401 {object} models.ErrorResponse
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func authenticateHost(response http.ResponseWriter, request *http.Request) {
|
|
var authRequest models.AuthParams
|
|
var errorResponse = models.ErrorResponse{
|
|
Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
|
|
}
|
|
|
|
decoder := json.NewDecoder(request.Body)
|
|
decoderErr := decoder.Decode(&authRequest)
|
|
defer request.Body.Close()
|
|
|
|
if decoderErr != nil {
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = decoderErr.Error()
|
|
logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
|
|
decoderErr.Error())
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
errorResponse.Code = http.StatusBadRequest
|
|
if authRequest.ID == "" {
|
|
errorResponse.Message = "W1R3: ID can't be empty"
|
|
logger.Log(0, request.Header.Get("user"), errorResponse.Message)
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
} else if authRequest.Password == "" {
|
|
errorResponse.Message = "W1R3: Password can't be empty"
|
|
logger.Log(0, request.Header.Get("user"), errorResponse.Message)
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
hostID, err := uuid.Parse(authRequest.ID)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(response, request, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(request.Context())
|
|
if err != nil {
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logger.Log(0, request.Header.Get("user"),
|
|
"error retrieving host: ", authRequest.ID, err.Error())
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
|
|
if err != nil {
|
|
errorResponse.Code = http.StatusUnauthorized
|
|
errorResponse.Message = "unauthorized"
|
|
logger.Log(0, request.Header.Get("user"),
|
|
"error validating user password: ", err.Error())
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
|
|
tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, "")
|
|
if tokenString == "" {
|
|
errorResponse.Code = http.StatusUnauthorized
|
|
errorResponse.Message = "unauthorized"
|
|
logger.Log(0, request.Header.Get("user"),
|
|
fmt.Sprintf("%s: %v", errorResponse.Message, err))
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
|
|
var successResponse = models.SuccessResponse{
|
|
Code: http.StatusOK,
|
|
Message: "W1R3: Host " + authRequest.ID + " Authorized",
|
|
Response: models.SuccessfulLoginResponse{
|
|
AuthToken: tokenString,
|
|
ID: authRequest.ID,
|
|
},
|
|
}
|
|
successJSONResponse, jsonError := json.Marshal(successResponse)
|
|
|
|
if jsonError != nil {
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logger.Log(0, request.Header.Get("user"),
|
|
"error marshalling resp: ", err.Error())
|
|
logic.ReturnErrorResponse(response, request, errorResponse)
|
|
return
|
|
}
|
|
go func() {
|
|
// Create EMQX creds
|
|
if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
|
|
if err := mq.GetEmqxHandler().CreateEmqxUser(host.ID.String(), authRequest.Password); err != nil {
|
|
slog.Error("failed to create host credentials for EMQX: ", err.Error())
|
|
}
|
|
}
|
|
}()
|
|
|
|
response.WriteHeader(http.StatusOK)
|
|
response.Header().Set("Content-Type", "application/json")
|
|
response.Write(successJSONResponse)
|
|
}
|
|
|
|
// @Summary Send signal to peer
|
|
// @Router /api/v1/host/{hostid}/signalpeer [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Param body body models.Signal true "Signal data"
|
|
// @Success 200 {object} models.Signal
|
|
// @Failure 400 {object} models.ErrorResponse
|
|
func signalPeer(w http.ResponseWriter, r *http.Request) {
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
// confirm host exists
|
|
err = (&schema.Host{
|
|
ID: hostID,
|
|
}).Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to get host:", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
var signal models.Signal
|
|
w.Header().Set("Content-Type", "application/json")
|
|
err = json.NewDecoder(r.Body).Decode(&signal)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
if signal.ToHostPubKey == "" {
|
|
msg := "insufficient data to signal peer"
|
|
logger.Log(0, r.Header.Get("user"), msg)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New(msg), "badrequest"))
|
|
return
|
|
}
|
|
signal.IsPro = servercfg.IsPro
|
|
hostID, err = uuid.Parse(signal.ToHostID)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
peerHost := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = peerHost.Get(r.Context())
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(errors.New("failed to signal, peer not found"), "badrequest"),
|
|
)
|
|
return
|
|
}
|
|
err = mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.SignalHost,
|
|
Host: *peerHost,
|
|
Signal: signal,
|
|
})
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(
|
|
w,
|
|
r,
|
|
logic.FormatError(
|
|
errors.New("failed to publish signal to peer: "+err.Error()),
|
|
"badrequest",
|
|
),
|
|
)
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(signal)
|
|
}
|
|
|
|
// @Summary Update keys for all hosts
|
|
// @Router /api/hosts/keys [put]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Success 200 {string} string "OK"
|
|
// @Failure 400 {object} models.ErrorResponse
|
|
func updateAllKeys(w http.ResponseWriter, r *http.Request) {
|
|
var errorResponse = models.ErrorResponse{}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
hosts, err := (&schema.Host{}).ListAll(r.Context())
|
|
if err != nil {
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logger.Log(0, r.Header.Get("user"),
|
|
"error retrieving hosts ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, errorResponse)
|
|
return
|
|
}
|
|
go func() {
|
|
hostUpdate := models.HostUpdate{}
|
|
hostUpdate.Action = models.UpdateKeys
|
|
for _, host := range hosts {
|
|
hostUpdate.Host = host
|
|
logger.Log(2, "updating host", host.ID.String(), " for a key update")
|
|
if err = mq.HostUpdate(&hostUpdate); err != nil {
|
|
logger.Log(
|
|
0,
|
|
"failed to send update to node during a network wide key update",
|
|
host.ID.String(),
|
|
err.Error(),
|
|
)
|
|
}
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.RefreshAllKeys,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: "All Devices",
|
|
Name: "All Devices",
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
logger.Log(2, r.Header.Get("user"), "updated keys for all hosts")
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
// @Summary Update keys for a host
|
|
// @Router /api/hosts/{hostid}/keys [put]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Success 200 {string} string "OK"
|
|
// @Failure 400 {object} models.ErrorResponse
|
|
func updateKeys(w http.ResponseWriter, r *http.Request) {
|
|
var errorResponse = models.ErrorResponse{}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
var params = mux.Vars(r)
|
|
hostIDStr := params["hostid"]
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, "failed to retrieve host", hostIDStr, err.Error())
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logger.Log(0, r.Header.Get("user"),
|
|
"error retrieving hosts ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, errorResponse)
|
|
return
|
|
}
|
|
go func() {
|
|
hostUpdate := models.HostUpdate{
|
|
Action: models.UpdateKeys,
|
|
Host: *host,
|
|
}
|
|
if err = mq.HostUpdate(&hostUpdate); err != nil {
|
|
logger.Log(0, "failed to send host key update", host.ID.String(), err.Error())
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.RefreshKey,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: host.ID.String(),
|
|
Name: host.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
logger.Log(2, r.Header.Get("user"), "updated key on host", host.Name)
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
// @Summary Requests all the hosts to pull
|
|
// @Router /api/hosts/sync [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Success 200 {string} string "sync all hosts request received"
|
|
func syncHosts(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
user := r.Header.Get("user")
|
|
|
|
go func() {
|
|
slog.Info("requesting all hosts to sync", "user", user)
|
|
|
|
hosts, err := (&schema.Host{}).ListAll(r.Context())
|
|
if err != nil {
|
|
slog.Error("failed to retrieve all hosts", "user", user, "error", err)
|
|
return
|
|
}
|
|
|
|
for _, host := range hosts {
|
|
go func(host schema.Host) {
|
|
hostUpdate := models.HostUpdate{
|
|
Action: models.RequestPull,
|
|
Host: host,
|
|
}
|
|
if err = mq.HostUpdate(&hostUpdate); err != nil {
|
|
slog.Error("failed to request host to sync", "user", user, "host", host.ID.String(), "error", err)
|
|
} else {
|
|
slog.Info("host sync requested", "user", user, "host", host.ID.String())
|
|
}
|
|
}(host)
|
|
time.Sleep(time.Millisecond * 100)
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.SyncAll,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: "All Devices",
|
|
Name: "All Devices",
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
slog.Info("sync all hosts request received", "user", user)
|
|
logic.ReturnSuccessResponse(w, r, "sync all hosts request received")
|
|
}
|
|
|
|
// @Summary Requests a host to pull
|
|
// @Router /api/hosts/{hostid}/sync [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Param hostid path string true "Host ID"
|
|
// @Success 200 {string} string "OK"
|
|
// @Failure 400 {object} models.ErrorResponse
|
|
func syncHost(w http.ResponseWriter, r *http.Request) {
|
|
hostIDStr := mux.Vars(r)["hostid"]
|
|
|
|
var errorResponse = models.ErrorResponse{}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(r.Context())
|
|
if err != nil {
|
|
slog.Error("failed to retrieve host", "user", r.Header.Get("user"), "error", err)
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logic.ReturnErrorResponse(w, r, errorResponse)
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
hostUpdate := models.HostUpdate{
|
|
Action: models.RequestPull,
|
|
Host: *host,
|
|
}
|
|
if err = mq.HostUpdate(&hostUpdate); err != nil {
|
|
slog.Error("failed to send host pull request", "host", host.ID.String(), "error", err)
|
|
}
|
|
}()
|
|
logic.LogEvent(&models.Event{
|
|
Action: schema.Sync,
|
|
Source: models.Subject{
|
|
ID: r.Header.Get("user"),
|
|
Name: r.Header.Get("user"),
|
|
Type: schema.UserSub,
|
|
},
|
|
TriggeredBy: r.Header.Get("user"),
|
|
Target: models.Subject{
|
|
ID: host.ID.String(),
|
|
Name: host.Name,
|
|
Type: schema.DeviceSub,
|
|
},
|
|
Origin: schema.Dashboard,
|
|
})
|
|
slog.Info("requested host pull", "user", r.Header.Get("user"), "host", host.ID.String())
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func delEmqxHosts(w http.ResponseWriter, r *http.Request) {
|
|
currentHosts, err := (&schema.Host{}).ListAll(r.Context())
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
for _, host := range currentHosts {
|
|
// delete EMQX credentials for host
|
|
if err := mq.GetEmqxHandler().DeleteEmqxUser(host.ID.String()); err != nil {
|
|
slog.Error("failed to remove host credentials from EMQX", "id", host.ID, "error", err)
|
|
}
|
|
}
|
|
err = mq.GetEmqxHandler().DeleteEmqxUser(servercfg.GetMqUserName())
|
|
if err != nil {
|
|
slog.Error(
|
|
"failed to remove server credentials from EMQX",
|
|
"user",
|
|
servercfg.GetMqUserName(),
|
|
"error",
|
|
err,
|
|
)
|
|
}
|
|
logic.ReturnSuccessResponse(w, r, "deleted hosts data on emqx")
|
|
}
|
|
|
|
// @Summary Fetches host peerinfo
|
|
// @Router /api/v1/host/{hostid}/peer_info [get]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param hostid path string true "Host ID"
|
|
// @Success 200 {object} models.HostPeerInfo
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func getHostPeerInfo(w http.ResponseWriter, r *http.Request) {
|
|
hostIDStr := mux.Vars(r)["hostid"]
|
|
var errorResponse = models.ErrorResponse{}
|
|
hostID, err := uuid.Parse(hostIDStr)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
host := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = host.Get(r.Context())
|
|
if err != nil {
|
|
slog.Error("failed to retrieve host", "error", err)
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logic.ReturnErrorResponse(w, r, errorResponse)
|
|
return
|
|
}
|
|
peerInfo, err := logic.GetHostPeerInfo(host)
|
|
if err != nil {
|
|
slog.Error("failed to retrieve host peerinfo", "error", err)
|
|
errorResponse.Code = http.StatusBadRequest
|
|
errorResponse.Message = err.Error()
|
|
logic.ReturnErrorResponse(w, r, errorResponse)
|
|
return
|
|
}
|
|
logic.ReturnSuccessResponseWithJson(w, r, peerInfo, "fetched host peer info")
|
|
}
|
|
|
|
// @Summary List pending hosts in a network
|
|
// @Router /api/v1/pending_hosts [get]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param network query string true "Network ID"
|
|
// @Success 200 {array} schema.PendingHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func getPendingHosts(w http.ResponseWriter, r *http.Request) {
|
|
netID := r.URL.Query().Get("network")
|
|
if netID == "" {
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("network id param is missing"), "badrequest"))
|
|
return
|
|
}
|
|
pendingHosts, err := (&schema.PendingHost{
|
|
Network: netID,
|
|
}).List(db.WithContext(r.Context()))
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
logger.Log(2, r.Header.Get("user"), "fetched all hosts")
|
|
logic.ReturnSuccessResponseWithJson(w, r, pendingHosts, "returned pending hosts in "+netID)
|
|
}
|
|
|
|
// @Summary Approve pending host in a network
|
|
// @Router /api/v1/pending_hosts/approve/{id} [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param id path string true "Pending Host ID"
|
|
// @Success 200 {object} models.ApiNode
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func approvePendingHost(w http.ResponseWriter, r *http.Request) {
|
|
id := mux.Vars(r)["id"]
|
|
p := &schema.PendingHost{ID: id}
|
|
err := p.Get(db.WithContext(r.Context()))
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
hostID, err := uuid.Parse(p.HostID)
|
|
if err != nil {
|
|
err = fmt.Errorf("failed to parse host id: %w", err)
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
|
|
return
|
|
}
|
|
h := &schema.Host{
|
|
ID: hostID,
|
|
}
|
|
err = h.Get(r.Context())
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
key := models.EnrollmentKey{}
|
|
json.Unmarshal(p.EnrollmentKey, &key)
|
|
newNode, err := logic.UpdateHostNetwork(h, p.Network, true)
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
if key.AutoAssignGateway {
|
|
newNode.AutoAssignGateway = true
|
|
}
|
|
if len(key.Groups) > 0 {
|
|
newNode.Tags = make(map[models.TagID]struct{})
|
|
for _, tagI := range key.Groups {
|
|
newNode.Tags[tagI] = struct{}{}
|
|
}
|
|
logic.UpsertNode(newNode)
|
|
}
|
|
if key.Relay != uuid.Nil && !newNode.IsRelayed {
|
|
// check if relay node exists and acting as relay
|
|
relaynode, err := logic.GetNodeByID(key.Relay.String())
|
|
if err == nil && relaynode.IsGw && relaynode.Network == newNode.Network {
|
|
slog.Error(fmt.Sprintf("adding relayed node %s to relay %s on network %s", newNode.ID.String(), key.Relay.String(), p.Network))
|
|
newNode.IsRelayed = true
|
|
newNode.RelayedBy = key.Relay.String()
|
|
updatedRelayNode := relaynode
|
|
updatedRelayNode.RelayedNodes = append(updatedRelayNode.RelayedNodes, newNode.ID.String())
|
|
logic.UpdateRelayed(&relaynode, &updatedRelayNode)
|
|
if err := logic.UpsertNode(&updatedRelayNode); err != nil {
|
|
slog.Error("failed to update node", "nodeid", key.Relay.String())
|
|
}
|
|
if err := logic.UpsertNode(newNode); err != nil {
|
|
slog.Error("failed to update node", "nodeid", key.Relay.String())
|
|
}
|
|
} else {
|
|
slog.Error("failed to relay node. maybe specified relay node is actually not a relay? Or the relayed node is not in the same network with relay?", "err", err)
|
|
}
|
|
}
|
|
|
|
logger.Log(1, "added new node", newNode.ID.String(), "to host", h.Name)
|
|
mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.JoinHostToNetwork,
|
|
Host: *h,
|
|
Node: *newNode,
|
|
})
|
|
if h.IsDefault {
|
|
// make host gateway
|
|
logic.CreateIngressGateway(p.Network, newNode.ID.String(), models.IngressRequest{})
|
|
logic.CreateRelay(models.RelayRequest{
|
|
NodeID: newNode.ID.String(),
|
|
NetID: p.Network,
|
|
})
|
|
}
|
|
p.Delete(db.WithContext(r.Context()))
|
|
go mq.PublishPeerUpdate(false)
|
|
logic.ReturnSuccessResponseWithJson(w, r, newNode.ConvertToAPINode(), "added pending host to "+p.Network)
|
|
}
|
|
|
|
// @Summary Reject pending host in a network
|
|
// @Router /api/v1/pending_hosts/reject/{id} [post]
|
|
// @Tags Hosts
|
|
// @Security oauth
|
|
// @Produce json
|
|
// @Param id path string true "Pending Host ID"
|
|
// @Success 200 {object} schema.PendingHost
|
|
// @Failure 500 {object} models.ErrorResponse
|
|
func rejectPendingHost(w http.ResponseWriter, r *http.Request) {
|
|
id := mux.Vars(r)["id"]
|
|
p := &schema.PendingHost{ID: id}
|
|
err := p.Get(db.WithContext(r.Context()))
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
err = p.Delete(db.WithContext(r.Context()))
|
|
if err != nil {
|
|
logic.ReturnErrorResponse(w, r, models.ErrorResponse{
|
|
Code: http.StatusBadRequest,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
logic.ReturnSuccessResponseWithJson(w, r, p, "deleted pending host from "+p.Network)
|
|
}
|
|
|
|
// addDefaultHostToNetworks enrolls a newly-made-default host into every
|
|
// existing network it is not already part of, applying the standard default
|
|
// host operations for each network.
|
|
func addDefaultHostToNetworks(host *schema.Host) {
|
|
networks, err := (&schema.Network{}).ListAll(db.WithContext(context.TODO()))
|
|
if err != nil {
|
|
logger.Log(0, "failed to get networks for default host ops:", err.Error())
|
|
return
|
|
}
|
|
for _, network := range networks {
|
|
if !network.AutoJoin {
|
|
continue
|
|
}
|
|
newNode, err := logic.UpdateHostNetwork(host, network.Name, true)
|
|
if err != nil {
|
|
logger.Log(2, "skipping network", network.Name, "for default host", host.Name, ":", err.Error())
|
|
continue
|
|
}
|
|
logger.Log(1, "added default host", host.Name, "to network", network.Name)
|
|
if len(host.Nodes) == 1 {
|
|
mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.RequestPull,
|
|
Host: *host,
|
|
Node: *newNode,
|
|
})
|
|
} else {
|
|
mq.HostUpdate(&models.HostUpdate{
|
|
Action: models.JoinHostToNetwork,
|
|
Host: *host,
|
|
Node: *newNode,
|
|
})
|
|
}
|
|
logic.CreateIngressGateway(network.Name, newNode.ID.String(), models.IngressRequest{})
|
|
logic.CreateRelay(models.RelayRequest{
|
|
NodeID: newNode.ID.String(),
|
|
NetID: network.Name,
|
|
})
|
|
}
|
|
}
|