package integration_test

import (
	"encoding/json"
	"os"
	"path/filepath"
	"testing"

	"github.com/KristianJBorgwarth/dendrite.daemon/core/handlers"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestCreateNoteHandler_NoTemplate_CreatesNoteFileAndReturnsPath(t *testing.T) {
	// Arrange
	handler := handlers.NewCreateNoteHandler()

	vaultPath := Fixture.VaultStore.Config.VaultPath()
	notePath := filepath.Join(vaultPath, "my-note.md")
	params, _ := json.Marshal(map[string]any{
		"title":        "My Note",
		"templateName": "",
		"directory":    "",
	})

	// Act
	result, err := handler.Handle(Fixture.TestContext, params)

	// Assert
	require.NoError(t, err)
	assert.Equal(t, notePath, result)

	var count int
	require.NoError(t, Fixture.DB.QueryRow(`SELECT COUNT(*) FROM note WHERE slug = ?`, "my-note").Scan(&count))
	assert.Equal(t, 1, count)

	_, statErr := os.Stat(notePath)
	assert.NoError(t, statErr)
}

func TestCreateNoteHandler_WithTemplate_CreatesNoteFileWithTagsAndReturnsPath(t *testing.T) {
	// Arrange
	handler := handlers.NewCreateNoteHandler()

	vaultPath := Fixture.VaultStore.Config.VaultPath()
	templateDir := Fixture.VaultStore.Config.TemplateDirectory()
	require.NoError(t, os.MkdirAll(templateDir, 0o755))

	templatePath := filepath.Join(templateDir, "template.md")
	require.NoError(t, os.WriteFile(templatePath, []byte("---\ntitle: Template\ntags: [go, testing]\n---\n"), 0o644))
	defer os.Remove(templatePath)

	subDir := "subdir"
	require.NoError(t, os.MkdirAll(filepath.Join(vaultPath, subDir), 0o755))
	defer os.RemoveAll(filepath.Join(vaultPath, subDir))
	notePath := filepath.Join(vaultPath, subDir, "templated-note.md")
	params, _ := json.Marshal(map[string]any{
		"title":        "Templated Note",
		"templateName": "template.md",
		"directory":    subDir,
	})

	// Act
	result, err := handler.Handle(Fixture.TestContext, params)

	// Assert
	require.NoError(t, err)
	assert.Equal(t, notePath, result)

	var noteCount int
	require.NoError(t, Fixture.DB.QueryRow(`SELECT COUNT(*) FROM note WHERE slug = ?`, "templated-note").Scan(&noteCount))
	assert.Equal(t, 1, noteCount)

	rows, err := Fixture.DB.Query(`SELECT name FROM tag WHERE name IN ('go', 'testing')`)
	require.NoError(t, err)
	defer rows.Close()

	var tags []string
	for rows.Next() {
		var name string
		require.NoError(t, rows.Scan(&name))
		tags = append(tags, name)
	}
	assert.Len(t, tags, 2)

	_, statErr := os.Stat(notePath)
	assert.NoError(t, statErr)
}

func TestCreateNoteHandler_DuplicateSlug_Upserts(t *testing.T) {
	// Arrange
	handler := handlers.NewCreateNoteHandler()

	vaultPath := Fixture.VaultStore.Config.VaultPath()
	subDir := "dup-dir"
	require.NoError(t, os.MkdirAll(filepath.Join(vaultPath, subDir), 0o755))
	defer os.RemoveAll(filepath.Join(vaultPath, subDir))

	params1, _ := json.Marshal(map[string]any{"title": "Dup Note", "templateName": "", "directory": subDir})
	_, err := handler.Handle(Fixture.TestContext, params1)
	require.NoError(t, err)

	subDir2 := "dup-dir-moved"
	require.NoError(t, os.MkdirAll(filepath.Join(vaultPath, subDir2), 0o755))
	defer os.RemoveAll(filepath.Join(vaultPath, subDir2))
	params2, _ := json.Marshal(map[string]any{"title": "Dup Note", "templateName": "", "directory": subDir2})

	// Act
	_, err = handler.Handle(Fixture.TestContext, params2)

	// Assert
	require.NoError(t, err)

	var count int
	require.NoError(t, Fixture.DB.QueryRow(`SELECT COUNT(*) FROM note WHERE slug = ?`, "dup-note").Scan(&count))
	assert.Equal(t, 1, count)

	expectedPath := filepath.Join(vaultPath, subDir2, "dup-note.md")
	var path string
	require.NoError(t, Fixture.DB.QueryRow(`SELECT path FROM note WHERE slug = ?`, "dup-note").Scan(&path))
	assert.Equal(t, expectedPath, path)
}

func TestCreateNoteHandler_InvalidJSON_ReturnsError(t *testing.T) {
	// Arrange
	handler := handlers.NewCreateNoteHandler()

	// Act
	_, err := handler.Handle(Fixture.TestContext, json.RawMessage(`{invalid json}`))

	// Assert
	assert.Error(t, err)
}

func TestCreateNoteHandler_NonExistentTemplateName_ReturnsError(t *testing.T) {
	// Arrange
	handler := handlers.NewCreateNoteHandler()

	params, _ := json.Marshal(map[string]any{
		"title":        "Ghost Note",
		"templateName": "non-existent-template",
		"directory":    "",
	})

	// Act
	_, err := handler.Handle(Fixture.TestContext, params)

	// Assert
	assert.Error(t, err)
}