package operations_test

import (
	"context"
	"github.com/alist-org/alist/v3/internal/db"
	"testing"

	"github.com/alist-org/alist/v3/internal/model"
	"github.com/alist-org/alist/v3/internal/operations"
	"github.com/alist-org/alist/v3/pkg/utils"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func init() {
	dB, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
	if err != nil {
		panic("failed to connect database")
	}
	db.Init(dB)
}

func TestCreateAccount(t *testing.T) {
	var accounts = []struct {
		account model.Account
		iserr   bool
	}{
		{account: model.Account{Driver: "Local", VirtualPath: "/local", Addition: `{"root_folder":"."}`}, iserr: false},
		{account: model.Account{Driver: "Local", VirtualPath: "/local", Addition: `{"root_folder":"."}`}, iserr: true},
		{account: model.Account{Driver: "None", VirtualPath: "/none", Addition: `{"root_folder":"."}`}, iserr: true},
	}
	for _, account := range accounts {
		err := operations.CreateAccount(context.Background(), account.account)
		if err != nil {
			if !account.iserr {
				t.Errorf("failed to create account: %+v", err)
			} else {
				t.Logf("expect failed to create account: %+v", err)
			}
		}
	}
}

func TestGetAccountVirtualFilesByPath(t *testing.T) {
	setupAccounts(t)
	virtualFiles := operations.GetAccountVirtualFilesByPath("/a")
	var names []string
	for _, virtualFile := range virtualFiles {
		names = append(names, virtualFile.GetName())
	}
	var expectedNames = []string{"b", "c", "d"}
	if utils.SliceEqual(names, expectedNames) {
		t.Logf("passed")
	} else {
		t.Errorf("expected: %+v, got: %+v", expectedNames, names)
	}
}

func TestGetBalancedAccount(t *testing.T) {
	setupAccounts(t)
	account := operations.GetBalancedAccount("/a/d/e")
	if account.GetAccount().VirtualPath != "/a/d/e" {
		t.Errorf("expected: /a/d/e, got: %+v", account.GetAccount().VirtualPath)
	}
	account = operations.GetBalancedAccount("/a/d/e")
	if account.GetAccount().VirtualPath != "/a/d/e.balance" {
		t.Errorf("expected: /a/d/e.balance, got: %+v", account.GetAccount().VirtualPath)
	}
}

func setupAccounts(t *testing.T) {
	var accounts = []model.Account{
		{Driver: "Local", VirtualPath: "/a/b", Index: 0, Addition: `{"root_folder":"."}`},
		{Driver: "Local", VirtualPath: "/a/c", Index: 1, Addition: `{"root_folder":"."}`},
		{Driver: "Local", VirtualPath: "/a/d", Index: 2, Addition: `{"root_folder":"."}`},
		{Driver: "Local", VirtualPath: "/a/d/e", Index: 3, Addition: `{"root_folder":"."}`},
		{Driver: "Local", VirtualPath: "/a/d/e.balance", Index: 4, Addition: `{"root_folder":"."}`},
	}
	for _, account := range accounts {
		err := operations.CreateAccount(context.Background(), account)
		if err != nil {
			t.Fatalf("failed to create account: %+v", err)
		}
	}
}