godns/handler/handler.go

package handler

import (
	"github.com/TimothyYe/godns"
	"github.com/TimothyYe/godns/handler/alidns"
	"github.com/TimothyYe/godns/handler/cloudflare"
	"github.com/TimothyYe/godns/handler/dnspod"
	"github.com/TimothyYe/godns/handler/dreamhost"
	"github.com/TimothyYe/godns/handler/duck"
	"github.com/TimothyYe/godns/handler/google"
	"github.com/TimothyYe/godns/handler/he"
)

// IHandler is the interface for all DNS handlers
type IHandler interface {
	SetConfiguration(*godns.Settings)
	DomainLoop(domain *godns.Domain, panicChan chan<- godns.Domain)
}

// CreateHandler creates DNS handler by different providers
func CreateHandler(provider string) IHandler {
	var handler IHandler

	switch provider {
	case godns.CLOUDFLARE:
		handler = IHandler(&cloudflare.Handler{})
	case godns.DNSPOD:
		handler = IHandler(&dnspod.Handler{})
	case godns.DREAMHOST:
		handler = IHandler(&dreamhost.Handler{})
	case godns.HE:
		handler = IHandler(&he.Handler{})
	case godns.ALIDNS:
		handler = IHandler(&alidns.Handler{})
	case godns.GOOGLE:
		handler = IHandler(&google.Handler{})
	case godns.DUCK:
		handler = IHandler(&duck.Handler{})
	}

	return handler
}