From 5022d1a0df1534b1cbec1cbee23568f201ea1cc6 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@sbcglobal.net>
Date: Mon, 9 Jan 2017 16:45:53 -0500
Subject: Implement actual file generation, go through fixing things.

---
 lib/sitegen.rb | 49 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

(limited to 'lib/sitegen.rb')

diff --git a/lib/sitegen.rb b/lib/sitegen.rb
index c6e89e9..765cfbd 100644
--- a/lib/sitegen.rb
+++ b/lib/sitegen.rb
@@ -1,5 +1,6 @@
 # coding: utf-8
 require 'date'
+require 'fileutils'
 require 'set'
 
 module Sitegen
@@ -60,6 +61,52 @@ module Sitegen
 	end
 
 	def self.generate(target)
-		# TODO
+		case
+		when @mk[target].nil?
+			raise "No rule to make target '#{target}'.  Stop."
+		when target.end_with?(".atom")
+			write_ifchanged(target) do |file|
+				file.puts('<?xml version="1.0" encoding="utf-8"?>')
+				file.print(@mk[target].atom)
+			end
+		when target.end_with?(".html")
+			write_ifchanged(target) do |file|
+				file.print(@mk[target].html)
+			end
+		else
+			raise "No rule to make target '#{target}'.  Stop."
+		end
+	end
+
+	def self.write_ifchanged(outfilename)
+		tmpfilename = "#{File::dirname(outfilename)}/.tmp#{File::basename(outfilename)}"
+
+		# Write our stuff to tmpfile
+		FileUtils::mkdir_p(File::dirname(tmpfilename))
+		tmpfile = File::new(tmpfilename, 'wb')
+		begin
+			yield tmpfile
+		rescue Exception => e
+			tmpfile.close
+			File::unlink(tmpfilename)
+			raise e
+		end
+		tmpfile.close
+
+		# Now see if we should replace outfile with tmpfile
+		same = false
+		begin
+			if FileUtils::compare_file(tmpfilename, outfilename)
+				same = true
+			end
+		rescue Errno::ENOENT
+		end
+
+		# And actually do so
+		if same
+			File::unlink(tmpfilename)
+		else
+			File::rename(tmpfilename, outfilename)
+		end
 	end
 end
-- 
cgit v1.2.3-2-g168b