module Lookbook::CodeFormatter

def beautify(source, language = "html")
  source = source.strip
  result = language.downcase == "html" ? HtmlBeautifier.beautify(source) : source
  result.strip.html_safe
end