// Copyright 2023 E99p1ant. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package db

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/tamakyi/TamaBox/internal/dbutil"
)

func TestQuestions(t *testing.T) {
	t.Parallel()

	db, cleanup := newTestDB(t)
	ctx := context.Background()

	questionsStore := NewQuestionsStore(db)

	for _, tc := range []struct {
		name string
		test func(t *testing.T, ctx context.Context, db *questions)
	}{
		{"Create", testQuestionsCreate},
		{"GetByID", testQuestionsGetByID},
		{"GetByUserID", testQuestionsGetByUserID},
		{"GetByAskUserID", testQuestionsGetByAskUserID},
		{"AnswerByID", testQuestionsAnswerByID},
		{"AnswerByIDPrivate", testQuestionsAnswerByIDPrivate},
		{"DeleteByID", testQuestionsDeleteByID},
		{"UpdateCensor", testQuestionsUpdateCensor},
		{"Count", testQuestionsCount},
		{"SetPrivate", testQuestionsSetPrivate},
		{"SetPublic", testQuestionsSetPublic},
		{"NextUnread", testQuestionsNextUnread},
	} {
		t.Run(tc.name, func(t *testing.T) {
			t.Cleanup(func() {
				if err := cleanup("questions"); err != nil {
					t.Fatal(err)
				}
			})
			tc.test(t, ctx, questionsStore.(*questions))
		})
	}
}

func testQuestionsCreate(t *testing.T, ctx context.Context, db *questions) {
	t.Run("normal", func(t *testing.T) {
		got, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Hello, world!",
			ReceiveReplyEmail: "i@github.red",
			AskerUserID:       0,
		})
		require.Nil(t, err)

		got.Token = ""
		got.CreatedAt = time.Time{}
		got.UpdatedAt = time.Time{}

		want := &Question{
			Model: dbutil.Model{
				ID: 1,
			},
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Hello, world!",
			ReceiveReplyEmail: "i@github.red",
		}
		require.Equal(t, got, want)
	})
}

