Add TOP-100 playlist

handlers.go

@@ -2,7 +2,9 @@ package main
 
 import (
 	"fmt"
+	"log"
 	"net/http"
+	"net/url"
 )
 
 func handleGenres(w http.ResponseWriter, r *http.Request) {
@@ -26,12 +28,13 @@ func handleGenres(w http.ResponseWriter, r *http.Request) {
 
 func handleGetM3u(w http.ResponseWriter, r *http.Request) {
 	genreCode := r.FormValue("genre")
-	if genreCode == "" {
-		http.Error(w, `"genre" field is not specified`, http.StatusBadRequest)
-		return
+	params := url.Values{}
+
+	if r.FormValue("top100") != "" {
+		params.Set("top100", "1")
 	}
 
-	tracks, err := tracksByGenre(genreCode)
+	tracks, err := tracksByGenre(genreCode, params)
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -39,6 +42,10 @@ func handleGetM3u(w http.ResponseWriter, r *http.Request) {
 
 	b := tracksToM3u(r.Host, tracks)
 
+	if genreCode == "" {
+		genreCode = "music"
+	}
+
 	w.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s.m3u8"`, genreCode))
 	w.Header().Set("Content-Type", "audio/x-mpegurl")
 	w.Header().Set("Accept-Ranges", "none")
@@ -47,5 +54,8 @@ func handleGetM3u(w http.ResponseWriter, r *http.Request) {
 }
 
 func handleStream(w http.ResponseWriter, r *http.Request) {
-	stream(r.FormValue("url"), w)
+	err := stream(r.FormValue("url"), w)
+	if err != nil {
+		log.Println(err)
+	}
 }

parser.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"fmt"
 	"net/url"
+	"strconv"
 	"strings"
 
 	"github.com/PuerkitoBio/goquery"
@@ -52,11 +53,19 @@ func updateGenreList() ([]Genre, error) {
 }
 
 // tracksByGenre возвращает список треков по указанному жанру
-func tracksByGenre(genre string) ([]TrackInfo, error) {
+func tracksByGenre(genre string, params url.Values) ([]TrackInfo, error) {
+	if params == nil {
+		params = url.Values{"download": []string{"1"}}
+	} else {
+		params.Set("download", "1") // only available tracks
+	}
+
 	var result []TrackInfo
 
 	for i := 1; i <= 50; i++ {
-		url := fmt.Sprintf("https://promodj.com/music/%s?download=1&page=%d", genre, i)
+		params.Set("page", strconv.Itoa(i))
+		//url := fmt.Sprintf("https://promodj.com/music/%s?download=1&page=%d", genre, i)
+		url := constructUrl(genre, params)
 
 		doc, err := goquery.NewDocument(url)
 		if err != nil {
@@ -118,3 +127,21 @@ func removeDuplicate(strSlice []TrackInfo) []TrackInfo {
 	}
 	return list
 }
+
+func constructUrl(genre string, params url.Values) string {
+	urlTemplate := fmt.Sprintf("https://promodj.com/music/%s", genre)
+
+	if genre == "" {
+		urlTemplate = "https://promodj.com/music"
+	}
+
+	u, err := url.Parse(urlTemplate)
+	if err != nil {
+		panic(err)
+	}
+
+	u.RawQuery = params.Encode()
+
+	//?download=1&page=%d
+	return u.String()
+}

parser_test.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConstructURL(t *testing.T) {
+	tests := []struct {
+		params   url.Values
+		expected string
+	}{
+		{nil, "https://promodj.com/music/testGenre"},
+		{url.Values{"download": []string{"1"}}, "https://promodj.com/music/testGenre?download=1"},
+		{url.Values{"download": []string{"1"}, "page": []string{"1"}}, "https://promodj.com/music/testGenre?download=1&page=1"},
+	}
+
+	for _, test := range tests {
+		assert.Equal(t, test.expected, constructUrl("testGenre", test.params))
+	}
+}

templates/index.htm

@@ -15,6 +15,7 @@
 		<main>
 			<h1>Список жанров</h1>
 			<ul id="genres-list">
+				<li><a href="/getm3u?top100=1">TOP 100</a></li>
 				{{range .Genres}}
 				<li><a href="/getm3u?genre={{.Code}}">{{.Name}}</a></li>
 				{{end}}