mirror of
https://github.com/hrfee/jfa-go.git
synced 2024-11-15 23:00:10 +00:00
Harvey Tindall
c4acb43cb8
user templates will become profiles. You will be able to make multiple, and assign them to invites individually. This commit migrates the separate template files into one profile entry called "Default", and lets you select them on invites. No way to create profiles has been added yet.
977 lines
30 KiB
Go
977 lines
30 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/knz/strtime"
|
|
"github.com/lithammer/shortuuid/v3"
|
|
"gopkg.in/ini.v1"
|
|
)
|
|
|
|
func (app *appContext) loadStrftime() {
|
|
app.datePattern = app.config.Section("email").Key("date_format").String()
|
|
app.timePattern = `%H:%M`
|
|
if val, _ := app.config.Section("email").Key("use_24h").Bool(); !val {
|
|
app.timePattern = `%I:%M %p`
|
|
}
|
|
return
|
|
}
|
|
|
|
func (app *appContext) prettyTime(dt time.Time) (date, time string) {
|
|
date, _ = strtime.Strftime(dt, app.datePattern)
|
|
time, _ = strtime.Strftime(dt, app.timePattern)
|
|
return
|
|
}
|
|
|
|
func (app *appContext) formatDatetime(dt time.Time) string {
|
|
d, t := app.prettyTime(dt)
|
|
return d + " " + t
|
|
}
|
|
|
|
// https://stackoverflow.com/questions/36530251/time-since-with-months-and-years/36531443#36531443 THANKS
|
|
func timeDiff(a, b time.Time) (year, month, day, hour, min, sec int) {
|
|
if a.Location() != b.Location() {
|
|
b = b.In(a.Location())
|
|
}
|
|
if a.After(b) {
|
|
a, b = b, a
|
|
}
|
|
y1, M1, d1 := a.Date()
|
|
y2, M2, d2 := b.Date()
|
|
|
|
h1, m1, s1 := a.Clock()
|
|
h2, m2, s2 := b.Clock()
|
|
|
|
year = int(y2 - y1)
|
|
month = int(M2 - M1)
|
|
day = int(d2 - d1)
|
|
hour = int(h2 - h1)
|
|
min = int(m2 - m1)
|
|
sec = int(s2 - s1)
|
|
|
|
// Normalize negative values
|
|
if sec < 0 {
|
|
sec += 60
|
|
min--
|
|
}
|
|
if min < 0 {
|
|
min += 60
|
|
hour--
|
|
}
|
|
if hour < 0 {
|
|
hour += 24
|
|
day--
|
|
}
|
|
if day < 0 {
|
|
// days in month:
|
|
t := time.Date(y1, M1, 32, 0, 0, 0, 0, time.UTC)
|
|
day += 32 - t.Day()
|
|
month--
|
|
}
|
|
if month < 0 {
|
|
month += 12
|
|
year--
|
|
}
|
|
return
|
|
}
|
|
|
|
func (app *appContext) checkInvites() {
|
|
current_time := time.Now()
|
|
app.storage.loadInvites()
|
|
changed := false
|
|
for code, data := range app.storage.invites {
|
|
expiry := data.ValidTill
|
|
if current_time.After(expiry) {
|
|
app.debug.Printf("Housekeeping: Deleting old invite %s", code)
|
|
notify := data.Notify
|
|
if app.config.Section("notifications").Key("enabled").MustBool(false) && len(notify) != 0 {
|
|
app.debug.Printf("%s: Expiry notification", code)
|
|
for address, settings := range notify {
|
|
if settings["notify-expiry"] {
|
|
go func() {
|
|
msg, err := app.email.constructExpiry(code, data, app)
|
|
if err != nil {
|
|
app.err.Printf("%s: Failed to construct expiry notification", code)
|
|
app.debug.Printf("Error: %s", err)
|
|
} else if err := app.email.send(address, msg); err != nil {
|
|
app.err.Printf("%s: Failed to send expiry notification", code)
|
|
app.debug.Printf("Error: %s", err)
|
|
} else {
|
|
app.info.Printf("Sent expiry notification to %s", address)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
changed = true
|
|
delete(app.storage.invites, code)
|
|
}
|
|
}
|
|
if changed {
|
|
app.storage.storeInvites()
|
|
}
|
|
}
|
|
|
|
func (app *appContext) checkInvite(code string, used bool, username string) bool {
|
|
current_time := time.Now()
|
|
app.storage.loadInvites()
|
|
changed := false
|
|
if inv, match := app.storage.invites[code]; match {
|
|
expiry := inv.ValidTill
|
|
if current_time.After(expiry) {
|
|
app.debug.Printf("Housekeeping: Deleting old invite %s", code)
|
|
notify := inv.Notify
|
|
if app.config.Section("notifications").Key("enabled").MustBool(false) && len(notify) != 0 {
|
|
app.debug.Printf("%s: Expiry notification", code)
|
|
for address, settings := range notify {
|
|
if settings["notify-expiry"] {
|
|
go func() {
|
|
msg, err := app.email.constructExpiry(code, inv, app)
|
|
if err != nil {
|
|
app.err.Printf("%s: Failed to construct expiry notification", code)
|
|
app.debug.Printf("Error: %s", err)
|
|
} else if err := app.email.send(address, msg); err != nil {
|
|
app.err.Printf("%s: Failed to send expiry notification", code)
|
|
app.debug.Printf("Error: %s", err)
|
|
} else {
|
|
app.info.Printf("Sent expiry notification to %s", address)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
changed = true
|
|
match = false
|
|
delete(app.storage.invites, code)
|
|
} else if used {
|
|
changed = true
|
|
del := false
|
|
newInv := inv
|
|
if newInv.RemainingUses == 1 {
|
|
del = true
|
|
delete(app.storage.invites, code)
|
|
} else if newInv.RemainingUses != 0 {
|
|
// 0 means infinite i guess?
|
|
newInv.RemainingUses -= 1
|
|
}
|
|
newInv.UsedBy = append(newInv.UsedBy, []string{username, app.formatDatetime(current_time)})
|
|
if !del {
|
|
app.storage.invites[code] = newInv
|
|
}
|
|
}
|
|
if changed {
|
|
app.storage.storeInvites()
|
|
}
|
|
return match
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Routes from now on!
|
|
|
|
type newUserReq struct {
|
|
Username string `json:"username"`
|
|
Password string `json:"password"`
|
|
Email string `json:"email"`
|
|
Code string `json:"code"`
|
|
}
|
|
|
|
func (app *appContext) NewUserAdmin(gc *gin.Context) {
|
|
var req newUserReq
|
|
gc.BindJSON(&req)
|
|
existingUser, _, _ := app.jf.userByName(req.Username, false)
|
|
if existingUser != nil {
|
|
msg := fmt.Sprintf("User already exists named %s", req.Username)
|
|
app.info.Printf("%s New user failed: %s", req.Username, msg)
|
|
respond(401, msg, gc)
|
|
return
|
|
}
|
|
user, status, err := app.jf.newUser(req.Username, req.Password)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("%s New user failed: Jellyfin responded with %d", req.Username, status)
|
|
respond(401, "Unknown error", gc)
|
|
return
|
|
}
|
|
var id string
|
|
if user["Id"] != nil {
|
|
id = user["Id"].(string)
|
|
}
|
|
if len(app.storage.policy) != 0 {
|
|
status, err = app.jf.setPolicy(id, app.storage.policy)
|
|
if !(status == 200 || status == 204) {
|
|
app.err.Printf("%s: Failed to set user policy: Code %d", req.Username, status)
|
|
}
|
|
}
|
|
if len(app.storage.configuration) != 0 && len(app.storage.displayprefs) != 0 {
|
|
status, err = app.jf.setConfiguration(id, app.storage.configuration)
|
|
if (status == 200 || status == 204) && err == nil {
|
|
status, err = app.jf.setDisplayPreferences(id, app.storage.displayprefs)
|
|
} else {
|
|
app.err.Printf("%s: Failed to set configuration template: Code %d", req.Username, status)
|
|
}
|
|
}
|
|
if app.config.Section("password_resets").Key("enabled").MustBool(false) {
|
|
app.storage.emails[id] = req.Email
|
|
app.storage.storeEmails()
|
|
}
|
|
if app.config.Section("ombi").Key("enabled").MustBool(false) {
|
|
app.storage.loadOmbiTemplate()
|
|
if len(app.storage.ombi_template) != 0 {
|
|
errors, code, err := app.ombi.newUser(req.Username, req.Password, req.Email, app.storage.ombi_template)
|
|
if err != nil || code != 200 {
|
|
app.info.Printf("Failed to create Ombi user (%d): %s", code, err)
|
|
app.debug.Printf("Errors reported by Ombi: %s", strings.Join(errors, ", "))
|
|
} else {
|
|
app.info.Println("Created Ombi user")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (app *appContext) NewUser(gc *gin.Context) {
|
|
var req newUserReq
|
|
gc.BindJSON(&req)
|
|
app.debug.Printf("%s: New user attempt", req.Code)
|
|
if !app.checkInvite(req.Code, false, "") {
|
|
app.info.Printf("%s New user failed: invalid code", req.Code)
|
|
gc.JSON(401, map[string]bool{"success": false})
|
|
gc.Abort()
|
|
return
|
|
}
|
|
validation := app.validator.validate(req.Password)
|
|
valid := true
|
|
for _, val := range validation {
|
|
if !val {
|
|
valid = false
|
|
}
|
|
}
|
|
if !valid {
|
|
// 200 bcs idk what i did in js
|
|
app.info.Printf("%s New user failed: Invalid password", req.Code)
|
|
gc.JSON(200, validation)
|
|
gc.Abort()
|
|
return
|
|
}
|
|
existingUser, _, _ := app.jf.userByName(req.Username, false)
|
|
if existingUser != nil {
|
|
msg := fmt.Sprintf("User already exists named %s", req.Username)
|
|
app.info.Printf("%s New user failed: %s", req.Code, msg)
|
|
respond(401, msg, gc)
|
|
return
|
|
}
|
|
user, status, err := app.jf.newUser(req.Username, req.Password)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("%s New user failed: Jellyfin responded with %d", req.Code, status)
|
|
respond(401, "Unknown error", gc)
|
|
return
|
|
}
|
|
app.checkInvite(req.Code, true, req.Username)
|
|
invite := app.storage.invites[req.Code]
|
|
if app.config.Section("notifications").Key("enabled").MustBool(false) {
|
|
for address, settings := range invite.Notify {
|
|
if settings["notify-creation"] {
|
|
go func() {
|
|
msg, err := app.email.constructCreated(req.Code, req.Username, req.Email, invite, app)
|
|
if err != nil {
|
|
app.err.Printf("%s: Failed to construct user creation notification", req.Code)
|
|
app.debug.Printf("%s: Error: %s", req.Code, err)
|
|
} else if err := app.email.send(address, msg); err != nil {
|
|
app.err.Printf("%s: Failed to send user creation notification", req.Code)
|
|
app.debug.Printf("%s: Error: %s", req.Code, err)
|
|
} else {
|
|
app.info.Printf("%s: Sent user creation notification to %s", req.Code, address)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
var id string
|
|
if user["Id"] != nil {
|
|
id = user["Id"].(string)
|
|
}
|
|
if invite.Profile != "" {
|
|
profile, ok := app.storage.profiles[invite.Profile]
|
|
if !ok {
|
|
profile = app.storage.profiles["Default"]
|
|
}
|
|
if len(profile.Policy) != 0 {
|
|
app.debug.Printf("Applying policy from profile \"%s\"", invite.Profile)
|
|
status, err = app.jf.setPolicy(id, profile.Policy)
|
|
if !(status == 200 || status == 204) {
|
|
app.err.Printf("%s: Failed to set user policy: Code %d", req.Code, status)
|
|
}
|
|
}
|
|
if len(profile.Configuration) != 0 && len(profile.Displayprefs) != 0 {
|
|
app.debug.Printf("Applying homescreen from profile \"%s\"", invite.Profile)
|
|
status, err = app.jf.setConfiguration(id, profile.Configuration)
|
|
if (status == 200 || status == 204) && err == nil {
|
|
status, err = app.jf.setDisplayPreferences(id, profile.Displayprefs)
|
|
} else {
|
|
app.err.Printf("%s: Failed to set configuration template: Code %d", req.Code, status)
|
|
}
|
|
}
|
|
}
|
|
if app.config.Section("password_resets").Key("enabled").MustBool(false) {
|
|
app.storage.emails[id] = req.Email
|
|
app.storage.storeEmails()
|
|
}
|
|
if app.config.Section("ombi").Key("enabled").MustBool(false) {
|
|
app.storage.loadOmbiTemplate()
|
|
if len(app.storage.ombi_template) != 0 {
|
|
errors, code, err := app.ombi.newUser(req.Username, req.Password, req.Email, app.storage.ombi_template)
|
|
if err != nil || code != 200 {
|
|
app.info.Printf("Failed to create Ombi user (%d): %s", code, err)
|
|
app.debug.Printf("Errors reported by Ombi: %s", strings.Join(errors, ", "))
|
|
} else {
|
|
app.info.Println("Created Ombi user")
|
|
}
|
|
}
|
|
}
|
|
gc.JSON(200, validation)
|
|
}
|
|
|
|
type deleteUserReq struct {
|
|
Users []string `json:"users"`
|
|
Notify bool `json:"notify"`
|
|
Reason string `json:"reason"`
|
|
}
|
|
|
|
func (app *appContext) DeleteUser(gc *gin.Context) {
|
|
var req deleteUserReq
|
|
gc.BindJSON(&req)
|
|
errors := map[string]string{}
|
|
for _, userID := range req.Users {
|
|
status, err := app.jf.deleteUser(userID)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
errors[userID] = fmt.Sprintf("%d: %s", status, err)
|
|
}
|
|
if req.Notify {
|
|
addr, ok := app.storage.emails[userID]
|
|
if addr != nil && ok {
|
|
go func(userID, reason, address string) {
|
|
msg, err := app.email.constructDeleted(reason, app)
|
|
if err != nil {
|
|
app.err.Printf("%s: Failed to construct account deletion email", userID)
|
|
app.debug.Printf("%s: Error: %s", userID, err)
|
|
} else if err := app.email.send(address, msg); err != nil {
|
|
app.err.Printf("%s: Failed to send to %s", userID, address)
|
|
app.debug.Printf("%s: Error: %s", userID, err)
|
|
} else {
|
|
app.info.Printf("%s: Sent invite email to %s", userID, address)
|
|
}
|
|
}(userID, req.Reason, addr.(string))
|
|
}
|
|
}
|
|
}
|
|
app.jf.cacheExpiry = time.Now()
|
|
if len(errors) == len(req.Users) {
|
|
respond(500, "Failed", gc)
|
|
app.err.Printf("Account deletion failed: %s", errors[req.Users[0]])
|
|
return
|
|
} else if len(errors) != 0 {
|
|
gc.JSON(500, errors)
|
|
return
|
|
}
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
type generateInviteReq struct {
|
|
Days int `json:"days"`
|
|
Hours int `json:"hours"`
|
|
Minutes int `json:"minutes"`
|
|
Email string `json:"email"`
|
|
MultipleUses bool `json:"multiple-uses"`
|
|
NoLimit bool `json:"no-limit"`
|
|
RemainingUses int `json:"remaining-uses"`
|
|
Profile string `json:"profile"`
|
|
}
|
|
|
|
func (app *appContext) GenerateInvite(gc *gin.Context) {
|
|
var req generateInviteReq
|
|
app.debug.Println("Generating new invite")
|
|
app.storage.loadInvites()
|
|
gc.BindJSON(&req)
|
|
current_time := time.Now()
|
|
valid_till := current_time.AddDate(0, 0, req.Days)
|
|
valid_till = valid_till.Add(time.Hour*time.Duration(req.Hours) + time.Minute*time.Duration(req.Minutes))
|
|
invite_code := shortuuid.New()
|
|
var invite Invite
|
|
invite.Created = current_time
|
|
if req.MultipleUses {
|
|
if req.NoLimit {
|
|
invite.NoLimit = true
|
|
} else {
|
|
invite.RemainingUses = req.RemainingUses
|
|
}
|
|
} else {
|
|
invite.RemainingUses = 1
|
|
}
|
|
invite.ValidTill = valid_till
|
|
if req.Email != "" && app.config.Section("invite_emails").Key("enabled").MustBool(false) {
|
|
app.debug.Printf("%s: Sending invite email", invite_code)
|
|
invite.Email = req.Email
|
|
msg, err := app.email.constructInvite(invite_code, invite, app)
|
|
if err != nil {
|
|
invite.Email = fmt.Sprintf("Failed to send to %s", req.Email)
|
|
app.err.Printf("%s: Failed to construct invite email", invite_code)
|
|
app.debug.Printf("%s: Error: %s", invite_code, err)
|
|
} else if err := app.email.send(req.Email, msg); err != nil {
|
|
invite.Email = fmt.Sprintf("Failed to send to %s", req.Email)
|
|
app.err.Printf("%s: %s", invite_code, invite.Email)
|
|
app.debug.Printf("%s: Error: %s", invite_code, err)
|
|
} else {
|
|
app.info.Printf("%s: Sent invite email to %s", invite_code, req.Email)
|
|
}
|
|
}
|
|
if req.Profile != "" {
|
|
if _, ok := app.storage.profiles[req.Profile]; ok {
|
|
invite.Profile = req.Profile
|
|
} else {
|
|
invite.Profile = "Default"
|
|
}
|
|
}
|
|
app.storage.invites[invite_code] = invite
|
|
app.storage.storeInvites()
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
type profileReq struct {
|
|
Invite string `json:"invite"`
|
|
Profile string `json:"profile"`
|
|
}
|
|
|
|
func (app *appContext) SetProfile(gc *gin.Context) {
|
|
var req profileReq
|
|
gc.BindJSON(&req)
|
|
app.debug.Printf("%s: Setting profile to \"%s\"", req.Invite, req.Profile)
|
|
if _, ok := app.storage.profiles[req.Profile]; !ok {
|
|
app.err.Printf("%s: Profile \"%s\" not found", req.Invite, req.Profile)
|
|
respond(500, "Profile not found", gc)
|
|
return
|
|
}
|
|
inv := app.storage.invites[req.Invite]
|
|
inv.Profile = req.Profile
|
|
app.storage.invites[req.Invite] = inv
|
|
app.storage.storeInvites()
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
func (app *appContext) GetInvites(gc *gin.Context) {
|
|
app.debug.Println("Invites requested")
|
|
current_time := time.Now()
|
|
app.storage.loadInvites()
|
|
app.checkInvites()
|
|
var invites []map[string]interface{}
|
|
for code, inv := range app.storage.invites {
|
|
_, _, days, hours, minutes, _ := timeDiff(inv.ValidTill, current_time)
|
|
invite := map[string]interface{}{
|
|
"code": code,
|
|
"days": days,
|
|
"hours": hours,
|
|
"minutes": minutes,
|
|
"created": app.formatDatetime(inv.Created),
|
|
"profile": inv.Profile,
|
|
}
|
|
if len(inv.UsedBy) != 0 {
|
|
invite["used-by"] = inv.UsedBy
|
|
}
|
|
if inv.NoLimit {
|
|
invite["no-limit"] = true
|
|
}
|
|
invite["remaining-uses"] = 1
|
|
if inv.RemainingUses != 0 {
|
|
invite["remaining-uses"] = inv.RemainingUses
|
|
}
|
|
if inv.Email != "" {
|
|
invite["email"] = inv.Email
|
|
}
|
|
if len(inv.Notify) != 0 {
|
|
var address string
|
|
if app.config.Section("ui").Key("jellyfin_login").MustBool(false) {
|
|
app.storage.loadEmails()
|
|
if addr := app.storage.emails[gc.GetString("jfId")]; addr != nil {
|
|
address = addr.(string)
|
|
}
|
|
} else {
|
|
address = app.config.Section("ui").Key("email").String()
|
|
}
|
|
if _, ok := inv.Notify[address]; ok {
|
|
for _, notifyType := range []string{"notify-expiry", "notify-creation"} {
|
|
if _, ok = inv.Notify[address][notifyType]; ok {
|
|
invite[notifyType] = inv.Notify[address][notifyType]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
invites = append(invites, invite)
|
|
}
|
|
profiles := make([]string, len(app.storage.profiles))
|
|
i := 0
|
|
for p := range app.storage.profiles {
|
|
profiles[i] = p
|
|
i++
|
|
}
|
|
resp := map[string]interface{}{
|
|
"profiles": profiles,
|
|
"invites": invites,
|
|
}
|
|
gc.JSON(200, resp)
|
|
}
|
|
|
|
func (app *appContext) SetNotify(gc *gin.Context) {
|
|
var req map[string]map[string]bool
|
|
gc.BindJSON(&req)
|
|
changed := false
|
|
for code, settings := range req {
|
|
app.debug.Printf("%s: Notification settings change requested", code)
|
|
app.storage.loadInvites()
|
|
app.storage.loadEmails()
|
|
invite, ok := app.storage.invites[code]
|
|
if !ok {
|
|
app.err.Printf("%s Notification setting change failed: Invalid code", code)
|
|
gc.JSON(400, map[string]string{"error": "Invalid invite code"})
|
|
gc.Abort()
|
|
return
|
|
}
|
|
var address string
|
|
if app.config.Section("ui").Key("jellyfin_login").MustBool(false) {
|
|
var ok bool
|
|
address, ok = app.storage.emails[gc.GetString("jfId")].(string)
|
|
if !ok {
|
|
app.err.Printf("%s: Couldn't find email address. Make sure it's set", code)
|
|
app.debug.Printf("%s: User ID \"%s\"", code, gc.GetString("jfId"))
|
|
gc.JSON(500, map[string]string{"error": "Missing user email"})
|
|
gc.Abort()
|
|
return
|
|
}
|
|
} else {
|
|
address = app.config.Section("ui").Key("email").String()
|
|
}
|
|
if invite.Notify == nil {
|
|
invite.Notify = map[string]map[string]bool{}
|
|
}
|
|
if _, ok := invite.Notify[address]; !ok {
|
|
invite.Notify[address] = map[string]bool{}
|
|
} /*else {
|
|
if _, ok := invite.Notify[address]["notify-expiry"]; !ok {
|
|
*/
|
|
if _, ok := settings["notify-expiry"]; ok && invite.Notify[address]["notify-expiry"] != settings["notify-expiry"] {
|
|
invite.Notify[address]["notify-expiry"] = settings["notify-expiry"]
|
|
app.debug.Printf("%s: Set \"notify-expiry\" to %t for %s", code, settings["notify-expiry"], address)
|
|
changed = true
|
|
}
|
|
if _, ok := settings["notify-creation"]; ok && invite.Notify[address]["notify-creation"] != settings["notify-creation"] {
|
|
invite.Notify[address]["notify-creation"] = settings["notify-creation"]
|
|
app.debug.Printf("%s: Set \"notify-creation\" to %t for %s", code, settings["notify-creation"], address)
|
|
changed = true
|
|
}
|
|
if changed {
|
|
app.storage.invites[code] = invite
|
|
}
|
|
}
|
|
if changed {
|
|
app.storage.storeInvites()
|
|
}
|
|
}
|
|
|
|
type deleteReq struct {
|
|
Code string `json:"code"`
|
|
}
|
|
|
|
func (app *appContext) DeleteInvite(gc *gin.Context) {
|
|
var req deleteReq
|
|
gc.BindJSON(&req)
|
|
app.debug.Printf("%s: Deletion requested", req.Code)
|
|
var ok bool
|
|
_, ok = app.storage.invites[req.Code]
|
|
if ok {
|
|
delete(app.storage.invites, req.Code)
|
|
app.storage.storeInvites()
|
|
app.info.Printf("%s: Invite deleted", req.Code)
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
return
|
|
}
|
|
app.err.Printf("%s: Deletion failed: Invalid code", req.Code)
|
|
respond(401, "Code doesn't exist", gc)
|
|
}
|
|
|
|
type dateToParse struct {
|
|
Parsed time.Time `json:"parseme"`
|
|
}
|
|
|
|
// json magically parses datetimes so why not
|
|
func parseDt(date string) time.Time {
|
|
timeJSON := []byte("{ \"parseme\": \"" + date + "\" }")
|
|
var parsed dateToParse
|
|
json.Unmarshal(timeJSON, &parsed)
|
|
return parsed.Parsed
|
|
}
|
|
|
|
type respUser struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Email string `json:"email,omitempty"`
|
|
// this magically parses a string to time.time
|
|
LastActive string `json:"last_active"`
|
|
Admin bool `json:"admin"`
|
|
}
|
|
|
|
type userResp struct {
|
|
UserList []respUser `json:"users"`
|
|
}
|
|
|
|
func (app *appContext) GetUsers(gc *gin.Context) {
|
|
app.debug.Println("Users requested")
|
|
var resp userResp
|
|
resp.UserList = []respUser{}
|
|
users, status, err := app.jf.getUsers(false)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get users from Jellyfin: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get users", gc)
|
|
return
|
|
}
|
|
for _, jfUser := range users {
|
|
var user respUser
|
|
user.LastActive = "n/a"
|
|
if jfUser["LastActivityDate"] != nil {
|
|
date := parseDt(jfUser["LastActivityDate"].(string))
|
|
user.LastActive = app.formatDatetime(date)
|
|
}
|
|
user.ID = jfUser["Id"].(string)
|
|
user.Name = jfUser["Name"].(string)
|
|
user.Admin = jfUser["Policy"].(map[string]interface{})["IsAdministrator"].(bool)
|
|
if email, ok := app.storage.emails[jfUser["Id"].(string)]; ok {
|
|
user.Email = email.(string)
|
|
}
|
|
|
|
resp.UserList = append(resp.UserList, user)
|
|
}
|
|
gc.JSON(200, resp)
|
|
}
|
|
|
|
type ombiUser struct {
|
|
Name string `json:"name,omitempty"`
|
|
ID string `json:"id"`
|
|
}
|
|
|
|
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: Code %d", status)
|
|
app.debug.Printf("Error: %s", 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, map[string][]ombiUser{"users": userlist})
|
|
}
|
|
|
|
func (app *appContext) ModifyEmails(gc *gin.Context) {
|
|
var req map[string]string
|
|
gc.BindJSON(&req)
|
|
fmt.Println(req)
|
|
app.debug.Println("Email modification requested")
|
|
users, status, err := app.jf.getUsers(false)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get users from Jellyfin: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get users", gc)
|
|
return
|
|
}
|
|
for _, jfUser := range users {
|
|
if address, ok := req[jfUser["Id"].(string)]; ok {
|
|
app.storage.emails[jfUser["Id"].(string)] = address
|
|
}
|
|
}
|
|
app.storage.storeEmails()
|
|
app.info.Println("Email list modified")
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
func (app *appContext) SetOmbiDefaults(gc *gin.Context) {
|
|
var req ombiUser
|
|
gc.BindJSON(&req)
|
|
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 %s", code, err)
|
|
respond(500, "Couldn't get user", gc)
|
|
return
|
|
}
|
|
app.storage.ombi_template = template
|
|
fmt.Println(app.storage.ombi_path)
|
|
app.storage.storeOmbiTemplate()
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
type defaultsReq struct {
|
|
From string `json:"from"`
|
|
ApplyTo []string `json:"apply_to"`
|
|
ID string `json:"id"`
|
|
Homescreen bool `json:"homescreen"`
|
|
}
|
|
|
|
func (app *appContext) SetDefaults(gc *gin.Context) {
|
|
var req defaultsReq
|
|
gc.BindJSON(&req)
|
|
userID := req.ID
|
|
user, status, err := app.jf.userById(userID, false)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get user from Jellyfin: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get user", gc)
|
|
return
|
|
}
|
|
app.info.Printf("Getting user defaults from \"%s\"", user["Name"].(string))
|
|
policy := user["Policy"].(map[string]interface{})
|
|
app.storage.policy = policy
|
|
app.storage.storePolicy()
|
|
app.debug.Println("User policy template stored")
|
|
if req.Homescreen {
|
|
configuration := user["Configuration"].(map[string]interface{})
|
|
var displayprefs map[string]interface{}
|
|
displayprefs, status, err = app.jf.getDisplayPreferences(userID)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get DisplayPrefs: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get displayprefs", gc)
|
|
return
|
|
}
|
|
app.storage.configuration = configuration
|
|
app.storage.displayprefs = displayprefs
|
|
app.storage.storeConfiguration()
|
|
app.debug.Println("Configuration template stored")
|
|
app.storage.storeDisplayprefs()
|
|
app.debug.Println("DisplayPrefs template stored")
|
|
}
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
func (app *appContext) ApplySettings(gc *gin.Context) {
|
|
var req defaultsReq
|
|
gc.BindJSON(&req)
|
|
applyingFrom := "template"
|
|
var policy, configuration, displayprefs map[string]interface{}
|
|
if req.From == "template" {
|
|
if len(app.storage.policy) == 0 {
|
|
respond(500, "No policy template available", gc)
|
|
return
|
|
}
|
|
app.storage.loadPolicy()
|
|
policy = app.storage.policy
|
|
if req.Homescreen {
|
|
if len(app.storage.configuration) == 0 || len(app.storage.displayprefs) == 0 {
|
|
respond(500, "No homescreen template available", gc)
|
|
return
|
|
}
|
|
configuration = app.storage.configuration
|
|
displayprefs = app.storage.displayprefs
|
|
}
|
|
} else if req.From == "user" {
|
|
applyingFrom = "user"
|
|
user, status, err := app.jf.userById(req.ID, false)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get user from Jellyfin: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get user", gc)
|
|
return
|
|
}
|
|
applyingFrom = "\"" + user["Name"].(string) + "\""
|
|
policy = user["Policy"].(map[string]interface{})
|
|
if req.Homescreen {
|
|
displayprefs, status, err = app.jf.getDisplayPreferences(req.ID)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to get DisplayPrefs: Code %d", status)
|
|
app.debug.Printf("Error: %s", err)
|
|
respond(500, "Couldn't get displayprefs", gc)
|
|
return
|
|
}
|
|
configuration = user["Configuration"].(map[string]interface{})
|
|
}
|
|
}
|
|
app.info.Printf("Applying settings to %d user(s) from %s", len(req.ApplyTo), applyingFrom)
|
|
errors := map[string]map[string]string{
|
|
"policy": map[string]string{},
|
|
"homescreen": map[string]string{},
|
|
}
|
|
for _, id := range req.ApplyTo {
|
|
status, err := app.jf.setPolicy(id, policy)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
errors["policy"][id] = fmt.Sprintf("%d: %s", status, err)
|
|
}
|
|
if req.Homescreen {
|
|
status, err = app.jf.setConfiguration(id, configuration)
|
|
errorString := ""
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
errorString += fmt.Sprintf("Configuration %d: %s ", status, err)
|
|
} else {
|
|
status, err = app.jf.setDisplayPreferences(id, displayprefs)
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
errorString += fmt.Sprintf("Displayprefs %d: %s ", status, err)
|
|
}
|
|
}
|
|
if errorString != "" {
|
|
errors["homescreen"][id] = errorString
|
|
}
|
|
}
|
|
}
|
|
code := 200
|
|
if len(errors["policy"]) == len(req.ApplyTo) || len(errors["homescreen"]) == len(req.ApplyTo) {
|
|
code = 500
|
|
}
|
|
gc.JSON(code, errors)
|
|
}
|
|
|
|
func (app *appContext) GetConfig(gc *gin.Context) {
|
|
app.info.Println("Config requested")
|
|
resp := map[string]interface{}{}
|
|
for section, settings := range app.configBase {
|
|
if section == "order" {
|
|
resp[section] = settings.([]interface{})
|
|
} else {
|
|
resp[section] = make(map[string]interface{})
|
|
for key, values := range settings.(map[string]interface{}) {
|
|
if key == "order" {
|
|
resp[section].(map[string]interface{})[key] = values.([]interface{})
|
|
} else {
|
|
resp[section].(map[string]interface{})[key] = values.(map[string]interface{})
|
|
if key != "meta" {
|
|
dataType := resp[section].(map[string]interface{})[key].(map[string]interface{})["type"].(string)
|
|
configKey := app.config.Section(section).Key(key)
|
|
if dataType == "number" {
|
|
if val, err := configKey.Int(); err == nil {
|
|
resp[section].(map[string]interface{})[key].(map[string]interface{})["value"] = val
|
|
}
|
|
} else if dataType == "bool" {
|
|
resp[section].(map[string]interface{})[key].(map[string]interface{})["value"] = configKey.MustBool(false)
|
|
} else {
|
|
resp[section].(map[string]interface{})[key].(map[string]interface{})["value"] = configKey.String()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
gc.JSON(200, resp)
|
|
}
|
|
|
|
func (app *appContext) ModifyConfig(gc *gin.Context) {
|
|
app.info.Println("Config modification requested")
|
|
var req map[string]interface{}
|
|
gc.BindJSON(&req)
|
|
tempConfig, _ := ini.Load(app.config_path)
|
|
for section, settings := range req {
|
|
_, err := tempConfig.GetSection(section)
|
|
if section != "restart-program" && err == nil {
|
|
for setting, value := range settings.(map[string]interface{}) {
|
|
tempConfig.Section(section).Key(setting).SetValue(value.(string))
|
|
}
|
|
}
|
|
}
|
|
tempConfig.SaveTo(app.config_path)
|
|
app.debug.Println("Config saved")
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
if req["restart-program"].(bool) {
|
|
app.info.Println("Restarting...")
|
|
err := app.Restart()
|
|
if err != nil {
|
|
app.err.Printf("Couldn't restart, try restarting manually. (%s)", err)
|
|
}
|
|
}
|
|
app.loadConfig()
|
|
// Reinitialize password validator on config change, as opposed to every applicable request like in python.
|
|
if _, ok := req["password_validation"]; ok {
|
|
app.debug.Println("Reinitializing validator")
|
|
validatorConf := ValidatorConf{
|
|
"characters": app.config.Section("password_validation").Key("min_length").MustInt(0),
|
|
"uppercase characters": app.config.Section("password_validation").Key("upper").MustInt(0),
|
|
"lowercase characters": app.config.Section("password_validation").Key("lower").MustInt(0),
|
|
"numbers": app.config.Section("password_validation").Key("number").MustInt(0),
|
|
"special characters": app.config.Section("password_validation").Key("special").MustInt(0),
|
|
}
|
|
if !app.config.Section("password_validation").Key("enabled").MustBool(false) {
|
|
for key := range validatorConf {
|
|
validatorConf[key] = 0
|
|
}
|
|
}
|
|
app.validator.init(validatorConf)
|
|
}
|
|
}
|
|
|
|
func (app *appContext) Logout(gc *gin.Context) {
|
|
cookie, err := gc.Cookie("refresh")
|
|
if err != nil {
|
|
app.debug.Printf("Couldn't get cookies: %s", err)
|
|
respond(500, "Couldn't fetch cookies", gc)
|
|
return
|
|
}
|
|
app.invalidTokens = append(app.invalidTokens, cookie)
|
|
gc.SetCookie("refresh", "invalid", -1, "/", gc.Request.URL.Hostname(), true, true)
|
|
gc.JSON(200, map[string]bool{"success": true})
|
|
}
|
|
|
|
// func Restart() error {
|
|
// defer func() {
|
|
// if r := recover(); r != nil {
|
|
// os.Exit(0)
|
|
// }
|
|
// }()
|
|
// cwd, err := os.Getwd()
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// args := os.Args
|
|
// // for _, key := range args {
|
|
// // fmt.Println(key)
|
|
// // }
|
|
// cmd := exec.Command(args[0], args[1:]...)
|
|
// cmd.Stdout = os.Stdout
|
|
// cmd.Stderr = os.Stderr
|
|
// cmd.Dir = cwd
|
|
// err = cmd.Start()
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// // cmd.Process.Release()
|
|
// panic(fmt.Errorf("restarting"))
|
|
// }
|
|
|
|
// func (app *appContext) Restart() error {
|
|
// defer func() {
|
|
// if r := recover(); r != nil {
|
|
// signal.Notify(app.quit, os.Interrupt)
|
|
// <-app.quit
|
|
// }
|
|
// }()
|
|
// args := os.Args
|
|
// // After a single restart, args[0] gets messed up and isnt the real executable.
|
|
// // JFA_DEEP tells the new process its a child, and JFA_EXEC is the real executable
|
|
// if os.Getenv("JFA_DEEP") == "" {
|
|
// os.Setenv("JFA_DEEP", "1")
|
|
// os.Setenv("JFA_EXEC", args[0])
|
|
// }
|
|
// env := os.Environ()
|
|
// err := syscall.Exec(os.Getenv("JFA_EXEC"), []string{""}, env)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// panic(fmt.Errorf("restarting"))
|
|
// }
|
|
|
|
// no need to syscall.exec anymore!
|
|
func (app *appContext) Restart() error {
|
|
RESTART <- true
|
|
return nil
|
|
}
|