jeremylightsmith-actionsite 0.1 → 0.3

data/Rakefile

@@ -3,7 +3,7 @@ $LOAD_PATH << File.expand_path(File.dirname(__FILE__) + "/lib")
 require 'action_site'
 require 'spec/rake/spectask'
 
-task :default => [:spec, :example]
+task :default => [:spec, :gemspec, :example]
 
 Spec::Rake::SpecTask.new(:spec) do |t|
   t.spec_files = FileList['spec/**/*_spec.rb']

data/lib/action_site.rb

@@ -15,3 +15,8 @@ require 'yaml'
 require 'action_site/extensions/string'
 
 require 'action_site/site'
+require 'action_site/link_checker'
+
+module ActionSite
+  VERSION = 0.3
+end

data/lib/action_site/generators/markaby_generator.rb

@@ -4,7 +4,7 @@ module ActionSite
       def process(context, content)
         builder = Markaby::Builder.new({}, context)
         builder.instance_eval content, context.file_name
-        content = builder.to_s
+        builder.to_s
       end
     end
   end

data/lib/action_site/generators/redcloth_generator.rb

@@ -3,7 +3,7 @@ module ActionSite
     class RedclothGenerator
       def process(context, content)
         redcloth = RedCloth.new(content)
-        content = redcloth.to_html(:html, :textile)
+        redcloth.to_html(:html, :textile)
       end
     end
   end

data/lib/action_site/generators/yaml_generator.rb

@@ -2,7 +2,7 @@ module ActionSite
   module Generators
     class YamlGenerator
       def process(context, content)
-        content = YAML::load(content) rescue raise("error reading #{context.file_name}: #{$!.message}")
+        YAML::load(content) rescue raise("error reading #{context.file_name}: #{$!.message}")
       end
     end
   end

data/lib/action_site/html_generator.rb

@@ -4,8 +4,9 @@ module ActionSite
   class HtmlGenerator
     attr_reader :template_directory
   
-    def initialize(site_context, template_directory)
-      @site_context, @template_directory = site_context, template_directory
+    def initialize(site_context, template_directory, generators)
+      @site_context, @template_directory, @generators = 
+        site_context, template_directory, generators
     end
 
     def process_file(file, context = new_context(file), apply_layout = true)
@@ -18,7 +19,7 @@ module ActionSite
     def process(file_name, context, content, apply_layout = true)
       file_name, extension = file_name.split_filename
 
-      generator = ActionSite.generators[extension]
+      generator = @generators[extension]
       if generator
         content = generator.process(context, content)
 

data/lib/action_site/link_checker.rb

