package main

import (
	"fmt"
	"log"

	"github.com/hashicorp/go-hclog"
	"github.com/hashicorp/go-plugin"
	"github.com/tus/tusd/cmd/tusd/cli/hooks"
)

// Here is a real implementation of Greeter
type MyHookHandler struct {
	logger hclog.Logger
}

func (g *MyHookHandler) Setup() error {
	log.Println("MyHookHandler.Setup is invoked")
	return nil
}

func (g *MyHookHandler) InvokeHook(req hooks.HookRequest) (res hooks.HookResponse, err error) {
	log.Println("MyHookHandler.InvokeHook is invoked")

	res.HTTPResponse.Headers = make(map[string]string)

	if req.Type == hooks.HookPreCreate {
		res.HTTPResponse.Headers["X-From-Pre-Create"] = "hello"

		if req.Event.Upload.Size > 10 {
			res.HTTPResponse.StatusCode = 413
			res.HTTPResponse.Body = `{"error":"upload size is too large"}`
			res.RejectUpload = true

			return res, nil
		}
	}

	if req.Type == hooks.HookPreFinish {
		res.HTTPResponse.Headers["X-From-Pre-Finish"] = "hello again"
		res.HTTPResponse.Body = "some information"
	}

	return res, nil
}

// handshakeConfigs are used to just do a basic handshake between
// a plugin and host. If the handshake fails, a user friendly error is shown.
// This prevents users from executing bad plugins or executing a plugin
// directory. It is a UX feature, not a security feature.
var handshakeConfig = plugin.HandshakeConfig{
	ProtocolVersion:  1,
	MagicCookieKey:   "TUSD_PLUGIN",
	MagicCookieValue: "yes",
}

func main() {
	myHandler := &MyHookHandler{}
	// pluginMap is the map of plugins we can dispense.
	var pluginMap = map[string]plugin.Plugin{
		"hookHandler": &hooks.HookHandlerPlugin{Impl: myHandler},
	}

	plugin.Serve(&plugin.ServeConfig{
		HandshakeConfig: handshakeConfig,
		Plugins:         pluginMap,
	})

	fmt.Println("DOONE")
}