Kiu/Semaphore
Kiu/Semaphore
1
0
Fork 0
mirror of https://github.com/semaphoreui/semaphore.git synced 2025-01-20 15:29:28 +01:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat(inv): implement attach/detach to tpl

Browse Source
This commit is contained in:
Denis Gukov 2024-12-18 00:16:22 +05:00
parent 0a5e1c269c
commit f75bbc2c5a
No known key found for this signature in database
GPG Key ID: 044381366A5D4731
3 changed files with 74 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
api/projects/templates.go
View File

@ -230,13 +230,68 @@ func RemoveTemplate(w http.ResponseWriter, r *http.Request) {
}
func SetTemplateInventory(w http.ResponseWriter, r *http.Request) {
tpl := context.Get(r, "template").(db.Template)
inv := context.Get(r, "inventory").(db.Inventory)
if !tpl.App.HasInventoryType(inv.Type) {
helpers.WriteErrorStatus(w, "Inventory type is not supported for this template", http.StatusBadRequest)
return
}
if tpl.App.IsTerraform() && (inv.TemplateID == nil || *inv.TemplateID != tpl.ID) {
helpers.WriteErrorStatus(w, "Inventory is not attached to this template", http.StatusBadRequest)
return
}
tpl.InventoryID = &inv.ID
err := helpers.Store(r).UpdateTemplate(tpl)
if err != nil {
helpers.WriteError(w, err)
return
}
w.WriteHeader(http.StatusNoContent)
}
func AttachInventory(w http.ResponseWriter, r *http.Request) {
tpl := context.Get(r, "template").(db.Template)
inv := context.Get(r, "inventory").(db.Inventory)
if inv.TemplateID != nil {
helpers.WriteErrorStatus(w, "Inventory is already attached to another template", http.StatusBadRequest)
return
}
if !tpl.App.HasInventoryType(inv.Type) {
helpers.WriteErrorStatus(w, "Inventory type is not supported for this template", http.StatusBadRequest)
return
}
inv.TemplateID = &tpl.ID
err := helpers.Store(r).UpdateInventory(inv)
if err != nil {
helpers.WriteError(w, err)
return
}
w.WriteHeader(http.StatusNoContent)
}
func DetachInventory(w http.ResponseWriter, r *http.Request) {
tpl := context.Get(r, "template").(db.Template)
inv := context.Get(r, "inventory").(db.Inventory)
if inv.TemplateID == nil || *inv.TemplateID != tpl.ID {
helpers.WriteErrorStatus(w, "Inventory is not attached to this template", http.StatusBadRequest)
return
}
inv.TemplateID = nil
err := helpers.Store(r).UpdateInventory(inv)
if err != nil {
helpers.WriteError(w, err)
return
}
w.WriteHeader(http.StatusNoContent)
}

9
api/router.go
View File

@ -320,9 +320,12 @@ func Route() *mux.Router {
projectTmplManagement.HandleFunc("/{template_id}/tasks/last", projects.GetLastTasks).Methods("GET")
projectTmplManagement.HandleFunc("/{template_id}/schedules", projects.GetTemplateSchedules).Methods("GET")
projectTmplManagement.HandleFunc("/{template_id}/stats", projects.GetTaskStats).Methods("GET")
projectTmplManagement.HandleFunc("/{template_id}/inventory", projects.AttachInventory).Methods("POST")
projectTmplManagement.HandleFunc("/{template_id}/inventory/{inventory_id}", projects.DetachInventory).Methods("DELETE")
projectTmplManagement.HandleFunc("/{template_id}/default_inventory", projects.SetTemplateInventory).Methods("PUT")
projectTmplInvManagement := projectTmplManagement.PathPrefix("/{template_id}/inventory").Subrouter()
projectTmplInvManagement.Use(projects.InventoryMiddleware)
projectTmplInvManagement.HandleFunc("/{inventory_id}/set_default", projects.SetTemplateInventory).Methods("POST")
projectTmplInvManagement.HandleFunc("/{inventory_id}/attach", projects.AttachInventory).Methods("POST")
projectTmplInvManagement.HandleFunc("/{inventory_id}/detach", projects.DetachInventory).Methods("POST")
projectTaskManagement := projectUserAPI.PathPrefix("/tasks").Subrouter()
projectTaskManagement.Use(projects.GetTaskMiddleware)

13
db/Template.go
View File

@ -37,6 +37,19 @@ func (t TemplateApp) InventoryTypes() ([]InventoryType, error) {
}
}
func (t TemplateApp) HasInventoryType(inventoryType InventoryType) bool {
types, err := t.InventoryTypes()
if err != nil {
return false
}
for _, typ := range types {
if typ == inventoryType {
return true
}
}
return false
}
func (t TemplateApp) IsTerraform() bool {
return t == AppTerraform || t == AppTofu
}