Add importer tool to import books directly from harddisk

tools/importer/search.go

@@ -0,0 +1,85 @@
+package main
+
+import (
+	"labix.org/v2/mgo/bson"
+	"net/http"
+	"strconv"
+	"strings"
+)
+
+func buildQuery(q string) bson.M {
+	var reg []bson.RegEx
+	query := bson.M{"active": true}
+	words := strings.Split(q, " ")
+	for _, w := range words {
+		tag := strings.SplitN(w, ":", 2)
+		if len(tag) > 1 {
+			query[tag[0]] = bson.RegEx{tag[1], "i"}
+		} else {
+			reg = append(reg, bson.RegEx{w, "i"})
+		}
+	}
+	if len(reg) > 0 {
+		query["keywords"] = bson.M{"$all": reg}
+	}
+	return query
+}
+
+type searchData struct {
+	S         Status
+	Found     int
+	Books     []Book
+	ItemsPage int
+	Page      int
+	Next      string
+	Prev      string
+}
+
+func searchHandler(h handler) {
+	err := h.r.ParseForm()
+	if err != nil {
+		http.Error(h.w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	req := strings.Join(h.r.Form["q"], " ")
+	page := 0
+	if len(h.r.Form["p"]) != 0 {
+		page, err = strconv.Atoi(h.r.Form["p"][0])
+		if err != nil {
+			page = 0
+		}
+	}
+	items_page := itemsPage(h.r)
+	res, num, _ := h.db.GetBooks(buildQuery(req), items_page, page*items_page)
+
+	var data searchData
+	data.S = GetStatus(h)
+	data.S.Search = req
+	data.Books = res
+	data.ItemsPage = items_page
+	data.Found = num
+	data.Page = page + 1
+	if num > (page+1)*items_page {
+		data.Next = "/search/?q=" + req + "&p=" + strconv.Itoa(page+1) + "&num=" + strconv.Itoa(items_page)
+	}
+	if page > 0 {
+		data.Prev = "/search/?q=" + req + "&p=" + strconv.Itoa(page-1) + "&num=" + strconv.Itoa(items_page)
+	}
+
+	format := h.r.Form["fmt"]
+	if (len(format) > 0) && (format[0] == "rss") {
+		loadTxtTemplate(h.w, "search_rss.xml", data)
+	} else {
+		loadTemplate(h.w, "search", data)
+	}
+}
+
+func itemsPage(r *http.Request) int {
+	if len(r.Form["num"]) > 0 {
+		items_page, err := strconv.Atoi(r.Form["num"][0])
+		if err == nil {
+			return items_page
+		}
+	}
+	return SEARCH_ITEMS_PAGE
+}