mccabe 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 72379d20b7c7ea824390cb61969e45cb5a9f505e
-  data.tar.gz: 1001d47499a6d291601a23564f439b13d493d8bc
+  metadata.gz: 6fedee59e685c053624c54bbd801a2bdc85b9438
+  data.tar.gz: fb8618f85ca60eb054de2b28935a5cdd5f5ad92a
 SHA512:
-  metadata.gz: 46f4510109cecbbf18c989bd796b5bf5f45526a5ad14c6943e23f5385462e02362dfbeb43f353843bc55b765195ad56b25a7912da102f6308ce64dfcdec49a5c
-  data.tar.gz: 755647bd30d9dc774c9bb9990ab8fad5ee18a0ccffc1583a44aecef6c75b46db64477b5ba09212a6ec58834820766aaa2f88c695f5de7424c65b986f9974d5b4
+  metadata.gz: 0210e6c28d5c8343443a0b0adf3bfd846ef7e2e1558c3e18eacafa0e9d47ee50714a77ec1430346a388a3349c459c3de7d8bb3333bdb870d37e5000dd5780025
+  data.tar.gz: 5e8da4b81d6668b98e0655a81ac81a95a7094ec63b55efb32c18bfae5b314eb78cb0d6760c5b19157a2412d11e6cc302aa66f9a403e4f235decbd892028504b3

data/lib/mccabe/analyzer.rb

@@ -3,6 +3,7 @@ require 'mccabe/parser'
 module McCabe
   class Analyzer
     BRANCH_TYPES = [:if, :while, :until, :for, :when, :and, :or]
+    ENUMERABLE_METHODS = Enumerable.instance_methods + [:each]
 
     def self.analyze(ast)
       results = {}
@@ -16,10 +17,28 @@ module McCabe
       nodes, complexity = [ast], 0
       until nodes.empty?
         node = nodes.shift
-        complexity += 1 if BRANCH_TYPES.include?(node.type)
+        if BRANCH_TYPES.include?(node.type)
+          complexity += 1
+        elsif node.type == :block
+          complexity += block_complexity node
+        elsif node.type == :send
+          complexity += send_complexity node
+        end
         nodes += node.children.select { |child| child.is_a? ::Parser::AST::Node }
       end
       complexity
     end
+
+    def self.block_complexity(block_node)
+      sent_method = block_node.children[0]
+      sent_method.children.length == 2 &&
+        ENUMERABLE_METHODS.include?(sent_method.children[1]) ? 1 : 0
+    end
+
+    def self.send_complexity(send_node)
+      send_node.children.length >= 3 &&
+        send_node.children[2].type == :block_pass &&
+        ENUMERABLE_METHODS.include?(send_node.children[1]) ? 1 : 0
+    end
   end
 end

data/lib/mccabe/version.rb

@@ -1,3 +1,3 @@
 module McCabe
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mccabe
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrew Mason
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-17 00:00:00.000000000 Z
+date: 2015-08-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser