$ ~/Scripts/cfile.rb ~/Projects/Kifu/code/kifu log tmp doc versions

#! /usr/bin/ruby

# cfile.rb
# Hilton Lipschitz
# Use and modify freely, attribution appreciated
#
# This script counts the number and kinds of files in a project. Any additional
# parameters are treated as exclude folder regexes
#
# Requirements:
# 
#
# Examples:
# cfile ~/Folder/DBFS
# cfile ~/Folder/DBFS libs # Will exclude any folders with /libs/
# ~/Scripts/cfile.rb ~/Projects/Kifu/code/kifu log tmp doc versions # To skip all unnecessary rails trees

require 'rubygems'
require 'fileutils'

# No need to report these individually, so merge the counts
@merge_keys = {
	".xcbkptlist" => ".xcodeproj",
	".xcscheme" => ".xcodeproj",
	".xcuserstate" => ".xcodeproj",
	".pbxproj" => ".xcodeproj",
	".xcworkspacedata" => ".xcodeproj",
	".xcsettings" => ".xcodeproj",
	".scssc" => ".scss",
	".fnt" => ".ttf",
	".woff" => ".ttf"
}

# Make nice names for these
@descriptive = {
	".xcodeproj" => "Xcode Project",
	".c" 		 => "C/Objective-C",
	".h" 		 => "C/C++ Headers",
	".m" 		 => "Objective-C",
	".pch" 		 => "Precompiled Headers",
	".plist" 	 => "Plists",
	".png" 		 => "PNG Images",
	".txt" 		 => "Plain Text",
	".strings" 	 => "Strings",
	".coffee"    => "CoffeeScript",
	".conf"      => "Configuration",
	".css"       => "CSS",
	".csv" 		 => "CSV",
	".erb"       => "ERB",
	".gif" 		 => "GIF Images",
	".god" 		 => "GOD Configs",
	".html" 	 => "HTML",
	".ico" 		 => "Icon Images",
	".js" 		 => "JavaScripts",
	".lock" 	 => "Locks",
	".log" 		 => "Logs",
	".markdown"  => "Markdown",
	".old" 		 => "OLD",
	".pdf" 		 => "PDFs",
	".rake" 	 => "Rake",
	".rb" 		 => "Ruby",
	".scss" 	 => "SCSS",
	".ttf"		 => "Fonts",
	".xml"		 => "XML",
	".yml"		 => "YML",
	".ru"		 => "Rackup",
	".xib"		 => "XIBs",
	".storyboard" => "StoryBoards",
}

# Group the ones we know
@categories = {
	"Code" => [".c", ".h", ".m", ".coffee", ".erb", ".js", ".rb", ".css", ".scss", ".pch", ".html", ".xib", ".storyboard"],
	"Image" => [".gif", ".ico", ".png", ".pdf"],
	"Configuration" => [".xcodeproj", ".plist", ".conf", ".yml", ".ru", "strings"],
	"Data" => [".xml", ".csv"],
	"Text" => [".txt", ".markdown"],
}

@excludes = []
@counter_hash = {}

# Increment the count
def increment_key(key, value = 1)
	if @counter_hash[key].nil?
		@counter_hash[key] = value
	else
		@counter_hash[key] = @counter_hash[key] + value
	end
end

# Analyze a file by extension
def analyze_file(source)
	extension = File.extname(source)
# 	return if extension.length == 0
	increment_key extension
end

def process_folder(source)

	@excludes.each do |pattern|
		return if source =~ Regexp.new(pattern)
	end

	source_dir = Dir.new(source)
	source_dir.entries.each do |source_file|
		# Skip dot files and hidden files
		next if source_file =~ /^\./
		full_path = File.expand_path(source_file, source_dir.path)
# 		puts full_path
		if File.directory?(full_path)
			process_folder full_path
		else
			analyze_file full_path
		end
	end
end

def merge_keys
	@merge_keys.keys.each do |key|
		unless @counter_hash[key].nil?
			increment_key @merge_keys[key], @counter_hash[key]
			@counter_hash.delete(key)
		end
	end
end

if ARGV.length == 0
  puts "USAGE: cfile ~/Folder/DBFS <excludes>"
  exit
end

source = ARGV[0]
if ARGV.length > 1
	@excludes = ARGV[1..(ARGV.length)]
# 	puts @excludes
end

unless File.directory?(source)
  puts "  ERROR: Parameter 1 must be a project folder..."
  exit
end

process_folder source

merge_keys

puts
puts "Project File Counter v0.1 ©hiltmon.com 2012 https://hiltmon.com"
puts
puts "Project: #{File.basename(source)}"

used_keys = []
counter = 0
@categories.keys.each do |category|
	array = @categories[category]
	header_printed = false
	@counter_hash.keys.sort.each do |key|
		next if array.index(key).nil?
		unless header_printed
			puts "-------------------------------------------------------"
			puts "#{category} Files:"
			header_printed = true
		end
		used_keys << key if used_keys.index(key).nil?
		descriptive = (@descriptive[key] || "#{key} files").ljust(22)
		puts "    #{descriptive}: #{@counter_hash[key].to_s.rjust(6)}"
		counter = counter + @counter_hash[key]
	end
end

header_printed = false
@counter_hash.keys.sort.each do |key|
	next unless used_keys.index(key).nil?
	unless header_printed
		puts "-------------------------------------------------------"
		puts "Other Files:"
		header_printed = true
	end
	descriptive = (@descriptive[key] || "#{key} files").ljust(22)
	descriptive = "Unknown Kind".ljust(22) if key.length == 0
	puts "    #{descriptive}: #{@counter_hash[key].to_s.rjust(6)}"
	counter = counter + @counter_hash[key]
end

puts "-------------------------------------------------------"
puts "TOTAL                     : #{counter.to_s.rjust(6)} Files"
puts "-------------------------------------------------------"