add playlist support

server/ctrladmin/ctrl.go

@@ -97,6 +97,7 @@ type templateData struct {
 	AllUsers      []*model.User
 	LastScanTime  time.Time
 	IsScanning    bool
+	Playlists     []*model.Playlist
 	//
 	CurrentLastFMAPIKey    string
 	CurrentLastFMAPISecret string

server/ctrladmin/handlers.go

@@ -59,6 +59,19 @@ func (c *Controller) ServeHome(r *http.Request) *Response {
 		data.LastScanTime = time.Unix(i, 0)
 	}
 	//
+	// playlists box
+	user := r.Context().Value(key.User).(*model.User)
+	c.DB.
+		Select("*, count(items.id) as track_count").
+		Joins(`
+            LEFT JOIN playlist_items items
+		    ON items.playlist_id = playlists.id
+		`).
+		Where("user_id = ?", user.ID).
+		Group("playlists.id").
+		Limit(20).
+		Find(&data.Playlists)
+	//
 	return &Response{
 		template: "home.tmpl",
 		data:     data,
@@ -257,6 +270,36 @@ func (c *Controller) ServeStartScanDo(r *http.Request) *Response {
 	}()
 	return &Response{
 		redirect: "/admin/home",
-		flashN:   "scan started. refresh for results",
+		flashN:   []string{"scan started. refresh for results"},
+	}
+}
+
+func (c *Controller) ServeUploadPlaylist(r *http.Request) *Response {
+	return &Response{template: "upload_playlist.tmpl"}
+}
+
+func (c *Controller) ServeUploadPlaylistDo(r *http.Request) *Response {
+	if err := r.ParseMultipartForm((1 << 10) * 24); nil != err {
+		return &Response{
+			err:  "couldn't parse mutlipart",
+			code: 500,
+		}
+	}
+	user := r.Context().Value(key.User).(*model.User)
+	var playlistCount int
+	var errors []string
+	for _, headers := range r.MultipartForm.File {
+		for _, header := range headers {
+			headerErrors, created := playlistParseUpload(c, user.ID, header)
+			if created {
+				playlistCount++
+			}
+			errors = append(errors, headerErrors...)
+		}
+	}
+	return &Response{
+		redirect: "/admin/home",
+		flashN:   []string{fmt.Sprintf("%d playlist(s) created", playlistCount)},
+		flashW:   errors,
 	}
 }

server/ctrladmin/playlist.go

@@ -0,0 +1,66 @@
+package ctrladmin
+
+import (
+	"bufio"
+	"fmt"
+	"mime/multipart"
+	"strings"
+
+	"github.com/jinzhu/gorm"
+	"github.com/pkg/errors"
+
+	"senan.xyz/g/gonic/model"
+)
+
+func playlistParseLine(c *Controller, playlistID int, path string) error {
+	if strings.HasPrefix(path, "#") || strings.TrimSpace(path) == "" {
+		return nil
+	}
+	track := &model.Track{}
+	query := c.DB.Raw(`
+		SELECT tracks.id FROM TRACKS
+		JOIN albums ON tracks.album_id = albums.id
+		WHERE (? || '/' || albums.left_path || albums.right_path || '/' || tracks.filename) = ?
+	`, c.MusicPath, path)
+	err := query.First(&track).Error
+	switch {
+	case gorm.IsRecordNotFoundError(err):
+		return fmt.Errorf("couldn't match track %q", path)
+	case err != nil:
+		return errors.Wrap(err, "while matching")
+	}
+	c.DB.Create(&model.PlaylistItem{
+		PlaylistID: playlistID,
+		TrackID:    track.ID,
+	})
+	return nil
+}
+
+func playlistParseUpload(c *Controller, userID int, header *multipart.FileHeader) ([]string, bool) {
+	file, err := header.Open()
+	if err != nil {
+		return []string{fmt.Sprintf("couldn't open file %q", header.Filename)}, false
+	}
+	playlistName := strings.TrimSuffix(header.Filename, ".m3u8")
+	if playlistName == "" {
+		return []string{fmt.Sprintf("invalid filename %q", header.Filename)}, false
+	}
+	playlist := &model.Playlist{}
+	c.DB.FirstOrCreate(playlist, model.Playlist{
+		Name:   playlistName,
+		UserID: userID,
+	})
+	c.DB.Delete(&model.PlaylistItem{}, "playlist_id = ?", playlist.ID)
+	var errors []string
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		path := scanner.Text()
+		if err := playlistParseLine(c, playlist.ID, path); err != nil {
+			errors = append(errors, err.Error())
+		}
+	}
+	if err := scanner.Err(); err != nil {
+		return []string{fmt.Sprintf("scanning line of playlist: %v", err)}, true
+	}
+	return errors, true
+}