discogs basic backend

cmd/serve/main.go

@@ -7,6 +7,7 @@ import (
 	"log"
 	"net/http"
 	"strings"
+	"time"
 
 	"git.yetaga.in/alazyreader/library/config"
 	"git.yetaga.in/alazyreader/library/database"
@@ -22,24 +23,44 @@ func max(a, b int) int {
 	return b
 }
 
+func obscureStr(in string, l int) string {
+	return in[0:max(l, len(in))] + strings.Repeat("*", max(0, len(in)-l))
+}
+
 type Library interface {
 	GetAllBooks(context.Context) ([]media.Book, error)
 }
 
+type RecordCollection interface {
+	GetAllRecords(context.Context) ([]media.Record, error)
+}
+
 type Router struct {
 	static fs.FS
 	lib    Library
+	rcol   RecordCollection
+}
+
+func writeJSON(w http.ResponseWriter, b []byte, status int) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.WriteHeader(status)
+	w.Write(b)
+	w.Write([]byte("\n"))
 }
 
 func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
-	if req.URL.Path == "/api" {
-		APIHandler(r.lib).ServeHTTP(w, req)
+	if req.URL.Path == "/api/records" {
+		RecordsAPIHandler(r.rcol).ServeHTTP(w, req)
+		return
+	}
+	if req.URL.Path == "/api/books" {
+		BooksAPIHandler(r.lib).ServeHTTP(w, req)
 		return
 	}
 	StaticHandler(r.static).ServeHTTP(w, req)
 }
 
-func APIHandler(l Library) http.Handler {
+func BooksAPIHandler(l Library) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		books, err := l.GetAllBooks(r.Context())
 		if err != nil {
@@ -50,11 +71,25 @@ func APIHandler(l Library) http.Handler {
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 			return
+
 		}
-		w.Header().Set("Content-Type", "application/json")
-		w.WriteHeader(http.StatusOK)
-		w.Write(b)
-		w.Write([]byte("\n"))
+		writeJSON(w, b, http.StatusOK)
+	})
+}
+
+func RecordsAPIHandler(l RecordCollection) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		books, err := l.GetAllRecords(r.Context())
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+		b, err := json.Marshal(books)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+		writeJSON(w, b, http.StatusOK)
 	})
 }
 
@@ -73,8 +108,11 @@ func main() {
 		log.Fatalln(err)
 	}
 	if c.DBUser == "" || c.DBPass == "" || c.DBHost == "" || c.DBPort == "" || c.DBName == "" {
-		if c.DBPass != "" { // obscure password
-			c.DBPass = c.DBPass[0:max(3, len(c.DBPass))] + strings.Repeat("*", max(0, len(c.DBPass)-3))
+		if c.DBPass != "" {
+			c.DBPass = obscureStr(c.DBPass, 3)
+		}
+		if c.Discogs_Token != "" {
+			c.Discogs_Token = obscureStr(c.Discogs_Token, 3)
 		}
 		log.Fatalf("vars: %+v", c)
 	}
@@ -91,10 +129,16 @@ func main() {
 		log.Fatalln(err)
 	}
 	log.Printf("latest migration: %d; migrations run: %d", latest, run)
+	discogsCache, err := database.NewDiscogsCache(c.Discogs_Token, time.Hour*24, "delta.mu.alpha")
+	if err != nil {
+		log.Fatalln(err)
+	}
+	go discogsCache.FlushCache(context.Background())
 	r := &Router{
 		static: f,
 		lib:    lib,
+		rcol:   discogsCache,
 	}
-	log.Println("Listening on http://localhost:8080/")
+	log.Println("Listening on http://0.0.0.0:8080/")
 	log.Fatalln(http.ListenAndServe(":8080", r))
 }