adding validated services? patching forcad_local.py
This commit is contained in:
73
OmCTF-2025/services/block_game/backend/utils/utils.go
Normal file
73
OmCTF-2025/services/block_game/backend/utils/utils.go
Normal file
@@ -0,0 +1,73 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"omctf.ru/block-game-backend/codegen/ent/level"
|
||||
"omctf.ru/block-game-backend/codegen/ent/predicate"
|
||||
"omctf.ru/block-game-backend/codegen/ent/user"
|
||||
"omctf.ru/block-game-backend/db"
|
||||
)
|
||||
|
||||
func BailInternalServerError(w http.ResponseWriter, err error) {
|
||||
http.Error(w, "Unexpected error", http.StatusInternalServerError)
|
||||
log.Printf("Unexpected error occured: %v\n", err)
|
||||
}
|
||||
|
||||
func RespondWithJSON(w http.ResponseWriter, o any) {
|
||||
data, err := json.Marshal(o)
|
||||
if err != nil {
|
||||
BailInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.Write(data)
|
||||
}
|
||||
|
||||
func GetJSONBody[T any](r *http.Request) (*T, error) {
|
||||
var req T
|
||||
err := json.NewDecoder(r.Body).Decode(&req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid request body: %w", err)
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
func MustMarshal(value any) json.RawMessage {
|
||||
data, err := json.Marshal(value)
|
||||
if err != nil {
|
||||
log.Panicf("Can't marshal value %v: %s", value, err)
|
||||
}
|
||||
return data
|
||||
}
|
||||
|
||||
func LevelAccessibleBy(userId int) predicate.Level {
|
||||
return level.Or( // one of
|
||||
level.HasOwnerWith(user.ID(userId)), // user owns the level
|
||||
level.HasInvitedPlayersWith(user.ID(userId)), // or user is invited to the level
|
||||
level.VisibilityEQ(level.VisibilityPublic), // or it's public
|
||||
)
|
||||
}
|
||||
|
||||
const (
|
||||
CleanUpDelay = time.Hour * 1
|
||||
)
|
||||
|
||||
func OccasionallyCleanUp() {
|
||||
ticker := time.NewTicker(CleanUpDelay / 4)
|
||||
for range ticker.C {
|
||||
cnt, err := db.Client.Level.Delete().Where(
|
||||
level.CreatedAtLT(time.Now().Add(-CleanUpDelay)),
|
||||
).Exec(context.Background())
|
||||
if err != nil {
|
||||
log.Printf("Error during levels clean up: %s", err)
|
||||
}
|
||||
log.Printf("Cleaned up %d levels", cnt)
|
||||
}
|
||||
}
|
||||
40
OmCTF-2025/services/block_game/backend/utils/xy/xy.go
Normal file
40
OmCTF-2025/services/block_game/backend/utils/xy/xy.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package xy
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type Point struct {
|
||||
X int `json:"x"`
|
||||
Y int `json:"y"`
|
||||
}
|
||||
|
||||
func (p Point) Go(dir Direction) (Point, error) {
|
||||
switch dir {
|
||||
case Up:
|
||||
return Point{X: p.X, Y: p.Y - 1}, nil
|
||||
case Down:
|
||||
return Point{X: p.X, Y: p.Y + 1}, nil
|
||||
case Left:
|
||||
return Point{X: p.X - 1, Y: p.Y}, nil
|
||||
case Right:
|
||||
return Point{X: p.X + 1, Y: p.Y}, nil
|
||||
}
|
||||
return p, fmt.Errorf("invalid direction: %s", dir)
|
||||
}
|
||||
|
||||
type Direction string
|
||||
|
||||
const (
|
||||
Up Direction = "up"
|
||||
Down Direction = "down"
|
||||
Left Direction = "left"
|
||||
Right Direction = "right"
|
||||
)
|
||||
|
||||
func (d Direction) Validate() error {
|
||||
if d == Up || d == Down || d == Left || d == Right {
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("invalid direction: %s (out of %v)", d, []Direction{Up, Down, Left, Right})
|
||||
}
|
||||
Reference in New Issue
Block a user