func testQuestionsGetByID(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Hello, world!",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	t.Run("normal", func(t *testing.T) {
		got, err := db.GetByID(ctx, 1)
		require.Nil(t, err)

		got.Token = ""
		got.CreatedAt = time.Time{}
		got.UpdatedAt = time.Time{}

		want := &Question{
			Model: dbutil.Model{
				ID: 1,
			},
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Hello, world!",
			ReceiveReplyEmail: "i@github.red",
		}
		require.Equal(t, got, want)
	})

	t.Run("not found", func(t *testing.T) {
		_, err := db.GetByID(ctx, 404)
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

func testQuestionsGetByUserID(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 1",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            2,
		Content:           "Content - 2",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 3",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	t.Run("all", func(t *testing.T) {
		got, err := db.GetByUserID(ctx, 1, GetQuestionsByUserIDOptions{})
		require.Nil(t, err)

		for _, g := range got {
			g.Token = ""
			g.CreatedAt = time.Time{}
			g.UpdatedAt = time.Time{}
		}

		want := []*Question{
			{
				Model: dbutil.Model{
					ID: 3,
				},
				FromIP:            "114.5.1.4",
				UserID:            1,
				Content:           "Content - 3",
				ReceiveReplyEmail: "i@github.red",
			},
			{
				Model: dbutil.Model{
					ID: 1,
				},
				FromIP:            "114.5.1.4",
				UserID:            1,
				Content:           "Content - 1",
				ReceiveReplyEmail: "i@github.red",
			},
		}
		require.Equal(t, want, got)
	})

	t.Run("cursor", func(t *testing.T) {
		got, err := db.GetByUserID(ctx, 1, GetQuestionsByUserIDOptions{
			Cursor: &dbutil.Cursor{
				Value:    3,
				PageSize: 10,
			},
		})
		require.Nil(t, err)

		for _, g := range got {
			g.Token = ""
			g.CreatedAt = time.Time{}
			g.UpdatedAt = time.Time{}
		}

		want := []*Question{
			{
				Model: dbutil.Model{
					ID: 1,
				},
				FromIP:            "114.5.1.4",
				UserID:            1,
				Content:           "Content - 1",
				ReceiveReplyEmail: "i@github.red",
			},
		}
		require.Equal(t, want, got)
	})

	t.Run("not found", func(t *testing.T) {
		_, err := db.GetByID(ctx, 404)
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

func testQuestionsGetByAskUserID(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 1",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       1,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            2,
		Content:           "Content - 2",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       2,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            2,
		Content:           "Content - 2",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       2,
	})
	require.Nil(t, err)

	t.Run("all", func(t *testing.T) {
		got, err := db.GetByAskUserID(ctx, 1, GetQuestionsByAskUserIDOptions{})
		require.Nil(t, err)

		for _, g := range got {
			g.Token = ""
			g.CreatedAt = time.Time{}
			g.UpdatedAt = time.Time{}
		}

		want := []*Question{
			{
				Model: dbutil.Model{
					ID: 1,
				},
				FromIP:            "114.5.1.4",
				UserID:            1,
				Content:           "Content - 1",
				ReceiveReplyEmail: "i@github.red",
				AskerUserID:       1,
			},
		}
		require.Equal(t, want, got)
	})

	t.Run("cursor", func(t *testing.T) {
		got, err := db.GetByAskUserID(ctx, 2, GetQuestionsByAskUserIDOptions{
			Cursor: &dbutil.Cursor{
				Value:    3,
				PageSize: 10,
			},
		})
		require.Nil(t, err)

		for _, g := range got {
			g.Token = ""
			g.CreatedAt = time.Time{}
			g.UpdatedAt = time.Time{}
		}

		want := []*Question{
			{
				Model: dbutil.Model{
					ID: 2,
				},
				FromIP:            "114.5.1.4",
				UserID:            2,
				Content:           "Content - 2",
				ReceiveReplyEmail: "i@github.red",
				AskerUserID:       2,
			},
		}
		require.Equal(t, want, got)
	})

	t.Run("not found", func(t *testing.T) {
		_, err := db.GetByID(ctx, 404)
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

func testQuestionsAnswerByID(t *testing.T, ctx context.Context, db *questions) {
	t.Run("normal", func(t *testing.T) {
		_, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Content - 1",
			ReceiveReplyEmail: "i@github.red",
			AskerUserID:       1,
		})
		require.Nil(t, err)

		err = db.AnswerByID(ctx, 1, "Answer - 1")
		require.Nil(t, err)

		got, err := db.GetByID(ctx, 1)
		require.Nil(t, err)
		require.Equal(t, "Answer - 1", got.Answer)
	})

	t.Run("not found", func(t *testing.T) {
		err := db.AnswerByID(ctx, 404, "Answer - 1")
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

// testQuestionsAnswerByIDPrivate 覆盖 L2「收窄公开」的写入语义：
// 回答与 is_private 必须在同一次写入里生效，不能出现「公开 + 已作答」的中间态。
func testQuestionsAnswerByIDPrivate(t *testing.T, ctx context.Context, db *questions) {
	t.Run("normal", func(t *testing.T) {
		_, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:      "114.5.1.4",
			UserID:      1,
			Content:     "Content - 1",
			AskerUserID: 1,
		})
		require.Nil(t, err)

		err = db.AnswerByIDPrivate(ctx, 1, "Answer - 1")
		require.Nil(t, err)

		got, err := db.GetByID(ctx, 1)
		require.Nil(t, err)
		require.Equal(t, "Answer - 1", got.Answer)
		require.True(t, got.IsPrivate)
	})

	t.Run("not found", func(t *testing.T) {
		err := db.AnswerByIDPrivate(ctx, 404, "Answer - 1")
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

func testQuestionsDeleteByID(t *testing.T, ctx context.Context, db *questions) {
	t.Run("normal", func(t *testing.T) {
		_, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Content - 1",
			ReceiveReplyEmail: "i@github.red",
			AskerUserID:       1,
		})
		require.Nil(t, err)

		err = db.DeleteByID(ctx, 1)
		require.Nil(t, err)

		_, err = db.GetByID(ctx, 1)
		require.Equal(t, ErrQuestionNotExist, err)
	})

	t.Run("not found", func(t *testing.T) {
		err := db.DeleteByID(ctx, 404)
		require.Equal(t, ErrQuestionNotExist, err)
	})
}

func testQuestionsUpdateCensor(t *testing.T, ctx context.Context, db *questions) {
	t.Run("normal", func(t *testing.T) {
		_, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:            "114.5.1.4",
			UserID:            1,
			Content:           "Content - 1",
			ReceiveReplyEmail: "i@github.red",
			AskerUserID:       1,
		})
		require.Nil(t, err)

		err = db.DeleteByID(ctx, 1)
		require.Nil(t, err)

		_, err = db.GetByID(ctx, 1)
		require.Equal(t, ErrQuestionNotExist, err)
	})

	t.Run("not found", func(t *testing.T) {
		err := db.UpdateCensor(ctx, 404, UpdateQuestionCensorOptions{})
		require.NotNil(t, err)
	})
}

func testQuestionsCount(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 1",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       1,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            2,
		Content:           "Content - 2",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	_, err = db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 3",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       0,
	})
	require.Nil(t, err)

	err = db.AnswerByID(ctx, 3, "Answer - 3")
	require.Nil(t, err)

	t.Run("normal", func(t *testing.T) {
		got, err := db.Count(ctx, 1, GetQuestionsCountOptions{})
		require.Nil(t, err)

		want := int64(2)
		require.Equal(t, want, got)
	})

	t.Run("filter", func(t *testing.T) {
		got, err := db.Count(ctx, 1, GetQuestionsCountOptions{FilterAnswered: true})
		require.Nil(t, err)

		want := int64(1)
		require.Equal(t, want, got)
	})

	t.Run("not found", func(t *testing.T) {
		got, err := db.Count(ctx, 404, GetQuestionsCountOptions{})
		require.Nil(t, err)

		want := int64(0)
		require.Equal(t, want, got)
	})
}

