package http

import (
	"io"
	"net/http"

	"git.capotej.com/capotej/communique/registry"
	"github.com/gin-gonic/gin"
)

type Router struct {
	registry *registry.Registry
}

func NewRouter(registry *registry.Registry) *Router {
	return &Router{registry: registry}
}

func render(c *gin.Context, resource map[string]interface{}, err error) {
	if resource == nil && err == nil {
		c.JSON(http.StatusNotFound, map[string]interface{}{})
	} else if err != nil {
		jsonErr := c.Error(err)
		c.JSON(http.StatusInternalServerError, jsonErr.JSON())
	} else {
		c.Writer.Header().Set("Content-Type", "application/activity+json")
		c.JSON(http.StatusOK, resource)
	}
}

func (s *Router) Start(zapWriter io.Writer) {
	router := gin.Default()
	router.SetTrustedProxies(nil)
	gin.DisableConsoleColor()
	gin.DefaultWriter = zapWriter // send gin logs to zap

	// Webfinger
	router.GET("/.well-known/webfinger", func(c *gin.Context) {
		resourceParam := c.Query("resource")
		resource, _ := s.registry.Webfinger(resourceParam)
		if resource != nil {
			c.JSON(http.StatusOK, resource)
		} else {
			c.JSON(http.StatusNotFound, map[string]interface{}{})
		}

	})

	// Actor
	router.GET("/actors/:actor", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Actor(actorParam)
		render(c, resource, err)
	})

	// Actor Followers
	router.GET("/actors/:actor/followers", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Followers(actorParam)
		render(c, resource, err)
	})

	// Actor Following
	router.GET("/actors/:actor/following", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Following(actorParam)
		render(c, resource, err)
	})

	// // Inbox
	// router.POST("/actors/:actor/inbox", func(c *gin.Context) {
	// 	actorParam := c.Param("actor")
	// 	resource := s.registry.Inbox(actorParam)
	// 	if resource != nil {
	// 		c.JSON(http.StatusOK, resource)
	// 	} else {
	// 		c.JSON(http.StatusNotFound, nil)
	// 	}

	// })

	// Actor Outbox
	router.GET("/actors/:actor/outbox", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.OutboxCollection(actorParam)
		render(c, resource, err)
	})

	// Actor Activity
	router.GET("/actors/:actor/activity/:id", func(c *gin.Context) {
		actorParam := c.Param("actor")
		idParam := c.Param("id")
		resource, err := s.registry.ActivityOrNote("activity", actorParam, idParam)
		render(c, resource, err)
	})

	// Activity Note
	router.GET("/actors/:actor/activity/:id/note", func(c *gin.Context) {
		actorParam := c.Param("actor")
		idParam := c.Param("id")
		resource, err := s.registry.ActivityOrNote("note", actorParam, idParam)
		render(c, resource, err)
	})

	router.Run()
}