| 
									
										
										
										
											2018-06-29 12:44:16 +03:00
										 |  |  | package metadata
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							| 
									
										
										
										
											2020-09-24 18:14:41 +02:00
										 |  |  | 	"github.com/coredns/caddy"
 | 
					
						
							| 
									
										
										
										
											2018-06-29 12:44:16 +03:00
										 |  |  | 	"github.com/coredns/coredns/core/dnsserver"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/plugin"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-20 08:02:30 +01:00
										 |  |  | func init() { plugin.Register("metadata", setup) }
 | 
					
						
							| 
									
										
										
										
											2018-06-29 12:44:16 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  | func setup(c *caddy.Controller) error {
 | 
					
						
							|  |  |  | 	m, err := metadataParse(c)
 | 
					
						
							|  |  |  | 	if err != nil {
 | 
					
						
							|  |  |  | 		return err
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
 | 
					
						
							|  |  |  | 		m.Next = next
 | 
					
						
							|  |  |  | 		return m
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	c.OnStartup(func() error {
 | 
					
						
							|  |  |  | 		plugins := dnsserver.GetConfig(c).Handlers()
 | 
					
						
							|  |  |  | 		for _, p := range plugins {
 | 
					
						
							|  |  |  | 			if met, ok := p.(Provider); ok {
 | 
					
						
							|  |  |  | 				m.Providers = append(m.Providers, met)
 | 
					
						
							|  |  |  | 			}
 | 
					
						
							|  |  |  | 		}
 | 
					
						
							|  |  |  | 		return nil
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return nil
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func metadataParse(c *caddy.Controller) (*Metadata, error) {
 | 
					
						
							|  |  |  | 	m := &Metadata{}
 | 
					
						
							|  |  |  | 	c.Next()
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-05-17 22:19:54 +02:00
										 |  |  | 	m.Zones = plugin.OriginsFromArgsOrServerBlock(c.RemainingArgs(), c.ServerBlockKeys)
 | 
					
						
							| 
									
										
										
										
											2018-06-29 12:44:16 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if c.NextBlock() || c.Next() {
 | 
					
						
							|  |  |  | 		return nil, plugin.Error("metadata", c.ArgErr())
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	return m, nil
 | 
					
						
							|  |  |  | }
 |