/*
 * SPDX-License-Identifier: AGPL-3.0-only
 * Copyright (c) 2023, v2rayA Organization <team@v2raya.org>
 */

package dns

import (
	"fmt"
	"github.com/sirupsen/logrus"
	"github.com/v2rayA/dae/component/routing"
	"github.com/v2rayA/dae/pkg/config_parser"
	"golang.org/x/net/dns/dnsmessage"
	"strings"
)

var typeNames = map[string]dnsmessage.Type{
	"A":     dnsmessage.TypeA,
	"NS":    dnsmessage.TypeNS,
	"CNAME": dnsmessage.TypeCNAME,
	"SOA":   dnsmessage.TypeSOA,
	"PTR":   dnsmessage.TypePTR,
	"MX":    dnsmessage.TypeMX,
	"TXT":   dnsmessage.TypeTXT,
	"AAAA":  dnsmessage.TypeAAAA,
	"SRV":   dnsmessage.TypeSRV,
	"OPT":   dnsmessage.TypeOPT,
	"WKS":   dnsmessage.TypeWKS,
	"HINFO": dnsmessage.TypeHINFO,
	"MINFO": dnsmessage.TypeMINFO,
	"AXFR":  dnsmessage.TypeAXFR,
	"ALL":   dnsmessage.TypeALL,
}

func TypeParserFactory(callback func(f *config_parser.Function, types []dnsmessage.Type, overrideOutbound *routing.Outbound) (err error)) routing.FunctionParser {
	return func(log *logrus.Logger, f *config_parser.Function, key string, paramValueGroup []string, overrideOutbound *routing.Outbound) (err error) {
		var types []dnsmessage.Type
		for _, v := range paramValueGroup {
			t, ok := typeNames[strings.ToUpper(v)]
			if !ok {
				return fmt.Errorf("unknown DNS request type: %v", v)
			}
			types = append(types, t)
		}
		return callback(f, types, overrideOutbound)
	}
}