func testQuestionsSetPrivate(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 1",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       1,
	})
	require.Nil(t, err)

	t.Run("normal", func(t *testing.T) {
		err = db.SetPrivate(ctx, 1)
		require.Nil(t, err)

		got, err := db.GetByID(ctx, 1)
		require.Nil(t, err)
		require.True(t, got.IsPrivate)
	})
}

func testQuestionsSetPublic(t *testing.T, ctx context.Context, db *questions) {
	_, err := db.Create(ctx, CreateQuestionOptions{
		FromIP:            "114.5.1.4",
		UserID:            1,
		Content:           "Content - 1",
		ReceiveReplyEmail: "i@github.red",
		AskerUserID:       1,
	})
	require.Nil(t, err)

	t.Run("normal", func(t *testing.T) {
		err = db.SetPublic(ctx, 1)
		require.Nil(t, err)

		got, err := db.GetByID(ctx, 1)
		require.Nil(t, err)
		require.False(t, got.IsPrivate)
	})
}

// NextUnread 必须满足三条：只取本人的提问、只取未读（is_read=false 且未回答）、按最早优先。
func testQuestionsNextUnread(t *testing.T, ctx context.Context, db *questions) {
	create := func(userID uint, content string) *Question {
		q, err := db.Create(ctx, CreateQuestionOptions{
			FromIP:  "114.5.1.4",
			UserID:  userID,
			Content: content,
		})
		require.Nil(t, err)
		return q
	}

	q1 := create(1, "Content - 1")
	q2 := create(1, "Content - 2")
	q3 := create(1, "Content - 3")
	other := create(2, "Content - other")

	t.Run("oldest first", func(t *testing.T) {
		got, err := db.NextUnread(ctx, 1, 0)
		require.Nil(t, err)
		require.Equal(t, q1.ID, got.ID)
	})

	t.Run("exclude current", func(t *testing.T) {
		got, err := db.NextUnread(ctx, 1, q1.ID)
		require.Nil(t, err)
		require.Equal(t, q2.ID, got.ID)
	})

	t.Run("only own questions", func(t *testing.T) {
		got, err := db.NextUnread(ctx, 2, 0)
		require.Nil(t, err)
		require.Equal(t, other.ID, got.ID)
	})

	t.Run("skip read and answered", func(t *testing.T) {
		require.Nil(t, db.SetRead(ctx, q1.ID))
		require.Nil(t, db.SetRead(ctx, q2.ID))
		require.Nil(t, db.AnswerByID(ctx, q3.ID, "Answer - 3"))

		got, err := db.NextUnread(ctx, 1, 0)
		require.Nil(t, err)
		require.Nil(t, got, "已读与已回答都不算未读")
	})

	t.Run("empty", func(t *testing.T) {
		got, err := db.NextUnread(ctx, 404, 0)
		require.Nil(t, err)
		require.Nil(t, got)
	})
}