mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 11:21:04 +01:00
8f98adddf6
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
130 lines
3.8 KiB
Go
130 lines
3.8 KiB
Go
package http
|
|
|
|
import (
|
|
"net/http"
|
|
"regexp"
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/netbirdio/netbird/management/server"
|
|
"github.com/netbirdio/netbird/management/server/geolocation"
|
|
"github.com/netbirdio/netbird/management/server/http/api"
|
|
"github.com/netbirdio/netbird/management/server/http/util"
|
|
"github.com/netbirdio/netbird/management/server/jwtclaims"
|
|
"github.com/netbirdio/netbird/management/server/status"
|
|
)
|
|
|
|
var (
|
|
countryCodeRegex = regexp.MustCompile("^[a-zA-Z]{2}$")
|
|
)
|
|
|
|
// GeolocationsHandler is a handler that returns locations.
|
|
type GeolocationsHandler struct {
|
|
accountManager server.AccountManager
|
|
geolocationManager *geolocation.Geolocation
|
|
claimsExtractor *jwtclaims.ClaimsExtractor
|
|
}
|
|
|
|
// NewGeolocationsHandlerHandler creates a new Geolocations handler
|
|
func NewGeolocationsHandlerHandler(accountManager server.AccountManager, geolocationManager *geolocation.Geolocation, authCfg AuthCfg) *GeolocationsHandler {
|
|
return &GeolocationsHandler{
|
|
accountManager: accountManager,
|
|
geolocationManager: geolocationManager,
|
|
claimsExtractor: jwtclaims.NewClaimsExtractor(
|
|
jwtclaims.WithAudience(authCfg.Audience),
|
|
jwtclaims.WithUserIDClaim(authCfg.UserIDClaim),
|
|
),
|
|
}
|
|
}
|
|
|
|
// GetAllCountries retrieves a list of all countries
|
|
func (l *GeolocationsHandler) GetAllCountries(w http.ResponseWriter, r *http.Request) {
|
|
if err := l.authenticateUser(r); err != nil {
|
|
util.WriteError(r.Context(), err, w)
|
|
return
|
|
}
|
|
|
|
if l.geolocationManager == nil {
|
|
// TODO: update error message to include geo db self hosted doc link when ready
|
|
util.WriteError(r.Context(), status.Errorf(status.PreconditionFailed, "Geo location database is not initialized"), w)
|
|
return
|
|
}
|
|
|
|
allCountries, err := l.geolocationManager.GetAllCountries()
|
|
if err != nil {
|
|
util.WriteError(r.Context(), err, w)
|
|
return
|
|
}
|
|
|
|
countries := make([]api.Country, 0, len(allCountries))
|
|
for _, country := range allCountries {
|
|
countries = append(countries, toCountryResponse(country))
|
|
}
|
|
util.WriteJSONObject(r.Context(), w, countries)
|
|
}
|
|
|
|
// GetCitiesByCountry retrieves a list of cities based on the given country code
|
|
func (l *GeolocationsHandler) GetCitiesByCountry(w http.ResponseWriter, r *http.Request) {
|
|
if err := l.authenticateUser(r); err != nil {
|
|
util.WriteError(r.Context(), err, w)
|
|
return
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
countryCode := vars["country"]
|
|
if !countryCodeRegex.MatchString(countryCode) {
|
|
util.WriteError(r.Context(), status.Errorf(status.InvalidArgument, "invalid country code"), w)
|
|
return
|
|
}
|
|
|
|
if l.geolocationManager == nil {
|
|
util.WriteError(r.Context(), status.Errorf(status.PreconditionFailed, "Geo location database is not initialized. "+
|
|
"Check the self-hosted Geo database documentation at https://docs.netbird.io/selfhosted/geo-support"), w)
|
|
return
|
|
}
|
|
|
|
allCities, err := l.geolocationManager.GetCitiesByCountry(countryCode)
|
|
if err != nil {
|
|
util.WriteError(r.Context(), err, w)
|
|
return
|
|
}
|
|
|
|
cities := make([]api.City, 0, len(allCities))
|
|
for _, city := range allCities {
|
|
cities = append(cities, toCityResponse(city))
|
|
}
|
|
util.WriteJSONObject(r.Context(), w, cities)
|
|
}
|
|
|
|
func (l *GeolocationsHandler) authenticateUser(r *http.Request) error {
|
|
claims := l.claimsExtractor.FromRequestContext(r)
|
|
_, userID, err := l.accountManager.GetAccountIDFromToken(r.Context(), claims)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
user, err := l.accountManager.GetUserByID(r.Context(), userID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !user.HasAdminPower() {
|
|
return status.Errorf(status.PermissionDenied, "user is not allowed to perform this action")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func toCountryResponse(country geolocation.Country) api.Country {
|
|
return api.Country{
|
|
CountryName: country.CountryName,
|
|
CountryCode: country.CountryISOCode,
|
|
}
|
|
}
|
|
|
|
func toCityResponse(city geolocation.City) api.City {
|
|
return api.City{
|
|
CityName: city.CityName,
|
|
GeonameId: city.GeoNameID,
|
|
}
|
|
}
|