Caleb Gardner
2afbd64dc2
Fixed issues with CORS not being able to do an OPTIONS request Fixed folders not being processed properly. Fixed some other issues.
73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package swassistant
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/CalebQ42/darkstorm-server/internal/backend"
|
|
"github.com/CalebQ42/darkstorm-server/internal/backend/db"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type SWBackend struct {
|
|
back *backend.Backend
|
|
db *mongo.Database
|
|
}
|
|
|
|
func NewSWBackend(db *mongo.Database) *SWBackend {
|
|
go func() {
|
|
for range time.Tick(time.Hour) {
|
|
log.Println("SWAssistant: Deleting expired profiles")
|
|
res, err := db.Collection("profiles").DeleteMany(context.Background(), bson.M{"expiration": bson.M{"$lt": time.Now().Unix()}})
|
|
if err == mongo.ErrNoDocuments {
|
|
continue
|
|
}
|
|
log.Println("SWAssistant: Deleted", res.DeletedCount, "profiles")
|
|
}
|
|
}()
|
|
return &SWBackend{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (s *SWBackend) AppID() string {
|
|
return "swassistant"
|
|
}
|
|
|
|
func (s *SWBackend) CountTable() backend.CountTable {
|
|
return db.NewMongoTable[backend.CountLog](s.db.Collection("logs"))
|
|
}
|
|
|
|
func (s *SWBackend) CrashTable() backend.CrashTable {
|
|
return db.NewMongoCrashTable(s.db.Collection("crashes"), s.db.Collection("crashArchive"))
|
|
}
|
|
|
|
func (s *SWBackend) AddBackend(b *backend.Backend) {
|
|
s.back = b
|
|
}
|
|
|
|
func (s *SWBackend) AddCrash(cr backend.IndividualCrash) bool {
|
|
res := s.db.Collection("versions").FindOne(context.Background(), bson.M{"version": cr.Version})
|
|
return res.Err() != mongo.ErrNoDocuments
|
|
}
|
|
|
|
func (s *SWBackend) Extension(mux *http.ServeMux) {
|
|
mux.HandleFunc("GET /swa/room", s.ListRooms)
|
|
mux.HandleFunc("POST /swa/room", s.NewRoom)
|
|
mux.HandleFunc("GET /swa/room/{roomID}", s.GetRoom)
|
|
|
|
mux.HandleFunc("POST /swa/profile", s.UploadProfile)
|
|
mux.HandleFunc("GET /swa/profile/{profileID}", s.GetProfile)
|
|
|
|
//Legacy (TODO: remove this after a month or two after the applciation gets updated)
|
|
mux.HandleFunc("GET /room/list", s.ListRooms)
|
|
mux.HandleFunc("POST /room/new", s.NewRoom)
|
|
mux.HandleFunc("GET /room/{roomID}", s.GetRoom)
|
|
|
|
mux.HandleFunc("POST /profile/upload", s.UploadProfile)
|
|
mux.HandleFunc("GET /profile/{profileID}", s.GetProfile)
|
|
}
|