Detect if a website is using an open source framework

lib/site-inspector/checks/sniffer.rb

@@ -2,6 +2,19 @@ class SiteInspector
   class Endpoint
     class Sniffer < Check
 
+      OPEN_SOURCE_FRAMEWORKS = [
+        # Sniffles
+        :drupal,
+        :joomla,
+        :movabletype,
+        :phpbb,
+        :wordpress,
+
+        # Internal
+        :php,
+        :expression_engine
+      ]
+
       def framework
         cms = sniff :cms
         return cms unless cms.nil?
@@ -10,6 +23,10 @@ def framework
         nil
       end
 
+      def open_source?
+        OPEN_SOURCE_FRAMEWORKS.include?(framework)
+      end
+
       def analytics
         sniff :analytics
       end

spec/checks/site_inspector_endpoint_sniffer_spec.rb

@@ -74,6 +74,10 @@ def set_cookie(key, value)
     it "detects advertising" do
       expect(subject.advertising).to eql(:adsense)
     end
+
+    it "knows wordpress is open source" do
+      expect(subject.open_source?).to eql(true)
+    end
   end
 
   context "no body" do
@@ -83,14 +87,21 @@ def set_cookie(key, value)
       SiteInspector::Endpoint::Sniffer.new(endpoint)
     end
 
+    it "knows when something isn't open source" do
+      set_cookie("foo", "bar")
+      expect(subject.open_source?).to eql(false)
+    end
+
     it "detects PHP" do
       set_cookie("PHPSESSID", "1234")
       expect(subject.framework).to eql(:php)
+      expect(subject.open_source?).to eql(true)
     end
 
     it "detects Expression Engine" do
       set_cookie("exp_csrf_token", "1234")
       expect(subject.framework).to eql(:expression_engine)
+      expect(subject.open_source?).to eql(true)
     end
   end
 end