| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | // Package fall handles the fallthrough logic used in plugins that support it.
 | 
					
						
							|  |  |  | package fall
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/plugin"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // F can be nil to allow for no fallthrough, empty allow all zones to fallthrough or
 | 
					
						
							|  |  |  | // contain a zone list that is checked.
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | type F struct {
 | 
					
						
							|  |  |  | 	Zones []string
 | 
					
						
							|  |  |  | }
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | // Through will check if we should fallthrough for qname. Note that we've named the
 | 
					
						
							|  |  |  | // variable in each plugin "Fall", so this then reads Fall.Through().
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | func (f F) Through(qname string) bool {
 | 
					
						
							|  |  |  | 	return plugin.Zones(f.Zones).Matches(qname) != ""
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | // setZones will set zones in f.
 | 
					
						
							|  |  |  | func (f *F) setZones(zones []string) {
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | 	for i := range zones {
 | 
					
						
							|  |  |  | 		zones[i] = plugin.Host(zones[i]).Normalize()
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 	f.Zones = zones
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | // SetZonesFromArgs sets zones in f to the passed value or to "." if the slice is empty.
 | 
					
						
							|  |  |  | func (f *F) SetZonesFromArgs(zones []string) {
 | 
					
						
							|  |  |  | 	if len(zones) == 0 {
 | 
					
						
							| 
									
										
										
										
											2018-01-09 22:29:19 +00:00
										 |  |  | 		f.setZones(Root.Zones)
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 		return
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | 	}
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 	f.setZones(zones)
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | // Equal returns true if f and g are equal.
 | 
					
						
							|  |  |  | func (f F) Equal(g F) bool {
 | 
					
						
							|  |  |  | 	if len(f.Zones) != len(g.Zones) {
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | 		return false
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 	for i := range f.Zones {
 | 
					
						
							|  |  |  | 		if f.Zones[i] != g.Zones[i] {
 | 
					
						
							|  |  |  | 			return false
 | 
					
						
							| 
									
										
										
										
											2018-01-07 16:32:59 +00:00
										 |  |  | 		}
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	return true
 | 
					
						
							|  |  |  | }
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  | // Zero returns a zero valued F.
 | 
					
						
							|  |  |  | var Zero = func() F {
 | 
					
						
							|  |  |  | 	return F{[]string{}}
 | 
					
						
							| 
									
										
										
										
											2018-01-09 22:29:19 +00:00
										 |  |  | }()
 | 
					
						
							| 
									
										
										
										
											2018-01-07 14:51:32 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  | // Root returns F set to only ".".
 | 
					
						
							|  |  |  | var Root = func() F {
 | 
					
						
							|  |  |  | 	return F{[]string{"."}}
 | 
					
						
							| 
									
										
										
										
											2018-01-09 22:29:19 +00:00
										 |  |  | }()
 |