mirror of https://github.com/Xhofe/alist
92 lines
1.7 KiB
Go
92 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Xhofe/alist/drivers"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
func GetAccounts(c *gin.Context) {
|
|
accounts, err := model.GetAccounts()
|
|
if err != nil {
|
|
ErrorResp(c, err, 500)
|
|
return
|
|
}
|
|
SuccessResp(c, accounts)
|
|
}
|
|
|
|
func CreateAccount(c *gin.Context) {
|
|
var req model.Account
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
ErrorResp(c, err, 400)
|
|
return
|
|
}
|
|
driver, ok := drivers.GetDriver(req.Type)
|
|
if !ok {
|
|
ErrorResp(c, fmt.Errorf("no [%s] driver", req.Type), 400)
|
|
return
|
|
}
|
|
now := time.Now()
|
|
req.UpdatedAt = &now
|
|
if err := model.CreateAccount(&req); err != nil {
|
|
ErrorResp(c, err, 500)
|
|
} else {
|
|
log.Debugf("new account: %+v", req)
|
|
err = driver.Save(&req, nil)
|
|
if err != nil {
|
|
ErrorResp(c, err, 500)
|
|
return
|
|
}
|
|
SuccessResp(c)
|
|
}
|
|
}
|
|
|
|
func SaveAccount(c *gin.Context) {
|
|
var req model.Account
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
ErrorResp(c, err, 400)
|
|
return
|
|
}
|
|
driver, ok := drivers.GetDriver(req.Type)
|
|
if !ok {
|
|
ErrorResp(c, fmt.Errorf("no [%s] driver", req.Type), 400)
|
|
return
|
|
}
|
|
old, ok := model.GetAccount(req.Name)
|
|
now := time.Now()
|
|
req.UpdatedAt = &now
|
|
if err := model.SaveAccount(&req); err != nil {
|
|
ErrorResp(c, err, 500)
|
|
} else {
|
|
log.Debugf("save account: %+v", req)
|
|
if ok {
|
|
err = driver.Save(&req, &old)
|
|
} else {
|
|
err = driver.Save(&req, nil)
|
|
}
|
|
if err != nil {
|
|
ErrorResp(c, err, 500)
|
|
return
|
|
}
|
|
SuccessResp(c)
|
|
}
|
|
}
|
|
|
|
func DeleteAccount(c *gin.Context) {
|
|
idStr := c.Query("id")
|
|
id, err := strconv.Atoi(idStr)
|
|
if err != nil {
|
|
ErrorResp(c, err, 400)
|
|
return
|
|
}
|
|
if err := model.DeleteAccount(uint(id)); err != nil {
|
|
ErrorResp(c, err, 500)
|
|
return
|
|
}
|
|
SuccessResp(c)
|
|
}
|