mirror of
https://github.com/renorris/openfsd
synced 2026-03-22 06:25:35 +08:00
Introduce SetIfNotExists, atomic Client, dynamic web config, ServerConfig, data APIs, and fixes.
This commit is contained in:
Reese Norris
355
web/data.go
Normal file
355
web/data.go
Normal file
@@ -0,0 +1,355 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
_ "embed"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/renorris/openfsd/db"
|
||||
"github.com/renorris/openfsd/fsd"
|
||||
"go.uber.org/atomic"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strings"
|
||||
"text/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
//go:embed data_templates/status.txt
|
||||
var statusTxtRawTemplate string
|
||||
var statusTxtTemplate *template.Template
|
||||
|
||||
//go:embed data_templates/servers.txt
|
||||
var serversTxtRawTemplate string
|
||||
var serversTxtTemplate *template.Template
|
||||
|
||||
func init() {
|
||||
var err error
|
||||
statusTxtTemplate = template.New("statustxt")
|
||||
if statusTxtTemplate, err = statusTxtTemplate.Parse(statusTxtRawTemplate); err != nil {
|
||||
panic("Unable to parse status.txt template: " + err.Error())
|
||||
}
|
||||
|
||||
serversTxtTemplate = template.New("serverstxt")
|
||||
if serversTxtTemplate, err = serversTxtTemplate.Parse(serversTxtRawTemplate); err != nil {
|
||||
panic("Unable to parse servers.txt template: " + err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) handleGetStatusTxt(c *gin.Context) {
|
||||
baseURL, ok := s.getBaseURLOrErr(c)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
// Generate a new status.txt
|
||||
statusTxt, err := generateStatusTxt(baseURL)
|
||||
if err != nil {
|
||||
c.Writer.WriteHeader(http.StatusInternalServerError)
|
||||
c.Writer.WriteString("Error generating status.txt")
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.Writer.WriteHeader(http.StatusOK)
|
||||
c.Writer.WriteString(statusTxt)
|
||||
}
|
||||
|
||||
func generateStatusTxt(baseURL string) (txt string, err error) {
|
||||
type TemplateData struct {
|
||||
ApiServerBaseURL string
|
||||
}
|
||||
|
||||
tmplData := TemplateData{ApiServerBaseURL: baseURL}
|
||||
|
||||
buf := bytes.Buffer{}
|
||||
buf.Grow(1024)
|
||||
if err = statusTxtTemplate.Execute(&buf, &tmplData); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Ensure all newlines have a carriage return
|
||||
txt = strings.ReplaceAll(buf.String(), "\n", "\r\n")
|
||||
return
|
||||
}
|
||||
|
||||
type DataJsonStatus struct {
|
||||
Data map[string][]string `json:"data"`
|
||||
}
|
||||
|
||||
func (s *Server) handleGetStatusJSON(c *gin.Context) {
|
||||
baseURL, ok := s.getBaseURLOrErr(c)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
statusJson := DataJsonStatus{
|
||||
Data: map[string][]string{
|
||||
"v3": {
|
||||
baseURL + "/api/v1/data/openfsd-data.json",
|
||||
},
|
||||
"servers": {
|
||||
baseURL + "/api/v1/data/openfsd-servers.json",
|
||||
},
|
||||
"servers_sweatbox": {
|
||||
baseURL + "/api/v1/data/sweatbox-servers.json",
|
||||
},
|
||||
"servers_all": {
|
||||
baseURL + "/api/v1/data/all-servers.json",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
res, err := json.Marshal(&statusJson)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
writePlaintext500Error(c, "Unable to marshal JSON")
|
||||
return
|
||||
}
|
||||
|
||||
c.Writer.Header().Set("Content-Type", "application/json")
|
||||
c.Writer.WriteHeader(http.StatusOK)
|
||||
c.Writer.Write(res)
|
||||
}
|
||||
|
||||
type DataJsonServer struct {
|
||||
Ident string `json:"ident"`
|
||||
HostnameOrIp string `json:"hostname_or_ip"`
|
||||
Location string `json:"location"`
|
||||
Name string `json:"name"`
|
||||
ClientsConnectionAllowed int `json:"clients_connection_allowed"`
|
||||
ClientConnectionsAllowed bool `json:"client_connections_allowed"`
|
||||
IsSweatbox bool `json:"is_sweatbox"`
|
||||
}
|
||||
|
||||
func (s *Server) handleGetServersJSON(c *gin.Context) {
|
||||
serverIdent, serverHostname, serverLocation, err := s.getFsdServerInfo()
|
||||
if err != nil {
|
||||
writePlaintext500Error(c, "Unable to load FSD server info from configuration")
|
||||
return
|
||||
}
|
||||
|
||||
_, isSweatbox := c.Get("is_sweatbox")
|
||||
|
||||
type ServersJson []DataJsonServer
|
||||
dataJson := ServersJson{
|
||||
{
|
||||
Ident: serverIdent,
|
||||
HostnameOrIp: serverHostname,
|
||||
Location: serverLocation,
|
||||
Name: serverIdent,
|
||||
ClientConnectionsAllowed: true,
|
||||
ClientsConnectionAllowed: 99,
|
||||
IsSweatbox: isSweatbox,
|
||||
},
|
||||
}
|
||||
|
||||
res, err := json.Marshal(&dataJson)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
writePlaintext500Error(c, "Unable to marshal JSON")
|
||||
return
|
||||
}
|
||||
|
||||
c.Writer.Header().Set("Content-Type", "application/json")
|
||||
c.Writer.WriteHeader(http.StatusOK)
|
||||
c.Writer.Write(res)
|
||||
}
|
||||
|
||||
func (s *Server) handleGetServersTxt(c *gin.Context) {
|
||||
serversTxt, err := s.generateServersTxt()
|
||||
if err != nil {
|
||||
c.Writer.WriteHeader(http.StatusInternalServerError)
|
||||
c.Writer.WriteString("Error generating status.txt")
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.Writer.WriteHeader(http.StatusOK)
|
||||
c.Writer.WriteString(serversTxt)
|
||||
}
|
||||
|
||||
func (s *Server) generateServersTxt() (txt string, err error) {
|
||||
serverIdent, serverHostname, serverLocation, err := s.getFsdServerInfo()
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
type TemplateData []DataJsonServer
|
||||
tmplData := TemplateData{
|
||||
{
|
||||
Ident: serverIdent,
|
||||
HostnameOrIp: serverHostname,
|
||||
Location: serverLocation,
|
||||
Name: serverIdent,
|
||||
ClientConnectionsAllowed: true,
|
||||
ClientsConnectionAllowed: 99,
|
||||
IsSweatbox: false,
|
||||
},
|
||||
}
|
||||
|
||||
buf := bytes.Buffer{}
|
||||
buf.Grow(1024)
|
||||
if err = serversTxtTemplate.Execute(&buf, &tmplData); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Ensure all newlines have a carriage return
|
||||
txt = strings.ReplaceAll(buf.String(), "\n", "\r\n")
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Server) getFsdServerInfo() (serverIdent string, serverHostname string, serverLocation string, err error) {
|
||||
serverIdent, err = s.dbRepo.ConfigRepo.Get(db.ConfigFsdServerIdent)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
serverHostname, err = s.dbRepo.ConfigRepo.Get(db.ConfigFsdServerHostname)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
serverLocation, err = s.dbRepo.ConfigRepo.Get(db.ConfigFsdServerLocation)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func writePlaintext500Error(c *gin.Context, msg string) {
|
||||
c.Writer.Header().Set("Content-Type", "text/plain")
|
||||
c.Writer.WriteHeader(http.StatusInternalServerError)
|
||||
c.Writer.WriteString(msg)
|
||||
}
|
||||
|
||||
func (s *Server) getBaseURLOrErr(c *gin.Context) (baseURL string, ok bool) {
|
||||
baseURL, err := s.dbRepo.ConfigRepo.Get(db.ConfigApiServerBaseURL)
|
||||
if err != nil {
|
||||
c.Writer.WriteHeader(http.StatusInternalServerError)
|
||||
if !errors.Is(err, db.ErrConfigKeyNotFound) {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
errMsg := "API server base URL is not set in the config"
|
||||
slog.Error(errMsg)
|
||||
c.Writer.WriteString(errMsg)
|
||||
return
|
||||
}
|
||||
|
||||
ok = true
|
||||
return
|
||||
}
|
||||
|
||||
type Datafeed struct {
|
||||
Pilots []fsd.OnlineUserPilot `json:"pilots"`
|
||||
ATC []fsd.OnlineUserATC `json:"atc"`
|
||||
}
|
||||
|
||||
type DatafeedCache struct {
|
||||
jsonStr string
|
||||
lastUpdated time.Time
|
||||
}
|
||||
|
||||
var datafeedCache atomic.Pointer[DatafeedCache]
|
||||
|
||||
func (s *Server) getDatafeed(c *gin.Context) {
|
||||
feed := datafeedCache.Load()
|
||||
if feed == nil {
|
||||
c.AbortWithStatus(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
c.Writer.Header().Set("Content-Type", "application/json")
|
||||
c.Writer.WriteHeader(http.StatusOK)
|
||||
c.Writer.WriteString(feed.jsonStr)
|
||||
}
|
||||
|
||||
func (s *Server) generateDatafeed() (feed *DatafeedCache, err error) {
|
||||
// Generate JWT bearer token
|
||||
customFields := fsd.CustomFields{
|
||||
TokenType: "fsd_service",
|
||||
CID: -1,
|
||||
NetworkRating: fsd.NetworkRatingAdministator,
|
||||
}
|
||||
token, err := fsd.MakeJwtToken(&customFields, 15*time.Minute)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
secretKey, err := s.dbRepo.ConfigRepo.Get(db.ConfigJwtSecretKey)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
tokenStr, err := token.SignedString([]byte(secretKey))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
client := http.Client{}
|
||||
req, err := http.NewRequest("GET", s.cfg.FsdHttpServiceAddress+"/online_users", nil)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
req.Header.Set("Authorization", "Bearer "+tokenStr)
|
||||
res, err := client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if res.StatusCode != http.StatusOK {
|
||||
err = errors.New("FSD HTTP service returned a non-200 status code")
|
||||
return
|
||||
}
|
||||
|
||||
decoder := json.NewDecoder(res.Body)
|
||||
onlineUsers := fsd.OnlineUsersResponseData{}
|
||||
if err = decoder.Decode(&onlineUsers); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
dataFeed := Datafeed{
|
||||
Pilots: onlineUsers.Pilots,
|
||||
ATC: onlineUsers.ATC,
|
||||
}
|
||||
buf := bytes.Buffer{}
|
||||
encoder := json.NewEncoder(&buf)
|
||||
if err = encoder.Encode(&dataFeed); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
feed = &DatafeedCache{
|
||||
jsonStr: buf.String(),
|
||||
lastUpdated: time.Now(),
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Server) runDatafeedWorker(ctx context.Context) {
|
||||
s.updateDataFeedCache()
|
||||
ticker := time.NewTicker(15 * time.Second)
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
s.updateDataFeedCache()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) updateDataFeedCache() {
|
||||
feed, err := s.generateDatafeed()
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
return
|
||||
}
|
||||
datafeedCache.Store(feed)
|
||||
}
|
||||
Reference in New Issue
Block a user