refactoring everything teehee (i'm so glad this isn't a team project)
Signed-off-by: ari melody <ari@arimelody.me>
This commit is contained in:
parent
c684f0c7ae
commit
10f5f51e76
17 changed files with 970 additions and 547 deletions
12
api/api.go
12
api/api.go
|
|
@ -1,12 +0,0 @@
|
|||
package api
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"html/template"
|
||||
)
|
||||
|
||||
func Handle(writer http.ResponseWriter, req *http.Request, root *template.Template) int {
|
||||
writer.WriteHeader(501)
|
||||
writer.Write([]byte("501 Not Implemented"))
|
||||
return 501;
|
||||
}
|
||||
|
|
@ -24,14 +24,12 @@ type (
|
|||
const TOKEN_LENGTH = 64
|
||||
const TOKEN_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||
|
||||
// TODO: consider relying *entirely* on env vars instead of hard-coded fallbacks
|
||||
var ADMIN_ID_DISCORD = func() string {
|
||||
envvar := os.Getenv("DISCORD_ADMIN_ID")
|
||||
if envvar != "" {
|
||||
return envvar
|
||||
} else {
|
||||
return "356210742200107009"
|
||||
if envvar == "" {
|
||||
fmt.Printf("DISCORD_ADMIN_ID was not provided. Admin login will be unavailable.\n")
|
||||
}
|
||||
return envvar
|
||||
}()
|
||||
|
||||
var sessions []*Session
|
||||
|
|
@ -48,30 +46,30 @@ func Handler() http.Handler {
|
|||
mux := http.NewServeMux()
|
||||
|
||||
mux.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(200)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("hello /admin!"))
|
||||
}))
|
||||
mux.Handle("/callback", global.HTTPLog(OAuthCallbackHandler()))
|
||||
mux.Handle("/login", global.HTTPLog(LoginHandler()))
|
||||
mux.Handle("/verify", global.HTTPLog(AuthorisedHandler(VerifyHandler())))
|
||||
mux.Handle("/logout", global.HTTPLog(AuthorisedHandler(LogoutHandler())))
|
||||
mux.Handle("/verify", global.HTTPLog(MustAuthorise(VerifyHandler())))
|
||||
mux.Handle("/logout", global.HTTPLog(MustAuthorise(LogoutHandler())))
|
||||
|
||||
return mux
|
||||
}
|
||||
|
||||
func AuthorisedHandler(next http.Handler) http.Handler {
|
||||
func MustAuthorise(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
auth := r.Header.Get("Authorization")
|
||||
if auth == "" || !strings.HasPrefix(auth, "Bearer ") {
|
||||
if strings.HasPrefix(auth, "Bearer ") {
|
||||
auth = auth[7:]
|
||||
} else {
|
||||
cookie, err := r.Cookie("token")
|
||||
if err != nil {
|
||||
w.WriteHeader(401)
|
||||
w.Write([]byte("Unauthorized"))
|
||||
http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
auth = cookie.Value
|
||||
}
|
||||
auth = auth[7:]
|
||||
|
||||
var session *Session
|
||||
for _, s := range sessions {
|
||||
|
|
@ -86,6 +84,7 @@ func AuthorisedHandler(next http.Handler) http.Handler {
|
|||
}
|
||||
continue
|
||||
}
|
||||
|
||||
if s.Token == auth {
|
||||
session = s
|
||||
break
|
||||
|
|
@ -93,46 +92,47 @@ func AuthorisedHandler(next http.Handler) http.Handler {
|
|||
}
|
||||
|
||||
if session == nil {
|
||||
w.WriteHeader(401)
|
||||
w.Write([]byte("Unauthorized"))
|
||||
http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := context.WithValue(r.Context(), "token", session.Token)
|
||||
ctx = context.WithValue(ctx, "role", "admin")
|
||||
next.ServeHTTP(w, r.WithContext(ctx))
|
||||
})
|
||||
}
|
||||
|
||||
func LoginHandler() http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if ADMIN_ID_DISCORD == "" {
|
||||
http.Error(w, http.StatusText(http.StatusServiceUnavailable), http.StatusServiceUnavailable)
|
||||
return
|
||||
}
|
||||
|
||||
code := r.URL.Query().Get("code")
|
||||
|
||||
if code == "" {
|
||||
w.Header().Add("Location", discord.REDIRECT_URI)
|
||||
w.WriteHeader(307)
|
||||
http.Redirect(w, r, discord.REDIRECT_URI, http.StatusTemporaryRedirect)
|
||||
return
|
||||
}
|
||||
|
||||
auth_token, err := discord.GetOAuthTokenFromCode(code)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to retrieve discord access token: %s\n", err)
|
||||
w.WriteHeader(500)
|
||||
w.Write([]byte("Internal server error"))
|
||||
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
discord_user, err := discord.GetDiscordUserFromAuth(auth_token)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to retrieve discord user information: %s\n", err)
|
||||
w.WriteHeader(500)
|
||||
w.Write([]byte("Internal server error"))
|
||||
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
if discord_user.Id != ADMIN_ID_DISCORD {
|
||||
// TODO: unauthorized user. revoke the token
|
||||
w.WriteHeader(401)
|
||||
w.Write([]byte("Unauthorized"))
|
||||
// TODO: unauthorized user; revoke the token
|
||||
http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
|
|
@ -149,7 +149,7 @@ func LoginHandler() http.Handler {
|
|||
cookie.Path = "/"
|
||||
http.SetCookie(w, &cookie)
|
||||
|
||||
w.WriteHeader(200)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(session.Token))
|
||||
})
|
||||
}
|
||||
|
|
@ -159,7 +159,7 @@ func LogoutHandler() http.Handler {
|
|||
token := r.Context().Value("token").(string)
|
||||
|
||||
if token == "" {
|
||||
w.WriteHeader(401)
|
||||
http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
|
|
@ -172,7 +172,8 @@ func LogoutHandler() http.Handler {
|
|||
return new_sessions
|
||||
}(token)
|
||||
|
||||
w.WriteHeader(200)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("OK"))
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -185,7 +186,8 @@ func VerifyHandler() http.Handler {
|
|||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
// this is an authorised endpoint, so you *must* supply a valid token
|
||||
// before accessing this route.
|
||||
w.WriteHeader(200)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("OK"))
|
||||
})
|
||||
}
|
||||
|
||||
|
|
|
|||
95
api/v1/music/artist.go
Normal file
95
api/v1/music/artist.go
Normal file
|
|
@ -0,0 +1,95 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type Artist struct {
|
||||
id string
|
||||
name string
|
||||
website string
|
||||
}
|
||||
|
||||
var Artists []Artist
|
||||
|
||||
func GetArtist(id string) *Artist {
|
||||
for _, artist := range Artists {
|
||||
if artist.GetID() == id {
|
||||
return &artist
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// GETTERS
|
||||
|
||||
func (artist Artist) GetID() string {
|
||||
return artist.id
|
||||
}
|
||||
|
||||
func (artist Artist) GetName() string {
|
||||
return artist.name
|
||||
}
|
||||
|
||||
func (artist Artist) GetWebsite() string {
|
||||
return artist.website
|
||||
}
|
||||
|
||||
// SETTERS
|
||||
|
||||
func (artist Artist) SetID(id string) error {
|
||||
artist.id = id
|
||||
return nil
|
||||
}
|
||||
|
||||
func (artist Artist) SetName(name string) error {
|
||||
artist.name = name
|
||||
return nil
|
||||
}
|
||||
|
||||
func (artist Artist) SetWebsite(website string) error {
|
||||
artist.website = website
|
||||
return nil
|
||||
}
|
||||
|
||||
// DATABASE
|
||||
|
||||
func (artist Artist) PushToDB(db *sqlx.DB) {
|
||||
fmt.Printf("Pushing artist [%s] to database...", artist.name)
|
||||
|
||||
db.MustExec("INSERT INTO artists (id, name, website) VALUES ($1, $2, $3) ON CONFLICT (id) DO UPDATE SET name=$2, website=$3",
|
||||
artist.id,
|
||||
artist.name,
|
||||
artist.website,
|
||||
)
|
||||
|
||||
fmt.Printf("done!\n")
|
||||
}
|
||||
|
||||
func PullAllArtists(db *sqlx.DB) ([]Artist, error) {
|
||||
artists := []Artist{}
|
||||
|
||||
rows, err := db.Query("SELECT id, name, website FROM artists")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for rows.Next() {
|
||||
var artist = Artist{}
|
||||
|
||||
err = rows.Scan(
|
||||
&artist.id,
|
||||
&artist.name,
|
||||
&artist.website,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
artists = append(artists, artist)
|
||||
}
|
||||
|
||||
return artists, nil
|
||||
}
|
||||
84
api/v1/music/credit.go
Normal file
84
api/v1/music/credit.go
Normal file
|
|
@ -0,0 +1,84 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type Credit struct {
|
||||
artist *Artist
|
||||
role string
|
||||
primary bool
|
||||
}
|
||||
|
||||
// GETTERS
|
||||
|
||||
func (credit Credit) GetArtist() Artist {
|
||||
return *credit.artist
|
||||
}
|
||||
|
||||
func (credit Credit) GetRole() string {
|
||||
return credit.role
|
||||
}
|
||||
|
||||
func (credit Credit) IsPrimary() bool {
|
||||
return credit.primary
|
||||
}
|
||||
|
||||
// SETTERS
|
||||
|
||||
func (credit Credit) SetArtist(artist *Artist) error {
|
||||
// TODO: update DB
|
||||
credit.artist = artist
|
||||
return nil
|
||||
}
|
||||
|
||||
func (credit Credit) SetRole(role string) error {
|
||||
// TODO: update DB
|
||||
credit.role = role
|
||||
return nil
|
||||
}
|
||||
|
||||
func (credit Credit) SetPrimary(primary bool) error {
|
||||
// TODO: update DB
|
||||
credit.primary = primary
|
||||
return nil
|
||||
}
|
||||
|
||||
// DATABASE
|
||||
|
||||
func PullReleaseCredits(db *sqlx.DB, releaseID string) ([]Credit, error) {
|
||||
var credits = []Credit{}
|
||||
|
||||
credit_rows, err := db.Query("SELECT artist, role, is_primary FROM musiccredits WHERE release=$1", releaseID)
|
||||
if err != nil {
|
||||
return []Credit{}, err
|
||||
}
|
||||
|
||||
for credit_rows.Next() {
|
||||
var artistID string
|
||||
var credit = Credit{}
|
||||
err = credit_rows.Scan(
|
||||
&artistID,
|
||||
&credit.role,
|
||||
&credit.primary,
|
||||
)
|
||||
if err != nil {
|
||||
fmt.Printf("Error while pulling credit for release %s: %s\n", releaseID, err)
|
||||
continue
|
||||
}
|
||||
|
||||
credit.artist = GetArtist(artistID)
|
||||
if credit.artist == nil {
|
||||
// this should absolutely not happen ever due to foreign key
|
||||
// constraints, but it doesn't hurt to be sure!
|
||||
fmt.Printf("Error while pulling credit for release %s: Artist %s does not exist\n", releaseID, artistID)
|
||||
continue
|
||||
}
|
||||
|
||||
credits = append(credits, credit)
|
||||
}
|
||||
|
||||
return credits, nil
|
||||
}
|
||||
73
api/v1/music/link.go
Normal file
73
api/v1/music/link.go
Normal file
|
|
@ -0,0 +1,73 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type Link struct {
|
||||
name string
|
||||
url string
|
||||
}
|
||||
|
||||
// GETTERS
|
||||
|
||||
func (link Link) GetName() string {
|
||||
return link.name
|
||||
}
|
||||
|
||||
func (link Link) GetURL() string {
|
||||
return link.url
|
||||
}
|
||||
|
||||
// SETTERS
|
||||
|
||||
func (link Link) SetName(name string) error {
|
||||
// TODO: update DB
|
||||
link.name = name
|
||||
return nil
|
||||
}
|
||||
|
||||
func (link Link) SetURL(url string) error {
|
||||
// TODO: update DB
|
||||
link.url = url
|
||||
return nil
|
||||
}
|
||||
|
||||
// MISC
|
||||
|
||||
func (link Link) NormaliseName() string {
|
||||
rgx := regexp.MustCompile(`[^a-z0-9]`)
|
||||
return strings.ToLower(rgx.ReplaceAllString(link.name, ""))
|
||||
}
|
||||
|
||||
// DATABASE
|
||||
|
||||
func PullReleaseLinks(db *sqlx.DB, releaseID string) ([]Link, error) {
|
||||
var links = []Link{}
|
||||
|
||||
link_rows, err := db.Query("SELECT name, url FROM musiclinks WHERE release=$1", releaseID);
|
||||
if err != nil {
|
||||
return []Link{}, err
|
||||
}
|
||||
|
||||
for link_rows.Next() {
|
||||
var link = Link{}
|
||||
|
||||
err = link_rows.Scan(
|
||||
&link.name,
|
||||
&link.url,
|
||||
)
|
||||
if err != nil {
|
||||
fmt.Printf("Error while pulling link for release %s: %s\n", releaseID, err)
|
||||
continue
|
||||
}
|
||||
|
||||
links = append(links, link)
|
||||
}
|
||||
|
||||
return links, nil
|
||||
}
|
||||
|
|
@ -1,37 +1,62 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"arimelody.me/arimelody.me/api/v1/admin"
|
||||
"arimelody.me/arimelody.me/global"
|
||||
)
|
||||
|
||||
var Releases []*MusicRelease;
|
||||
var Artists []*Artist;
|
||||
// func make_date_work(date string) time.Time {
|
||||
// res, err := time.Parse("2-Jan-2006", date)
|
||||
// if err != nil {
|
||||
// fmt.Printf("somehow we failed to parse %s! falling back to epoch :]\n", date)
|
||||
// return time.Unix(0, 0)
|
||||
// }
|
||||
// return res
|
||||
// }
|
||||
|
||||
func make_date_work(date string) time.Time {
|
||||
res, err := time.Parse("2-Jan-2006", date)
|
||||
if err != nil {
|
||||
fmt.Printf("somehow we failed to parse %s! falling back to epoch :]\n", date)
|
||||
return time.Unix(0, 0)
|
||||
}
|
||||
return res
|
||||
}
|
||||
// HTTP HANDLER METHODS
|
||||
|
||||
func GetRelease(id string) (*MusicRelease, error) {
|
||||
for _, release := range Releases {
|
||||
if release.Id == id {
|
||||
return release, nil
|
||||
func ServeCatalog() http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
releases := []Release{}
|
||||
authorised := r.Context().Value("role") != nil && r.Context().Value("role") == "admin"
|
||||
for _, release := range Releases {
|
||||
if !release.IsReleased() && !authorised {
|
||||
continue
|
||||
}
|
||||
releases = append(releases, release)
|
||||
}
|
||||
}
|
||||
return nil, errors.New(fmt.Sprintf("Release %s not found", id))
|
||||
|
||||
global.ServeTemplate("music.html", releases).ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func GetArtist(id string) (*Artist, error) {
|
||||
for _, artist := range Artists {
|
||||
if artist.Id == id {
|
||||
return artist, nil
|
||||
func ServeGateway() http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
id := r.URL.Path[1:]
|
||||
release := GetRelease(id)
|
||||
if release == nil {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
}
|
||||
return nil, errors.New(fmt.Sprintf("Artist %s not found", id))
|
||||
|
||||
// only allow authorised users to view unreleased releases
|
||||
authorised := r.Context().Value("role") != nil && r.Context().Value("role") == "admin"
|
||||
if !release.IsReleased() && !authorised {
|
||||
admin.MustAuthorise(ServeGateway()).ServeHTTP(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
lrw := global.LoggingResponseWriter{w, http.StatusOK}
|
||||
|
||||
global.ServeTemplate("music-gateway.html", release).ServeHTTP(&lrw, r)
|
||||
|
||||
if lrw.Code != http.StatusOK {
|
||||
fmt.Printf("Error loading music gateway for %s\n", id)
|
||||
return
|
||||
}
|
||||
})
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,133 +0,0 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type (
|
||||
Artist struct {
|
||||
Id string
|
||||
Name string
|
||||
Website string
|
||||
}
|
||||
|
||||
MusicRelease struct {
|
||||
Id string
|
||||
Title string
|
||||
Type string
|
||||
ReleaseDate time.Time
|
||||
Artwork string
|
||||
Buyname string
|
||||
Buylink string
|
||||
Links []MusicLink
|
||||
Description string
|
||||
Credits []MusicCredit
|
||||
Tracks []MusicTrack
|
||||
}
|
||||
|
||||
MusicLink struct {
|
||||
Name string
|
||||
Url string
|
||||
}
|
||||
|
||||
MusicCredit struct {
|
||||
Artist *Artist
|
||||
Role string
|
||||
Primary bool
|
||||
}
|
||||
|
||||
MusicTrack struct {
|
||||
Number int
|
||||
Title string
|
||||
Description string
|
||||
Lyrics string
|
||||
PreviewUrl string
|
||||
}
|
||||
)
|
||||
|
||||
func (release MusicRelease) GetUniqueArtists(include_non_primary bool) []*Artist {
|
||||
res := []*Artist{}
|
||||
for _, credit := range release.Credits {
|
||||
if !include_non_primary && !credit.Primary {
|
||||
continue
|
||||
}
|
||||
|
||||
exists := false
|
||||
for _, a := range res {
|
||||
if a == credit.Artist {
|
||||
exists = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if exists {
|
||||
continue
|
||||
}
|
||||
|
||||
res = append(res, credit.Artist)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func (release MusicRelease) GetUniqueArtistNames(include_non_primary bool) []string {
|
||||
artists := release.GetUniqueArtists(include_non_primary)
|
||||
names := []string{}
|
||||
for _, artist := range artists {
|
||||
names = append(names, artist.Name)
|
||||
}
|
||||
|
||||
return names
|
||||
}
|
||||
|
||||
func (release MusicRelease) PrintPrimaryArtists(include_non_primary bool, ampersand bool) string {
|
||||
names := release.GetUniqueArtistNames(include_non_primary)
|
||||
if len(names) == 0 {
|
||||
return "Unknown Artist"
|
||||
} else if len(names) == 1 {
|
||||
return names[0]
|
||||
}
|
||||
if ampersand {
|
||||
res := strings.Join(names[:len(names)-1], ", ")
|
||||
res += " & " + names[len(names)-1]
|
||||
return res
|
||||
} else {
|
||||
return strings.Join(names[:], ", ")
|
||||
}
|
||||
}
|
||||
|
||||
func (release MusicRelease) ResolveType() string {
|
||||
if release.Type != "" {
|
||||
return release.Type
|
||||
}
|
||||
return "unknown"
|
||||
}
|
||||
|
||||
func (release MusicRelease) ResolveArtwork() string {
|
||||
if release.Artwork != "" {
|
||||
return release.Artwork
|
||||
}
|
||||
return "/img/music-artwork/default.png"
|
||||
}
|
||||
|
||||
func (release MusicRelease) PrintReleaseDate() string {
|
||||
return release.ReleaseDate.Format("02 January 2006")
|
||||
}
|
||||
|
||||
func (release MusicRelease) GetReleaseYear() int {
|
||||
return release.ReleaseDate.Year()
|
||||
}
|
||||
|
||||
func (link MusicLink) NormaliseName() string {
|
||||
re := regexp.MustCompile(`[^a-z0-9]`)
|
||||
return strings.ToLower(re.ReplaceAllString(link.Name, ""))
|
||||
}
|
||||
|
||||
func (release MusicRelease) IsSingle() bool {
|
||||
return len(release.Tracks) == 1;
|
||||
}
|
||||
|
||||
func (credit MusicCredit) ResolveArtist() Artist {
|
||||
return *credit.Artist
|
||||
}
|
||||
368
api/v1/music/release.go
Normal file
368
api/v1/music/release.go
Normal file
|
|
@ -0,0 +1,368 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type ReleaseType string
|
||||
|
||||
const (
|
||||
Single ReleaseType = "Single"
|
||||
Album ReleaseType = "Album"
|
||||
EP ReleaseType = "EP"
|
||||
Compilation ReleaseType = "Compilation"
|
||||
)
|
||||
|
||||
type Release struct {
|
||||
id string
|
||||
title string
|
||||
releaseType ReleaseType
|
||||
releaseDate time.Time
|
||||
artwork string
|
||||
buyname string
|
||||
buylink string
|
||||
links []Link
|
||||
description string
|
||||
credits []Credit
|
||||
tracks []Track
|
||||
}
|
||||
|
||||
var Releases []Release;
|
||||
|
||||
// GETTERS
|
||||
|
||||
func (release Release) GetID() string {
|
||||
return release.id
|
||||
}
|
||||
|
||||
func (release Release) GetTitle() string {
|
||||
return release.title
|
||||
}
|
||||
|
||||
func (release Release) GetType() ReleaseType {
|
||||
return release.releaseType
|
||||
}
|
||||
|
||||
func (release Release) GetReleaseDate() time.Time {
|
||||
return release.releaseDate
|
||||
}
|
||||
|
||||
func (release Release) GetArtwork() string {
|
||||
if release.artwork == "" {
|
||||
return "/img/music-artwork/default.png"
|
||||
}
|
||||
return release.artwork
|
||||
}
|
||||
|
||||
func (release Release) GetBuyName() string {
|
||||
return release.buyname
|
||||
}
|
||||
|
||||
func (release Release) GetBuyLink() string {
|
||||
return release.buylink
|
||||
}
|
||||
|
||||
func (release Release) GetLinks() []Link {
|
||||
return release.links
|
||||
}
|
||||
|
||||
func (release Release) GetDescription() string {
|
||||
return release.description
|
||||
}
|
||||
|
||||
func (release Release) GetCredits() []Credit {
|
||||
return release.credits
|
||||
}
|
||||
|
||||
func (release Release) GetTracks() []Track {
|
||||
return release.tracks
|
||||
}
|
||||
|
||||
// SETTERS
|
||||
|
||||
func (release Release) SetID(id string) error {
|
||||
// TODO: update DB
|
||||
release.id = id
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetTitle(title string) error {
|
||||
// TODO: update DB
|
||||
release.title = title
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetType(releaseType ReleaseType) error {
|
||||
// TODO: update DB
|
||||
release.releaseType = releaseType
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetReleaseDate(releaseDate time.Time) error {
|
||||
// TODO: update DB
|
||||
release.releaseDate = releaseDate
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetArtwork(artwork string) error {
|
||||
// TODO: update DB
|
||||
release.artwork = artwork
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetBuyName(buyname string) error {
|
||||
// TODO: update DB
|
||||
release.buyname = buyname
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetBuyLink(buylink string) error {
|
||||
// TODO: update DB
|
||||
release.buylink = buylink
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetLinks(links []Link) error {
|
||||
// TODO: update DB
|
||||
release.links = links
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetDescription(description string) error {
|
||||
// TODO: update DB
|
||||
release.description = description
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetCredits(credits []Credit) error {
|
||||
// TODO: update DB
|
||||
release.credits = credits
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) SetTracks(tracks []Track) error {
|
||||
// TODO: update DB
|
||||
release.tracks = tracks
|
||||
return nil
|
||||
}
|
||||
|
||||
// MISC
|
||||
|
||||
func GetRelease(id string) *Release {
|
||||
for _, release := range Releases {
|
||||
if release.GetID() == id {
|
||||
return &release
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) PrintReleaseDate() string {
|
||||
return release.releaseDate.Format("02 January 2006")
|
||||
}
|
||||
|
||||
func (release Release) GetReleaseYear() int {
|
||||
return release.releaseDate.Year()
|
||||
}
|
||||
|
||||
func (release Release) IsSingle() bool {
|
||||
return len(release.tracks) == 1;
|
||||
}
|
||||
|
||||
func (release Release) IsReleased() bool {
|
||||
return release.releaseDate.Before(time.Now())
|
||||
}
|
||||
|
||||
func (release Release) GetUniqueArtists(only_primary bool) []Artist {
|
||||
var artists = []Artist{}
|
||||
|
||||
for _, credit := range release.credits {
|
||||
if only_primary && !credit.primary {
|
||||
continue
|
||||
}
|
||||
|
||||
exists := false
|
||||
for _, a := range artists {
|
||||
if a.id == credit.artist.id {
|
||||
exists = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if exists {
|
||||
continue
|
||||
}
|
||||
|
||||
artists = append(artists, *credit.artist)
|
||||
}
|
||||
|
||||
return artists
|
||||
}
|
||||
|
||||
func (release Release) GetUniqueArtistNames(only_primary bool) []string {
|
||||
var names = []string{}
|
||||
for _, artist := range release.GetUniqueArtists(only_primary) {
|
||||
names = append(names, artist.GetName())
|
||||
}
|
||||
|
||||
return names
|
||||
}
|
||||
|
||||
func (release Release) PrintArtists(only_primary bool, ampersand bool) string {
|
||||
names := release.GetUniqueArtistNames(only_primary)
|
||||
|
||||
if len(names) == 0 {
|
||||
return "Unknown Artist"
|
||||
} else if len(names) == 1 {
|
||||
return names[0]
|
||||
}
|
||||
|
||||
if ampersand {
|
||||
res := strings.Join(names[:len(names)-1], ", ")
|
||||
res += " & " + names[len(names)-1]
|
||||
return res
|
||||
} else {
|
||||
return strings.Join(names[:], ", ")
|
||||
}
|
||||
}
|
||||
|
||||
// DATABASE
|
||||
|
||||
func (release Release) PushToDB(db *sqlx.DB) error {
|
||||
// fmt.Printf("Pushing release [%s] to database...", release.id)
|
||||
|
||||
tx, err := db.Begin()
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to initiate transaction: %s", err))
|
||||
}
|
||||
|
||||
_, err = tx.Exec("INSERT INTO musicreleases (id, title, type, release_date, artwork, buyname, buylink) VALUES ($1, $2, $3, $4, $5, $6, $7) "+
|
||||
"ON CONFLICT (id) DO UPDATE SET title=$2, type=$3, release_date=$4, artwork=$5, buyname=$6, buylink=$7",
|
||||
release.id, release.title, release.releaseType, release.releaseDate.Format("2-Jan-2006"), release.artwork, release.buyname, release.buylink)
|
||||
|
||||
for _, link := range release.links {
|
||||
_, err = tx.Exec(
|
||||
"INSERT INTO musiclinks (release, name, url) "+
|
||||
"VALUES ($1, $2, $3) "+
|
||||
"ON CONFLICT (release, name) "+
|
||||
"DO UPDATE SET url=$3 ",
|
||||
release.id,
|
||||
link.name,
|
||||
link.url,
|
||||
)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to add music link to transaction: %s", err))
|
||||
}
|
||||
}
|
||||
for _, credit := range release.credits {
|
||||
_, err = tx.Exec(
|
||||
"INSERT INTO musiccredits (release, artist, role, is_primary) "+
|
||||
"VALUES ($1, $2, $3, $4) "+
|
||||
"ON CONFLICT (release, artist) "+
|
||||
"DO UPDATE SET role=$3, is_primary=$4",
|
||||
release.id,
|
||||
credit.artist.id,
|
||||
credit.role,
|
||||
credit.primary,
|
||||
)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to add music credit to transaction: %s", err))
|
||||
}
|
||||
}
|
||||
for _, track := range release.tracks {
|
||||
_, err = tx.Exec(
|
||||
"INSERT INTO musictracks (release, number, title, description, lyrics, preview_url) "+
|
||||
"VALUES ($1, $2, $3, $4, $5, $6) "+
|
||||
"ON CONFLICT (release, number) "+
|
||||
"DO UPDATE SET title=$3, description=$4, lyrics=$5, preview_url=$6",
|
||||
release.id,
|
||||
track.number,
|
||||
track.title,
|
||||
track.description,
|
||||
track.lyrics,
|
||||
track.previewURL,
|
||||
)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to add music track to transaction: %s", err))
|
||||
}
|
||||
}
|
||||
|
||||
err = tx.Commit()
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to commit transaction: %s", err))
|
||||
}
|
||||
|
||||
// fmt.Printf("done!\n")
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (release Release) DeleteFromDB(db *sqlx.DB) error {
|
||||
// this probably doesn't need to be a transaction;
|
||||
// i just felt like making it one
|
||||
tx, err := db.Begin()
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to initiate transaction: %s", err))
|
||||
}
|
||||
|
||||
_, err = tx.Exec("DELETE FROM musicreleases WHERE id=$1", release.id)
|
||||
|
||||
err = tx.Commit()
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to commit transaction: %s", err))
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func PullAllReleases(db *sqlx.DB) ([]Release, error) {
|
||||
releases := []Release{}
|
||||
|
||||
rows, err := db.Query("SELECT id, title, description, type, release_date, artwork, buyname, buylink FROM musicreleases")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for rows.Next() {
|
||||
var release = Release{}
|
||||
|
||||
err = rows.Scan(
|
||||
&release.id,
|
||||
&release.title,
|
||||
&release.description,
|
||||
&release.releaseType,
|
||||
&release.releaseDate,
|
||||
&release.artwork,
|
||||
&release.buyname,
|
||||
&release.buylink,
|
||||
)
|
||||
if err != nil {
|
||||
fmt.Printf("Error while pulling a release: %s\n", err)
|
||||
continue
|
||||
}
|
||||
|
||||
release.credits, err = PullReleaseCredits(db, release.id)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to pull credits for %s: %v\n", release.id, err)
|
||||
}
|
||||
|
||||
release.links, err = PullReleaseLinks(db, release.id)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to pull links for %s: %v\n", release.id, err)
|
||||
}
|
||||
|
||||
release.tracks, err = PullReleaseTracks(db, release.id)
|
||||
if err != nil {
|
||||
return nil, errors.New(fmt.Sprintf("error pulling tracks for %s: %v\n", release.id, err))
|
||||
}
|
||||
|
||||
releases = append(releases, release)
|
||||
}
|
||||
|
||||
return releases, nil
|
||||
}
|
||||
100
api/v1/music/track.go
Normal file
100
api/v1/music/track.go
Normal file
|
|
@ -0,0 +1,100 @@
|
|||
package music
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type Track struct {
|
||||
number int
|
||||
title string
|
||||
description string
|
||||
lyrics string
|
||||
previewURL string
|
||||
}
|
||||
|
||||
// GETTERS
|
||||
|
||||
func (track Track) GetNumber() int {
|
||||
return track.number
|
||||
}
|
||||
|
||||
func (track Track) GetTitle() string {
|
||||
return track.title
|
||||
}
|
||||
|
||||
func (track Track) GetDescription() string {
|
||||
return track.description
|
||||
}
|
||||
|
||||
func (track Track) GetLyrics() string {
|
||||
return track.lyrics
|
||||
}
|
||||
|
||||
func (track Track) GetPreviewURL() string {
|
||||
return track.previewURL
|
||||
}
|
||||
|
||||
// SETTERS
|
||||
|
||||
func (track Track) SetNumber(number int) error {
|
||||
// TODO: update DB
|
||||
track.number = number
|
||||
return nil
|
||||
}
|
||||
|
||||
func (track Track) SetTitle(title string) error {
|
||||
// TODO: update DB
|
||||
track.title = title
|
||||
return nil
|
||||
}
|
||||
|
||||
func (track Track) SetDescription(description string) error {
|
||||
// TODO: update DB
|
||||
track.description = description
|
||||
return nil
|
||||
}
|
||||
|
||||
func (track Track) SetLyrics(lyrics string) error {
|
||||
// TODO: update DB
|
||||
track.lyrics = lyrics
|
||||
return nil
|
||||
}
|
||||
|
||||
func (track Track) SetPreviewURL(previewURL string) error {
|
||||
// TODO: update DB
|
||||
track.previewURL = previewURL
|
||||
return nil
|
||||
}
|
||||
|
||||
// DATABASE
|
||||
|
||||
func PullReleaseTracks(db *sqlx.DB, releaseID string) ([]Track, error) {
|
||||
var tracks = []Track{}
|
||||
|
||||
track_rows, err := db.Query("SELECT number, title, description, lyrics, preview_url FROM musictracks WHERE release=$1", releaseID)
|
||||
if err != nil {
|
||||
return []Track{}, err
|
||||
}
|
||||
|
||||
for track_rows.Next() {
|
||||
var track = Track{}
|
||||
|
||||
err = track_rows.Scan(
|
||||
&track.number,
|
||||
&track.title,
|
||||
&track.description,
|
||||
&track.lyrics,
|
||||
&track.previewURL,
|
||||
)
|
||||
if err != nil {
|
||||
fmt.Printf("Error while pulling track for release %s: %s\n", releaseID, err)
|
||||
continue
|
||||
}
|
||||
|
||||
tracks = append(tracks, track)
|
||||
}
|
||||
|
||||
return tracks, nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue