headcount 0.5.0 → 0.6.0

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    headcount (0.0.1)
+    headcount (0.6.0)
       rails (~> 3.2.2)
 
 GEM

data/MIT-LICENSE

@@ -1,4 +1,4 @@
-Copyright 2012 YOURNAME
+Copyright 2012 Ryan Mohr
 
 Permission is hereby granted, free of charge, to any person obtaining
 a copy of this software and associated documentation files (the

data/README.md

@@ -33,6 +33,9 @@ Headcount.configure do |config|
   
   # you can specify the key yourself if you'd like to override the default
   count User.active, :as => :active_members
+  
+  # if you're fine with using default keys you can pass a list of queries instead
+  count User, Account
 end
 ```
 
@@ -44,11 +47,13 @@ If you'd like to access or reconfigure Headcount outside of the initializer you
 # Usage
 
 Fire up `rails c` and run:
+
 ```
 Headcount.count # will return current headcount as a hash
 ```
 
 If you'd like to have the results written to disk instead just use the bang version:
+
 ```
 Headcount.count! # will append the results to the output file
 ```
@@ -59,7 +64,11 @@ Your options are open as far as scheduling goes.  If you like [crontab](http://c
 
 You have two options for triggering a headcount.
 
-1. Call `Headcount.count!` directly
-2. Use `rake headcount`
+1. Call it directly: `Headcount.count!`
+2. Use the rake task: `rake headcount`
+
+In the future I may write a daemon to simplify the scheduling process with an upstart script to boot, but for now you're on your own.
+
+# Credit
 
-In the future I may write a daemon to simplify the scheduling process with an upstart script to boot, but for now you're on your own.
+Headcount was created for use at Kumu (http://kumupowered.com)

data/lib/headcount/configuration.rb

@@ -34,17 +34,38 @@ module Headcount
         instance_exec(self, &block)
       end
 
-      def count(query, options = {}, &block)
-        raise UnsupportedQuery, 'query does not respond to :count' unless query.respond_to?(:count)
+      def count(*arguments, &block)
+        if list_given?(*arguments)
+          count_each(arguments.flatten)
+        else
+          query   = arguments[0]
+          options = arguments[1] || {}
+          
+          if query.respond_to?(:count)
+            key = options[:as] || Headcount::Support.key_for(query)
 
-        key = options[:as] || Headcount::Support.key_for(query)
+            # just ignore blocks for now
+            # not really finding a need for nested declarations
+            #if block_given?
+            #  evaluate(&block)
+            #end
 
-        # just ignore blocks for now
-        #if block_given?
-        #  evaluate(&block)
-        #end
-
-        Headcount.register(key, query)
+            Headcount.register(key, query)
+          else
+            raise UnsupportedQuery, 'query does not respond to :count'
+          end
+        end
+      end
+      
+      private
+      def list_given?(*arguments)
+        arguments[0].is_a?(Array) || (arguments.length > 1 && !arguments[1].is_a?(Hash))
+      end
+      
+      def count_each(queries)
+        queries.each do |query|
+          count(query)
+        end
       end
     end
   end

data/lib/headcount/version.rb

@@ -1,3 +1,3 @@
 module Headcount
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

metadata

@@ -2,7 +2,7 @@
 name: headcount
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors: 
 - Ryan Mohr
@@ -89,7 +89,7 @@ dependencies:
         version: "0"
   type: :development
   version_requirements: *id007
-description: ""
+description: If you can write a query for it, headcount can track it for you.
 email: 
 - ryan.mohr@gmail.com
 executables: []
@@ -142,6 +142,6 @@ rubyforge_project: headcount
 rubygems_version: 1.8.21
 signing_key: 
 specification_version: 3
-summary: A simple stat collector for rails apps
+summary: A simple stat collector for rails apps.
 test_files: []