50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"runtime"
|
|
|
|
"github.com/mr-karan/doggo/pkg/resolvers"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
// loadResolver checks
|
|
func (hub *Hub) loadResolver(c *cli.Context) error {
|
|
// check if DOH flag is set.
|
|
if hub.QueryFlags.IsDOH {
|
|
rslvr, err := resolvers.NewDOHResolver(hub.QueryFlags.Nameservers.Value())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hub.Resolver = rslvr
|
|
return nil
|
|
}
|
|
// check if DOT flag is set.
|
|
|
|
// check if TCP flag is set.
|
|
|
|
// fallback to good ol UDP.
|
|
if len(hub.QueryFlags.Nameservers.Value()) == 0 {
|
|
if runtime.GOOS == "windows" {
|
|
// TODO: Add a method for reading system default nameserver in windows.
|
|
} else {
|
|
rslvr, err := resolvers.NewResolverFromResolvFile(resolvers.DefaultResolvConfPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hub.Resolver = rslvr
|
|
return nil
|
|
}
|
|
} else {
|
|
rslvr, err := resolvers.NewClassicResolver(hub.QueryFlags.Nameservers.Value(), resolvers.ClassicResolverOpts{
|
|
UseIPv4: hub.QueryFlags.UseIPv4,
|
|
UseIPv6: hub.QueryFlags.UseIPv6,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hub.Resolver = rslvr
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|