package server

import (
	"fmt"
	"github.com/Xhofe/alist/drivers"
	"github.com/Xhofe/alist/model"
	"github.com/gofiber/fiber/v2"
	"time"
)

func GetAccounts(ctx *fiber.Ctx) error {
	accounts, err := model.GetAccounts()
	if err != nil {
		return ErrorResp(ctx, err, 500)
	}
	return SuccessResp(ctx, accounts)
}

func SaveAccount(ctx *fiber.Ctx) error {
	var req model.Account
	if err := ctx.BodyParser(&req); err != nil {
		return ErrorResp(ctx, err, 400)
	}
	if err := validate.Struct(req); err != nil {
		return ErrorResp(ctx, err, 400)
	}
	driver, ok := drivers.GetDriver(req.Type)
	if !ok {
		return ErrorResp(ctx, fmt.Errorf("no [%s] driver", req.Type), 400)
	}
	old, ok := model.GetAccount(req.Name)
	now := time.Now()
	req.UpdatedAt = &now
	if err := model.SaveAccount(req); err != nil {
		return ErrorResp(ctx, err, 500)
	} else {
		if ok {
			err = driver.Save(&req, &old)
		} else {
			err = driver.Save(&req, nil)
		}
		if err != nil {
			return ErrorResp(ctx, err, 500)
		}
		return SuccessResp(ctx)
	}
}

func DeleteAccount(ctx *fiber.Ctx) error {
	name := ctx.Query("name")
	if err := model.DeleteAccount(name); err != nil {
		return ErrorResp(ctx, err, 500)
	}
	return SuccessResp(ctx)
}