@@ -0,0 +1,106 @@
+require 'rubygems'
+require 'hpricot'
+require 'net/http'
+require 'uri'
+
+module ActionSite
+  class LinkChecker
+    def initialize(options = {})
+      @options = options
+      @checked_urls = []
+    end
+  
+    def check(url, host = nil, indent = "")
+      url = url.sub(/\#.*/, '')
+      return if url.blank?
+      return if @options[:local] && host && !local?(host, url)
+      return if @checked_urls.include?(url)
+      @checked_urls << url
+      
+      puts "#{indent}checking #{url}"
+
+      begin
+        url, html = fetch(url)
+      rescue
+        puts "#{url} not found"
+        raise
+      end
+      
+      host ||= host_for(url)
+      doc = Hpricot(html)
+
+      if local?(host, url)
+        (doc / "a").each do |link|
+          if child_link = expand_link(url, link["href"])
+            begin
+              check child_link, host, indent + "   "
+            rescue
+              puts "from #{url} as #{link}"
+              raise
+            end
+          end
+        end
+      end
+    end
+  
+    def expand_link(from, to)
+      case to
+      when nil
+        nil
+      when /^mailto:.+\@.+\..+/
+        nil
+      when /^https?\:\/\//
+        to
+      when /^\//
+        File.join(host_for(from), to)
+      else
+        relative_link(from, to)
+      end
+    end
+    
+    def relative_link(from, to)
+      from += "." if from.ends_with?("/")
+      
+      from = from.split("/")
+      to = to.split("/")
+      
+      from.pop
+      
+      while to[0] == ".."
+        to.shift
+        from.pop
+      end
+      
+      while to[0] == "."
+        to.shift
+      end
+      
+      (from + to).join("/")
+    end
+
+    def fetch(url, limit = 10)
+      # You should choose better exception.
+      raise ArgumentError, 'HTTP redirect too deep' if limit == 0
+
+      response = Net::HTTP.get_response(URI.parse(url))
+      case response
+      when Net::HTTPSuccess     then [url, response.body]
+      when Net::HTTPRedirection then fetch(response['location'], limit - 1)
+      else
+        response.error!
+      end
+    end
+    
+    def host_for(url)
+      if url =~ /^(https?\:\/\/[^\/]+\/)/
+        $1
+      else
+        raise "don't know how to get the host for #{url}"
+      end
+    end
+    
+    def local?(host, url)
+      url.starts_with?(host)
+    end
+  end
+end

data/lib/action_site/page_context.rb

@@ -17,6 +17,7 @@ module ActionSite
       @html_generator, @global_context, @file_name = 
         html_generator, global_context, file_name
       layout(:application) rescue nil
+      helper(:application, false)
     end
   
     def layout(name)
@@ -33,17 +34,18 @@ module ActionSite
       @html_generator.process_file(*args)
     end
   
-    def helper(name)
+    def helper(name, assert_loaded = true)
       file_name = "#{name}_helper"
+      fully_qualified_file = File.join(@html_generator.template_directory, "helpers", file_name + ".rb")
+      return unless File.exist?(fully_qualified_file) || assert_loaded
+      
       class_name = file_name.classify
-      helper = begin
-        class_name.constantize
-      rescue NameError # this constant hasn't been loaded yet
-        require File.join(@html_generator.template_directory, "helpers", file_name)
-        class_name.constantize
+      if Object.const_defined?(class_name)
+        Object.send(:remove_const, class_name)
       end
-    
-      metaclass.send(:include, helper)
+
+      load fully_qualified_file
+      metaclass.send(:include, class_name.constantize)
     end
   
     def content_for(name)

data/lib/action_site/site.rb

@@ -9,30 +9,32 @@ require 'action_site/generators/yaml_generator'
 
 module ActionSite
   EXCLUDED_DIRECTORIES = %w(layouts helpers)
-  RESOURCE_EXTENSIONS = %w(css ico gif jpg png js)
-
-  def self.generators
-    @generators ||= {
-      "erb" => Generators::ErbGenerator.new,
-      "mab" => Generators::MarkabyGenerator.new,
-      "red" => Generators::RedclothGenerator.new,
-      "yml" => Generators::YamlGenerator.new,
-      "yaml" => Generators::YamlGenerator.new
-    }
-  end
-
+  RESOURCE_EXTENSIONS = %w(css ico gif jpg png js pdf)
+  DEFAULT_GENERATORS = {
+    "erb" => Generators::ErbGenerator.new,
+    "mab" => Generators::MarkabyGenerator.new,
+    "red" => Generators::RedclothGenerator.new,
+    "yml" => Generators::YamlGenerator.new,
+    "yaml" => Generators::YamlGenerator.new
+  }
+  
   class Site
     attr_reader :context
-    
+
     def initialize(in_dir, out_dir)
       @context = Context.new
-      @generator = HtmlGenerator.new(@context, in_dir)
+      @generator = HtmlGenerator.new(@context, in_dir, generators)
       @in_dir, @out_dir = in_dir, out_dir
       puts "\nGENERATING #{File.basename(in_dir).upcase}\n\n"
       rm_rf out_dir
       mkdir_p out_dir
     end
     
+    # add / remove / change generators to change the behavior of the html generation
+    def generators
+      @generators ||= DEFAULT_GENERATORS.dup
+    end
+    
     def generate(in_dir = @in_dir, out_dir = @out_dir)
       Dir[in_dir + "/*"].each do |in_file|
         out_file = in_file.gsub(in_dir, out_dir)

data/spec/action_site/html_generator_spec.rb

@@ -5,12 +5,14 @@ describe ActionSite::HtmlGenerator do
   attr_reader :generator
   
   before do
-    @generator = ActionSite::HtmlGenerator.new(ActionSite::Context.new, sandbox.root)
+    @generator = ActionSite::HtmlGenerator.new(ActionSite::Context.new, 
+                                               sandbox.root, 
+                                               ActionSite::DEFAULT_GENERATORS)
   end
   
   describe "default generators" do
     it "should have erb, mab, yaml, red" do
-      ActionSite.generators.keys.sort.should == %w(erb mab red yaml yml)
+      ActionSite::DEFAULT_GENERATORS.keys.sort.should == %w(erb mab red yaml yml)
     end
   end
   
@@ -55,6 +57,15 @@ describe ActionSite::HtmlGenerator do
             'hello world'
           end
         end"
+
+      sandbox.new :file => 'helpers/application_helper.rb', 
+                  :with_contents => "
+        module ApplicationHelper
+          def bob
+            'the builder'
+          end
+        end"
+        
       sandbox.new :file => "layouts/application.html.erb", 
                   :with_contents => "<%= content %>"
     end
@@ -63,6 +74,25 @@ describe ActionSite::HtmlGenerator do
       process("*.html.erb", "% helper 'test'\n<%= some_test_method %>").
                 should == "hello world"
     end
+    
+    it "should allow multiple helpers with same name between sites" do
+      process("*.html.erb", "<%= bob %>").should == "the builder"
+
+      sandbox.new :file => 'helpers/application_helper.rb', 
+                  :with_contents => "
+        module ApplicationHelper
+          def jack
+            'the bricklayer'
+          end
+        end"
+
+        process("*.html.erb", "<%= jack %>").should == "the bricklayer"
+        proc {process("*.html.erb", "<%= bob %>")}.should raise_error
+    end
+    
+    it "should include application_helper.rb if it is there" do
+      process("*.html.erb", "<%= bob %>").should == "the builder"
+    end
   end
   
   describe "content_for" do

data/spec/action_site/link_checker_spec.rb

@@ -0,0 +1,40 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe ActionSite::LinkChecker do
+  attr_reader :links
+  
+  before do
+    @links = ActionSite::LinkChecker.new
+  end
+  
+  it "should calculate correct links" do
+    expand_link("http://localhost/bob", "http://google.com").should == "http://google.com"
+    expand_link("http://localhost/bob", "http://google.com").should == "http://google.com"
+    expand_link("http://localhost/bob", "george").should == "http://localhost/george"
+    expand_link("http://localhost/bob/", "george").should == "http://localhost/bob/george"
+    expand_link("http://localhost/bob/", ".").should == "http://localhost/bob"
+    expand_link("http://localhost/bob", ".").should == "http://localhost"
+    expand_link("http://localhost/bob", "..").should == "http:/"
+    expand_link("http://localhost/bob/", "..").should == "http://localhost"
+    expand_link("http://localhost/a/blues_hero/", "/register.html").should == "http://localhost/register.html"
+  end
+  
+  it "should find get redirects" do
+    url, html = links.fetch("http://google.com")
+    
+    url.should == "http://www.google.com/"
+  end
+  
+  it "should find host for urls" do
+    links.host_for("http://foo.bar:3000/").should == "http://foo.bar:3000/"
+    links.host_for("http://google.com/").should == "http://google.com/"
+    links.host_for("https://google.com/").should == "https://google.com/"
+    links.host_for("http://google.com/a/b/c").should == "http://google.com/"
+    links.host_for("https://google.com/a/b/c").should == "https://google.com/"
+    proc {links.host_for("foo")}.should raise_error
+  end
+  
+  def expand_link(from, to)
+    links.expand_link(from, to)
+  end
+end

data/spec/action_site/page_context_spec.rb

@@ -2,10 +2,14 @@ require File.dirname(__FILE__) + '/../spec_helper'
 
 module ActionSite
   describe PageContext do
+    include FileSandbox
     attr_reader :context
     
     before do
-      @context = PageContext.new(nil, nil, nil)
+      @generator = ActionSite::HtmlGenerator.new(ActionSite::Context.new, 
+                                                 sandbox.root, 
+                                                 ActionSite::DEFAULT_GENERATORS)
+      @context = @generator.new_context(nil)
     end
     
     it "should allow setting new variables and accessing them" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jeremylightsmith-actionsite
 version: !ruby/object:Gem::Version 
-  version: "0.1"
+  version: "0.3"
 platform: ruby
 authors: 
 - Jeremy Lightsmith
@@ -64,6 +64,7 @@ files:
 - lib/action_site/helpers/markaby_helper.rb
 - lib/action_site/helpers/url_helper.rb
 - lib/action_site/html_generator.rb
+- lib/action_site/link_checker.rb
 - lib/action_site/page_context.rb
 - lib/action_site/site.rb
 - lib/action_site.rb
@@ -133,5 +134,6 @@ test_files:
 - spec/action_site/helpers/form_helper_spec.rb
 - spec/action_site/helpers/url_helper_spec.rb
 - spec/action_site/html_generator_spec.rb
+- spec/action_site/link_checker_spec.rb
 - spec/action_site/page_context_spec.rb
 - spec/spec_helper.rb