package main

import (
	"bytes"
	"io"
	"path"
	"strings"
	"text/template"
)

func newTemplate(filenames ...string) *template.Template {
	return template.Must(template.New(path.Base(filenames[0])).
		Funcs(template.FuncMap{
			"istext": istext,
		}).
		ParseFiles(filenames...))
}

var (
	tmplPage     = newTemplate("got/page.html.got")
	tmplViewTree = newTemplate("got/view_tree.got")
	tmplViewBlob = newTemplate("got/view_blob.got")
	tmplModified = newTemplate("got/modified.got")
	tmplDeleted  = newTemplate("got/deleted.got")
)

func renderPage(w io.Writer, title, head, body string) error {
	return tmplPage.Execute(w, map[string]string{
		"title": title,
		"head":  head,
		"body":  body,
	})
}

func renderViewTree(w io.Writer, upath string, tree GitTree) error {
	// Pre-processing
	files := make(GitTree)
	for name, file := range tree {
		if !strings.HasPrefix(name, upath) {
			continue
		}
		name = strings.TrimPrefix(name, upath)
		if len(name) == 0 || strings.Contains(name, "/") {
			continue
		}
		files[name] = file

	}
	// Component Render
	var buf bytes.Buffer
	err := tmplViewTree.Execute(&buf, map[string]interface{}{
		"path":  upath,
		"files": files,
	})
	if err != nil {
		return err
	}
	// Page render
	return renderPage(w, upath, "", buf.String())
}

func renderViewBlob(w io.Writer, upath string, file GitFile) error {
	// Pre-processing
	content, err := file.Cat()
	if err != nil {
		return err
	}
	ctype := getctype(upath, content)
	// Component render
	var buf bytes.Buffer
	err = tmplViewBlob.Execute(&buf, map[string]string{
		"path":    upath,
		"ctype":   ctype,
		"content": string(content),
	})
	if err != nil {
		return err
	}
	head := ""
	if ctype == "text/markdown" {
		head += "<link rel=\"stylesheet\" href=\"/static/font-awesome/css/font-awesome.min.css\">\n"
		head += "<link rel=\"stylesheet\" href=\"/static/simplemde/dist/simplemde.min.css\">\n"
		head += "<script src=\"/static/simplemde/dist/simplemde.min.js\"></script>\n"
	}
	// Page render
	return renderPage(w, upath, head, buf.String())
}

func renderModified(w io.Writer, upath string) error {
	// Component render
	var buf bytes.Buffer
	err := tmplModified.Execute(&buf, map[string]string{
		"path": upath,
	})
	if err != nil {
		return err
	}
	// Page render
	return renderPage(w, upath, "", buf.String())
}

func renderDeleted(w io.Writer, upath string) error {
	// Component render
	var buf bytes.Buffer
	err := tmplDeleted.Execute(&buf, map[string]string{
		"path": upath,
	})
	if err != nil {
		return err
	}
	// Page render
	return renderPage(w, upath, "", buf.String())
}