package page

import (
	"bytes"
	"net/http"
	"path/filepath"
	"strings"
)

func getURLPath(r *http.Request) string {
	u := r.URL.Path
	if u == "/" {
		u = "/index"
	}

	return u
}

// Render5xx is automatically called if any render fails,
// additionally you can call it for your own uses. It will
// try to use the 5xx template but will fall back to a plain
// page if that fails.
func Render5xx(w http.ResponseWriter, r *http.Request,
	vars map[string]interface{}, statusCode int) {
	u := getURLPath(r)

	Logger.Printf("%s %s %d %s",
		r.RemoteAddr,
		r.Method,
		statusCode,
		u)

	p := NewPage(TemplateDirectory + "/5xx")

	buf := &bytes.Buffer{}

	err := p.Render(buf)
	if err != nil {
		Logger.Printf("%s %s path: %s while trying 5xx: %s",
			r.RemoteAddr,
			r.Method,
			u,
			err)
		http.Error(w, "Internal server error", statusCode)

		return
	}

	w.WriteHeader(statusCode)

	_, _ = w.Write(buf.Bytes())

	Logger.Printf("%s %s %d %s", r.RemoteAddr, r.Method, statusCode, u)
}

// Render4xx is mostly used to render a 404 page, pulls from 404 template.
// automatically called by Render if a page is missing.
func Render4xx(w http.ResponseWriter, r *http.Request,
	vars map[string]interface{}, statusCode int) {
	u := getURLPath(r)

	Logger.Printf("%s %s %d %s",
		r.RemoteAddr,
		r.Method,
		statusCode,
		u)

	p := NewPage(TemplateDirectory + "/4xx")

	buf := &bytes.Buffer{}

	err := p.Render(buf)
	if err != nil {
		Logger.Printf("%s %s path: %s while trying 404: %s",
			r.RemoteAddr,
			r.Method,
			u,
			err)
		Render5xx(w, r, vars, http.StatusInternalServerError)

		return
	}

	w.WriteHeader(statusCode)

	_, err = w.Write(buf.Bytes())
	if err != nil {
		Render5xx(w, r, nil, http.StatusInternalServerError)

		return
	}

	Logger.Printf("%s %s %d %s", r.RemoteAddr, r.Method, statusCode, u)
}

// Render is a lower level option, allowing you to specify local
// variables and the status code in which to return.
func Render(w http.ResponseWriter, r *http.Request,
	path string, vars map[string]interface{}, statusCode int) {
	u := getURLPath(r)

	u = filepath.Join(".", u)

	// Sepcifically use the specified path for the page
	p := NewPage(path)

	if vars != nil {
		p.Vars = vars
	}

	buf := &bytes.Buffer{}

	err := p.Render(buf)
	if err != nil {
		if strings.HasSuffix(err.Error(), "no such file or directory") {
			Render4xx(w, r, vars, http.StatusNotFound)

			return
		}

		Logger.Printf("%s %s path: %s rendering encountered: %s",
			r.RemoteAddr,
			r.Method,
			u,
			err)
		Render5xx(w, r, vars, http.StatusInternalServerError)

		return
	}

	w.WriteHeader(statusCode)

	_, err = w.Write(buf.Bytes())
	if err != nil {
		Logger.Printf("%s %s %d %s: while writing buf: %s",
			r.RemoteAddr,
			r.Method,
			statusCode,
			u,
			err)
		Render5xx(w, r, nil, http.StatusInternalServerError)

		return
	}

	Logger.Printf("%s %s %d %s", r.RemoteAddr, r.Method, statusCode, u)
}

// RenderWithVars allows you to specify a specific page and whether or not
// you wish to override vars. If left nil they will not be overridden.
// Also see RenderForPath if you don't need to override them.
func RenderWithVars(w http.ResponseWriter, r *http.Request,
	path string, vars map[string]interface{}) {
	Render(w, r, path, vars, http.StatusOK)
}

// RenderForPath takes the path to a page and finish up the rendering
// Allowing you to place logic on what page is rendered by your handlers.
func RenderForPath(w http.ResponseWriter, r *http.Request, path string) {
	RenderWithVars(w, r, path, nil)
}