Reader link tracking

1 files changed, 35 insertions, 2 deletions

diff --git a/src/java/com/juick/http/www/Reader.java b/src/java/com/juick/http/www/Reader.java
index dc356482b..75a0e7745 100644
--- a/src/java/com/juick/http/www/Reader.java
+++ b/src/java/com/juick/http/www/Reader.java
@@ -54,11 +54,11 @@ public class Reader {
             PreparedStatement stmt = null;
             ResultSet rs = null;
             try {
-                stmt = sql.prepareStatement("SELECT rss_id,url,title FROM reader_links ORDER BY ts DESC LIMIT 100");
+                stmt = sql.prepareStatement("SELECT link_id,rss_id,url,title FROM reader_links ORDER BY ts DESC LIMIT 100");
                 rs = stmt.executeQuery();
                 rs.beforeFirst();
                 while (rs.next()) {
-                    out.println("<li><img src=\"//i.juick.com/rss-icons/" + rs.getInt(1) + ".png\" width=\"16\" height=\"16\"/><a href=\"" + rs.getString(2) + "\" rel=\"nofollow\" target=\"_blank\">" + rs.getString(3) + "</a></li>");
+                    out.println("<li><img src=\"//i.juick.com/rss-icons/" + rs.getInt(2) + ".png\" width=\"16\" height=\"16\"/><a href=\"" + rs.getString(3) + "\" data-lid=\""+rs.getInt(1)+"\" onmousedown=\"readerLinkReplace(this)\" rel=\"nofollow\" target=\"_blank\">" + rs.getString(4) + "</a></li>");
                 }
             } catch (SQLException e) {
                 System.err.println(e);
@@ -75,4 +75,37 @@ public class Reader {
             out.close();
         }
     }
+
+    protected void doGetOut(Connection sql, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
+        com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
+        int lid = Utils.parseInt(request.getParameter("lid"), 0);
+        if (lid == 0) {
+            response.sendError(404);
+            return;
+        }
+
+        String url = null;
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT url FROM reader_links WHERE link_id=?");
+            stmt.setInt(1, lid);
+            rs = stmt.executeQuery();
+            if (rs.first()) {
+                url = rs.getString(1);
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+
+        if (url == null) {
+            response.sendError(404);
+            return;
+        }
+
+        Utils.sendTemporaryRedirect(response, url);
+    }
 }