📦 kurokobo / caddy-dns-azure

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105package azure

import (
	azure "github.com/kurokobo/libdns-azure"

	"github.com/caddyserver/caddy/v2"
	"github.com/caddyserver/caddy/v2/caddyconfig/caddyfile"
)

// Provider wraps the provider implementation as a Caddy module.
type Provider struct{ *azure.Provider }

func init() {
	caddy.RegisterModule(Provider{})
}

// CaddyModule returns the Caddy module information.
func (Provider) CaddyModule() caddy.ModuleInfo {
	return caddy.ModuleInfo{
		ID: "dns.providers.azure",
		New: func() caddy.Module {
			return &Provider{new(azure.Provider)}
		},
	}
}

// Provision implements the Provisioner interface to initialize the Azure client
func (p *Provider) Provision(ctx caddy.Context) error {
	repl := caddy.NewReplacer()
	p.Provider.TenantId = repl.ReplaceAll(p.Provider.TenantId, "")
	p.Provider.ClientId = repl.ReplaceAll(p.Provider.ClientId, "")
	p.Provider.ClientSecret = repl.ReplaceAll(p.Provider.ClientSecret, "")
	p.Provider.SubscriptionId = repl.ReplaceAll(p.Provider.SubscriptionId, "")
	p.Provider.ResourceGroupName = repl.ReplaceAll(p.Provider.ResourceGroupName, "")

	return nil
}

// UnmarshalCaddyfile sets up the DNS provider from Caddyfile tokens. Syntax:
//
// azure {
//     tenant_id <string>
//     client_id <string>
//     client_secret <string>
//     subscription_id <string>
//     resource_group_name <string>
// }
//
func (p *Provider) UnmarshalCaddyfile(d *caddyfile.Dispenser) error {
	for d.Next() {
		if d.NextArg() {
			return d.ArgErr()
		}
		for nesting := d.Nesting(); d.NextBlock(nesting); {
			switch d.Val() {
			case "tenant_id":
				if d.NextArg() {
					p.Provider.TenantId = d.Val()
				}
				if d.NextArg() {
					return d.ArgErr()
				}
			case "client_id":
				if d.NextArg() {
					p.Provider.ClientId = d.Val()
				}
				if d.NextArg() {
					return d.ArgErr()
				}
			case "client_secret":
				if d.NextArg() {
					p.Provider.ClientSecret = d.Val()
				}
				if d.NextArg() {
					return d.ArgErr()
				}
			case "subscription_id":
				if d.NextArg() {
					p.Provider.SubscriptionId = d.Val()
				}
				if d.NextArg() {
					return d.ArgErr()
				}
			case "resource_group_name":
				if d.NextArg() {
					p.Provider.ResourceGroupName = d.Val()
				}
				if d.NextArg() {
					return d.ArgErr()
				}
			default:
				return d.Errf("unrecognized subdirective '%s'", d.Val())
			}
		}
	}

	return nil
}

// Interface guards
var (
	_ caddyfile.Unmarshaler = (*Provider)(nil)
	_ caddy.Provisioner     = (*Provider)(nil)
)