package dns

// A structure for handling zone data

import (
	"github.com/miekg/radix"
)

type Zone struct {
	Name         string // Name of the zone
	*radix.Radix        // Zone data
}

type ZoneData struct {
	Name       string          // Domain name for this node
	RR         map[uint16][]RR // Map of the RR type to the RR
	Signatures []*RR_RRSIG     // DNSSEC signatures
	Glue       bool            // True if the A and AAAA record are glue
}

// New ...
func New(name string) *Zone {
	z := new(Zone)
	z.Name = name
	z.Radix = radix.New()
	return z
}

func (z *Zone) Insert(r RR) {
	zd := z.Radix.Find(r.Header().Name)
	if zd == nil {
		zd := new(ZoneData)
		zd.Name = r.Header().Name
		switch t := r.Header().Rrtype; t {
		case TypeRRSIG:
			zd.Signatures = append(zd.Signatures, r.(*RR_RRSIG))
		default:
			zd.RR[t] = append(zd.RR[t], r)
		}
		z.Radix.Insert(r.Header().Name)
		return
	}
	switch t := r.Header().Rrtype; t {
	case TypeRRSIG:
		zd.(*ZoneData).Signatures = append(zd.(*ZoneData).Signatures, r.(*RR_RRSIG))
	default:
		zd.(*ZoneData).RR[t] = append(zd.(*ZoneData).RR[t], r)
	}
	// TODO(mg): Glue
	return
}

func (z *Zone) Remove(r RR) {

}