Methods to set content type and content

src/amber/controller/render.cr

@@ -37,5 +37,43 @@ module Amber::Controller
         Kilt.render("#{{{path}}}/{{filename.id}}")
       {% end %}
     end
+
+    protected def respond_with(html : String? = nil, json : Hash | String? = nil, xml : String? = nil, text : String? = nil)
+      accepts = context.request.headers["Accept"].split(";").try(&.split(/,|,\s/))
+
+      if accepts.includes?("text/html") && html
+        respond_with_html(html)
+      elsif accepts.includes?("application/json") && json
+        respond_with_json(json.is_a?(Hash) ? json.to_json : json)
+      elsif accepts.includes?("application/xml") && xml
+        respond_with_xml(xml)
+      elsif accepts.includes?("text/plain") && text
+        respond_with_text(text)
+      else
+        respond_with_text("Response not acceptable", 406)
+      end
+    end
+
+    protected def respond_with_html(body, status_code = 200)
+      set_response(body, status_code, "text/html")
+    end
+
+    protected def respond_with_text(body, status_code = 200)
+      set_response(body, status_code, "text/plain")
+    end
+
+    protected def respond_with_json(body, status_code = 200)
+      set_response(body, status_code, "application/json")
+    end
+
+    protected def respond_with_xml(body, status_code = 200)
+      set_response(body, status_code, "application/xml")
+    end
+
+    private def set_response(body, status_code = 200, content_type = "text/html")
+      context.response.status_code = status_code
+      context.response.content_type = content_type 
+      context.content = body
+    end
   end
 end