Do not produce prototypes for FuncDecl outside sketch source code

Author: cmaglie

ArduinoDiagnosticConsumer.cpp

@@ -32,6 +32,7 @@
 #include "ArduinoDiagnosticConsumer.hpp"
 #include "CommandLine.hpp"
 #include "JsonImpl.hpp"
+#include "utils.hpp"
 
 using namespace clang;
 
@@ -56,6 +57,13 @@ void ArduinoDiagnosticConsumer::HandleDiagnostic(DiagnosticsEngine::Level level,
             outs() << presumedFilename << ") ";
         }
 
+        if (!cStrEndsWith(presumedFilename, ".ino")) {
+            if (debugOutput) {
+                outs() << "Ignoring non .ino source file\n";
+            }
+            return;
+        }
+
         unsigned id = info.getID();
         if (id == 3441 || id == 3442 /* use of undeclared identifier */) {
             // It seems that the only way to retrieve the undeclared symbol

utils.hpp

@@ -62,3 +62,16 @@ inline bool startsWith(const string &in, const string &prefix) {
     int l = prefix.length();
     return in.find(prefix.c_str(), 0, l) == 0;
 }
+
+inline bool cStrEndsWith(const char *str, const char *suffix) {
+  if (str == NULL || suffix == NULL)
+    return false;
+
+  size_t strLen = strlen(str);
+  size_t suffixLen = strlen(suffix);
+
+  if (suffixLen > strLen)
+    return false;
+
+  return strncmp(str + strLen - suffixLen, suffix, suffixLen) == 0;
+}