package main

import (
	"bytes"
	"crypto/tls"
	"crypto/x509/pkix"
	gmi "git.sr.ht/~adnano/go-gemini"
	"io"
	"io/ioutil"
	"log"
	"os"
	"path"
	"path/filepath"
	"strings"
	"text/template"
	"time"
)

var gt *template.Template

func generateGemfeedPage(user string) string {
	feed := generateFeedFromUser(user)
	data := struct {
		Host        string
		Title       string
		FeedEntries []FeedEntry
	}{c.Host, feed.Title, feed.Entries}
	var buff bytes.Buffer
	gt.ExecuteTemplate(&buff, "gemfeed.gmi", data)
	return buff.String()
}

func generateFolderPage(fullpath string) string {
	files, _ := ioutil.ReadDir(fullpath)
	var renderedFiles = []File{}
	for _, file := range files {
		// Very awkward
		res := fileFromPath(path.Join(fullpath, file.Name()))
		renderedFiles = append(renderedFiles, res)
	}
	var buff bytes.Buffer
	data := struct {
		Host   string
		Folder string
		Files  []File
	}{c.Host, getLocalPath(fullpath), renderedFiles}
	err := gt.ExecuteTemplate(&buff, "folder.gmi", data)
	if err != nil {
		log.Println(err)
		return ""
	}
	return buff.String()
}

func gmiIndex(w *gmi.ResponseWriter, r *gmi.Request) {
	logGemini(r) // TODO move into wrapper
	t, err := template.ParseFiles("templates/index.gmi")
	if err != nil {
		log.Fatal(err)
	}
	files, err := getIndexFiles(false)
	users, err := getActiveUserNames()
	if err != nil {
		log.Println(err)
		w.WriteHeader(40, "Internal server error")
	}
	data := struct {
		Host      string
		SiteTitle string
		Files     []*File
		Users     []string
	}{
		Host:      c.Host,
		SiteTitle: c.SiteTitle,
		Files:     files,
		Users:     users,
	}
	t.Execute(w, data)
}

func gmiPage(w *gmi.ResponseWriter, r *gmi.Request) {
	logGemini(r) // TODO move into wrapper
	var userName string
	custom := domains[r.URL.Host]
	if custom != "" {
		userName = custom
	} else {
		userName = filepath.Clean(strings.Split(r.URL.Host, ".")[0]) // clean probably unnecessary
	}
	fileName := filepath.Clean(r.URL.Path)
	if fileName == "/" {
		fileName = "index.gmi"
	} else if strings.HasPrefix(fileName, "/"+HiddenFolder) {
		w.WriteStatus(gmi.StatusNotFound)
		return
	}
	fullPath := path.Join(c.FilesDirectory, userName, fileName)
	if fileName == "/gemlog" { // temp hack
		_, err := os.Stat(path.Join(fullPath, "index.gmi"))
		if err != nil {
			w.SetMediaType("text/gemini")
			io.Copy(w, strings.NewReader(generateGemfeedPage(userName)))
			return
		}
	} else if fileName == "/gemlog/atom.xml" {
		_, err := os.Stat(fullPath)
		if err != nil {
			w.SetMediaType("application/atom+xml")
			feed := generateFeedFromUser(userName)
			atomString := feed.toAtomFeed()
			io.Copy(w, strings.NewReader(atomString))
			return
		}
	}

	gmi.ServeFile(w, gmi.Dir(path.Join(c.FilesDirectory, userName)), fileName)
}

func runGeminiServer() {
	log.Println("Starting gemini server")
	var err error
	gt, err = template.ParseGlob(path.Join(c.TemplatesDirectory, "*.gmi"))
	if err != nil {
		log.Fatal(err)
	}
	var server gmi.Server
	server.ReadTimeout = 1 * time.Minute
	server.WriteTimeout = 2 * time.Minute

	hostname := strings.SplitN(c.Host, ":", 2)[0]
	// is this necc?
	err = server.Certificates.Load(c.GeminiCertStore)
	if err != nil {
	}
	server.CreateCertificate = func(h string) (tls.Certificate, error) {
		log.Println("Generating certificate for", h)
		return gmi.CreateCertificate(gmi.CertificateOptions{
			Subject: pkix.Name{
				CommonName: h,
			},
			DNSNames: []string{h},
			Duration: time.Hour * 8760 * 100, // 100 years
		})
	}

	var mux gmi.ServeMux
	// replace with wildcard cert
	mux.HandleFunc("/", gmiIndex)

	var wildcardMux gmi.ServeMux
	wildcardMux.HandleFunc("/", gmiPage)
	server.Register(hostname, &mux)
	server.Register("*."+hostname, &wildcardMux)
	for k, _ := range domains { // TODO fix
		server.Register(k, &wildcardMux)
	}

	err = server.ListenAndServe()
	if err != nil {
		log.Fatal(err)
	}
}