diff --git a/doWITCOM.go b/doWITCOM.go index c5e8ddc..c68408a 100644 --- a/doWITCOM.go +++ b/doWITCOM.go @@ -103,6 +103,9 @@ func scanForGoFiles(startDir string) ([]string, error) { if strings.HasSuffix(path, ".go") { files = append(files, path) } + if strings.HasSuffix(path, ".proto") { + files = append(files, path) + } // don't go into any directories if info.IsDir() {