package main

import (
	"database/sql"
	"encoding/xml"
	"log"
	"time"

	_ "github.com/mattn/go-sqlite3"
)

type Scrobble struct {
	Artist      Correctable `xml:"artist" json:"artist"`
	AlbumArtist Correctable `xml:"albumArtist" json:"albumArtist"`
	TrackName   Correctable `xml:"track" json:"track"`
	Album       Correctable `xml:"album" json:"album"`
	TrackNumber int         `xml:"-" json:"-"`
	Duration    int         `xml:"-" json:"-"`
	Time        time.Time   `xml:"timestamp" json:"timestamp,string"`
	Chosen      bool        `xml:"-" json:"-"`
	Mbid        string      `xml:"-" json:"-"`
	MbidComp    string
	Session     string `xml:"-" json:"-"`
	User        string
	Image       string
}

type Session struct {
	XMLName       xml.Name `json:"-" xml:"session"`
	User          string   `json:"name" xml:"name"`
	Key           string   `json:"key" xml:"key"`
	Client        string
	ClientVersion string
	Protocol      string
	Created       time.Time
	Subscriber    int64 `json:"subscriber" xml:"subscriber"`
}

type DataStore interface {
	PutSession(*Session) error
	GetSession(key string) (*Session, error)
	GetPassword(userName string) (string, error)
	PutScrobbles([]Scrobble) error
	RecentScrobbles(lfmName string) []*Scrobble
	Api
}

type SqlStore struct {
	*sql.DB
}

func (store *SqlStore) PutSession(s *Session) error {
	query := `
	INSERT INTO scrobbling_sessions (lfm_name, session_key, client, client_version, protocol)
	VALUES ($1, $2, $3, $4, $5)`
	_, err := store.Exec(query, s.User, s.Key, s.Client, s.ClientVersion, s.Protocol)
	return err
}

func (store *SqlStore) GetSession(key string) (*Session, error) {
	query := `
	SELECT lfm_name, session_key, client, client_version, protocol, created
	FROM scrobbling_sessions WHERE session_key = $1`
	row := store.QueryRow(query, key)
	s := &Session{}
	err := row.Scan(&s.User, &s.Key, &s.Client, &s.ClientVersion, &s.Protocol,
		&s.Created)
	return s, err
}

func (store *SqlStore) GetPassword(name string) (string, error) {
	var password string
	row := store.QueryRow("SELECT lfm_password FROM users WHERE lfm_name = $1",
		name)
	err := row.Scan(&password)
	return password, err
}

func (store *SqlStore) PutScrobbles(scrobbles []Scrobble) error {
	tx, err := store.Begin()
	if err != nil {
		return err
	}
	query := `
	INSERT INTO scrobbles
	(artist, albumartist, trackname, album, tracknumber, duration, time,
	chosen, mbid, mbid_computed, session_key, lfm_name)
	VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`
	st, err := tx.Prepare(query)
	if err != nil {
		return err
	}
	for _, s := range scrobbles {
		_, err = st.Exec(s.Artist, s.AlbumArtist, s.TrackName, s.Album,
			s.TrackNumber, s.Duration, s.Time, s.Chosen, s.Mbid, s.MbidComp,
			s.Session, s.User)
		if err != nil {
			tx.Rollback()
			return err
		}
	}
	return tx.Commit()
}

func (store *SqlStore) RecentScrobbles(lfmName string) []*Scrobble {
	scrobbles := make([]*Scrobble, 0, 10)
	query := `
	SELECT artist, album, trackname, time, image
	FROM scrobbles 
	LEFT JOIN songs ON COALESCE(scrobbles.mbid, scrobbles.mbid_computed) = songs.mbid 
	WHERE lfm_name=$1 ORDER BY time DESC LIMIT 10`
	rows, err := store.Query(query, lfmName)
	if err != nil {
		log.Println(err)
	}

	for rows.Next() {
		scrobble := new(Scrobble)
		rows.Scan(&scrobble.Artist.Name, &scrobble.Album.Name,
			&scrobble.TrackName.Name, &scrobble.Time, &scrobble.Image)
		scrobbles = append(scrobbles, scrobble)
	}
	rows.Close()
	return scrobbles
}