import java.io.IOException; import java.util.Collection; import java.util.Iterator; import java.util.regex.Pattern; import net.yacy.cora.document.UTF8; import net.yacy.cora.protocol.RequestHeader; import net.yacy.data.UserDB; import net.yacy.data.ymark.YMarkCrawlStart; import net.yacy.data.ymark.YMarkDate; import net.yacy.data.ymark.YMarkEntry; import net.yacy.data.ymark.YMarkTables; import net.yacy.data.ymark.YMarkUtil; import net.yacy.data.ymark.YMarkTables.TABLES; import net.yacy.kelondro.blob.Tables; import net.yacy.kelondro.blob.Tables.Row; import net.yacy.kelondro.logging.Log; import net.yacy.search.Switchboard; import net.yacy.server.serverObjects; import net.yacy.server.serverSwitch; public class get_ymark { private static Switchboard sb = null; private static serverObjects prop = null; final static String FOLDER_IMG = ""; public static serverObjects respond(final RequestHeader header, final serverObjects post, final serverSwitch env) { sb = (Switchboard) env; prop = new serverObjects(); int rp; // items per page int page; // page int total; String sortorder; String sortname; String qtype; String query; final UserDB.Entry user = sb.userDB.getUser(header); final boolean isAdmin = (sb.verifyAuthentication(header)); final boolean isAuthUser = user!= null && user.hasRight(UserDB.AccessRight.BOOKMARK_RIGHT); Iterator bookmarks = null; if(isAdmin || isAuthUser) { final String bmk_user = (isAuthUser ? user.getUserName() : YMarkTables.USER_ADMIN); query = ".*"; qtype = YMarkEntry.BOOKMARK.TITLE.key(); page = 1; rp = 100; total = 0; sortname = YMarkEntry.BOOKMARK.TITLE.key(); sortorder = "asc"; if(post != null) { rp = (post.containsKey("rp")) ? post.getInt("rp", 10) : 10; page = (post.containsKey("page")) ? post.getInt("page", 1): 1; query = (post.containsKey("query")) ? post.get("query", ".*") : ".*"; qtype = (post.containsKey("qtype")) ? post.get("qtype", YMarkEntry.BOOKMARK.TAGS.key()) : YMarkEntry.BOOKMARK.TAGS.key(); sortname = (post.containsKey("sortname")) ? post.get("sortname", YMarkEntry.BOOKMARK.TITLE.key()) : YMarkEntry.BOOKMARK.TITLE.key(); sortorder = (post.containsKey("sortorder")) ? post.get("sortorder", "asc") : "asc"; } try { final String bmk_table = TABLES.BOOKMARKS.tablename(bmk_user); final Collection result; if(!query.isEmpty()) { if(!qtype.isEmpty()) { if(qtype.equals("_tags")) { final String tags = YMarkUtil.cleanTagsString(query); result = sb.tables.bookmarks.orderBookmarksBy(sb.tables.bookmarks.getBookmarksByTag(bmk_user, tags), sortname, sortorder); } else if(qtype.equals("_folder")) { result = sb.tables.bookmarks.orderBookmarksBy(sb.tables.bookmarks.getBookmarksByFolder(bmk_user, query), sortname, sortorder); } else { result = sb.tables.bookmarks.orderBookmarksBy(sb.tables.iterator(bmk_table, qtype, Pattern.compile(query)), sortname, sortorder); } } else { result = sb.tables.bookmarks.orderBookmarksBy(sb.tables.iterator(bmk_table, Pattern.compile(query)), sortname, sortorder); } } else { result = sb.tables.bookmarks.orderBookmarksBy(sb.tables.iterator(bmk_table), sortname, sortorder); } total = result.size(); bookmarks = result.iterator(); } catch (final IOException e) { Log.logException(e); } prop.put("page", page); prop.put("total", total); putProp(bookmarks, rp, page); } else { prop.put(serverObjects.ACTION_AUTHENTICATE, YMarkTables.USER_AUTHENTICATE_MSG); } // return rewrite properties return prop; } private static void putProp(final Iterator bit, final int rp, final int page) { Tables.Row bmk_row; int count = 0; int offset = 0; if (page > 1) { offset = ((page - 1) * rp) + 1; } while(count < offset && bit.hasNext()) { bmk_row = bit.next(); count++; } count = 0; while (count < rp && bit.hasNext()) { bmk_row = bit.next(); if (bmk_row != null) { // put JSON prop.put("json_"+count+"_id", count); prop.put("json_"+count+"_hash", UTF8.String(bmk_row.getPK())); for (final YMarkEntry.BOOKMARK bmk : YMarkEntry.BOOKMARK.values()) { if(bmk == YMarkEntry.BOOKMARK.PUBLIC) prop.put("json_"+count+"_"+bmk.key(), bmk_row.get(bmk.key(),bmk.deflt()).equals("false") ? 1 : 0); else if(bmk == YMarkEntry.BOOKMARK.TAGS) prop.putJSON("json_"+count+"_"+bmk.key(), bmk_row.get(bmk.key(),bmk.deflt()).replaceAll(YMarkUtil.TAGS_SEPARATOR, ", ")); else if(bmk == YMarkEntry.BOOKMARK.FOLDERS) prop.putJSON("json_"+count+"_"+bmk.key(), bmk_row.get(bmk.key(),bmk.deflt()).replaceAll(YMarkUtil.TAGS_SEPARATOR, "
"+FOLDER_IMG)); else if(bmk == YMarkEntry.BOOKMARK.DATE_ADDED || bmk == YMarkEntry.BOOKMARK.DATE_MODIFIED || bmk == YMarkEntry.BOOKMARK.DATE_VISITED) prop.putJSON("json_"+count+"_"+bmk.key(), (new YMarkDate(bmk_row.get(bmk.key()))).toISO8601().replaceAll("T", "
")); else prop.putJSON("json_"+count+"_"+bmk.key(), bmk_row.get(bmk.key(),bmk.deflt())); } final YMarkCrawlStart crawlstart = new YMarkCrawlStart(sb.tables, bmk_row.get(YMarkEntry.BOOKMARK.URL.key(),YMarkEntry.BOOKMARK.URL.deflt())); int crawl = 0; if (!crawlstart.isEmpty()) { crawl = 1; prop.put("json_"+count+"_crawlstart_info", "Crawl last executed: "+YMarkDate.ISO8601(crawlstart.date_last_exec())); } if (crawlstart.hasSchedule()) { crawl = 2; prop.put("json_"+count+"_crawlstart_info", "Crawl scheduled: "+YMarkDate.ISO8601(crawlstart.date_next_exec())); } if (crawlstart.isRunning(sb.crawler)) { crawl = 3; prop.put("json_"+count+"_crawlstart_info", "Crawl is running ..."); } prop.put("json_"+count+"_crawlstart", crawl); prop.put("json_"+count+"_apicall_pk", crawlstart.getPK()); prop.put("json_"+count+"_date_recording", YMarkDate.ISO8601(crawlstart.date_recording()).replaceAll("T", "
")); prop.put("json_"+count+"_date_next_exec", YMarkDate.ISO8601(crawlstart.date_next_exec()).replaceAll("T", "
")); prop.put("json_"+count+"_date_last_exec", YMarkDate.ISO8601(crawlstart.date_last_exec()).replaceAll("T", "
")); prop.put("json_"+count+"_comma", ","); // put XML prop.putXML("xml_"+count+"_id", UTF8.String(bmk_row.getPK())); for (final YMarkEntry.BOOKMARK bmk : YMarkEntry.BOOKMARK.values()) { prop.putXML("xml_"+count+"_"+bmk.key(), bmk_row.get(bmk.key(),bmk.deflt())); } count++; } } // eliminate the trailing comma for Json output prop.put("json_" + (count - 1) + "_comma", ""); prop.put("json", count); prop.put("xml", count); } }