Tag tab

1 files changed, 89 insertions, 2 deletions

diff --git a/main.go b/main.go
index a89a096..29b08d3 100644
--- a/main.go
+++ b/main.go
@@ -367,6 +367,92 @@ func handleAPIBrowse(w http.ResponseWriter, r *http.Request) {
 
 // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
 
+type webTagNamespace struct {
+	Description string           `json:"description"`
+	Tags        map[string]int64 `json:"tags"`
+}
+
+func getTags(nsID int64) (result map[string]int64, err error) {
+	rows, err := db.Query(`
+		SELECT t.name, COUNT(ta.tag) AS count
+		FROM tag AS t
+		LEFT JOIN tag_assignment AS ta ON t.id = ta.tag
+		WHERE t.space = ?
+		GROUP BY t.id`, nsID)
+	if err != nil {
+		return
+	}
+	defer rows.Close()
+
+	result = make(map[string]int64)
+	for rows.Next() {
+		var (
+			name  string
+			count int64
+		)
+		if err = rows.Scan(&name, &count); err != nil {
+			return
+		}
+		result[name] = count
+	}
+	return result, rows.Err()
+}
+
+func getTagNamespaces(match *string) (
+	result map[string]webTagNamespace, err error) {
+	var rows *sql.Rows
+	if match != nil {
+		rows, err = db.Query(`SELECT id, name, IFNULL(description, '')
+			FROM tag_space WHERE name = ?`, *match)
+	} else {
+		rows, err = db.Query(`SELECT id, name, IFNULL(description, '')
+			FROM tag_space`)
+	}
+	if err != nil {
+		return
+	}
+	defer rows.Close()
+
+	result = make(map[string]webTagNamespace)
+	for rows.Next() {
+		var (
+			id   int64
+			name string
+			ns   webTagNamespace
+		)
+		if err = rows.Scan(&id, &name, &ns.Description); err != nil {
+			return
+		}
+		if ns.Tags, err = getTags(id); err != nil {
+			return
+		}
+		result[name] = ns
+	}
+	return result, rows.Err()
+}
+
+func handleAPITags(w http.ResponseWriter, r *http.Request) {
+	var params struct {
+		Namespace *string
+	}
+	if err := json.NewDecoder(r.Body).Decode(&params); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	result, err := getTagNamespaces(params.Namespace)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	if err := json.NewEncoder(w).Encode(result); err != nil {
+		log.Println(err)
+	}
+}
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
 func getImageDimensions(sha1 string) (w int64, h int64, err error) {
 	err = db.QueryRow(`SELECT width, height FROM image WHERE sha1 = ?`,
 		sha1).Scan(&w, &h)
@@ -832,10 +918,11 @@ func cmdWeb(args []string) error {
 	http.HandleFunc("/image/", handleImages)
 	http.HandleFunc("/thumb/", handleThumbs)
 	http.HandleFunc("/api/browse", handleAPIBrowse)
-	http.HandleFunc("/api/info", handleAPIInfo)
-	http.HandleFunc("/api/similar", handleAPISimilar)
+	http.HandleFunc("/api/tags", handleAPITags)
 	http.HandleFunc("/api/duplicates", handleAPIDuplicates)
 	http.HandleFunc("/api/orphans", handleAPIOrphans)
+	http.HandleFunc("/api/info", handleAPIInfo)
+	http.HandleFunc("/api/similar", handleAPISimilar)
 
 	host, port, err := net.SplitHostPort(address)
 	if err != nil {