package vault

import (
	"encoding/json"
	"errors"
	"os"
	"path/filepath"
	"sync"
	"testing"
	"time"
)

func trashAt(t *testing.T, v *Vault, path string, when time.Time) TrashItem {
	t.Helper()
	if _, err := v.Write(path, []byte("content"), Precondition{}); err != nil {
		t.Fatal(err)
	}
	item, err := v.Delete(path)
	if err != nil {
		t.Fatal(err)
	}
	meta, err := v.readTrashMetadata(item.ID)
	if err != nil {
		t.Fatal(err)
	}
	meta.DeletedAt = when.UnixMilli()
	data, err := json.Marshal(meta)
	if err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(v.trash, item.ID, "meta.json"), data, 0600); err != nil {
		t.Fatal(err)
	}
	return meta.TrashItem
}

func TestTrashRetentionAndInvalidMetadata(t *testing.T) {
	v := newVault(t)
	now := time.Now()
	old := trashAt(t, v, "old.md", now.Add(-31*24*time.Hour))
	boundary := trashAt(t, v, "boundary.md", now.Add(-30*24*time.Hour))
	fresh := trashAt(t, v, "fresh.md", now.Add(-29*24*time.Hour))
	future := trashAt(t, v, "future.md", now.Add(time.Hour))
	broken := trashAt(t, v, "broken.md", now.Add(-40*24*time.Hour))
	if err := os.WriteFile(filepath.Join(v.trash, broken.ID, "meta.json"), []byte(`{"deletedAt":0}`), 0600); err != nil {
		t.Fatal(err)
	}
	result, err := v.PurgeTrash(30, now)
	if err != nil || len(result.Deleted) != 2 || len(result.Failed) != 1 || result.Failed[0].ID != broken.ID {
		t.Fatalf("结果异常: %+v, %v", result, err)
	}
	for _, item := range []TrashItem{old, boundary} {
		if _, err := os.Stat(filepath.Join(v.trash, item.ID)); !errors.Is(err, os.ErrNotExist) {
			t.Fatalf("过期条目未删除: %v", err)
		}
	}
	for _, item := range []TrashItem{fresh, future, broken} {
		if _, err := os.Stat(filepath.Join(v.trash, item.ID)); err != nil {
			t.Fatalf("不应删除未过期/无效条目: %v", err)
		}
	}
	// 极大天数不应溢出成过期期限。
	result, err = v.PurgeTrash(1<<53-1, now)
	if err != nil || len(result.Deleted) != 0 || len(result.Failed) != 1 {
		t.Fatalf("极大期限: %+v, %v", result, err)
	}
	result, err = v.PurgeTrash(0, now)
	if err != nil || len(result.Deleted) != 2 || len(result.Failed) != 1 {
		t.Fatalf("清空全部应独立报告损坏条目: %+v, %v", result, err)
	}
}

func TestTrashRestoreAndPurgeSerialize(t *testing.T) {
	for range 20 {
		v := newVault(t)
		item := trashAt(t, v, "note.md", time.Now().Add(-40*24*time.Hour))
		var restoreErr, purgeErr error
		var result TrashPurgeResult
		var wg sync.WaitGroup
		wg.Add(2)
		go func() { defer wg.Done(); _, restoreErr = v.Restore(item.ID) }()
		go func() { defer wg.Done(); result, purgeErr = v.PurgeTrash(30, time.Now()) }()
		wg.Wait()
		if purgeErr != nil || len(result.Failed) != 0 {
			t.Fatalf("并发清理失败: %+v %v", result, purgeErr)
		}
		if restoreErr == nil {
			if len(result.Deleted) != 0 {
				t.Fatal("已恢复的内容不可再次清理")
			}
			if body, err := os.ReadFile(filepath.Join(v.Root(), item.Path)); err != nil || string(body) != "content" {
				t.Fatalf("恢复内容丢失: %v", err)
			}
		} else if !errors.Is(restoreErr, ErrNotFound) || len(result.Deleted) != 1 {
			t.Fatalf("清理先完成时应明确不存在: %v %+v", restoreErr, result)
		}
	}
}