mirror of
https://github.com/semaphoreui/semaphore.git
synced 2025-01-21 07:49:34 +01:00
d1b7ad021e
* feat(runners): add register endpoint * feat(runners): add remote runner * refactor(runners): move functionality TaskRunner -> AnsibleJobRunner * fix(runners): init job * chore(runners): remote unused field * feat(runners): use external logger from AnsibleJobRunner * refactor(runners): remove status field * refactor(runners): remove mutation from job * feat(runners): pass username and verison to task * test(runners): fix tests * fix(runners): params for Run * feat(runners): implement runner selection * feat(runners): fill required fields * fix(runners): session block * feat(runners): kill process * refactor(runners): rename fields to public * feat(runners): remote runner functionallity * refactor(runners): remove unused class * fix(runners): send json * feat(runners): runner registration * feat(runners): logging * feat(runners): server <-> running communication works * feat(runners): pass creds to runenr
103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package helpers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/ansible-semaphore/semaphore/services/tasks"
|
|
"net/http"
|
|
"net/url"
|
|
"runtime/debug"
|
|
"strconv"
|
|
"strings"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/gorilla/context"
|
|
|
|
"github.com/ansible-semaphore/semaphore/db"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func Store(r *http.Request) db.Store {
|
|
return context.Get(r, "store").(db.Store)
|
|
}
|
|
|
|
func TaskPool(r *http.Request) *tasks.TaskPool {
|
|
return context.Get(r, "task_pool").(*tasks.TaskPool)
|
|
}
|
|
|
|
func isXHR(w http.ResponseWriter, r *http.Request) bool {
|
|
accept := r.Header.Get("Accept")
|
|
return !strings.Contains(accept, "text/html")
|
|
}
|
|
|
|
// GetIntParam fetches a parameter from the route variables as an integer
|
|
// redirects to a 404 or writes bad request state depending on error state
|
|
func GetIntParam(name string, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
intParam, err := strconv.Atoi(mux.Vars(r)[name])
|
|
|
|
if err != nil {
|
|
if !isXHR(w, r) {
|
|
http.Redirect(w, r, "/404", http.StatusFound)
|
|
} else {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
}
|
|
|
|
return 0, err
|
|
}
|
|
|
|
return intParam, nil
|
|
}
|
|
|
|
// H just a string-to-anything map
|
|
type H map[string]interface{}
|
|
|
|
// Bind decodes json into object
|
|
func Bind(w http.ResponseWriter, r *http.Request, out interface{}) bool {
|
|
err := json.NewDecoder(r.Body).Decode(out)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
}
|
|
|
|
return err == nil
|
|
}
|
|
|
|
// WriteJSON writes object as JSON
|
|
func WriteJSON(w http.ResponseWriter, code int, out interface{}) {
|
|
w.Header().Set("content-type", "application/json")
|
|
w.WriteHeader(code)
|
|
|
|
if err := json.NewEncoder(w).Encode(out); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func WriteError(w http.ResponseWriter, err error) {
|
|
if err == db.ErrNotFound {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
if err == db.ErrInvalidOperation {
|
|
w.WriteHeader(http.StatusConflict)
|
|
return
|
|
}
|
|
|
|
switch e := err.(type) {
|
|
case *db.ValidationError:
|
|
WriteJSON(w, http.StatusBadRequest, map[string]string{
|
|
"error": e.Error(),
|
|
})
|
|
default:
|
|
log.Error(err)
|
|
debug.PrintStack()
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
func QueryParams(url *url.URL) db.RetrieveQueryParams {
|
|
return db.RetrieveQueryParams{
|
|
SortBy: url.Query().Get("sort"),
|
|
SortInverted: url.Query().Get("order") == "desc",
|
|
}
|
|
}
|