package server

import (
	"errors"
	"log"
	"net"
)

type RouteHandler func(req *Request) *Response

type Router struct {
	Routes map[string]RouteHandler
}

func NewRouter() *Router {
	return &Router{
		Routes: make(map[string]RouteHandler),
	}
}

func (r *Router) AddRoute(pathname string, handler RouteHandler) {
	if _, ok := r.Routes[pathname]; ok {
		log.Printf("WARN: Rewriting existing route '%s'\n", pathname)
	}
	r.Routes[pathname] = handler
}

func (r *Router) Handle(conn net.Conn) error {
	defer conn.Close()

	log.Printf("Got new connection from %s\n", conn.RemoteAddr().String())

	dupl := NewPacketDuplex(conn)

	// TODO: Ability to keep connection alive
	packet, err := dupl.ReadPacket()
	if err != nil {
		return err
	}

	if route, ok := packet.Headers["route"]; ok {
		log.Printf("[ROUTER] New connection to '%s'", route)
		if handler, ok := r.Routes[route]; ok {
			req := NewRequest(conn, packet)
			res := handler(req)
			_, err := dupl.WritePacket(res.ToPacket())
			return err
		}
	}
	return errors.New("Not Found")
}