mirror of
https://github.com/hrfee/jfa-go.git
synced 2024-12-23 09:30:12 +00:00
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/gin-gonic/gin"
|
||
|
)
|
||
|
|
||
|
func (app *appContext) getOmbiUser(jfID string) (map[string]interface{}, int, error) {
|
||
|
ombiUsers, code, err := app.ombi.GetUsers()
|
||
|
if err != nil || code != 200 {
|
||
|
return nil, code, err
|
||
|
}
|
||
|
jfUser, code, err := app.jf.UserByID(jfID, false)
|
||
|
if err != nil || code != 200 {
|
||
|
return nil, code, err
|
||
|
}
|
||
|
username := jfUser.Name
|
||
|
email := ""
|
||
|
if e, ok := app.storage.emails[jfID]; ok {
|
||
|
email = e.Addr
|
||
|
}
|
||
|
for _, ombiUser := range ombiUsers {
|
||
|
ombiAddr := ""
|
||
|
if a, ok := ombiUser["emailAddress"]; ok && a != nil {
|
||
|
ombiAddr = a.(string)
|
||
|
}
|
||
|
if ombiUser["userName"].(string) == username || (ombiAddr == email && email != "") {
|
||
|
return ombiUser, code, err
|
||
|
}
|
||
|
}
|
||
|
return nil, 400, fmt.Errorf("Couldn't find user")
|
||
|
}
|
||
|
|
||
|
// @Summary Get a list of Ombi users.
|
||
|
// @Produce json
|
||
|
// @Success 200 {object} ombiUsersDTO
|
||
|
// @Failure 500 {object} stringResponse
|
||
|
// @Router /ombi/users [get]
|
||
|
// @Security Bearer
|
||
|
// @tags Ombi
|
||
|
func (app *appContext) OmbiUsers(gc *gin.Context) {
|
||
|
app.debug.Println("Ombi users requested")
|
||
|
users, status, err := app.ombi.GetUsers()
|
||
|
if err != nil || status != 200 {
|
||
|
app.err.Printf("Failed to get users from Ombi (%d): %v", status, err)
|
||
|
respond(500, "Couldn't get users", gc)
|
||
|
return
|
||
|
}
|
||
|
userlist := make([]ombiUser, len(users))
|
||
|
for i, data := range users {
|
||
|
userlist[i] = ombiUser{
|
||
|
Name: data["userName"].(string),
|
||
|
ID: data["id"].(string),
|
||
|
}
|
||
|
}
|
||
|
gc.JSON(200, ombiUsersDTO{Users: userlist})
|
||
|
}
|
||
|
|
||
|
// @Summary Store Ombi user template in an existing profile.
|
||
|
// @Produce json
|
||
|
// @Param ombiUser body ombiUser true "User to source settings from"
|
||
|
// @Param profile path string true "Name of profile to store in"
|
||
|
// @Success 200 {object} boolResponse
|
||
|
// @Failure 400 {object} boolResponse
|
||
|
// @Failure 500 {object} stringResponse
|
||
|
// @Router /profiles/ombi/{profile} [post]
|
||
|
// @Security Bearer
|
||
|
// @tags Ombi
|
||
|
func (app *appContext) SetOmbiProfile(gc *gin.Context) {
|
||
|
var req ombiUser
|
||
|
gc.BindJSON(&req)
|
||
|
profileName := gc.Param("profile")
|
||
|
profile, ok := app.storage.profiles[profileName]
|
||
|
if !ok {
|
||
|
respondBool(400, false, gc)
|
||
|
return
|
||
|
}
|
||
|
template, code, err := app.ombi.TemplateByID(req.ID)
|
||
|
if err != nil || code != 200 || len(template) == 0 {
|
||
|
app.err.Printf("Couldn't get user from Ombi (%d): %v", code, err)
|
||
|
respond(500, "Couldn't get user", gc)
|
||
|
return
|
||
|
}
|
||
|
profile.Ombi = template
|
||
|
app.storage.profiles[profileName] = profile
|
||
|
if err := app.storage.storeProfiles(); err != nil {
|
||
|
respond(500, "Failed to store profile", gc)
|
||
|
app.err.Printf("Failed to store profiles: %v", err)
|
||
|
return
|
||
|
}
|
||
|
respondBool(204, true, gc)
|
||
|
}
|
||
|
|
||
|
// @Summary Remove ombi user template from a profile.
|
||
|
// @Produce json
|
||
|
// @Param profile path string true "Name of profile to store in"
|
||
|
// @Success 200 {object} boolResponse
|
||
|
// @Failure 400 {object} boolResponse
|
||
|
// @Failure 500 {object} stringResponse
|
||
|
// @Router /profiles/ombi/{profile} [delete]
|
||
|
// @Security Bearer
|
||
|
// @tags Ombi
|
||
|
func (app *appContext) DeleteOmbiProfile(gc *gin.Context) {
|
||
|
profileName := gc.Param("profile")
|
||
|
profile, ok := app.storage.profiles[profileName]
|
||
|
if !ok {
|
||
|
respondBool(400, false, gc)
|
||
|
return
|
||
|
}
|
||
|
profile.Ombi = nil
|
||
|
app.storage.profiles[profileName] = profile
|
||
|
if err := app.storage.storeProfiles(); err != nil {
|
||
|
respond(500, "Failed to store profile", gc)
|
||
|
app.err.Printf("Failed to store profiles: %v", err)
|
||
|
return
|
||
|
}
|
||
|
respondBool(204, true, gc)
|
||
|
}
|