package forward

import (
	"context"
	"net"
	"net/netip"
	"time"

	"github.com/coredns/coredns/plugin/dnstap/msg"
	"github.com/coredns/coredns/request"

	tap "github.com/dnstap/golang-dnstap"
	"github.com/miekg/dns"
)

// toDnstap will send the forward and received message to the dnstap plugin.
func toDnstap(ctx context.Context, f *Forward, host string, localAddr net.Addr, proto string, state request.Request, reply *dns.Msg, start time.Time) {
	ap, _ := netip.ParseAddrPort(host) // this is preparsed and can't err here
	ip := net.IP(ap.Addr().AsSlice())
	port := int(ap.Port())

	var ta net.Addr = &net.UDPAddr{
		IP:   ip,
		Port: port,
	}
	if proto == "tcp" {
		ta = &net.TCPAddr{IP: ip, Port: port}
	}

	for _, t := range f.tapPlugins {
		// Query
		q := new(tap.Message)
		msg.SetQueryTime(q, start)
		// Forwarder dnstap messages are from the perspective of the
		// downstream DNS server (current CoreDNS) to an upstream DNS
		// server.
		if localAddr != nil {
			msg.SetQueryAddress(q, localAddr)
		}
		msg.SetResponseAddress(q, ta)
		if t.IncludeRawMessage {
			buf, _ := state.Req.Pack()
			q.QueryMessage = buf
		}
		msg.SetType(q, tap.Message_FORWARDER_QUERY)
		t.TapMessageWithMetadata(ctx, q, state)

		// Response
		if reply != nil {
			r := new(tap.Message)
			if t.IncludeRawMessage {
				buf, _ := reply.Pack()
				r.ResponseMessage = buf
			}
			msg.SetQueryTime(r, start)
			if localAddr != nil {
				msg.SetQueryAddress(r, localAddr)
			}
			msg.SetResponseAddress(r, ta)
			msg.SetResponseTime(r, time.Now())
			msg.SetType(r, tap.Message_FORWARDER_RESPONSE)
			t.TapMessageWithMetadata(ctx, r, state)
		}
	}
}