diff options
| -rw-r--r-- | directives_generate.go | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/directives_generate.go b/directives_generate.go index 74f7e2be2..e70eede23 100644 --- a/directives_generate.go +++ b/directives_generate.go @@ -43,7 +43,7 @@ func main() { md[priority] = items[1] mi[items[1]] = middlewarePath + items[2] // Default, unless overriden by 3rd arg - if strings.Contains(items[2], "/") { // External package has been given + if _, err := os.Stat(middlewareFSPath + items[2]); err != nil { // External package has been given mi[items[1]] = items[2] } } @@ -113,7 +113,8 @@ func fatalIfErr(err error) { } const ( - middlewarePath = "github.com/coredns/coredns/middleware/" - middlewareFile = "middleware.cfg" - header = "// generated by directives_generate.go; DO NOT EDIT\n\n" + middlewarePath = "github.com/coredns/coredns/middleware/" + middlewareFile = "middleware.cfg" + middlewareFSPath = "middleware/" // Where the middleware packages are located on the file system + header = "// generated by directives_generate.go; DO NOT EDIT\n\n" ) |