package main

import (
	"bytes"
	"context"
	"io"
	"log"
	"os"
	"os/exec"

	"github.com/go-kit/log/level"
	sr "tildegit.org/tjp/sliderule"
	"tildegit.org/tjp/sliderule/gemini"
	"tildegit.org/tjp/sliderule/logging"
)

func main() {
	// Get TLS files from the environment
	certfile, keyfile := envConfig()

	// build a TLS configuration suitable for gemini
	tlsconf, err := gemini.FileTLS(certfile, keyfile)
	if err != nil {
		log.Fatal(err)
	}

	baseLog := logging.Base()

	// add request logging to the request handler
	handler := logging.LogRequests(level.Info(baseLog))(cowsayHandler)

	// run the server
	server, err := gemini.NewServer(context.Background(), "localhost", "tcp4", ":1965", handler, baseLog, tlsconf)
	if err != nil {
		log.Fatal(err)
	}
	server.Serve()
}

var cowsayHandler = sr.HandlerFunc(func(ctx context.Context, req *sr.Request) *sr.Response {
	// prompt for a query if there is none already
	if req.RawQuery == "" {
		return gemini.Input("enter a phrase")
	}

	// find the "cowsay" executable
	binpath, err := exec.LookPath("cowsay")
	if err != nil {
		return gemini.Failure(err)
	}

	// build the command and set the query to be passed to its stdin
	cmd := exec.CommandContext(ctx, binpath)
	cmd.Stdin = bytes.NewBufferString(req.UnescapedQuery())

	// set up a pipe so we can read the command's stdout
	rd, err := cmd.StdoutPipe()
	if err != nil {
		return gemini.Failure(err)
	}

	// start the command
	if err := cmd.Start(); err != nil {
		return gemini.Failure(err)
	}

	// read the complete stdout contents, clean up the process on error
	buf, err := io.ReadAll(rd)
	if err != nil {
		cmd.Process.Kill()
		cmd.Wait()
		return gemini.Failure(err)
	}

	// wait for the process to close
	cmd.Wait()

	// pass the buffer to the response wrapped in ``` toggles,
	// and include a link to start over
	out := io.MultiReader(
		bytes.NewBufferString("```\n"),
		bytes.NewBuffer(buf),
		bytes.NewBufferString("\n```\n=> . again"),
	)
	return gemini.Success("text/gemini", out)
})

func envConfig() (string, string) {
	certfile, ok := os.LookupEnv("SERVER_CERTIFICATE")
	if !ok {
		log.Fatal("missing SERVER_CERTIFICATE environment variable")
	}

	keyfile, ok := os.LookupEnv("SERVER_PRIVATEKEY")
	if !ok {
		log.Fatal("missing SERVER_PRIVATEKEY environment variable")
	}

	return certfile, keyfile
}