From 41d2634304185a58e87403923efa928dfd0e08cb Mon Sep 17 00:00:00 2001
From: albert <r888888888@gmail.com>
Date: Fri, 11 Nov 2011 17:50:26 -0500
Subject: [PATCH] added post atom feed

---
 app/controllers/posts_controller.rb |  4 +++-
 app/views/posts/index.atom.erb      | 35 +++++++++++++++++++++++++++++
 config/danbooru_default_config.rb   |  2 +-
 3 files changed, 39 insertions(+), 2 deletions(-)
 create mode 100644 app/views/posts/index.atom.erb

diff --git a/app/controllers/posts_controller.rb b/app/controllers/posts_controller.rb
index d77f77a67..9c1e4c073 100644
--- a/app/controllers/posts_controller.rb
+++ b/app/controllers/posts_controller.rb
@@ -6,7 +6,9 @@ class PostsController < ApplicationController
   def index
     @post_set = PostSets::Post.new(tag_query, params[:page])
     @posts = @post_set.posts
-    respond_with(@posts)
+    respond_with(@posts) do |format|
+      format.atom
+    end
   end
   
   def show
diff --git a/app/views/posts/index.atom.erb b/app/views/posts/index.atom.erb
new file mode 100644
index 000000000..13248a37e
--- /dev/null
+++ b/app/views/posts/index.atom.erb
@@ -0,0 +1,35 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<feed xmlns="http://www.w3.org/2005/Atom">
+	<title><%= Danbooru.config.app_name %></title>
+	<link href="http://<%= Danbooru.config.hostname %>/post/atom" rel="self"/>
+  <link href="http://<%= Danbooru.config.hostname %>/post/index" rel="alternate"/>
+  <id>http://<%= Danbooru.config.hostname %>/posts.atom?tags=<%= params[:tags] %></id>
+  <% if @posts.any? %>
+    <updated><%= @posts[0].created_at.gmtime.xmlschema %></updated>
+  <% end %>
+  <author><name><%= Danbooru.config.app_name %></name></author>
+
+  <% Danbooru.config.select_posts_visible_to_user(CurrentUser.user, @posts).each do |post| %>
+    <entry>
+      <title><%= post.tag_string %></title>
+      <link href="http://<%= Danbooru.config.hostname %>/posts/<%= post.id %>" rel="alternate"/>
+      <% if post.source =~ /^http/ %>
+        <link href="<%= post.source %>" rel="related"/>
+      <% end %>
+      <id>http://<%= Danbooru.config.hostname %>/posts/<%= post.id %></id>
+      <updated><%= post.created_at.gmtime.xmlschema %></updated>
+      <summary><%= post.tag_string %></summary>
+      <content type="xhtml">
+        <div xmlns="http://www.w3.org/1999/xhtml">
+          <a href="http://<%= Danbooru.config.hostname %>/posts/<%= post.id %>">
+            <img src="http://<%= Danbooru.config.hostname %><%= post.preview_file_url %>"/>
+          </a>
+        </div>
+      </content>
+      <author>
+        <name><%= post.uploader_name %></name>
+      </author>
+    </entry>
+  <% end %>
+</feed>
diff --git a/config/danbooru_default_config.rb b/config/danbooru_default_config.rb
index fe41f1b10..6eab69e1f 100644
--- a/config/danbooru_default_config.rb
+++ b/config/danbooru_default_config.rb
@@ -232,7 +232,7 @@ module Danbooru
     end
     
     def select_posts_visible_to_user(user, posts)
-      posts.select {|x| can_user_see_post?(x)}
+      posts.select {|x| can_user_see_post?(user, x)}
     end
     
     def pixiv_login