golia 1.0 → 1.1

data/bin/golia

@@ -1,71 +1,10 @@
 #!/usr/bin/env ruby
 require 'rubygems'
-require 'open-uri'
-require 'net/http'
-require 'fileutils'
-require 'benchmark'
-require 'tmpdir'
-
-puts "You need to pass a domain ex: www.lipsiasoft.com" and exit unless ARGV[0]
-
-@host = ARGV[0]
-@host = "http://#{@host}" unless @host =~ /^http/
-@pid  = "#{Dir.tmpdir}/golia-#{@host.gsub(/^http:\/\//, '')}"
-@checked, @links, @invalid, @valid, @long, @ms = [], [], [], [], [], []
-
-if File.exist?(@pid)
-  puts "<= Founded staled pid"
-  Process.kill(9, File.read(@pid).to_i) rescue nil
-end
-
-trap("INT") { puts "<= Golia has ended his set (crowd applauds)"; remove_pid; Process.kill(9, Process.pid) }
-File.open(@pid, "w") { |f| f.write Process.pid }
-
-def remove_pid
-  FileUtils.rm_rf(@pid)
-end
-
-def parse!(url)
-  begun_at = Time.now
-  response = open(url)
-  @ms << Time.now-begun_at
-  body  = response.read
-  links = body.scan(/<a.+?href=["'](.+?)["']/m).flatten
-  links.reject! { |link| link =~ /^\/$|^http|^mailto|^javascript|#/ || File.extname(link) != "" }
-  links.map! { |link| link = "/"+link if link !~ /^\//; @host+link }
-  @links.concat(links-@checked)
-end
-
-parse!(@host)
-
-loop do
-  break if @links.empty?
-  @links.each do |link|
-    begin
-      @checked << link
-      parse!(link)
-      @valid << link
-      @long  << link if @ms.last > 1
-      puts "\e[32mValid\e[0m (%0.2fms) %s" % [@ms.last, link]
-    rescue Exception => e
-      @invalid << link
-      puts "\e[31mInvalid\e[0m %s - %s" % [link, e.message]
-    ensure
-      @links.delete(link)
-    end
-  end
-end
-puts
-puts "======== SUMMARY ========"
-puts "Valid Links: %d" % @valid.size
-puts "Invalid Links: %d" % @invalid.size
-@invalid.each do |link|
-  puts "  #{link}"
-end
-puts "Long requests: %d" % @long.size
-@long.each do |link|
-  puts "  #{link}"
-end
-puts "Average load time %0.2fms" % (@ms.inject(0) { |memo, ms| memo+=ms; memo }/@ms.size)
-puts
-remove_pid
+require File.dirname(__FILE__) + '/../lib/golia'
+
+if ARGV[0]
+  golia = Golia.new(ARGV[0])
+  golia.start!
+else
+  puts "<= You need to pass a domain ex: www.lipsiasoft.com"
+end

data/golia.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |s|
   s.description = "Golia is an website performance analyzer. Check speed and dead links."
   s.default_executable = "golia"
   s.executables = ["golia"]
-  s.version = "1.0"
+  s.version = "1.1"
   s.date = Time.now.strftime("%Y-%m-%d")
   s.extra_rdoc_files = Dir["*.rdoc"]
-  s.files = %w(README.rdoc Rakefile golia.gemspec)
+  s.files = %w(README.rdoc Rakefile golia.gemspec) + Dir["lib/**/*"]
 end

data/lib/golia.rb

@@ -0,0 +1,72 @@
+require 'open-uri'
+require 'net/http'
+require 'fileutils'
+require 'benchmark'
+require 'tmpdir'
+
+class Golia
+  def initialize(host)
+    @host = host
+    @host = "http://#{@host}" unless @host =~ /^http/
+    @pid  = "#{Dir.tmpdir}/golia-#{@host.gsub(/^http:\/\//, '')}"
+    @checked, @links, @invalid, @valid, @long, @ms = [], [], [], [], [], []
+
+    if File.exist?(@pid)
+      puts "<= Founded staled pid"
+      Process.kill(9, File.read(@pid).to_i) rescue nil
+    end
+
+    trap("INT") { puts "<= Golia has ended his set (crowd applauds)"; kill; Process.kill(9, Process.pid) }
+
+    parse!(@host)
+  end
+
+  def parse!(url)
+    begun_at = Time.now
+    response = open(url)
+    @ms << Time.now-begun_at
+    body  = response.read
+    links = body.scan(/<a.+?href=["'](.+?)["']/m).flatten
+    links.reject! { |link| link =~ /^\/$|^http|^mailto|^javascript|#/ || File.extname(link) != "" }
+    links.map! { |link| link = "/"+link if link !~ /^\//; @host+link }
+    @links.concat(links-@checked)
+  end
+
+  def kill
+    FileUtils.rm_rf(@pid)
+  end
+
+  def start!
+    loop do
+      break if @links.empty?
+      @links.each do |link|
+        begin
+          @checked << link
+          parse!(link)
+          @valid << link
+          @long  << link if @ms.last > 1
+          puts "\e[32mValid\e[0m (%0.2fms) %s" % [@ms.last, link]
+        rescue Exception => e
+          @invalid << link
+          puts "\e[31mInvalid\e[0m %s - %s" % [link, e.message]
+        ensure
+          @links.delete(link)
+        end
+      end
+    end
+    puts
+    puts "======== SUMMARY ========"
+    puts "Valid Links: %d" % @valid.size
+    puts "Invalid Links: %d" % @invalid.size
+    @invalid.each do |link|
+      puts "  #{link}"
+    end
+    puts "Long requests: %d" % @long.size
+    @long.each do |link|
+      puts "  #{link}"
+    end
+    puts "Average load time %0.2fms" % [@ms.inject(0) { |memo, ms| memo+=ms; memo }/@ms.size]
+    puts
+    kill
+  end
+end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: golia
 version: !ruby/object:Gem::Version 
-  hash: 15
+  hash: 13
   prerelease: 
   segments: 
   - 1
-  - 0
-  version: "1.0"
+  - 1
+  version: "1.1"
 platform: ruby
 authors: 
 - DAddYE
@@ -30,6 +30,7 @@ files:
 - README.rdoc
 - Rakefile
 - golia.gemspec
+- lib/golia.rb
 - bin/golia
 has_rdoc: true
 homepage: http://www.padrinorb.com