package dhclient

import (
	"github.com/insomniacslk/dhcp/dhcpv4"
	"github.com/insomniacslk/dhcp/dhcpv4/nclient4"
	"github.com/insomniacslk/dhcp/dhcpv6/nclient6"
	"github.com/rs/zerolog"
)

type dhcpLogger struct {
	// Printfer is used for actual output of the logger
	nclient4.Printfer

	l *zerolog.Logger
}

// Printf prints a log message as-is via predefined Printfer
func (s dhcpLogger) Printf(format string, v ...interface{}) {
	s.l.Info().Msgf(format, v...)
}

// PrintMessage prints a DHCP message in the short format via predefined Printfer
func (s dhcpLogger) PrintMessage(prefix string, message *dhcpv4.DHCPv4) {
	s.l.Info().Str("prefix", prefix).Str("message", message.String()).Msg("DHCP message")
}

func (c *Client) getDHCP4Logger(ifname string) nclient4.ClientOpt {
	logger := c.l.With().Str("interface", ifname).Logger()

	return nclient4.WithLogger(dhcpLogger{
		l: &logger,
	})
}

// TODO: nclient6 doesn't implement the WithLogger option,
// we might need to open a PR to add it

func (c *Client) getDHCP6Logger() nclient6.ClientOpt {
	return nclient6.WithSummaryLogger()
}