package git

import (
	"bytes"
	"log/slog"
	"net/http"
	"path/filepath"

	"git.gabrielgio.me/cerrado/pkg/service"
	"git.gabrielgio.me/cerrado/templates"
	"github.com/alecthomas/chroma/v2"
	"github.com/alecthomas/chroma/v2/formatters/html"
	"github.com/alecthomas/chroma/v2/lexers"
	"github.com/alecthomas/chroma/v2/styles"
	"github.com/go-git/go-git/v5/plumbing"
	"github.com/go-git/go-git/v5/plumbing/object"
)

type (
	GitHandler struct {
		gitService gitService
	}

	gitService interface {
		ListRepositories() ([]*service.Repository, error)
		ListCommits(name string, ref string) ([]*object.Commit, error)
		GetHead(name string) (*plumbing.Reference, error)
		GetTree(name, ref, path string) (*object.Tree, error)
		GetFileContent(name, ref, path string) (string, error)
		ListTags(name string) ([]*object.Tag, error)
		ListBranches(name string) ([]*plumbing.Reference, error)
	}
)

func NewGitHandler(gitService gitService) *GitHandler {
	return &GitHandler{
		gitService: gitService,
	}
}

func (g *GitHandler) List(w http.ResponseWriter, _ *http.Request) {
	repos, err := g.gitService.ListRepositories()
	if err != nil {
		slog.Error("Error listing repo", "error", err)
		return
	}

	gitList := &templates.GitListPage{repos}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) Summary(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")
	ref, err := g.gitService.GetHead(name)
	if err != nil {
		slog.Error("Error loading head", "error", err)
		return
	}

	gitList := &templates.GitItemPage{
		Name:        name,
		Ref:         ref.Name().Short(),
		GitItemBase: &templates.GitItemSummaryPage{},
	}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) About(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")
	ref, err := g.gitService.GetHead(name)
	if err != nil {
		slog.Error("Error loading head", "error", err)
		return
	}
	gitList := &templates.GitItemPage{
		Name:        name,
		Ref:         ref.Name().Short(),
		GitItemBase: &templates.GitItemAboutPage{},
	}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) Refs(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")

	tags, err := g.gitService.ListTags(name)
	if err != nil {
		slog.Error("Error loading tags", "error", err)
		return
	}

	branches, err := g.gitService.ListBranches(name)
	if err != nil {
		slog.Error("Error loading branches", "error", err)
		return
	}

	ref, err := g.gitService.GetHead(name)
	if err != nil {
		slog.Error("Error loading head", "error", err)
		return
	}

	gitList := &templates.GitItemPage{
		Name: name,
		Ref:  ref.Name().Short(),
		GitItemBase: &templates.GitItemRefsPage{
			Tags:     tags,
			Branches: branches,
		},
	}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) Tree(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")
	ref := r.PathValue("ref")
	rest := r.PathValue("rest")

	tree, err := g.gitService.GetTree(name, ref, rest)
	if err != nil {
		slog.Error("Error loading tree", "error", err)
		return
	}

	gitList := &templates.GitItemPage{
		Name: name,
		Ref:  ref,
		GitItemBase: &templates.GitItemTreePage{
			CurrentPath: rest,
			Tree:        tree,
			Ref:         ref,
			Name:        name,
		},
	}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) Blob(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")
	ref := r.PathValue("ref")
	rest := r.PathValue("rest")

	file, err := g.gitService.GetFileContent(name, ref, rest)
	if err != nil {
		slog.Error("Error loading blob", "error", err)
		return
	}

	filename := filepath.Base(rest)
	lexer := GetLexers(filename)
	style := styles.Get("xcode")
	formatter := html.New(
		html.WithLineNumbers(true),
	)
	iterator, err := lexer.Tokenise(nil, file)
	if err != nil {
		slog.Error("Error tokenise", "error", err)
		return
	}

	var code bytes.Buffer
	err = formatter.Format(&code, style, iterator)
	if err != nil {
		slog.Error("Error format", "error", err)
		return
	}

	gitList := &templates.GitItemPage{
		Name: name,
		Ref:  ref,
		GitItemBase: &templates.GitItemBlobPage{
			File:    rest,
			Content: code.Bytes(),
		},
	}
	templates.WritePageTemplate(w, gitList)
}

func (g *GitHandler) Log(w http.ResponseWriter, r *http.Request) {
	name := r.PathValue("name")
	ref := r.PathValue("ref")

	commits, err := g.gitService.ListCommits(name, ref)
	if err != nil {
		slog.Error("Error loading commits", "error", err)
		return
	}

	gitList := &templates.GitItemPage{
		Name: name,
		Ref:  ref,
		GitItemBase: &templates.GitItemLogPage{
			Commits: commits,
		},
	}
	templates.WritePageTemplate(w, gitList)
}

func GetLexers(filename string) chroma.Lexer {
	if filename == "APKBUILD" {
		return lexers.Get("sh")
	}

	lexer := lexers.Get(filename)

	if lexer == nil {
		lexer = lexers.Get("txt")
	}
	return lexer
}