fix unix line endings
This commit is contained in:
parent
b53e0bb9c7
commit
c955b3ee3c
24 changed files with 1044 additions and 1044 deletions
4
.gitignore
vendored
4
.gitignore
vendored
|
|
@ -1,2 +1,2 @@
|
||||||
config.yml
|
config.yml
|
||||||
build/
|
build/
|
||||||
|
|
|
||||||
|
|
@ -1,22 +1,22 @@
|
||||||
|
|
||||||
image: golang:1.23
|
image: golang:1.23
|
||||||
|
|
||||||
stages:
|
stages:
|
||||||
- test
|
- test
|
||||||
- build
|
- build
|
||||||
|
|
||||||
unit-test:
|
unit-test:
|
||||||
stage: test
|
stage: test
|
||||||
script:
|
script:
|
||||||
- go test -v ./...
|
- go test -v ./...
|
||||||
|
|
||||||
compile:
|
compile:
|
||||||
stage: build
|
stage: build
|
||||||
script:
|
script:
|
||||||
- mkdir -p build
|
- mkdir -p build
|
||||||
- GOOS=linux GOARCH=amd64 go build -o build/dnsupdater-linux-amd64 cmd/dnsupdater/main.go
|
- GOOS=linux GOARCH=amd64 go build -o build/dnsupdater-linux-amd64 cmd/dnsupdater/main.go
|
||||||
- GOOS=linux GOARCH=mips GOMIPS=softfloat go build -o build/dnsupdater-linux-mips cmd/dnsupdater/main.go
|
- GOOS=linux GOARCH=mips GOMIPS=softfloat go build -o build/dnsupdater-linux-mips cmd/dnsupdater/main.go
|
||||||
- GOOS=linux GOARCH=mipsle GOMIPS=softfloat go build -o build/dnsupdater-linux-mipsle cmd/dnsupdater/main.go
|
- GOOS=linux GOARCH=mipsle GOMIPS=softfloat go build -o build/dnsupdater-linux-mipsle cmd/dnsupdater/main.go
|
||||||
artifacts:
|
artifacts:
|
||||||
paths:
|
paths:
|
||||||
- build
|
- build
|
||||||
|
|
|
||||||
24
Makefile
24
Makefile
|
|
@ -1,12 +1,12 @@
|
||||||
GO=go
|
GO=go
|
||||||
VERSION=$(shell git describe --always --tags --dirty --match="v*")
|
VERSION=$(shell git describe --always --tags --dirty --match="v*")
|
||||||
GOLDFLAGS=-v -s -w -X main.version="$(VERSION)"
|
GOLDFLAGS=-v -s -w -X main.version="$(VERSION)"
|
||||||
GOBUILDFLAGS=-v -p $(shell nproc) -ldflags="$(GOLDFLAGS)"
|
GOBUILDFLAGS=-v -p $(shell nproc) -ldflags="$(GOLDFLAGS)"
|
||||||
|
|
||||||
.PHONY: build test
|
.PHONY: build test
|
||||||
|
|
||||||
build :
|
build :
|
||||||
$(GO) build $(GOBUILDFLAGS) -o build/dnsupdater cmd/dnsupdater/main.go
|
$(GO) build $(GOBUILDFLAGS) -o build/dnsupdater cmd/dnsupdater/main.go
|
||||||
|
|
||||||
test :
|
test :
|
||||||
$(GO) test -v ./...
|
$(GO) test -v ./...
|
||||||
|
|
|
||||||
118
app/app.go
118
app/app.go
|
|
@ -1,59 +1,59 @@
|
||||||
package app
|
package app
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
dnsservice "dnsupdater/dns/service"
|
dnsservice "dnsupdater/dns/service"
|
||||||
"dnsupdater/ip"
|
"dnsupdater/ip"
|
||||||
"dnsupdater/ip/resolver"
|
"dnsupdater/ip/resolver"
|
||||||
)
|
)
|
||||||
|
|
||||||
// WAN_IFACE Name for the virtual WAN interface
|
// WAN_IFACE Name for the virtual WAN interface
|
||||||
const WAN_IFACE = "wan"
|
const WAN_IFACE = "wan"
|
||||||
|
|
||||||
type App struct {
|
type App struct {
|
||||||
cache *ip.Cache
|
cache *ip.Cache
|
||||||
|
|
||||||
cacheDefaultCallback ip.CacheDefaultCallback
|
cacheDefaultCallback ip.CacheDefaultCallback
|
||||||
|
|
||||||
// DNS service manager
|
// DNS service manager
|
||||||
DnsServiceMgr *dnsservice.Manager
|
DnsServiceMgr *dnsservice.Manager
|
||||||
}
|
}
|
||||||
|
|
||||||
func makeCacheCallback(service resolver.Service) ip.CacheDefaultCallback {
|
func makeCacheCallback(service resolver.Service) ip.CacheDefaultCallback {
|
||||||
return func(name string) (net.IP, error) {
|
return func(name string) (net.IP, error) {
|
||||||
if name == WAN_IFACE {
|
if name == WAN_IFACE {
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
return service.Lookup(ctx)
|
return service.Lookup(ctx)
|
||||||
}
|
}
|
||||||
return ip.GetInterfaceIP(name)
|
return ip.GetInterfaceIP(name)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewApp(config *Config) (*App, error) {
|
func NewApp(config *Config) (*App, error) {
|
||||||
dnsServiceMgr := dnsservice.NewManager()
|
dnsServiceMgr := dnsservice.NewManager()
|
||||||
err := dnsServiceMgr.RegisterFromConfig(config.Providers)
|
err := dnsServiceMgr.RegisterFromConfig(config.Providers)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
ipService := resolver.Get(config.Services.IPLookup)
|
ipService := resolver.Get(config.Services.IPLookup)
|
||||||
|
|
||||||
if ipService == nil {
|
if ipService == nil {
|
||||||
return nil, fmt.Errorf("failed to load lookup service: %s", config.Services.IPLookup)
|
return nil, fmt.Errorf("failed to load lookup service: %s", config.Services.IPLookup)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &App{
|
return &App{
|
||||||
DnsServiceMgr: dnsServiceMgr,
|
DnsServiceMgr: dnsServiceMgr,
|
||||||
cache: ip.NewCache(),
|
cache: ip.NewCache(),
|
||||||
cacheDefaultCallback: makeCacheCallback(ipService),
|
cacheDefaultCallback: makeCacheCallback(ipService),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a App) GetIP(iface_name string) (net.IP, error) {
|
func (a App) GetIP(iface_name string) (net.IP, error) {
|
||||||
return a.cache.GetWithDefault(iface_name, a.cacheDefaultCallback)
|
return a.cache.GetWithDefault(iface_name, a.cacheDefaultCallback)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,46 +1,46 @@
|
||||||
package app
|
package app
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"gopkg.in/yaml.v3"
|
"gopkg.in/yaml.v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
DomainRecords map[string]string
|
DomainRecords map[string]string
|
||||||
Domain map[string]DomainRecords
|
Domain map[string]DomainRecords
|
||||||
)
|
)
|
||||||
|
|
||||||
type DigitalOceanService struct {
|
type DigitalOceanService struct {
|
||||||
Token string `yaml:"token"`
|
Token string `yaml:"token"`
|
||||||
Domains map[string]DomainRecords `yaml:"domains"`
|
Domains map[string]DomainRecords `yaml:"domains"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Providers struct {
|
type Providers struct {
|
||||||
Token string `yaml:"token"`
|
Token string `yaml:"token"`
|
||||||
Domains map[string]DomainRecords `yaml:"domains"`
|
Domains map[string]DomainRecords `yaml:"domains"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Services struct {
|
type Services struct {
|
||||||
IPLookup string `yaml:"IPLookup"`
|
IPLookup string `yaml:"IPLookup"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Services Services `yaml:"services"`
|
Services Services `yaml:"services"`
|
||||||
Providers map[string]map[string]any
|
Providers map[string]map[string]any
|
||||||
Updates map[string]Domain
|
Updates map[string]Domain
|
||||||
}
|
}
|
||||||
|
|
||||||
func LoadConfig(filename string) (*Config, error) {
|
func LoadConfig(filename string) (*Config, error) {
|
||||||
cfg := Config{
|
cfg := Config{
|
||||||
Services: Services{
|
Services: Services{
|
||||||
IPLookup: "ipecho",
|
IPLookup: "ipecho",
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
data, err := os.ReadFile(filename)
|
data, err := os.ReadFile(filename)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
err = yaml.Unmarshal(data, &cfg)
|
err = yaml.Unmarshal(data, &cfg)
|
||||||
}
|
}
|
||||||
return &cfg, err
|
return &cfg, err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,84 +1,84 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
App "dnsupdater/app"
|
App "dnsupdater/app"
|
||||||
|
|
||||||
"github.com/rs/zerolog"
|
"github.com/rs/zerolog"
|
||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
var version string = "(unknown)"
|
var version string = "(unknown)"
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
configFile := flag.String("config", "./config.yml", "configuration file")
|
configFile := flag.String("config", "./config.yml", "configuration file")
|
||||||
versionFlag := flag.Bool("v", false, "Prints the version")
|
versionFlag := flag.Bool("v", false, "Prints the version")
|
||||||
|
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
if *versionFlag {
|
if *versionFlag {
|
||||||
fmt.Println(version)
|
fmt.Println(version)
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Logger = log.Output(zerolog.ConsoleWriter{
|
log.Logger = log.Output(zerolog.ConsoleWriter{
|
||||||
Out: os.Stderr,
|
Out: os.Stderr,
|
||||||
TimeFormat: time.RFC3339,
|
TimeFormat: time.RFC3339,
|
||||||
})
|
})
|
||||||
|
|
||||||
config, err := App.LoadConfig(*configFile)
|
config, err := App.LoadConfig(*configFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal().Err(err).Str("file", *configFile).Msg("Failed to load config")
|
log.Fatal().Err(err).Str("file", *configFile).Msg("Failed to load config")
|
||||||
}
|
}
|
||||||
|
|
||||||
app, err := App.NewApp(config)
|
app, err := App.NewApp(config)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal().Err(err).Msg("Failed to initialize application")
|
log.Fatal().Err(err).Msg("Failed to initialize application")
|
||||||
}
|
}
|
||||||
|
|
||||||
for service_name, domains := range config.Updates {
|
for service_name, domains := range config.Updates {
|
||||||
|
|
||||||
// Get DNS Service
|
// Get DNS Service
|
||||||
dnsService := app.DnsServiceMgr.Get(service_name)
|
dnsService := app.DnsServiceMgr.Get(service_name)
|
||||||
|
|
||||||
if dnsService == nil {
|
if dnsService == nil {
|
||||||
log.Warn().Str("service", service_name).Msg("Invalid DNS service")
|
log.Warn().Str("service", service_name).Msg("Invalid DNS service")
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Info().Str("service", service_name).Msg("Begin update for service")
|
log.Info().Str("service", service_name).Msg("Begin update for service")
|
||||||
|
|
||||||
updater := App.NewUpdater(dnsService)
|
updater := App.NewUpdater(dnsService)
|
||||||
|
|
||||||
for domain, records := range domains {
|
for domain, records := range domains {
|
||||||
for name, data := range records {
|
for name, data := range records {
|
||||||
|
|
||||||
logger := log.With().
|
logger := log.With().
|
||||||
Str("service", service_name).
|
Str("service", service_name).
|
||||||
Str("domain", domain).
|
Str("domain", domain).
|
||||||
Str("record", name).
|
Str("record", name).
|
||||||
Str("interface", data).
|
Str("interface", data).
|
||||||
Logger()
|
Logger()
|
||||||
|
|
||||||
ip, err := app.GetIP(data)
|
ip, err := app.GetIP(data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error().Err(err).Msg("Failed to fetch ip")
|
logger.Error().Err(err).Msg("Failed to fetch ip")
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
logger = logger.With().IPAddr("ip", ip).Logger()
|
logger = logger.With().IPAddr("ip", ip).Logger()
|
||||||
|
|
||||||
err = updater.Update(domain, name, ip)
|
err = updater.Update(domain, name, ip)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error().Err(err).Msg("Failed to update record")
|
logger.Error().Err(err).Msg("Failed to update record")
|
||||||
} else {
|
} else {
|
||||||
logger.Info().Msg("Record updated")
|
logger.Info().Msg("Record updated")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,26 +1,26 @@
|
||||||
|
|
||||||
services:
|
services:
|
||||||
IPLookup: ipecho
|
IPLookup: ipecho
|
||||||
|
|
||||||
providers:
|
providers:
|
||||||
digitalocean:
|
digitalocean:
|
||||||
token: xxxx
|
token: xxxx
|
||||||
vultr:
|
vultr:
|
||||||
token: xxxx
|
token: xxxx
|
||||||
|
|
||||||
updates:
|
updates:
|
||||||
digitalocean:
|
digitalocean:
|
||||||
domain1.com:
|
domain1.com:
|
||||||
www: wan
|
www: wan
|
||||||
box: 10.140.14.2
|
box: 10.140.14.2
|
||||||
domain2.com:
|
domain2.com:
|
||||||
www: wan
|
www: wan
|
||||||
mail: wan
|
mail: wan
|
||||||
static: 84.24.254.21
|
static: 84.24.254.21
|
||||||
vultr:
|
vultr:
|
||||||
example1.com:
|
example1.com:
|
||||||
www: wan
|
www: wan
|
||||||
example2.com:
|
example2.com:
|
||||||
www: wan
|
www: wan
|
||||||
ftp: 88.212.99.90
|
ftp: 88.212.99.90
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,109 +1,109 @@
|
||||||
package digitalocean
|
package digitalocean
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/digitalocean/godo"
|
"github.com/digitalocean/godo"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
type mock struct {
|
type mock struct {
|
||||||
t *testing.T
|
t *testing.T
|
||||||
|
|
||||||
records_by_type map[string][]godo.DomainRecord
|
records_by_type map[string][]godo.DomainRecord
|
||||||
|
|
||||||
edit_record_request *godo.DomainRecordEditRequest
|
edit_record_request *godo.DomainRecordEditRequest
|
||||||
edit_record_error error
|
edit_record_error error
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) List(context.Context, *godo.ListOptions) ([]godo.Domain, *godo.Response, error) {
|
func (m mock) List(context.Context, *godo.ListOptions) ([]godo.Domain, *godo.Response, error) {
|
||||||
m.t.Error("List called when it should not have been")
|
m.t.Error("List called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) Get(context.Context, string) (*godo.Domain, *godo.Response, error) {
|
func (m mock) Get(context.Context, string) (*godo.Domain, *godo.Response, error) {
|
||||||
m.t.Error("Get called when it should not have been")
|
m.t.Error("Get called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) Create(context.Context, *godo.DomainCreateRequest) (*godo.Domain, *godo.Response, error) {
|
func (m mock) Create(context.Context, *godo.DomainCreateRequest) (*godo.Domain, *godo.Response, error) {
|
||||||
m.t.Error("Create called when it should not have been")
|
m.t.Error("Create called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) Delete(context.Context, string) (*godo.Response, error) {
|
func (m mock) Delete(context.Context, string) (*godo.Response, error) {
|
||||||
m.t.Error("Delete called when it should not have been")
|
m.t.Error("Delete called when it should not have been")
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) Records(context.Context, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
func (m mock) Records(context.Context, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
||||||
m.t.Error("Records called when it should not have been")
|
m.t.Error("Records called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) RecordsByType(_ context.Context, name string, t string, opt *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
func (m mock) RecordsByType(_ context.Context, name string, t string, opt *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
// Only care about "A" records
|
// Only care about "A" records
|
||||||
assert.Equal(m.t, "A", t)
|
assert.Equal(m.t, "A", t)
|
||||||
|
|
||||||
r, ok := m.records_by_type[name]
|
r, ok := m.records_by_type[name]
|
||||||
if !ok {
|
if !ok {
|
||||||
err = errors.New("Record not found")
|
err = errors.New("Record not found")
|
||||||
}
|
}
|
||||||
return r, nil, err
|
return r, nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) RecordsByName(context.Context, string, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
func (m mock) RecordsByName(context.Context, string, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
||||||
m.t.Error("RecordsByName called when it should not have been")
|
m.t.Error("RecordsByName called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) RecordsByTypeAndName(context.Context, string, string, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
func (m mock) RecordsByTypeAndName(context.Context, string, string, string, *godo.ListOptions) ([]godo.DomainRecord, *godo.Response, error) {
|
||||||
m.t.Error("RecordsByTypeAndName called when it should not have been")
|
m.t.Error("RecordsByTypeAndName called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) Record(context.Context, string, int) (*godo.DomainRecord, *godo.Response, error) {
|
func (m mock) Record(context.Context, string, int) (*godo.DomainRecord, *godo.Response, error) {
|
||||||
m.t.Error("Record called when it should not have been")
|
m.t.Error("Record called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) DeleteRecord(context.Context, string, int) (*godo.Response, error) {
|
func (m mock) DeleteRecord(context.Context, string, int) (*godo.Response, error) {
|
||||||
m.t.Error("DeleteRecord called when it should not have been")
|
m.t.Error("DeleteRecord called when it should not have been")
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) EditRecord(_ context.Context, domain string, id int, req *godo.DomainRecordEditRequest) (*godo.DomainRecord, *godo.Response, error) {
|
func (m mock) EditRecord(_ context.Context, domain string, id int, req *godo.DomainRecordEditRequest) (*godo.DomainRecord, *godo.Response, error) {
|
||||||
if m.edit_record_request == nil {
|
if m.edit_record_request == nil {
|
||||||
m.t.Error("EditRecord called with empty request")
|
m.t.Error("EditRecord called with empty request")
|
||||||
}
|
}
|
||||||
|
|
||||||
if m.edit_record_error != nil {
|
if m.edit_record_error != nil {
|
||||||
return nil, nil, m.edit_record_error
|
return nil, nil, m.edit_record_error
|
||||||
}
|
}
|
||||||
|
|
||||||
assert.Equal(m.t, m.edit_record_request, req)
|
assert.Equal(m.t, m.edit_record_request, req)
|
||||||
|
|
||||||
record := godo.DomainRecord{
|
record := godo.DomainRecord{
|
||||||
ID: id,
|
ID: id,
|
||||||
Type: req.Type,
|
Type: req.Type,
|
||||||
Name: req.Name,
|
Name: req.Name,
|
||||||
Data: req.Data,
|
Data: req.Data,
|
||||||
Priority: req.Priority,
|
Priority: req.Priority,
|
||||||
Port: req.Port,
|
Port: req.Port,
|
||||||
TTL: req.TTL,
|
TTL: req.TTL,
|
||||||
Weight: req.Weight,
|
Weight: req.Weight,
|
||||||
Flags: req.Flags,
|
Flags: req.Flags,
|
||||||
Tag: req.Tag,
|
Tag: req.Tag,
|
||||||
}
|
}
|
||||||
|
|
||||||
return &record, nil, nil
|
return &record, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mock) CreateRecord(context.Context, string, *godo.DomainRecordEditRequest) (*godo.DomainRecord, *godo.Response, error) {
|
func (m mock) CreateRecord(context.Context, string, *godo.DomainRecordEditRequest) (*godo.DomainRecord, *godo.Response, error) {
|
||||||
m.t.Error("CreateRecord called when it should not have been")
|
m.t.Error("CreateRecord called when it should not have been")
|
||||||
return nil, nil, nil
|
return nil, nil, nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,67 +1,67 @@
|
||||||
package digitalocean
|
package digitalocean
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"net"
|
"net"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"dnsupdater/dns"
|
"dnsupdater/dns"
|
||||||
|
|
||||||
"github.com/digitalocean/godo"
|
"github.com/digitalocean/godo"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
api godo.DomainsService
|
api godo.DomainsService
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(token string) Service {
|
func New(token string) Service {
|
||||||
return Service{
|
return Service{
|
||||||
api: godo.NewFromToken(token).Domains,
|
api: godo.NewFromToken(token).Domains,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func Factory(args map[string]any) (any, error) {
|
func Factory(args map[string]any) (any, error) {
|
||||||
t, ok := args["token"]
|
t, ok := args["token"]
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, errors.New("did not find token")
|
return nil, errors.New("did not find token")
|
||||||
}
|
}
|
||||||
|
|
||||||
token, ok := t.(string)
|
token, ok := t.(string)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, errors.New("token must be a string")
|
return nil, errors.New("token must be a string")
|
||||||
}
|
}
|
||||||
|
|
||||||
return New(token), nil
|
return New(token), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d Service) List(domain_name string) (dns.RecordList, error) {
|
func (d Service) List(domain_name string) (dns.RecordList, error) {
|
||||||
fetchedRecords, _, err := d.api.RecordsByType(context.Background(), domain_name, "A", &godo.ListOptions{
|
fetchedRecords, _, err := d.api.RecordsByType(context.Background(), domain_name, "A", &godo.ListOptions{
|
||||||
PerPage: 50,
|
PerPage: 50,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
records := dns.RecordList{}
|
records := dns.RecordList{}
|
||||||
for _, rec := range fetchedRecords {
|
for _, rec := range fetchedRecords {
|
||||||
records.Add(dns.Record{
|
records.Add(dns.Record{
|
||||||
Id: strconv.Itoa(rec.ID),
|
Id: strconv.Itoa(rec.ID),
|
||||||
Name: rec.Name,
|
Name: rec.Name,
|
||||||
Ip: net.ParseIP(rec.Data),
|
Ip: net.ParseIP(rec.Data),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
return records, nil
|
return records, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d Service) Update(domain, recordID, ip string) error {
|
func (d Service) Update(domain, recordID, ip string) error {
|
||||||
id, err := strconv.Atoi(recordID)
|
id, err := strconv.Atoi(recordID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, _, err = d.api.EditRecord(context.Background(), domain, id, &godo.DomainRecordEditRequest{
|
_, _, err = d.api.EditRecord(context.Background(), domain, id, &godo.DomainRecordEditRequest{
|
||||||
Data: ip,
|
Data: ip,
|
||||||
})
|
})
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,49 +1,49 @@
|
||||||
package service
|
package service
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"dnsupdater/dns/service/digitalocean"
|
"dnsupdater/dns/service/digitalocean"
|
||||||
"dnsupdater/dns/service/vultr"
|
"dnsupdater/dns/service/vultr"
|
||||||
)
|
)
|
||||||
|
|
||||||
var factories = map[string]Factory{
|
var factories = map[string]Factory{
|
||||||
"digitalocean": digitalocean.Factory,
|
"digitalocean": digitalocean.Factory,
|
||||||
"vultr": vultr.Factory,
|
"vultr": vultr.Factory,
|
||||||
}
|
}
|
||||||
|
|
||||||
type Manager struct {
|
type Manager struct {
|
||||||
services map[string]Service
|
services map[string]Service
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewManager() *Manager {
|
func NewManager() *Manager {
|
||||||
return &Manager{
|
return &Manager{
|
||||||
services: make(map[string]Service),
|
services: make(map[string]Service),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Manager) Get(name string) Service {
|
func (m Manager) Get(name string) Service {
|
||||||
if service, ok := m.services[name]; ok {
|
if service, ok := m.services[name]; ok {
|
||||||
return service
|
return service
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Manager) RegisterFromConfig(providers map[string]map[string]any) error {
|
func (m Manager) RegisterFromConfig(providers map[string]map[string]any) error {
|
||||||
for name, args := range providers {
|
for name, args := range providers {
|
||||||
if factory, ok := factories[name]; ok {
|
if factory, ok := factories[name]; ok {
|
||||||
|
|
||||||
provider, err := factory(args)
|
provider, err := factory(args)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("could not create provider '%s': %v", name, err)
|
return fmt.Errorf("could not create provider '%s': %v", name, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
m.Register(name, provider.(Service))
|
m.Register(name, provider.(Service))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Manager) Register(name string, provider Service) {
|
func (m Manager) Register(name string, provider Service) {
|
||||||
m.services[name] = provider
|
m.services[name] = provider
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,10 +1,10 @@
|
||||||
package service
|
package service
|
||||||
|
|
||||||
import "dnsupdater/dns"
|
import "dnsupdater/dns"
|
||||||
|
|
||||||
type Service interface {
|
type Service interface {
|
||||||
List(domain string) (dns.RecordList, error)
|
List(domain string) (dns.RecordList, error)
|
||||||
Update(domain, recordID, ip string) error
|
Update(domain, recordID, ip string) error
|
||||||
}
|
}
|
||||||
|
|
||||||
type Factory func(map[string]any) (any, error)
|
type Factory func(map[string]any) (any, error)
|
||||||
|
|
|
||||||
54
http/get.go
54
http/get.go
|
|
@ -1,27 +1,27 @@
|
||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Perform a HTTP Get request.
|
// Perform a HTTP Get request.
|
||||||
func Get(ctx context.Context, url string, headers http.Header) (*http.Response, error) {
|
func Get(ctx context.Context, url string, headers http.Header) (*http.Response, error) {
|
||||||
req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
|
req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
req.Header = headers
|
req.Header = headers
|
||||||
|
|
||||||
resp, err := http.DefaultClient.Do(req)
|
resp, err := http.DefaultClient.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if resp.StatusCode >= 400 {
|
if resp.StatusCode >= 400 {
|
||||||
return nil, fmt.Errorf("HTTP Response: %s", resp.Status)
|
return nil, fmt.Errorf("HTTP Response: %s", resp.Status)
|
||||||
}
|
}
|
||||||
return resp, nil
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
86
ip/cache.go
86
ip/cache.go
|
|
@ -1,43 +1,43 @@
|
||||||
package ip
|
package ip
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"net"
|
"net"
|
||||||
)
|
)
|
||||||
|
|
||||||
type CacheDefaultCallback func(name string) (net.IP, error)
|
type CacheDefaultCallback func(name string) (net.IP, error)
|
||||||
|
|
||||||
type Cache struct {
|
type Cache struct {
|
||||||
items map[string]net.IP
|
items map[string]net.IP
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewCache() *Cache {
|
func NewCache() *Cache {
|
||||||
return &Cache{
|
return &Cache{
|
||||||
items: make(map[string]net.IP),
|
items: make(map[string]net.IP),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Cache) Get(name string) (net.IP, error) {
|
func (c Cache) Get(name string) (net.IP, error) {
|
||||||
// Return cached entry.
|
// Return cached entry.
|
||||||
if cached, ok := c.items[name]; ok {
|
if cached, ok := c.items[name]; ok {
|
||||||
return cached, nil
|
return cached, nil
|
||||||
}
|
}
|
||||||
return nil, errors.New("key did not exist")
|
return nil, errors.New("key did not exist")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Cache) GetWithDefault(name string, callback CacheDefaultCallback) (net.IP, error) {
|
func (c Cache) GetWithDefault(name string, callback CacheDefaultCallback) (net.IP, error) {
|
||||||
// Return cached entry.
|
// Return cached entry.
|
||||||
if cached, ok := c.items[name]; ok {
|
if cached, ok := c.items[name]; ok {
|
||||||
return cached, nil
|
return cached, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
ip, err := callback(name)
|
ip, err := callback(name)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
c.Set(name, ip)
|
c.Set(name, ip)
|
||||||
}
|
}
|
||||||
return ip, err
|
return ip, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Cache) Set(name string, ip net.IP) {
|
func (c *Cache) Set(name string, ip net.IP) {
|
||||||
c.items[name] = ip
|
c.items[name] = ip
|
||||||
}
|
}
|
||||||
|
|
|
||||||
152
ip/cache_test.go
152
ip/cache_test.go
|
|
@ -1,76 +1,76 @@
|
||||||
package ip
|
package ip
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"net"
|
"net"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
func defaultCallback(t *testing.T, expected_name string, ip net.IP, err error) CacheDefaultCallback {
|
func defaultCallback(t *testing.T, expected_name string, ip net.IP, err error) CacheDefaultCallback {
|
||||||
return func(name string) (net.IP, error) {
|
return func(name string) (net.IP, error) {
|
||||||
assert.Equal(t, expected_name, name)
|
assert.Equal(t, expected_name, name)
|
||||||
return ip, err
|
return ip, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func dontCallDefaultCallback(t *testing.T) CacheDefaultCallback {
|
func dontCallDefaultCallback(t *testing.T) CacheDefaultCallback {
|
||||||
return func(name string) (net.IP, error) {
|
return func(name string) (net.IP, error) {
|
||||||
t.Error("Should not have been called")
|
t.Error("Should not have been called")
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCache_Get(t *testing.T) {
|
func TestCache_Get(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
c *Cache
|
c *Cache
|
||||||
iface string
|
iface string
|
||||||
want net.IP
|
want net.IP
|
||||||
wantErr bool
|
wantErr bool
|
||||||
}{
|
}{
|
||||||
{"Exists in cache", &Cache{items: map[string]net.IP{"eth0": net.IPv4(10, 4, 0, 1)}}, "eth0", net.IPv4(10, 4, 0, 1), false},
|
{"Exists in cache", &Cache{items: map[string]net.IP{"eth0": net.IPv4(10, 4, 0, 1)}}, "eth0", net.IPv4(10, 4, 0, 1), false},
|
||||||
{"Did not exist in cache", &Cache{items: map[string]net.IP{}}, "eth0", nil, true},
|
{"Did not exist in cache", &Cache{items: map[string]net.IP{}}, "eth0", nil, true},
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
got, err := tt.c.Get(tt.iface)
|
got, err := tt.c.Get(tt.iface)
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("Cache.Get() error = %v, wantErr %v", err, tt.wantErr)
|
t.Errorf("Cache.Get() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
t.Errorf("Cache.Get() = %v, want %v", got, tt.want)
|
t.Errorf("Cache.Get() = %v, want %v", got, tt.want)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCache_GetWithDefault(t *testing.T) {
|
func TestCache_GetWithDefault(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
c *Cache
|
c *Cache
|
||||||
def CacheDefaultCallback
|
def CacheDefaultCallback
|
||||||
iface string
|
iface string
|
||||||
want net.IP
|
want net.IP
|
||||||
wantErr bool
|
wantErr bool
|
||||||
}{
|
}{
|
||||||
{"Exists in cache", &Cache{items: map[string]net.IP{"eth0": net.IPv4(10, 4, 0, 1)}}, dontCallDefaultCallback(t), "eth0", net.IPv4(10, 4, 0, 1), false},
|
{"Exists in cache", &Cache{items: map[string]net.IP{"eth0": net.IPv4(10, 4, 0, 1)}}, dontCallDefaultCallback(t), "eth0", net.IPv4(10, 4, 0, 1), false},
|
||||||
{"Did not exists in cache", NewCache(), defaultCallback(t, "eth1", net.IPv4(192, 172, 44, 25), nil), "eth1", net.IPv4(192, 172, 44, 25), false},
|
{"Did not exists in cache", NewCache(), defaultCallback(t, "eth1", net.IPv4(192, 172, 44, 25), nil), "eth1", net.IPv4(192, 172, 44, 25), false},
|
||||||
{"Callback returns error", NewCache(), defaultCallback(t, "eth1", nil, errors.New("some error")), "eth1", nil, true},
|
{"Callback returns error", NewCache(), defaultCallback(t, "eth1", nil, errors.New("some error")), "eth1", nil, true},
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
got, err := tt.c.GetWithDefault(tt.iface, tt.def)
|
got, err := tt.c.GetWithDefault(tt.iface, tt.def)
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("Cache.Get() error = %v, wantErr %v", err, tt.wantErr)
|
t.Errorf("Cache.Get() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
t.Errorf("Cache.Get() = %v, want %v", got, tt.want)
|
t.Errorf("Cache.Get() = %v, want %v", got, tt.want)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,46 +1,46 @@
|
||||||
package ip
|
package ip
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"net"
|
"net"
|
||||||
)
|
)
|
||||||
|
|
||||||
func GetInterfaceIP(iface_name string) (net.IP, error) {
|
func GetInterfaceIP(iface_name string) (net.IP, error) {
|
||||||
ip := net.IP{}
|
ip := net.IP{}
|
||||||
iface, err := net.InterfaceByName(iface_name)
|
iface, err := net.InterfaceByName(iface_name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ip, err
|
return ip, err
|
||||||
}
|
}
|
||||||
|
|
||||||
addrs, err := iface.Addrs()
|
addrs, err := iface.Addrs()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ip, err
|
return ip, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return GetPublicIp(addrs)
|
return GetPublicIp(addrs)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetPublicIp(list []net.Addr) (net.IP, error) {
|
func GetPublicIp(list []net.Addr) (net.IP, error) {
|
||||||
for _, addr := range list {
|
for _, addr := range list {
|
||||||
ip, err := AddrToIP(addr)
|
ip, err := AddrToIP(addr)
|
||||||
if err == nil && !ip.IsPrivate() {
|
if err == nil && !ip.IsPrivate() {
|
||||||
return ip, nil
|
return ip, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil, errors.New("no public ip found on interface")
|
return nil, errors.New("no public ip found on interface")
|
||||||
}
|
}
|
||||||
|
|
||||||
func AddrToIP(addr net.Addr) (net.IP, error) {
|
func AddrToIP(addr net.Addr) (net.IP, error) {
|
||||||
switch v := addr.(type) {
|
switch v := addr.(type) {
|
||||||
case *net.IPNet:
|
case *net.IPNet:
|
||||||
return v.IP, nil
|
return v.IP, nil
|
||||||
case *net.IPAddr:
|
case *net.IPAddr:
|
||||||
return v.IP, nil
|
return v.IP, nil
|
||||||
case *net.UDPAddr:
|
case *net.UDPAddr:
|
||||||
return v.IP, nil
|
return v.IP, nil
|
||||||
case *net.TCPAddr:
|
case *net.TCPAddr:
|
||||||
return v.IP, nil
|
return v.IP, nil
|
||||||
}
|
}
|
||||||
return nil, errors.New("could not find ip")
|
return nil, errors.New("could not find ip")
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,69 +1,69 @@
|
||||||
package ip
|
package ip
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestGetPublicIp(t *testing.T) {
|
func TestGetPublicIp(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
list []string
|
list []string
|
||||||
want string
|
want string
|
||||||
wantErr bool
|
wantErr bool
|
||||||
}{
|
}{
|
||||||
{"empty", []string{}, "", true},
|
{"empty", []string{}, "", true},
|
||||||
{"find", []string{"99.140.96.132"}, "99.140.96.132", false},
|
{"find", []string{"99.140.96.132"}, "99.140.96.132", false},
|
||||||
{"findfirst", []string{"23.114.115.197", "251.78.128.148"}, "23.114.115.197", false},
|
{"findfirst", []string{"23.114.115.197", "251.78.128.148"}, "23.114.115.197", false},
|
||||||
{"dontfindprivate", []string{"192.168.0.22", "88.12.32.44"}, "88.12.32.44", false},
|
{"dontfindprivate", []string{"192.168.0.22", "88.12.32.44"}, "88.12.32.44", false},
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
list := []net.Addr{}
|
list := []net.Addr{}
|
||||||
for _, item := range tt.list {
|
for _, item := range tt.list {
|
||||||
list = append(list, &net.IPAddr{IP: net.ParseIP(item)})
|
list = append(list, &net.IPAddr{IP: net.ParseIP(item)})
|
||||||
}
|
}
|
||||||
|
|
||||||
want := net.ParseIP(tt.want)
|
want := net.ParseIP(tt.want)
|
||||||
|
|
||||||
got, err := GetPublicIp(list)
|
got, err := GetPublicIp(list)
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("GetPublicIp() error = %v, wantErr %v", err, tt.wantErr)
|
t.Errorf("GetPublicIp() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !reflect.DeepEqual(got, want) {
|
if !reflect.DeepEqual(got, want) {
|
||||||
t.Errorf("GetPublicIp() = %v, want %v", got, want)
|
t.Errorf("GetPublicIp() = %v, want %v", got, want)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestAddrToIP(t *testing.T) {
|
func TestAddrToIP(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
addr net.Addr
|
addr net.Addr
|
||||||
want net.IP
|
want net.IP
|
||||||
wantErr bool
|
wantErr bool
|
||||||
}{
|
}{
|
||||||
{"IPNet", &net.IPNet{IP: net.IPv4(177, 171, 44, 1)}, net.IPv4(177, 171, 44, 1), false},
|
{"IPNet", &net.IPNet{IP: net.IPv4(177, 171, 44, 1)}, net.IPv4(177, 171, 44, 1), false},
|
||||||
{"IPAddr", &net.IPAddr{IP: net.IPv4(240, 23, 119, 171)}, net.IPv4(240, 23, 119, 171), false},
|
{"IPAddr", &net.IPAddr{IP: net.IPv4(240, 23, 119, 171)}, net.IPv4(240, 23, 119, 171), false},
|
||||||
{"TCPAddr", &net.TCPAddr{IP: net.IPv4(139, 231, 35, 221)}, net.IPv4(139, 231, 35, 221), false},
|
{"TCPAddr", &net.TCPAddr{IP: net.IPv4(139, 231, 35, 221)}, net.IPv4(139, 231, 35, 221), false},
|
||||||
{"UDPAddr", &net.UDPAddr{IP: net.IPv4(167, 147, 140, 119)}, net.IPv4(167, 147, 140, 119), false},
|
{"UDPAddr", &net.UDPAddr{IP: net.IPv4(167, 147, 140, 119)}, net.IPv4(167, 147, 140, 119), false},
|
||||||
{"UnixAddr", &net.UnixAddr{}, nil, true},
|
{"UnixAddr", &net.UnixAddr{}, nil, true},
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
got, err := AddrToIP(tt.addr)
|
got, err := AddrToIP(tt.addr)
|
||||||
|
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("AddrToIP() error = %v, wantErr %v", err, tt.wantErr)
|
t.Errorf("AddrToIP() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
t.Errorf("AddrToIP() = %v, want %v", got, tt.want)
|
t.Errorf("AddrToIP() = %v, want %v", got, tt.want)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,15 +1,15 @@
|
||||||
package internal
|
package internal
|
||||||
|
|
||||||
import "net"
|
import "net"
|
||||||
|
|
||||||
func ParseIP(s string) (net.IP, error) {
|
func ParseIP(s string) (net.IP, error) {
|
||||||
var err error = nil
|
var err error = nil
|
||||||
ip := net.ParseIP(s)
|
ip := net.ParseIP(s)
|
||||||
if ip == nil {
|
if ip == nil {
|
||||||
err = &net.ParseError{
|
err = &net.ParseError{
|
||||||
Type: "IP address",
|
Type: "IP address",
|
||||||
Text: s,
|
Text: s,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ip, err
|
return ip, err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,35 +1,35 @@
|
||||||
package internal
|
package internal
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestParseIP(t *testing.T) {
|
func TestParseIP(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
input string
|
input string
|
||||||
want net.IP
|
want net.IP
|
||||||
wantErr bool
|
wantErr bool
|
||||||
}{
|
}{
|
||||||
{"localhost", "127.0.0.1", net.IPv4(127, 0, 0, 1), false},
|
{"localhost", "127.0.0.1", net.IPv4(127, 0, 0, 1), false},
|
||||||
{"Private#1", "10.4.0.11", net.IPv4(10, 4, 0, 11), false},
|
{"Private#1", "10.4.0.11", net.IPv4(10, 4, 0, 11), false},
|
||||||
{"Private#2", "192.168.1.12", net.IPv4(192, 168, 1, 12), false},
|
{"Private#2", "192.168.1.12", net.IPv4(192, 168, 1, 12), false},
|
||||||
{"Public#1", "82.249.10.254", net.IPv4(82, 249, 10, 254), false},
|
{"Public#1", "82.249.10.254", net.IPv4(82, 249, 10, 254), false},
|
||||||
{"Public#2", "57.167.50.222", net.IPv4(57, 167, 50, 222), false},
|
{"Public#2", "57.167.50.222", net.IPv4(57, 167, 50, 222), false},
|
||||||
{"Invalid", "xx", nil, true},
|
{"Invalid", "xx", nil, true},
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
got, err := ParseIP(tt.input)
|
got, err := ParseIP(tt.input)
|
||||||
if (err != nil) != tt.wantErr {
|
if (err != nil) != tt.wantErr {
|
||||||
t.Errorf("ParseIP() error = %v, wantErr %v", err, tt.wantErr)
|
t.Errorf("ParseIP() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !reflect.DeepEqual(got, tt.want) {
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
t.Errorf("ParseIP() = %v, want %v", got, tt.want)
|
t.Errorf("ParseIP() = %v, want %v", got, tt.want)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,16 +1,16 @@
|
||||||
package decoder
|
package decoder
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"io"
|
"io"
|
||||||
)
|
)
|
||||||
|
|
||||||
func Jsonip(r io.Reader) (string, error) {
|
func Jsonip(r io.Reader) (string, error) {
|
||||||
var v struct {
|
var v struct {
|
||||||
Ip string `json:"ip"`
|
Ip string `json:"ip"`
|
||||||
Location string `json:"geo-ip"`
|
Location string `json:"geo-ip"`
|
||||||
Help string `json:"API Help"`
|
Help string `json:"API Help"`
|
||||||
}
|
}
|
||||||
err := json.NewDecoder(r).Decode(&v)
|
err := json.NewDecoder(r).Decode(&v)
|
||||||
return v.Ip, err
|
return v.Ip, err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,42 +1,42 @@
|
||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
httputils "dnsupdater/http"
|
httputils "dnsupdater/http"
|
||||||
"dnsupdater/ip/internal"
|
"dnsupdater/ip/internal"
|
||||||
"dnsupdater/ip/resolver/decoder"
|
"dnsupdater/ip/resolver/decoder"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Decoder func(io.Reader) (string, error)
|
type Decoder func(io.Reader) (string, error)
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
ServiceName string
|
ServiceName string
|
||||||
Url string
|
Url string
|
||||||
Headers http.Header
|
Headers http.Header
|
||||||
Decoder Decoder
|
Decoder Decoder
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s Service) Name() string {
|
func (s Service) Name() string {
|
||||||
return s.ServiceName
|
return s.ServiceName
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s Service) Lookup(ctx context.Context) (net.IP, error) {
|
func (s Service) Lookup(ctx context.Context) (net.IP, error) {
|
||||||
resp, err := httputils.Get(ctx, s.Url, s.Headers)
|
resp, err := httputils.Get(ctx, s.Url, s.Headers)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.Decoder == nil {
|
if s.Decoder == nil {
|
||||||
s.Decoder = decoder.Text
|
s.Decoder = decoder.Text
|
||||||
}
|
}
|
||||||
|
|
||||||
body, err := s.Decoder(resp.Body)
|
body, err := s.Decoder(resp.Body)
|
||||||
if err != nil && err != io.EOF {
|
if err != nil && err != io.EOF {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return internal.ParseIP(string(body))
|
return internal.ParseIP(string(body))
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,85 +1,85 @@
|
||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestService_Name(t *testing.T) {
|
func TestService_Name(t *testing.T) {
|
||||||
s := Service{ServiceName: "my_service"}
|
s := Service{ServiceName: "my_service"}
|
||||||
|
|
||||||
assert.Equal(t, "my_service", s.Name())
|
assert.Equal(t, "my_service", s.Name())
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestService_Lookup(t *testing.T) {
|
func TestService_Lookup(t *testing.T) {
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
_, err := w.Write([]byte("255.240.85.2"))
|
_, err := w.Write([]byte("255.240.85.2"))
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}))
|
}))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
s := Service{Url: server.URL}
|
s := Service{Url: server.URL}
|
||||||
|
|
||||||
ip, err := s.Lookup(context.Background())
|
ip, err := s.Lookup(context.Background())
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, net.IPv4(255, 240, 85, 2), ip)
|
assert.Equal(t, net.IPv4(255, 240, 85, 2), ip)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestService_Lookup_WithHeaders(t *testing.T) {
|
func TestService_Lookup_WithHeaders(t *testing.T) {
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
|
assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
|
||||||
|
|
||||||
_, err := w.Write([]byte("125.74.233.13"))
|
_, err := w.Write([]byte("125.74.233.13"))
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}))
|
}))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
s := Service{
|
s := Service{
|
||||||
Url: server.URL,
|
Url: server.URL,
|
||||||
Headers: http.Header{
|
Headers: http.Header{
|
||||||
"Content-Type": []string{"application/json"},
|
"Content-Type": []string{"application/json"},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
ip, err := s.Lookup(context.Background())
|
ip, err := s.Lookup(context.Background())
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, net.IPv4(125, 74, 233, 13), ip)
|
assert.Equal(t, net.IPv4(125, 74, 233, 13), ip)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestService_Lookup_HTTPError(t *testing.T) {
|
func TestService_Lookup_HTTPError(t *testing.T) {
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
w.WriteHeader(404)
|
w.WriteHeader(404)
|
||||||
}))
|
}))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
s := Service{
|
s := Service{
|
||||||
Url: server.URL,
|
Url: server.URL,
|
||||||
}
|
}
|
||||||
|
|
||||||
ip, err := s.Lookup(context.Background())
|
ip, err := s.Lookup(context.Background())
|
||||||
assert.EqualError(t, err, "HTTP Response: 404 Not Found")
|
assert.EqualError(t, err, "HTTP Response: 404 Not Found")
|
||||||
assert.Nil(t, ip)
|
assert.Nil(t, ip)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestService_Lookup_ParseError(t *testing.T) {
|
func TestService_Lookup_ParseError(t *testing.T) {
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
_, err := w.Write([]byte("random_string"))
|
_, err := w.Write([]byte("random_string"))
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}))
|
}))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
s := Service{
|
s := Service{
|
||||||
Url: server.URL,
|
Url: server.URL,
|
||||||
}
|
}
|
||||||
|
|
||||||
ip, err := s.Lookup(context.Background())
|
ip, err := s.Lookup(context.Background())
|
||||||
assert.EqualError(t, err, "invalid IP address: random_string")
|
assert.EqualError(t, err, "invalid IP address: random_string")
|
||||||
assert.Nil(t, ip)
|
assert.Nil(t, ip)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,70 +1,70 @@
|
||||||
package resolver
|
package resolver
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"dnsupdater/ip/resolver/decoder"
|
"dnsupdater/ip/resolver/decoder"
|
||||||
httpres "dnsupdater/ip/resolver/http"
|
httpres "dnsupdater/ip/resolver/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
var services []Service
|
var services []Service
|
||||||
|
|
||||||
func Provide(service Service) {
|
func Provide(service Service) {
|
||||||
services = append(services, service)
|
services = append(services, service)
|
||||||
}
|
}
|
||||||
|
|
||||||
func Get(name string) Service {
|
func Get(name string) Service {
|
||||||
for _, service := range services {
|
for _, service := range services {
|
||||||
if service.Name() == name {
|
if service.Name() == name {
|
||||||
return service
|
return service
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "jsonip",
|
ServiceName: "jsonip",
|
||||||
Url: "https://jsonip.com",
|
Url: "https://jsonip.com",
|
||||||
Decoder: decoder.Jsonip,
|
Decoder: decoder.Jsonip,
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "ifconfig.me",
|
ServiceName: "ifconfig.me",
|
||||||
Url: "https://ifconfig.me/ip",
|
Url: "https://ifconfig.me/ip",
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "ip.me",
|
ServiceName: "ip.me",
|
||||||
Url: "https://ip.me",
|
Url: "https://ip.me",
|
||||||
Headers: http.Header{
|
Headers: http.Header{
|
||||||
"User-Agent": []string{"curl"},
|
"User-Agent": []string{"curl"},
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "ipecho",
|
ServiceName: "ipecho",
|
||||||
Url: "http://ipecho.net/plain",
|
Url: "http://ipecho.net/plain",
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "icanhazip",
|
ServiceName: "icanhazip",
|
||||||
Url: "https://icanhazip.com",
|
Url: "https://icanhazip.com",
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "ipify",
|
ServiceName: "ipify",
|
||||||
Url: "https://api.ipify.org",
|
Url: "https://api.ipify.org",
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "myip",
|
ServiceName: "myip",
|
||||||
Url: "https://api.myip.com",
|
Url: "https://api.myip.com",
|
||||||
Decoder: decoder.MyIP,
|
Decoder: decoder.MyIP,
|
||||||
})
|
})
|
||||||
|
|
||||||
Provide(&httpres.Service{
|
Provide(&httpres.Service{
|
||||||
ServiceName: "my-ip",
|
ServiceName: "my-ip",
|
||||||
Url: "https://api.my-ip.io/v2/ip.txt",
|
Url: "https://api.my-ip.io/v2/ip.txt",
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,19 +1,19 @@
|
||||||
package mock
|
package mock
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net"
|
"net"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
IP net.IP
|
IP net.IP
|
||||||
Error error
|
Error error
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s Service) Name() string {
|
func (s Service) Name() string {
|
||||||
return "mock"
|
return "mock"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s Service) Lookup(ctx context.Context) (net.IP, error) {
|
func (s Service) Lookup(ctx context.Context) (net.IP, error) {
|
||||||
return s.IP, s.Error
|
return s.IP, s.Error
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,15 +1,15 @@
|
||||||
package resolver
|
package resolver
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net"
|
"net"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Interface that IP Lookup Services must implement.
|
// Interface that IP Lookup Services must implement.
|
||||||
type Service interface {
|
type Service interface {
|
||||||
// Get the name of the serivce
|
// Get the name of the serivce
|
||||||
Name() string
|
Name() string
|
||||||
|
|
||||||
// Lookup the public ip.
|
// Lookup the public ip.
|
||||||
Lookup(ctx context.Context) (net.IP, error)
|
Lookup(ctx context.Context) (net.IP, error)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue