middleman--middleman/bin/mm-build

#!/usr/bin/env ruby

require 'templater'

ENV['MM_ENV'] = "build"

# Require app
require File.join(File.dirname(__FILE__), "..", "lib", "middleman")
require 'middleman/builder'

Middleman::Base.init!

module Generators
  extend Templater::Manifold
  desc "Build a staticmatic site"

  class Builder < Templater::Generator
    # Define source and desintation
    def self.source_root; Dir.pwd; end
    def destination_root; File.join(Dir.pwd, Middleman::Base.build_dir); end

    # Override template to ask middleman for the correct extension to output
    def self.template(name, *args, &block)
      return if args[0].include?('layout')
      
      args.first.split('/').each do |part|
        return if part[0,1] == '_'
      end
      
      if (args[0] === args[1])
        args[1] = args[0].gsub("#{File.basename(Middleman::Base.views)}/", "")
                         .gsub("#{File.basename(Middleman::Base.public)}/", "")
        if File.extname(args[1]) != ".js"
          args[1] = args[1].gsub!(File.extname(args[1]), "") if File.basename(args[1]).split('.').length > 2
        end
      end
      
      super(name, *args, &block)
    end

    def self.file(name, *args, &block)
      if (args[0] === args[1])
        args[1] = args[0].gsub("#{File.basename(Middleman::Base.views)}/", "")
                         .gsub("#{File.basename(Middleman::Base.public)}/", "")
      end
      super(name, *args, &block)
    end
    
    glob! File.basename(Middleman::Base.public), Middleman::Base.supported_formats
    glob! File.basename(Middleman::Base.views),  Middleman::Base.supported_formats
  end

  add :build, Builder
end

# Monkey-patch to use a dynamic renderer
class Templater::Actions::Template
  def render
    ::Middleman::Builder.render_file(source, destination)
  end
end

Generators.run_cli(Dir.pwd, 'mm-build', 1,  %w(build --force).concat(ARGV))