77 lines
2.0 KiB
Go
77 lines
2.0 KiB
Go
//go:generate oapi-codegen -generate types -o openapi_types.gen.go -package api flamenco-manager.yaml
|
|
//go:generate oapi-codegen -generate gin -o openapi_gin.gen.go -package api flamenco-manager.yaml
|
|
//go:generate oapi-codegen -generate spec -o openapi_spec.gen.go -package api flamenco-manager.yaml
|
|
//go:generate oapi-codegen -generate client -o openapi_client.gen.go -package api flamenco-manager.yaml
|
|
|
|
package api
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
type Flamenco struct {
|
|
}
|
|
|
|
var _ ServerInterface = (*Flamenco)(nil)
|
|
|
|
func NewFlamenco() *Flamenco {
|
|
return &Flamenco{}
|
|
}
|
|
|
|
func (f *Flamenco) RegisterWorker(c *gin.Context) {
|
|
remoteIP, isTrustedProxy := c.RemoteIP()
|
|
|
|
logger := log.With().
|
|
Str("ip", remoteIP.String()).
|
|
Bool("trustedProxy", isTrustedProxy).
|
|
Logger()
|
|
|
|
var req RegisterWorkerJSONBody
|
|
err := c.Bind(&req)
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("bad request received")
|
|
sendAPIError(c, http.StatusBadRequest, "invalid format")
|
|
return
|
|
}
|
|
|
|
logger.Info().Str("nickname", req.Nickname).Msg("registering new worker")
|
|
|
|
c.JSON(http.StatusOK, &RegisteredWorker{
|
|
Id: uuid.New().String(),
|
|
Nickname: req.Nickname,
|
|
Platform: req.Platform,
|
|
Address: remoteIP.String(),
|
|
})
|
|
}
|
|
|
|
func (f *Flamenco) PostTask(c *gin.Context) {
|
|
c.JSON(http.StatusOK, &AssignedTask{
|
|
Id: uuid.New().String(),
|
|
Commands: []Command{
|
|
{"echo", gin.H{"payload": "Simon says \"Shaders!\""}},
|
|
{"blender", gin.H{"blender_cmd": "/shared/bin/blender"}},
|
|
},
|
|
Job: uuid.New().String(),
|
|
JobPriority: 50,
|
|
JobType: "blender-render",
|
|
Name: "A1032",
|
|
Priority: 50,
|
|
Status: "active",
|
|
TaskType: "blender-render",
|
|
})
|
|
}
|
|
|
|
// sendPetstoreError wraps sending of an error in the Error format, and
|
|
// handling the failure to marshal that.
|
|
func sendAPIError(c *gin.Context, code int, message string) {
|
|
petErr := Error{
|
|
Code: int32(code),
|
|
Message: message,
|
|
}
|
|
c.JSON(code, petErr)
|
|
}
|