bdimcheff-staticmatic 0.10.1

data/lib/staticmatic/mixins/build.rb

@@ -0,0 +1,46 @@
+module StaticMatic::BuildMixin
+    
+  def build
+    build_css
+    build_html
+  end
+    
+  # Build HTML from the source files
+  def build_html
+    Dir["#{@src_dir}/pages/**/*.haml"].each do |path|
+      next if File.basename(path) =~ /^\_/  # skip partials
+      file_dir, template = source_template_from_path(path.sub(/^#{@src_dir}\/pages/, ''))
+      save_page(File.join(file_dir, template), generate_html_with_layout(template, file_dir))
+    end
+  end
+
+  # Build CSS from the source files
+  def build_css
+    Dir["#{@src_dir}/stylesheets/**/*.sass"].each do |path|
+      file_dir, template = source_template_from_path(path.sub(/^#{@src_dir}\/stylesheets/, ''))
+      save_stylesheet(File.join(file_dir, template), generate_css(template, file_dir))
+    end
+  end
+  
+  def copy_file(from, to)
+    FileUtils.cp(from, to)
+  end
+
+  def save_page(filename, content)
+    generate_site_file(filename, 'html', content)
+  end
+
+  def save_stylesheet(filename, content)
+    generate_site_file(File.join('stylesheets', filename), 'css', content)
+  end
+
+  def generate_site_file(filename, extension, content)
+    path = File.join(@site_dir,"#{filename}.#{extension}")
+    FileUtils.mkdir_p(File.dirname(path))
+    File.open(path, 'w+') do |f|
+      f << content
+    end
+    
+    puts "created #{path}"
+  end
+end

data/lib/staticmatic/mixins/helpers.rb

@@ -0,0 +1,15 @@
+module StaticMatic::HelpersMixin    
+  # Loads any helpers present in the helpers dir and mixes them into the template helpers
+  def load_helpers
+
+    Dir["#{@src_dir}/helpers/**/*_helper.rb"].each do |helper|
+      load_helper(helper)
+    end
+  end
+
+  def load_helper(helper)
+    load helper
+    module_name = File.basename(helper, '.rb').gsub(/(^|\_)./) { |c| c.upcase }.gsub(/\_/, '')
+    Haml::Helpers.class_eval("include #{module_name}")
+  end
+end

data/lib/staticmatic/mixins/render.rb

@@ -0,0 +1,125 @@
+module StaticMatic::RenderMixin
+  
+  def source_for_layout
+    if layout_exists?(@layout)
+      File.read(full_layout_path(@layout))
+    else
+      raise StaticMatic::Error.new("", full_layout_path(@layout), "Layout not found")
+    end
+  end
+
+  # Generate html from source file:
+  # generate_html("index")
+  def generate_html(source_file, source_dir = '')
+    full_file_path = File.join(@src_dir, 'pages', source_dir, "#{source_file}.haml")
+
+    begin
+      # clear all scope variables except @staticmatic
+      @scope.instance_variables.each do |var|
+        @scope.instance_variable_set(var, nil) unless var == '@staticmatic' 
+      end
+      html = generate_html_from_template_source(File.read(full_file_path))
+  
+      @layout = determine_layout(source_dir)
+    rescue StaticMatic::TemplateError => e
+      raise e # re-raise inline errors
+    rescue Exception => e
+      raise StaticMatic::TemplateError.new(full_file_path, e)
+    end
+
+    html
+  end
+
+  def generate_html_with_layout(source, source_dir = '')
+    @current_page = File.join(source_dir, "#{source}.html")
+    @current_file_stack.unshift(File.join(source_dir, "#{source}.haml"))
+    begin 
+      template_content = generate_html(source, source_dir)
+      @layout = determine_layout(source_dir)
+      generate_html_from_template_source(source_for_layout) { template_content }
+    rescue Exception => e
+      render_rescue_from_error(e)
+    ensure
+      @current_page = nil
+      @current_file_stack.shift
+    end
+  end
+
+  def generate_partial(name, options = {})
+    partial_dir, partial_name = File.dirname(self.current_file), name  # default relative to current file
+    partial_dir, partial_name = File.split(name) if name.index('/') # contains a path so it's absolute from src/pages dir
+    partial_name = "_#{partial_name}.haml"
+
+    partial_path = File.join(@src_dir, 'pages', partial_dir, partial_name)
+    unless File.exists?(partial_path)
+      # couldn't find it in the pages subdirectory tree so try old way (ignoring the path)
+      partial_dir = 'partials'
+      partial_name = "#{File.basename(name)}.haml"
+      partial_path = File.join(@src_dir, partial_dir, partial_name)
+    end
+  
+    if File.exists?(partial_path)
+      partial_rel_path = "/#{partial_dir}/#{partial_name}".gsub(/\/+/, '/')
+      @current_file_stack.unshift(partial_rel_path)
+      begin
+        generate_html_from_template_source(File.read(partial_path), options)
+      rescue Exception => e
+        raise StaticMatic::TemplateError.new(partial_path, e)
+      ensure
+        @current_file_stack.shift
+      end
+    else
+      raise StaticMatic::Error.new("", name, "Partial not found")
+    end
+  end
+
+  def generate_css(source, source_dir = '')
+    full_file_path = File.join(@src_dir, 'stylesheets', source_dir, "#{source}.sass")
+    begin
+      sass_options = { :load_paths => [ File.join(@src_dir, 'stylesheets') ] }.merge(self.configuration.sass_options)
+      stylesheet = Sass::Engine.new(File.read(full_file_path), sass_options)
+      stylesheet.to_css
+    rescue Exception => e
+      render_rescue_from_error(StaticMatic::TemplateError.new(full_file_path, e))
+    end
+  end
+
+  # Generates html from the passed source string
+  #
+  # generate_html_from_template_source("%h1 Welcome to My Site") -> "<h1>Welcome to My Site</h1>"
+  #
+  # Pass a block containing a string to yield within in the passed source:
+  #
+  # generate_html_from_template_source("content:\n= yield") { "blah" } -> "content: blah"
+  #
+  def generate_html_from_template_source(source, options = {})
+    html = Haml::Engine.new(source, self.configuration.haml_options.merge(options))
+    locals = options[:locals] || {}
+    html.render(@scope, locals) { yield }
+  end
+
+  def determine_layout(dir = '')
+    layout_name = "application"
+  
+    if @scope.instance_variable_get("@layout")
+      layout_name = @scope.instance_variable_get("@layout")
+    elsif dir
+      dirs = dir.split("/")
+      dir_layout_name = dirs[1]
+    
+      if layout_exists?(dir_layout_name)
+        layout_name = dir_layout_name
+      end
+    end
+
+    layout_name
+  end
+    
+  # Returns a raw template name from a source file path:
+  # source_template_from_path("/path/to/site/src/stylesheets/application.sass")  ->  "application"
+  def source_template_from_path(path)
+    file_dir, file_name = File.split(path)
+    file_name.chomp!(File.extname(file_name))
+    [ file_dir, file_name ]
+  end
+end

data/lib/staticmatic/mixins/rescue.rb

@@ -0,0 +1,12 @@
+module StaticMatic::RescueMixin
+  # Pass back an error template for the given exception
+  def render_rescue_from_error(exception)
+    rescue_template     = (exception.is_a?(StaticMatic::TemplateError)) ? "template" : "default"
+      
+    error_template_path = File.expand_path(File.dirname(__FILE__) + "/../templates/rescues/#{rescue_template}.haml")
+    
+    @scope.instance_variable_set("@exception", exception)
+    
+    generate_html_from_template_source(File.read(error_template_path))
+  end
+end

data/lib/staticmatic/mixins/server.rb

@@ -0,0 +1,6 @@
+module StaticMatic::ServerMixin
+  def preview
+    puts "StaticMatic Preview Server Starting..."
+    StaticMatic::Server.start(self)
+  end
+end

data/lib/staticmatic/mixins/setup.rb

@@ -0,0 +1,20 @@
+module StaticMatic::SetupMixin
+  
+  def setup
+    Dir.mkdir(@base_dir) unless File.exists?(@base_dir)
+  
+    StaticMatic::BASE_DIRS.each do |directory|
+      directory = "#{@base_dir}/#{directory}"
+      if !File.exists?(directory)
+        Dir.mkdir(directory)
+        puts "created #{directory}"
+      end
+    end
+
+    StaticMatic::TEMPLATES.each do |template, destination|
+      copy_file("#{@templates_dir}/#{template}", "#{@src_dir}/#{destination}")
+    end
+
+    puts "Done"
+  end
+end

data/lib/staticmatic/server.rb

@@ -0,0 +1,94 @@
+module StaticMatic
+  class Server < Mongrel::HttpHandler
+    @@file_only_methods = ["GET","HEAD"]
+ 
+    def initialize(staticmatic)
+      @files = Mongrel::DirHandler.new(staticmatic.site_dir, false)
+      @staticmatic = staticmatic
+    end
+  
+    def process(request, response)
+      @staticmatic.load_helpers
+      path_info = request.params[Mongrel::Const::PATH_INFO]
+      get_or_head = @@file_only_methods.include? request.params[Mongrel::Const::REQUEST_METHOD]
+      
+      file_dir, file_name, file_ext = expand_path(path_info)
+
+      # remove stylesheets/ directory if applicable
+      file_dir.gsub!(/^\/stylesheets\/?/, "")
+      
+      file_dir = CGI::unescape(file_dir)
+      file_name = CGI::unescape(file_name)
+      
+      if file_ext && file_ext.match(/html|css/)
+        response.start(200) do |head, out|
+          head["Content-Type"] = "text/#{file_ext}"
+          output = ""
+
+          if @staticmatic.template_exists?(file_name, file_dir) && !(File.basename(file_name) =~ /^\_/)
+
+            begin
+              if file_ext == "css"
+                output = @staticmatic.generate_css(file_name, file_dir)
+              else
+                output = @staticmatic.generate_html_with_layout(file_name, file_dir)
+              end
+            rescue StaticMatic::Error => e
+              output = e.message
+            end
+          else
+            if @files.can_serve(path_info)
+              @files.process(request,response)
+            else
+              output = "File not Found"
+            end
+          end
+          out.write output
+        end
+      else
+        # try to serve static file from site dir
+        if @files.can_serve(path_info)
+          @files.process(request,response)
+        end
+      end
+    end
+
+    def expand_path(path_info)
+      dirname, basename = File.split(path_info)
+
+      extname = File.extname(path_info).sub(/^\./, '')
+      filename = basename.chomp(".#{extname}")
+
+      if extname.empty?
+        dir = File.join(dirname, filename)
+        is_dir = path_info[-1, 1] == '/' || (@staticmatic.template_directory?(dir) && !@staticmatic.template_exists?(filename, dirname))
+        if is_dir
+          dirname = dir
+          filename = 'index'
+        end
+        extname = 'html'
+      end
+
+      [ dirname, filename, extname ]
+    end
+    
+    class << self
+      # Starts the StaticMatic preview server
+      def start(staticmatic)
+        port = staticmatic.configuration.preview_server_port || 3000
+        
+        host = staticmatic.configuration.preview_server_host || ""
+        
+        config = Mongrel::Configurator.new :host => host do
+          puts "Running Preview of #{staticmatic.base_dir} on #{host}:#{port}"
+          listener :port => port do
+            uri "/", :handler => Server.new(staticmatic)
+          end
+          trap("INT") { stop }
+          run
+        end
+        config.join
+      end
+    end
+  end
+end

data/lib/staticmatic/template_error.rb

@@ -0,0 +1,40 @@
+class StaticMatic::TemplateError < StandardError
+  SOURCE_CODE_RADIUS = 3
+  
+  attr_reader :original_exception, :backtrace
+
+  def initialize(template, original_exception)
+    @template, @original_exception = template, original_exception
+    @backtrace = original_exception.backtrace
+
+    @source = File.read(template)
+  end
+  
+  # TODO: Replace 'haml|sass' with any registered engines
+  def line_number
+    @line_number ||= $2 if backtrace.find { |line| line =~ /\((haml|sass)\)\:(\d+)/ }
+  end
+  
+  def filename
+    @template
+  end
+  
+  def source_extract(indentation = 0)
+    return unless num = line_number
+    num = num.to_i
+
+    source_code = @source.split("\n")
+
+    start_on_line = [ num - SOURCE_CODE_RADIUS - 1, 0 ].max
+    end_on_line   = [ num + SOURCE_CODE_RADIUS - 1, source_code.length].min
+
+    indent = ' ' * indentation
+    line_counter = start_on_line
+    return unless source_code = source_code[start_on_line..end_on_line]
+
+    source_code.collect do |line|
+      line_counter += 1
+      "#{indent}#{line_counter}: #{line}\n"
+    end.to_s
+  end
+end

data/lib/staticmatic/templates/default/application.haml

@@ -0,0 +1,7 @@
+!!!
+%html
+  %head
+    %title StaticMatic
+    = stylesheets
+  %body
+    = yield

data/lib/staticmatic/templates/default/application.sass

@@ -0,0 +1,4 @@
+body
+  :font
+    :family Verdana
+    :size 10pt

data/lib/staticmatic/templates/default/index.haml

@@ -0,0 +1 @@
+%h1 StaticMatic!

data/lib/staticmatic/templates/rescues/default.haml

@@ -0,0 +1,7 @@
+%h1= @exception.class.name
+
+%p= @exception.message
+
+%h2 Stack Trace
+
+= @exception.backtrace.join("<br/>")

data/lib/staticmatic/templates/rescues/template.haml

@@ -0,0 +1,18 @@
+%html
+  %head
+    %style{:type => "text/css"}
+      body { font-family: helvetica, verdana, arial; font-size: 10pt;}
+  %body
+    %h1
+      = @exception.class.name
+      in
+      = @exception.filename
+
+    %p= @exception.original_exception.message
+  
+    = @exception.source_extract.gsub(/\n/, "<br/>")
+
+
+    %h2 Backtrace
+
+    = @exception.backtrace.join("<br/>")

data/test/base_test.rb

@@ -0,0 +1,13 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+class StaticMatic::BaseTest < Test::Unit::TestCase
+  def setup
+    setup_staticmatic
+  end
+
+  should "set initial configuration settings" do
+    assert_equal true, @staticmatic.configuration.use_extensions_for_page_links
+    assert_equal 3000, @staticmatic.configuration.preview_server_port
+  end
+
+end

data/test/helpers_test.rb

@@ -0,0 +1,12 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+class StaticMatic::HelpersTest < Test::Unit::TestCase
+  def setup
+    setup_staticmatic
+  end
+  
+  should "include custom helper" do
+    content = @staticmatic.generate_html_with_layout("index")
+    assert_match "Hello, Steve!", content
+  end
+end