create custom resolvers from config

ipfs · Apr 12, 2021 · 45b9be5 · 45b9be5
1 parent 957809a
commit 45b9be5
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/core/node/dns.go b/core/node/dns.go
@@ -2,10 +2,17 @@ package node
 
 import (
 	config "github.com/ipfs/go-ipfs-config"
+	doh "github.com/libp2p/go-doh-resolver"
 	madns "github.com/multiformats/go-multiaddr-dns"
 )
 
 func DNSResolver(cfg *config.Config) (*madns.Resolver, error) {
-	// TODO custom resolvers from config
-	return madns.DefaultResolver, nil
+	var opts []madns.Option
+	if cfg.DNS.DefaultResolver != "" {
+		opts = append(opts, madns.WithDefaultResolver(doh.NewResolver(cfg.DNS.DefaultResolver)))
+	}
+	for domain, url := range cfg.DNS.CustomResolvers {
+		opts = append(opts, madns.WithDomainResolver(domain, doh.NewResolver(url)))
+	}
+	return madns.NewResolver(opts...)
 }