package main

import (
	"database/sql"
	"encoding/xml"
	"fmt"
	"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:"-"`
	Session     string      `xml:"-" json:"-"`
	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
	Protocol   string
	Created    time.Time
	Subscriber int64 `json:"subscriber" xml:"subscriber"`
}

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

type SqlStore struct {
	*sql.DB
}

func NewSession(user string, client string, protocol string) *Session {
	return &Session{
		User:     user,
		Key:      randomToken(16),
		Client:   client,
		Protocol: protocol,
		Created:  time.Now(),
	}
}

func (store *SqlStore) PutSession(s *Session) {
	_, err := store.Exec("INSERT INTO scrobbling_sessions VALUES ($1, $2, $3, $4, $5)",
		s.User, s.Key, s.Client, s.Protocol, s.Created)
	if err != nil {
		log.Println(err)
	}
}

func (store *SqlStore) GetSession(key string) (*Session, error) {
	s := &Session{}
	row := store.QueryRow("SELECT * FROM scrobbling_sessions WHERE key = $1", key)
	err := row.Scan(&s.User, &s.Key, &s.Client, &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) {
	for _, s := range scrobbles {
		if _, err := store.Exec(
			"INSERT INTO scrobbles VALUES ($1, $2, $3, $4,$5, $6, $7, $8, $9, $10)",
			s.Artist,
			s.AlbumArtist,
			s.TrackName,
			s.Album,
			s.TrackNumber,
			s.Duration,
			s.Time,
			s.Chosen,
			s.Mbid,
			s.Session,
		); err != nil {
			fmt.Printf("error : %v\n", err)
		}
	}
}

func (store *SqlStore) RecentScrobbles(lfmName string) []*Scrobble {
	scrobbles := make([]*Scrobble, 0, 10)
	rows, err := store.Query("SELECT artist, album, trackname, time, image "+
		"FROM scrobbles JOIN scrobbling_sessions ON session = key "+
		"LEFT JOIN songs ON scrobbles.mbid = songs.mbid "+
		"WHERE lfm_name=$1 ORDER BY time DESC LIMIT 10",
		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
}