edison 0.0.1

data/bin/edison

@@ -0,0 +1,262 @@
+#!/usr/bin/env ruby
+
+require 'fileutils'
+require 'listen'
+require 'kramdown'
+require 'hashie'
+require 'mustache'
+require 'pp'
+require 'webrick'
+require 'yaml'
+
+module Edison
+end
+
+module Edison::YAMLFrontMatter
+	def self.read(fname)
+		contents = File.read(fname).strip
+		if contents =~ /\A---/
+			_, yaml, body = contents.split("---", 3)
+			yaml = YAML.load(yaml)
+			body ||= ""
+			body.strip!
+			if yaml.include? "body"
+				raise Exception, "YAML Front Matter can't contain a 'body' key. Put it after the front matter."
+			end
+			{"body" => body}.merge(yaml)
+		else
+			{"body" => contents}
+		end
+	end
+end
+
+class Edison::ModelsLoader
+	def load(directory)
+		klasses = Dir[File.join(directory, "*")].map &File.method(:basename)
+		hash = Hashie::Mash.new
+		klasses.each do |klass|
+			objects = Dir[File.join(directory, klass, "*")]
+			objects.map! do |fname|
+				puts "Loading model #{File.join(klass, File.basename(fname))}..."
+
+				data = Hashie::Mash.new(case File.extname(fname)
+				when /(html)|(md)$/
+					Edison::YAMLFrontMatter.read(fname)
+				when /ya?ml$/
+					YAML.load_file(fname)
+				else
+					{}
+				end)
+
+				if data._ext
+					raise Exception, "Restricted key '_ext' appears in model!"
+				end
+				data._ext = File.extname(fname)
+
+				if data._ext == ".md"
+					data.body = Kramdown::Document.new(data.body).to_html
+				end
+
+				if data._fname
+					raise Exception, "Restricted key '_fname' appears in model!"
+				end
+				data._fname = File.basename(fname).sub(/#{File.extname(fname)}$/,'')
+
+				data
+			end
+			hash[klass] = objects
+		end
+		hash
+	end
+end
+
+class Edison::Renderer
+	attr_accessor :templates
+	def initialize(templates)
+		self.templates = templates
+	end
+	def render(template_name, data)
+		template = self.templates[template_name]
+		if template.nil?
+			raise Exception, "Called for template #{template_name}, but _templates/#{template_name} does not exist"
+		end
+		body = Mustache.render(template.body, data)
+		if template.layout
+			newdata = data.merge("yield" => body)
+			self.render(template.layout, newdata)
+		else
+			body
+		end
+	end
+end
+
+class Edison::Router
+	attr_reader :urls, :static
+	def initialize
+		@urls = []
+		@static = []
+	end
+	def url(url, template=nil, data={}, &b)
+		if b and template and data
+			raise Exception, "Please pass data or block, not both!"
+		end
+		if b
+			b.call(@urls.find { |(url, _, _)| url == url}[2])
+		else
+			@urls << [url, template, Hashie::Mash.new(data)]
+		end
+	end
+	def copy(url)
+		@static << url
+	end
+end
+
+module Edison
+	class <<self
+		attr_reader :models, :routes, :renderer
+	end
+	def self.initialize!(directory)
+		@directory = directory
+
+		@routes = Router.new
+
+		loader = ModelsLoader.new
+		@models = loader.load(File.join(directory, "_models"))
+
+		templates = Hash[Dir[File.join(directory, "_templates", "*")].map do |fname|
+			puts "Loading template _templates/#{File.basename(fname)}..."
+			name = File.basename(fname).sub(/\.[^\.]+$/,'')
+			data = Hashie::Mash.new Edison::YAMLFrontMatter.read(fname)
+			[name, data]
+		end]
+
+		static = Dir[File.join(directory, "**/*")]
+		blacklist = ["Gemfile", "Gemfile.lock", "config.rb", "gen.rb", "serve.rb"]
+		static.reject! do |fname|
+			name = fname.sub(/^#{directory}\//,'')
+			dirname = File.dirname(fname)
+			blacklist.include?(name) or name =~ /^_/ or File.directory?(fname)
+		end
+		static.each do |fname|
+			name = fname.sub(/^#{directory}\//,'')
+			if %w{.html .md}.include?(File.extname(fname))
+				data = Hashie::Mash.new Edison::YAMLFrontMatter.read(fname)
+				url = name.sub(/\.md$/,'')
+				templates[url] = data
+				@routes.url url, url, data
+			else
+				@routes.copy name
+			end
+		end
+
+		@renderer = Renderer.new(templates)
+	end
+	def self.config(&b)
+		instance_eval &b
+	end
+	def self.generate!
+		site = File.join(@directory, "_site")
+
+		FileUtils.rm_rf(site)
+		Dir.mkdir(site)
+		puts "Copying static files..."
+		routes.static.each do |name|
+			src = File.join(@directory, name)
+			dest = File.join(site, name)
+			FileUtils.mkdir_p File.dirname(dest)
+			FileUtils.copy src, dest
+		end
+		routes.urls.each do |(url, template_name, data)|
+			puts "Creating /#{url}..."
+
+			fname = File.join(site, url)
+			FileUtils.mkdir_p(File.dirname(fname))
+			File.open(fname, "w") do |f|
+				f.write @renderer.render(template_name, data)
+			end
+		end
+	end
+end
+
+module Edison::Helpers
+	def self.date_from_filename(data)
+		if data.date
+			raise Exception, "Date will be inferred from filename, but found in data"
+		end
+
+		if File.basename(data._fname) =~ /^(\d{4}-\d{1,2}-\d{1,2})/
+			data.date = Date.parse($1)
+		else
+			raise Exception, "Expected filename to start with date (YYYY-M?M-D?D)"
+		end
+
+		data
+	end
+
+	def self.default_layout(data)
+		unless data.layout
+			data.layout = "default"
+		end
+
+		data
+	end
+end
+
+def build
+	directory = Dir.pwd
+	puts "Running in #{directory}..."
+	Edison.initialize!(directory)
+	load File.join(directory, "config.rb")
+	site = File.join(directory, "_site")
+	Edison.generate!
+	puts "Done!"
+end
+
+require 'commander/import'
+
+program :name, "Edison"
+program :version, "0.0.1"
+program :description, "Static website generator"
+
+default_command :help
+
+command :up do |c|
+	c.description = "Build static files"
+	c.action do |args, options|
+		build
+	end
+end
+command :serve do |c|
+	c.description = "Auto-generating webserver"
+	c.action do |args, options|
+		directory = Dir.pwd
+
+		begin
+			build
+		rescue => e
+			pp e
+		end
+
+		listener = Listen.to(directory, :ignore => /^_site\//)
+		server = WEBrick::HTTPServer.new(
+			:Port => 4000,
+			:DocumentRoot => File.expand_path("_site", directory)
+		)
+		listener.change do |*args|
+			t = Time.now.strftime("%Y-%m-%d %H:%M:%S")
+			puts "Regenerating at #{t}"
+			begin
+				build
+			rescue => e
+				pp e
+			end
+		end
+		listener.start(false)
+
+		Signal.trap("INT") do
+			server.shutdown
+			listener.stop
+		end
+		server.start
+	end
+end

metadata

@@ -0,0 +1,135 @@
+--- !ruby/object:Gem::Specification 
+name: edison
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Michael Maltese
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-03-17 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: commander
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: listen
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: hashie
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: kramdown
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: mustache
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id005
+description: ""
+email: michael.maltese@pomona.edu
+executables: 
+- edison
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/edison
+- README.md
+homepage: http://github.com/michaelmaltese/jefe
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- - lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: A static (website) generator.
+test_files: []
+